mcp-reverse-proxy (#575)

* mcp-reverse-proxy

Summary:

- MCP server pass through to `postgres` connection.
- Added robot test `Concurrent psql and Reverse Proxy MCP HTTP Server Query Tool`.
- Uplifted wirre protocol server.

* - Linter issue.

Author: general-kroll-4-life

.vscode/launch.json

@@ -512,6 +512,30 @@
         "--pgsrv.debug.enable=${input:serverDebugPublish}",
       ],
     },
+    {
+      "name": "run server AND mcp",
+      "type": "go",
+      "request": "launch",
+      "envFile": "${workspaceFolder}/.vscode/.env",
+      "mode": "debug",
+      "program": "${workspaceFolder}/stackql",
+      "args": [
+        "srv",
+        "--pgsrv.port=5885",
+        "--tls.allowInsecure",
+        "--auth=${input:authString}",
+        "--session=${input:sessionString}",
+        "--gc=${input:gcString}",
+        "--registry=${input:registryString}",
+        "--namespaces=${input:namespaceString}",
+        "--sqlBackend=${input:sqlBackendString}",
+        "--dbInternal=${input:dbInternalString}",
+        "--export.alias=${input:exportAliasString}",
+        "--pgsrv.debug.enable=${input:serverDebugPublish}",
+        "--mcp.server.type=reverse_proxy",
+        "--mcp.config={\"server\": {\"transport\": \"http\", \"address\": \"127.0.0.1:9914\"}, \"backend\": {\"dsn\": \"postgres://stackql:stackql@127.0.0.1:5885?default_query_exec_mode=simple_protocol&standard_conforming_strings=on&client_encoding=UTF8\"}}"
+      ],
+    },
     {
       "name": "run mTLS server",
       "type": "go",

go.mod

@@ -19,7 +19,7 @@ require (
 	github.com/spf13/viper v1.10.1
 	github.com/stackql/any-sdk v0.2.2-beta07
 	github.com/stackql/go-suffix-map v0.0.1-alpha01
-	github.com/stackql/psql-wire v0.1.1-beta25
+	github.com/stackql/psql-wire v0.1.2-alpha01
 	github.com/stackql/stackql-parser v0.0.15-alpha06
 	github.com/stretchr/testify v1.10.0
 	golang.org/x/sync v0.15.0

go.sum

@@ -469,8 +469,8 @@ github.com/stackql/any-sdk v0.2.2-beta07 h1:c/MaT8p4lB30xslJo9LQm3JDWMMfzwheGXqf
 github.com/stackql/any-sdk v0.2.2-beta07/go.mod h1:m1o5TCfyKkdt2bREB3itwPv1MhM+lk4eu24KpPohFoY=
 github.com/stackql/go-suffix-map v0.0.1-alpha01 h1:TDUDS8bySu41Oo9p0eniUeCm43mnRM6zFEd6j6VUaz8=
 github.com/stackql/go-suffix-map v0.0.1-alpha01/go.mod h1:QAi+SKukOyf4dBtWy8UMy+hsXXV+yyEE4vmBkji2V7g=
-github.com/stackql/psql-wire v0.1.1-beta25 h1:DFBLjtz9N1S9gIYhqsjVZtVZMVSg7c0vvirPT29+S3s=
-github.com/stackql/psql-wire v0.1.1-beta25/go.mod h1:a44Wd8kDC3irFLpGutarKDBqhJ/aqXlj1aMzO5bVJYg=
+github.com/stackql/psql-wire v0.1.2-alpha01 h1:RMBRURGspmSNqm2/sgoEc+D6Sri2y/3drjl4nKlOOi4=
+github.com/stackql/psql-wire v0.1.2-alpha01/go.mod h1:a44Wd8kDC3irFLpGutarKDBqhJ/aqXlj1aMzO5bVJYg=
 github.com/stackql/readline v0.0.2-alpha05 h1:ID4QzGdplFBsrSnTuz8pvKzWw96JbrJg8fsLry2UriU=
 github.com/stackql/readline v0.0.2-alpha05/go.mod h1:OFAYOdXk/X4+5GYiDXFfaGrk+bCN6Qv0SYY5HNzD2E0=
 github.com/stackql/stackql-go-sqlite3 v1.0.4-stackql h1:fp70Vdw+PCVEoPrAhkyqPuAlrIiHT79mght/0rlR4oY=

internal/stackql/acid/tsm_physio/best_effort_coordinator.go

@@ -1,4 +1,4 @@
-package tsm_physio //nolint:stylecheck // prefer this nomenclature
+package tsm_physio //nolint:stylecheck,revive // prefer this nomenclature
 
 import (
 	"fmt"

internal/stackql/cmd/mcp.go

@@ -21,13 +21,17 @@ import (
 
 	"github.com/spf13/cobra"
 
+	"github.com/stackql/any-sdk/pkg/db/db_util"
+	"github.com/stackql/any-sdk/pkg/dto"
 	"github.com/stackql/any-sdk/pkg/logging"
 	"github.com/stackql/stackql/internal/stackql/acid/tsm_physio"
 	"github.com/stackql/stackql/internal/stackql/entryutil"
 	"github.com/stackql/stackql/internal/stackql/handler"
 	"github.com/stackql/stackql/internal/stackql/iqlerror"
 	"github.com/stackql/stackql/internal/stackql/mcpbackend"
 	"github.com/stackql/stackql/pkg/mcp_server"
+
+	_ "github.com/jackc/pgx/v5" //nolint:revive // canonical driver pattern
 )
 
 //nolint:gochecknoglobals // cobra pattern
@@ -68,27 +72,45 @@ func runMCPServer(handlerCtx handler.HandlerContext) {
 	if config.Server.IsReadOnly != nil {
 		isReadOnly = *config.Server.IsReadOnly
 	}
-	orchestrator, orchestratorErr := tsm_physio.NewOrchestrator(handlerCtx)
-	iqlerror.PrintErrorAndExitOneIfError(orchestratorErr)
-	iqlerror.PrintErrorAndExitOneIfNil(orchestrator, "orchestrator is unexpectedly nil")
-	// handlerCtx.SetTSMOrchestrator(orchestrator)
-	backend, backendErr := mcpbackend.NewStackqlMCPBackendService(
-		isReadOnly,
-		orchestrator,
-		handlerCtx,
-		logging.GetLogger(),
-	)
-	iqlerror.PrintErrorAndExitOneIfError(backendErr)
-	iqlerror.PrintErrorAndExitOneIfNil(backend, "mcp backend is unexpectedly nil")
-
+	var backend mcp_server.Backend
+	var backendErr error
+	if mcpServerType == "reverse_proxy" {
+		config.Server.Transport = "http"
+		dsn := config.GetBackendConnectionString()
+		// conn
+		var cfg dto.SQLBackendCfg
+		cfg.DSN = dsn
+		cfg.InitMaxRetries = 5
+		cfg.InitRetryInitialDelay = 2
+		db, err := db_util.GetDB("pgx", "postgres", cfg)
+		iqlerror.PrintErrorAndExitOneIfError(err)
+		backend, backendErr = mcpbackend.NewStackqlMCPReverseProxyService(
+			isReadOnly,
+			dsn,
+			db,
+			handlerCtx,
+			logging.GetLogger(),
+		)
+		iqlerror.PrintErrorAndExitOneIfError(backendErr)
+	} else {
+		orchestrator, orchestratorErr := tsm_physio.NewOrchestrator(handlerCtx)
+		iqlerror.PrintErrorAndExitOneIfError(orchestratorErr)
+		iqlerror.PrintErrorAndExitOneIfNil(orchestrator, "orchestrator is unexpectedly nil")
+		// handlerCtx.SetTSMOrchestrator(orchestrator)
+		backend, backendErr = mcpbackend.NewStackqlMCPBackendService(
+			isReadOnly,
+			orchestrator,
+			handlerCtx,
+			logging.GetLogger(),
+		)
+		iqlerror.PrintErrorAndExitOneIfError(backendErr)
+		iqlerror.PrintErrorAndExitOneIfNil(backend, "mcp backend is unexpectedly nil")
+	}
 	server, serverErr := mcp_server.NewAgnosticBackendServer(
 		backend,
 		&config,
 		logging.GetLogger(),
 	)
-	// server, serverErr := mcp_server.NewExampleHTTPBackendServer(
-	// 	logging.GetLogger(),
-	// )
 	iqlerror.PrintErrorAndExitOneIfError(serverErr)
 	server.Start(context.Background()) //nolint:errcheck // TODO: investigate
 }