fix(issue-34): support only allow http2 requests if config is set

z5labs · Dec 18, 2023 · d8f356f · d8f356f
1 parent d5459f4
commit d8f356f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/http/http.go b/http/http.go
@@ -180,7 +180,13 @@ func (rt *Runtime) Run(ctx context.Context) error {
 
 	s := &http.Server{
 		Handler: otelhttp.NewHandler(
-			rt.h,
+			http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+				if rt.http2Only && r.ProtoMajor < 2 {
+					w.WriteHeader(http.StatusUnauthorized)
+					return
+				}
+				rt.h.ServeHTTP(w, r)
+			}),
 			"server",
 			otelhttp.WithMessageEvents(otelhttp.ReadEvents, otelhttp.WriteEvents),
 		),