fix: fix logic to detect GCP in log

reearth · Feb 6, 2024 · 13a660b · 13a660b
1 parent 56c43c2
commit 13a660b
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/log/gcp.go b/log/gcp.go
@@ -7,7 +7,11 @@ import (
 )
 
 var GCP = false
-var GCPEnv = "GOOGLE_CLOUD_PROJECT"
+var GCPEnv = []string{
+	"CLOUD_RUN_JOB",
+	"K_SERVICE",
+	"GOOGLE_CLOUD_PROJECT",
+}
 
 type severity string
 
@@ -55,6 +59,11 @@ func isGCP() bool {
 		return true
 	}
 
-	gcp, _ := os.LookupEnv(GCPEnv)
-	return gcp != ""
+	for _, key := range GCPEnv {
+		if v, ok := os.LookupEnv(key); ok && v != "" {
+			return true
+		}
+	}
+
+	return false
 }