Merge branch 'v2' of github.com:tus/tusd into v2

This commit is contained in:
Marius 2022-10-10 20:49:03 +02:00
commit 70ba0f5540
1 changed files with 12 additions and 1 deletions

View File

@ -2,6 +2,7 @@ package hooks
import ( import (
"context" "context"
"net/http"
"time" "time"
grpc_retry "github.com/grpc-ecosystem/go-grpc-middleware/retry" grpc_retry "github.com/grpc-ecosystem/go-grpc-middleware/retry"
@ -66,12 +67,22 @@ func marshal(hookReq HookRequest) *pb.HookRequest {
Method: event.HTTPRequest.Method, Method: event.HTTPRequest.Method,
Uri: event.HTTPRequest.URI, Uri: event.HTTPRequest.URI,
RemoteAddr: event.HTTPRequest.RemoteAddr, RemoteAddr: event.HTTPRequest.RemoteAddr,
// TODO: HEaders Header: getHeaders(event.HTTPRequest.Header),
}, },
}, },
} }
} }
func getHeaders(httpHeader http.Header) (hookHeader map[string]string) {
hookHeader = make(map[string]string)
for key, val := range httpHeader {
if key != "" && val != nil && len(val) > 0 {
hookHeader[key] = val[0]
}
}
return hookHeader
}
func unmarshal(res *pb.HookResponse) (hookRes HookResponse) { func unmarshal(res *pb.HookResponse) (hookRes HookResponse) {
hookRes.RejectUpload = res.RejectUpload hookRes.RejectUpload = res.RejectUpload
hookRes.StopUpload = res.StopUpload hookRes.StopUpload = res.StopUpload