Add envvars HAMMER_REST_API_ENABLED and HAMMER_METRICS_ENABLED

README.md

@@ -220,3 +220,19 @@ curl -X GET http://localhost:8000/v1/delivery-attempts/01E8HX1D6PYFB1HJFG0S7WEKB
 ```
 docker build -f Dockerfile -t hammer .
 ```
+
+## Disable REST API
+
+To disable the rest api, set the environment variable **HAMMER_REST_API_ENABLED** to false.
+
+```bash
+export HAMMER_REST_API_ENABLED='false'
+```
+
+## Disable Prometheus metrics
+
+To disable prometheus metrics, set the environment variable **HAMMER_METRICS_ENABLED** to false.
+
+```bash
+export HAMMER_METRICS_ENABLED='false'
+```

cmd/hammer/main.go

@@ -76,6 +76,10 @@ func newAppContext() appContext {
 }
 
 func gatewayServer() {
+	gatewayEnabled := env.GetBool("HAMMER_REST_API_ENABLED", true)
+	if !gatewayEnabled {
+		return
+	}
 	ctx := context.Background()
 	ctx, cancel := context.WithCancel(ctx)
 	defer cancel()
@@ -94,6 +98,10 @@ func gatewayServer() {
 }
 
 func metricsServer() {
+	metricsEnabled := env.GetBool("HAMMER_METRICS_ENABLED", true)
+	if !metricsEnabled {
+		return
+	}
 	port := env.GetInt("HAMMER_METRICS_PORT", 4001)
 	http.Handle("/metrics", promhttp.Handler())
 	err := http.ListenAndServe(fmt.Sprintf(":%d", port), nil)

local.env

@@ -9,3 +9,5 @@ HAMMER_WORKER_DATABASE_DELAY='5'
 HAMMER_WORKER_DEFAULT_FETCH_LIMIT='100'
 # See https://github.com/golang-migrate/migrate/tree/master/source/file
 HAMMER_DATABASE_MIGRATION_DIR='file:///db/migrations'
+HAMMER_REST_API_ENABLED='true'
+HAMMER_METRICS_ENABLED='true'