docd: fallback to AppEngine env vars when flags not provided (#128)

sajari · Sep 20, 2022 · 106aaad · 106aaad
1 parent f481425
commit 106aaad
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/docd/main.go b/docd/main.go
@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"log"
 	"net/http"
+	"os"
 
 	"cloud.google.com/go/errorreporting"
 
@@ -40,6 +41,12 @@ func main() {
 
 	var er internal.ErrorReporter = &internal.NopErrorReporter{}
 	if *errorReporting {
+		if *errorReportingGCPProjectID == "" {
+			*errorReportingGCPProjectID = os.Getenv("GOOGLE_CLOUD_PROJECT")
+		}
+		if *errorReportingAppEngineService == "" {
+			*errorReportingAppEngineService = os.Getenv("GAE_SERVICE")
+		}
 		var err error
 		er, err = errorreporting.NewClient(context.Background(), *errorReportingGCPProjectID, errorreporting.Config{
 			ServiceName: *errorReportingAppEngineService,