Wait for app to exit before letting the cloud proxy terminate (#6)

ricardoarchive · Nov 25, 2019 · 943fd08 · 943fd08
1 parent 4279a42
commit 943fd08
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -171,6 +171,12 @@ func getCloudContainer() v1.Container {
 		cloudSQLProxyContainer.Resources = v1.ResourceRequirements{Requests: requestResources, Limits: limitResources}
 		cloudSQLProxyContainer.SecurityContext = &securityContext
 		cloudSQLProxyContainer.VolumeMounts = append(cloudSQLProxyContainer.VolumeMounts, volumeMount)
+
+		cloudSQLProxyContainer.Lifecycle = &v1.Lifecycle{PreStop: &v1.Handler{
+			Exec: &v1.ExecAction{
+				Command: []string{"/bin/sh", "-c", "while [ $(netstat -plunt | grep tcp | grep -v envoy | grep -v cloud_sql_proxy | wc -l | xargs) -ne 0 ]; do sleep 1; done"},
+			},
+		}}
 	}
 
 	return cloudSQLProxyContainer