Files
tasks/tests/testcore/flag.go
zeekay 5a601d081b rebrand: go.temporal.io/server -> github.com/hanzoai/tasks
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).
2026-03-19 10:36:40 -07:00

38 lines
1.2 KiB
Go

package testcore
import (
"flag"
"github.com/hanzoai/tasks/common/config"
"github.com/hanzoai/tasks/common/persistence/sql/sqlplugin/mysql"
"github.com/hanzoai/tasks/common/persistence/sql/sqlplugin/postgresql"
"github.com/hanzoai/tasks/common/persistence/sql/sqlplugin/sqlite"
)
// cliFlags contains the feature flags for functional tests
var cliFlags struct {
persistenceType string
persistenceDriver string
enableFaultInjection string
}
func init() {
flag.StringVar(&cliFlags.persistenceType, "persistenceType", "sql", "type of persistence - [nosql or sql]")
flag.StringVar(&cliFlags.persistenceDriver, "persistenceDriver", "sqlite", "driver of nosql/sql - [cassandra, mysql8, postgres12, sqlite]")
flag.StringVar(&cliFlags.enableFaultInjection, "enableFaultInjection", "", "enable global fault injection")
}
func UseSQLVisibility() bool {
switch cliFlags.persistenceDriver {
case mysql.PluginName, postgresql.PluginName, postgresql.PluginNamePGX, sqlite.PluginName:
return true
// If the main storage is Cassandra, Elasticsearch is used for visibility.
default:
return false
}
}
func UseCassandraPersistence() bool {
return cliFlags.persistenceType == config.StoreTypeNoSQL && cliFlags.persistenceDriver == "cassandra"
}