mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
143 lines
3.3 KiB
Go
143 lines
3.3 KiB
Go
package sqlplugin
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/hanzoai/tasks/common/config"
|
|
"github.com/hanzoai/tasks/common/log"
|
|
"github.com/hanzoai/tasks/common/metrics"
|
|
"github.com/hanzoai/tasks/common/resolver"
|
|
)
|
|
|
|
type (
|
|
DbKind int
|
|
)
|
|
|
|
const (
|
|
DbKindUnknown DbKind = iota
|
|
DbKindMain
|
|
DbKindVisibility
|
|
)
|
|
|
|
type VersionedBlob struct {
|
|
Version int64
|
|
Data []byte
|
|
DataEncoding string
|
|
}
|
|
|
|
type (
|
|
// Plugin defines the interface for any SQL database that needs to implement
|
|
Plugin interface {
|
|
CreateDB(dbKind DbKind, cfg *config.SQL, r resolver.ServiceResolver, l log.Logger, mh metrics.Handler) (GenericDB, error)
|
|
|
|
GetVisibilityQueryConverter() VisibilityQueryConverter
|
|
}
|
|
|
|
// TableCRUD defines the API for interacting with the database tables
|
|
TableCRUD interface {
|
|
ClusterMetadata
|
|
Namespace
|
|
Visibility
|
|
QueueMessage
|
|
QueueMetadata
|
|
QueueV2Message
|
|
QueueV2Metadata
|
|
|
|
MatchingTask
|
|
MatchingTaskV2
|
|
MatchingTaskQueue
|
|
MatchingTaskQueueUserData
|
|
|
|
NexusEndpoints
|
|
|
|
HistoryNode
|
|
HistoryTree
|
|
|
|
HistoryShard
|
|
|
|
HistoryExecution
|
|
HistoryExecutionBuffer
|
|
HistoryExecutionActivity
|
|
HistoryExecutionChildWorkflow
|
|
HistoryExecutionTimer
|
|
HistoryExecutionRequestCancel
|
|
HistoryExecutionSignal
|
|
HistoryExecutionSignalRequest
|
|
HistoryExecutionChasm
|
|
|
|
HistoryImmediateTask
|
|
HistoryScheduledTask
|
|
HistoryTransferTask
|
|
HistoryTimerTask
|
|
HistoryReplicationTask
|
|
HistoryReplicationDLQTask
|
|
HistoryVisibilityTask
|
|
}
|
|
|
|
// AdminCRUD defines admin operations for CLI and test suites
|
|
AdminCRUD interface {
|
|
CreateSchemaVersionTables() error
|
|
ReadSchemaVersion(database string) (string, error)
|
|
UpdateSchemaVersion(database string, newVersion string, minCompatibleVersion string) error
|
|
WriteSchemaUpdateLog(oldVersion string, newVersion string, manifestMD5 string, desc string) error
|
|
ListTables(database string) ([]string, error)
|
|
DropTable(table string) error
|
|
DropAllTables(database string) error
|
|
CreateDatabase(database string) error
|
|
DropDatabase(database string) error
|
|
Exec(stmt string, args ...any) error
|
|
}
|
|
|
|
// Tx defines the API for a SQL transaction
|
|
Tx interface {
|
|
TableCRUD
|
|
Commit() error
|
|
Rollback() error
|
|
}
|
|
|
|
// DB defines the API for regular SQL operations of a Temporal server
|
|
DB interface {
|
|
TableCRUD
|
|
GenericDB
|
|
BeginTx(ctx context.Context) (Tx, error)
|
|
IsDupEntryError(err error) bool
|
|
}
|
|
|
|
// AdminDB defines the API for admin SQL operations for CLI and testing suites
|
|
AdminDB interface {
|
|
AdminCRUD
|
|
GenericDB
|
|
ExpectedVersion() string
|
|
VerifyVersion() error
|
|
}
|
|
|
|
GenericDB interface {
|
|
DbName() string
|
|
PluginName() string
|
|
Close() error
|
|
}
|
|
|
|
// Conn defines the API for a single database connection
|
|
Conn interface {
|
|
Rebind(query string) string
|
|
ExecContext(ctx context.Context, query string, args ...any) (sql.Result, error)
|
|
NamedExecContext(ctx context.Context, query string, arg any) (sql.Result, error)
|
|
GetContext(ctx context.Context, dest any, query string, args ...any) error
|
|
SelectContext(ctx context.Context, dest any, query string, args ...any) error
|
|
PrepareNamedContext(ctx context.Context, query string) (*sqlx.NamedStmt, error)
|
|
}
|
|
)
|
|
|
|
func (k DbKind) String() string {
|
|
switch k {
|
|
case DbKindMain:
|
|
return "main"
|
|
case DbKindVisibility:
|
|
return "visibility"
|
|
default:
|
|
return "unknown"
|
|
}
|
|
}
|