feat: aiproxy deploy support one db (#5241)

labring · Nov 28, 2024 · d98d1a9 · d98d1a9
1 parent 98fdc1a
commit d98d1a9
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/service/aiproxy/README.md b/service/aiproxy/README.md
@@ -5,3 +5,12 @@ sealos run ghcr.io/labring/sealos-cloud-aiproxy-service:latest \
     -e ADMIN_KEY=<admin-key> \
     -e cloudDomain=<cloud-domain>
 ```
+
+# Use One PostgreSQL
+
+```bash
+sealos run ghcr.io/labring/sealos-cloud-aiproxy-service:latest \
+    -e ADMIN_KEY=<admin-key> \
+    -e cloudDomain=<cloud-domain> \
+    -e LOG_SQL_DSN=""
+```
diff --git a/service/aiproxy/deploy/scripts/init.sh b/service/aiproxy/deploy/scripts/init.sh
@@ -69,6 +69,18 @@ if grep "<sql-placeholder>" manifests/aiproxy-config.yaml >/dev/null 2>&1; then
     # Update config
     sed -i "s|<sql-placeholder>|${SQL_DSN}|g" manifests/aiproxy-config.yaml
     sed -i "s|<sql-log-placeholder>|${LOG_SQL_DSN}|g" manifests/aiproxy-config.yaml
+  elif grep "LOG_SQL_DSN: \"\"" manifests/aiproxy-config.yaml >/dev/null 2>&1; then
+    # Deploy PostgreSQL resources
+    kubectl apply -f manifests/pgsql.yaml -n aiproxy-system
+
+    # Wait for secrets
+    wait_for_secret "aiproxy-conn-credential"
+
+    # Build connection strings
+    SQL_DSN=$(build_postgres_dsn "aiproxy-conn-credential") || exit $?
+
+    # Update config
+    sed -i "s|<sql-placeholder>|${SQL_DSN}|g" manifests/aiproxy-config.yaml
   else
     echo "Error: LOG_SQL_DSN is not allowed to be passed alone, please provide both SQL_DSN and LOG_SQL_DSN or provide SQL_DSN only or neither."
     exit 1