mirror of
https://github.com/hanzoai/o11y.git
synced 2026-10-15 12:58:33 +00:00
zeekay
b36a8788a5
Full rebrand of the SigNoz observability platform: - Go module: github.com/SigNoz/signoz → github.com/hanzoai/o11y - Frontend: @signozhq/* → @hanzo/o11y-*, posthog-js → @hanzo/insights - Docker: signoz/* → ghcr.io/hanzoai/o11y* - ClickHouse → Hanzo Datastore (hanzoai/datastore) - Env vars: SIGNOZ_* → HANZO_* - DB tables: signoz_* → o11y_* - All branding, URLs, docs updated - 2172 files changed
69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
package sqlmigration
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/hanzoai/o11y/pkg/factory"
|
|
"github.com/uptrace/bun"
|
|
"github.com/uptrace/bun/migrate"
|
|
)
|
|
|
|
// SQLMigration is the interface for a single migration.
|
|
type SQLMigration interface {
|
|
// Register registers the migration with the given migrations. Each migration needs to be registered
|
|
//in a dedicated `*.go` file so that the correct migration semantics can be detected.
|
|
Register(*migrate.Migrations) error
|
|
|
|
// Up runs the migration.
|
|
Up(context.Context, *bun.DB) error
|
|
|
|
// Down rolls back the migration.
|
|
Down(context.Context, *bun.DB) error
|
|
}
|
|
|
|
var (
|
|
OrgReference = "org"
|
|
UserReference = "user"
|
|
UserReferenceNoCascade = "user_no_cascade"
|
|
FactorPasswordReference = "factor_password"
|
|
CloudIntegrationReference = "cloud_integration"
|
|
AgentConfigVersionReference = "agent_config_version"
|
|
)
|
|
|
|
func New(
|
|
ctx context.Context,
|
|
settings factory.ProviderSettings,
|
|
config Config,
|
|
factories factory.NamedMap[factory.ProviderFactory[SQLMigration, Config]],
|
|
) (*migrate.Migrations, error) {
|
|
migrations := migrate.NewMigrations()
|
|
|
|
for _, factory := range factories.GetInOrder() {
|
|
migration, err := factory.New(ctx, settings, config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = migration.Register(migrations)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return migrations, nil
|
|
}
|
|
|
|
func MustNew(
|
|
ctx context.Context,
|
|
settings factory.ProviderSettings,
|
|
config Config,
|
|
factories factory.NamedMap[factory.ProviderFactory[SQLMigration, Config]],
|
|
) *migrate.Migrations {
|
|
migrations, err := New(ctx, settings, config, factories)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return migrations
|
|
}
|