mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-29 04:01:59 +08:00
52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package migration
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/uptrace/bun"
|
|
"github.com/uptrace/bun/migrate"
|
|
"go.signoz.io/signoz/pkg/factory"
|
|
"go.signoz.io/signoz/pkg/sqlmigrator"
|
|
)
|
|
|
|
type addPipelines struct{}
|
|
|
|
func NewAddPipelinesFactory() factory.ProviderFactory[sqlmigrator.SQLMigration, sqlmigrator.Config] {
|
|
return factory.NewProviderFactory(factory.MustNewName("add_pipelines"), newAddPipelines)
|
|
}
|
|
|
|
func newAddPipelines(_ context.Context, _ factory.ProviderSettings, _ sqlmigrator.Config) (sqlmigrator.SQLMigration, error) {
|
|
return &addPipelines{}, nil
|
|
}
|
|
|
|
func (migration *addPipelines) Register(migrations *migrate.Migrations) error {
|
|
if err := migrations.Register(migration.Up, migration.Down); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (migration *addPipelines) Up(ctx context.Context, db *bun.DB) error {
|
|
if _, err := db.ExecContext(ctx, `CREATE TABLE IF NOT EXISTS pipelines(
|
|
id TEXT PRIMARY KEY,
|
|
order_id INTEGER,
|
|
enabled BOOLEAN,
|
|
created_by TEXT,
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|
name VARCHAR(400) NOT NULL,
|
|
alias VARCHAR(20) NOT NULL,
|
|
description TEXT,
|
|
filter TEXT NOT NULL,
|
|
config_json TEXT
|
|
);`); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (migration *addPipelines) Down(ctx context.Context, db *bun.DB) error {
|
|
return nil
|
|
}
|