feat: support extra headder

ShiningRush · Mar 15, 2023 · 940bc4d · 940bc4d
1 parent 8993ec7
commit 940bc4d
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -21,7 +21,7 @@ mock:
 .PHONY: tidy
 tidy:
 	go mod tidy && go fmt ./...
-	$(foreach var,$(WRAPPER_SUBMODULES),cd $(PWD)/wrapper/$(var) && go mod tidy && go fmt ./...;)
+	@$(foreach var,$(WRAPPER_SUBMODULES),cd $(PWD)/wrapper/$(var) && go mod tidy && go fmt ./...;)
 
 .PHONY: test
 test:

diff --git a/data/resp.go b/data/resp.go
@@ -84,6 +84,8 @@ type FileResponse struct {
 	ContentReader io.ReadCloser
 	Size          int
 	StatusCode    int
+
+	ExtraHeader map[string]string
 }
 
 func (r *FileResponse) Get() *FileResponse {

diff --git a/wrapper/wrapper.go b/wrapper/wrapper.go
@@ -93,6 +93,11 @@ func writeFileResp(input HandleHttpInPipelineInput, fr data.HttpFileResponse) {
 	if fileResp.Size != 0 {
 		input.RespWriter.SetHeader("Content-Length", strconv.Itoa(fileResp.Size))
 	}
+	if fileResp.ExtraHeader != nil {
+		for k, v := range fileResp.ExtraHeader {
+			input.RespWriter.SetHeader(k, v)
+		}
+	}
 
 	if fileResp.ContentReader != nil {
 		if sw, ok := input.RespWriter.(data.StreamSetter); ok {