set request headers

chyroc · May 9, 2024 · a2bdf1f · a2bdf1f
1 parent 9b44343
commit a2bdf1f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/impl_request.go b/impl_request.go
@@ -129,6 +129,11 @@ func (r *Lark) parseRawHttpRequest(ctx context.Context, req *RawRequestReq) (*ra
 	if req.MethodOption.timeout > 0 {
 		rawHttpReq.Timeout = req.MethodOption.timeout
 	}
+	if len(req.MethodOption.headers) > 0 {
+		for k, v := range req.MethodOption.headers {
+			rawHttpReq.Headers[k] = v
+		}
+	}
 
 	// 1 headers
 	if err := rawHttpReq.parseHeader(ctx, r, req); err != nil {

diff --git a/lark.go b/lark.go
@@ -174,10 +174,18 @@ func WithRequestTimeout(timeout time.Duration) MethodOptionFunc {
 	}
 }
 
+// WithRequestHeaders set request headers
+func WithRequestHeaders(headers map[string]string) MethodOptionFunc {
+	return func(option *MethodOption) {
+		option.headers = headers
+	}
+}
+
 // MethodOption method option
 type MethodOption struct {
 	userAccessToken string
 	timeout         time.Duration
+	headers         map[string]string
 }
 
 func newMethodOption(options []MethodOptionFunc) *MethodOption {