fixes

Author: bwagner5

cmd/v1/main.go

@@ -1,3 +1,17 @@
+/*
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
 package main
 
 import (
@@ -12,10 +26,11 @@ import (
 	"github.com/aws/aws-sdk-go/service/ec2"
 	"github.com/aws/aws-sdk-go/service/eks"
 	"github.com/aws/aws-sdk-go/service/s3"
-	"github.com/bwagner5/aws-sdk-go-metrics/pkg/awsmetrics"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"github.com/samber/lo"
+
+	"github.com/bwagner5/aws-sdk-go-metrics/pkg/awsmetrics"
 )
 
 type Options struct {
@@ -47,8 +62,15 @@ func main() {
 		registry,
 		promhttp.HandlerOpts{EnableOpenMetrics: false},
 	))
+	srv := &http.Server{
+		ReadTimeout:       1 * time.Second,
+		WriteTimeout:      1 * time.Second,
+		IdleTimeout:       30 * time.Second,
+		ReadHeaderTimeout: 2 * time.Second,
+		Addr:              fmt.Sprintf("127.0.0.1:%d", opts.Port),
+	}
 	log.Printf("Serving prometheus metrics at http://127.0.0.1:%d/metrics", opts.Port)
-	lo.Must0(http.ListenAndServe(fmt.Sprintf("127.0.0.1:%d", opts.Port), nil))
+	lo.Must0(srv.ListenAndServe())
 }
 
 func demo(sess *session.Session) {

cmd/v2/main.go

@@ -1,3 +1,17 @@
+/*
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
 package main
 
 import (
@@ -16,10 +30,11 @@ import (
 	ekstypes "github.com/aws/aws-sdk-go-v2/service/eks/types"
 	"github.com/aws/aws-sdk-go-v2/service/s3"
 	s3types "github.com/aws/aws-sdk-go-v2/service/s3/types"
-	"github.com/bwagner5/aws-sdk-go-metrics/pkg/awsmetricsv2"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"github.com/samber/lo"
+
+	"github.com/bwagner5/aws-sdk-go-metrics/pkg/awsmetricsv2"
 )
 
 type Options struct {
@@ -49,8 +64,15 @@ func main() {
 		registry,
 		promhttp.HandlerOpts{EnableOpenMetrics: false},
 	))
+	srv := &http.Server{
+		ReadTimeout:       1 * time.Second,
+		WriteTimeout:      1 * time.Second,
+		IdleTimeout:       30 * time.Second,
+		ReadHeaderTimeout: 2 * time.Second,
+		Addr:              fmt.Sprintf("127.0.0.1:%d", opts.Port),
+	}
 	log.Printf("Serving prometheus metrics at http://127.0.0.1:%d/metrics", opts.Port)
-	lo.Must0(http.ListenAndServe(fmt.Sprintf("127.0.0.1:%d", opts.Port), nil))
+	lo.Must0(srv.ListenAndServe())
 }
 
 func demo(cfg aws.Config) {