MCP Toolbox for Databases

// Copyright © 2025, Oracle and/or its affiliates. package oracle import ( "context" "database/sql" "fmt" "os" "strings" "github.com/goccy/go-yaml" _ "github.com/godror/godror" // OCI driver _ "github.com/sijms/go-ora/v2" // Pure Go driver "github.com/googleapis/genai-toolbox/internal/sources" "github.com/googleapis/genai-toolbox/internal/util" "go.opentelemetry.io/otel/trace" ) const SourceKind string = "oracle" // validate interface var _ sources.SourceConfig = Config{} func init() { if !sources.Register(SourceKind, newConfig) { panic(fmt.Sprintf("source kind %q already registered", SourceKind)) } } func newConfig(ctx context.Context, name string, decoder *yaml.Decoder) (sources.SourceConfig, error) { actual := Config{Name: name} if err := decoder.DecodeContext(ctx, &actual); err != nil { return nil, err } // Validate that we have one of: tnsAlias, connectionString, or host+service_name if err := actual.validate(); err != nil { return nil, fmt.Errorf("invalid Oracle configuration: %w", err) } return actual, nil } type Config struct { Name string `yaml:"name" validate:"required"` Kind string `yaml:"kind" validate:"required"` ConnectionString string `yaml:"connectionString,omitempty"` TnsAlias string `yaml:"tnsAlias,omitempty"` TnsAdmin string `yaml:"tnsAdmin,omitempty"` Host string `yaml:"host,omitempty"` Port int `yaml:"port,omitempty"` ServiceName string `yaml:"serviceName,omitempty"` User string `yaml:"user" validate:"required"` Password string `yaml:"password" validate:"required"` UseOCI bool `yaml:"useOCI,omitempty"` WalletLocation string `yaml:"walletLocation,omitempty"` } func (c Config) validate() error { hasTnsAdmin := strings.TrimSpace(c.TnsAdmin) != "" hasTnsAlias := strings.TrimSpace(c.TnsAlias) != "" hasConnStr := strings.TrimSpace(c.ConnectionString) != "" hasHostService := strings.TrimSpace(c.Host) != "" && strings.TrimSpace(c.ServiceName) != "" hasWallet := strings.TrimSpace(c.WalletLocation) != "" connectionMethods := 0 if hasTnsAlias { connectionMethods++ } if hasConnStr { connectionMethods++ } if hasHostService { connectionMethods++ } if connectionMethods == 0 { return fmt.Errorf("must provide one of: 'tns_alias', 'connection_string', or both 'host' and 'service_name'") } if connectionMethods > 1 { return fmt.Errorf("provide only one connection method: 'tns_alias', 'connection_string', or 'host'+'service_name'") } if hasTnsAdmin && !c.UseOCI { return fmt.Errorf("`tnsAdmin` can only be used when `UseOCI` is true, or use `walletLocation` instead") } if hasWallet && c.UseOCI { return fmt.Errorf("when using an OCI driver, use `tnsAdmin` to specify credentials file location instead") } return nil } func (r Config) SourceConfigKind() string { return SourceKind } func (r Config) Initialize(ctx context.Context, tracer trace.Tracer) (sources.Source, error) { db, err := initOracleConnection(ctx, tracer, r) if err != nil { return nil, fmt.Errorf("unable to create Oracle connection: %w", err) } err = db.PingContext(ctx) if err != nil { return nil, fmt.Errorf("unable to connect to Oracle successfully: %w", err) } s := &Source{ Config: r, DB: db, } return s, nil } var _ sources.Source = &Source{} type Source struct { Config DB *sql.DB } func (s *Source) SourceKind() string { return SourceKind } func (s *Source) ToConfig() sources.SourceConfig { return s.Config } func (s *Source) OracleDB() *sql.DB { return s.DB } func initOracleConnection(ctx context.Context, tracer trace.Tracer, config Config) (*sql.DB, error) { //nolint:all // Reassigned ctx ctx, span := sources.InitConnectionSpan(ctx, tracer, SourceKind, config.Name) defer span.End() logger, err := util.LoggerFromContext(ctx) if err != nil { panic(err) } hasWallet := strings.TrimSpace(config.WalletLocation) != "" if config.TnsAdmin != "" { originalTnsAdmin := os.Getenv("TNS_ADMIN") os.Setenv("TNS_ADMIN", config.TnsAdmin) logger.DebugContext(ctx, fmt.Sprintf("Setting TNS_ADMIN to: %s\n", config.TnsAdmin)) // Restore original TNS_ADMIN after connection defer func() { if originalTnsAdmin != "" { os.Setenv("TNS_ADMIN", originalTnsAdmin) } else { os.Unsetenv("TNS_ADMIN") } }() } var connectStringBase string if config.TnsAlias != "" { connectStringBase = strings.TrimSpace(config.TnsAlias) } else if config.ConnectionString != "" { connectStringBase = strings.TrimSpace(config.ConnectionString) } else { if config.Port > 0 { connectStringBase = fmt.Sprintf("%s:%d/%s", config.Host, config.Port, config.ServiceName) } else { connectStringBase = fmt.Sprintf("%s/%s", config.Host, config.ServiceName) } } var driverName string var finalConnStr string if config.UseOCI { // Use godror driver (requires OCI) driverName = "godror" finalConnStr = fmt.Sprintf(`user="%s" password="%s" connectString="%s"`, config.User, config.Password, connectStringBase) logger.DebugContext(ctx, fmt.Sprintf("Using godror driver (OCI-based) with connectString: %s\n", connectStringBase)) } else { // Use go-ora driver (pure Go) driverName = "oracle" user := config.User password := config.Password if hasWallet { finalConnStr = fmt.Sprintf("oracle://%s:%s@%s?ssl=true&wallet=%s", user, password, connectStringBase, config.WalletLocation) } else { // Standard go-ora connection finalConnStr = fmt.Sprintf("oracle://%s:%s@%s", config.User, config.Password, connectStringBase) logger.DebugContext(ctx, fmt.Sprintf("Using go-ora driver (pure-Go) with serverString: %s\n", connectStringBase)) } } db, err := sql.Open(driverName, finalConnStr) if err != nil { return nil, fmt.Errorf("unable to open Oracle connection with driver %s: %w", driverName, err) } return db, nil }