- Agnistic backend for MCP.

Author: general-kroll-4-life

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-beta23
+	github.com/stackql/psql-wire v0.1.1-beta25
 	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-beta23 h1:1ayYMjZArfDcIMyEOKnm+Bp1zRCISw8pguvTFuUhhVQ=
-github.com/stackql/psql-wire v0.1.1-beta23/go.mod h1:a44Wd8kDC3irFLpGutarKDBqhJ/aqXlj1aMzO5bVJYg=
+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/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/txn_provider.go

@@ -24,15 +24,15 @@ const (
 // that orchestrates transaction managers.
 type Provider interface {
 	// Create a new transaction manager.
-	GetOrchestrator(handler.HandlerContext) (Orchestrator, error)
+	getOrchestrator(handler.HandlerContext) (Orchestrator, error)
 	GetTSM(handlerCtx handler.HandlerContext) (tsm.TSM, error)
 }
 
 type standardProvider struct {
 	ctx txn_context.ITransactionCoordinatorContext
 }
 
-func (sp *standardProvider) GetOrchestrator(handlerCtx handler.HandlerContext) (Orchestrator, error) {
+func (sp *standardProvider) getOrchestrator(handlerCtx handler.HandlerContext) (Orchestrator, error) {
 	tsmInstance, walError := GetTSM(handlerCtx)
 	if walError != nil {
 		return nil, walError
@@ -67,3 +67,12 @@ func GetProviderInstance(ctx txn_context.ITransactionCoordinatorContext) (Provid
 	})
 	return providerSingleton, err
 }
+
+func NewOrchestrator(handlerCtx handler.HandlerContext) (Orchestrator, error) {
+	txnProvider, txnProviderErr := GetProviderInstance(
+		handlerCtx.GetTxnCoordinatorCtx())
+	if txnProviderErr != nil {
+		return nil, txnProviderErr
+	}
+	return txnProvider.getOrchestrator(handlerCtx)
+}

internal/stackql/cmd/mcp.go

@@ -22,8 +22,10 @@ import (
 	"github.com/spf13/cobra"
 
 	"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/iqlerror"
+	"github.com/stackql/stackql/internal/stackql/mcpbackend"
 	"github.com/stackql/stackql/pkg/mcp_server"
 )
 
@@ -53,7 +55,25 @@ var mcpSrvCmd = &cobra.Command{
 		var config mcp_server.Config
 		json.Unmarshal([]byte(mcpConfig), &config) //nolint:errcheck // TODO: investigate
 		config.Server.Transport = mcpServerType
-		server, serverErr := mcp_server.NewExampleBackendServer(
+		var isReadOnly bool
+		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")
+
+		server, serverErr := mcp_server.NewAgnosticBackendServer(
+			backend,
 			&config,
 			logging.GetLogger(),
 		)

internal/stackql/driver/driver.go

@@ -46,11 +46,7 @@ func (sdf *basicStackQLDriverFactory) newSQLDriver() (StackQLDriver, error) {
 	if err != nil {
 		return nil, err
 	}
-	txnProvider, txnProviderErr := tsm_physio.GetProviderInstance(sdf.handlerCtx.GetTxnCoordinatorCtx())
-	if txnProviderErr != nil {
-		return nil, txnProviderErr
-	}
-	txnOrchestrator, orcErr := txnProvider.GetOrchestrator(sdf.handlerCtx)
+	txnOrchestrator, orcErr := tsm_physio.NewOrchestrator(sdf.handlerCtx)
 	if orcErr != nil {
 		return nil, orcErr
 	}
@@ -186,12 +182,7 @@ func (dr *basicStackQLDriver) SplitCompoundQuery(s string) ([]string, error) {
 }
 
 func NewStackQLDriver(handlerCtx handler.HandlerContext) (StackQLDriver, error) {
-	txnProvider, txnProviderErr := tsm_physio.GetProviderInstance(
-		handlerCtx.GetTxnCoordinatorCtx())
-	if txnProviderErr != nil {
-		return nil, txnProviderErr
-	}
-	txnOrchestrator, orcErr := txnProvider.GetOrchestrator(handlerCtx)
+	txnOrchestrator, orcErr := tsm_physio.NewOrchestrator(handlerCtx)
 	if orcErr != nil {
 		return nil, orcErr
 	}