mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Alex ShtinandGitHub
8e8f1b0f7c
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.
## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
130 lines
3.3 KiB
Go
130 lines
3.3 KiB
Go
package schema
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
|
|
"github.com/blang/semver/v4"
|
|
"go.temporal.io/server/common/log"
|
|
"go.temporal.io/server/common/log/tag"
|
|
"go.temporal.io/server/common/persistence"
|
|
dbschemas "go.temporal.io/server/schema"
|
|
)
|
|
|
|
// SetupTask represents a task
|
|
// that sets up cassandra schema on
|
|
// a specified keyspace
|
|
type SetupTask struct {
|
|
db DB
|
|
config *SetupConfig
|
|
logger log.Logger
|
|
}
|
|
|
|
// NewSetupSchemaTask returns a new instance of SetupTask
|
|
func NewSetupSchemaTask(db DB, config *SetupConfig, logger log.Logger) *SetupTask {
|
|
return &SetupTask{
|
|
db: db,
|
|
config: config,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
// Run executes the task
|
|
func (task *SetupTask) Run() error {
|
|
config := task.config
|
|
task.logger.Info("Starting schema setup", tag.NewAnyTag("config", config))
|
|
|
|
if config.Overwrite {
|
|
err := task.db.DropAllTables()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if !config.DisableVersioning {
|
|
task.logger.Debug("Setting up version tables")
|
|
if err := task.db.CreateSchemaVersionTables(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if len(config.SchemaFilePath) > 0 || len(config.SchemaName) > 0 {
|
|
var schemaBuf []byte
|
|
var err error
|
|
var schemaFilePath string
|
|
if len(config.SchemaName) > 0 {
|
|
fsys := dbschemas.Assets()
|
|
schemaFilePath = path.Join(config.SchemaName, "schema"+schemaFileEnding(config.SchemaName))
|
|
schemaBuf, err = fs.ReadFile(fsys, schemaFilePath)
|
|
} else {
|
|
schemaFilePath, err = filepath.Abs(config.SchemaFilePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
schemaBuf, err = os.ReadFile(schemaFilePath)
|
|
}
|
|
if err != nil {
|
|
return fmt.Errorf("error reading file %s: %w", schemaFilePath, err)
|
|
}
|
|
stmts, err := persistence.LoadAndSplitQueryFromReaders([]io.Reader{bytes.NewBuffer(schemaBuf)})
|
|
if err != nil {
|
|
return fmt.Errorf("error parsing query: %v", err)
|
|
}
|
|
task.logger.Debug("----- Creating types and tables -----")
|
|
for _, stmt := range stmts {
|
|
task.logger.Debug(rmspaceRegex.ReplaceAllString(stmt, " "))
|
|
if err := task.db.Exec(stmt); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
task.logger.Debug("----- Done -----")
|
|
}
|
|
|
|
if !config.DisableVersioning {
|
|
currVer, err := task.db.ReadSchemaVersion()
|
|
if err != nil {
|
|
currVer = "0.0"
|
|
}
|
|
|
|
currVerParsed, err := semver.ParseTolerant(currVer)
|
|
if err != nil {
|
|
task.logger.Fatal("Unable to parse current version",
|
|
tag.NewStringTag("current version", currVer),
|
|
tag.Error(err),
|
|
)
|
|
}
|
|
|
|
initialVersionParsed, err := semver.ParseTolerant(config.InitialVersion)
|
|
if err != nil {
|
|
task.logger.Fatal("Unable to parse initial version",
|
|
tag.NewStringTag("initial version", config.InitialVersion),
|
|
tag.Error(err),
|
|
)
|
|
}
|
|
|
|
if currVerParsed.GT(initialVersionParsed) {
|
|
task.logger.Debug(fmt.Sprintf("Current database schema version %v is greater than initial schema version %v. Skip version upgrade", currVer, config.InitialVersion))
|
|
} else {
|
|
task.logger.Debug(fmt.Sprintf("Setting initial schema version to %v", config.InitialVersion))
|
|
err := task.db.UpdateSchemaVersion(config.InitialVersion, config.InitialVersion)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
task.logger.Debug("Updating schema update log")
|
|
err = task.db.WriteSchemaUpdateLog("0", config.InitialVersion, "", "initial version")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
task.logger.Info("Schema setup complete")
|
|
|
|
return nil
|
|
}
|