mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-06-04 11:25:52 +08:00
56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package sqlmigration
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/SigNoz/signoz/pkg/factory"
|
|
"github.com/uptrace/bun"
|
|
"github.com/uptrace/bun/migrate"
|
|
)
|
|
|
|
type addPipelines struct{}
|
|
|
|
func NewAddPipelinesFactory() factory.ProviderFactory[SQLMigration, Config] {
|
|
return factory.NewProviderFactory(factory.MustNewName("add_pipelines"), newAddPipelines)
|
|
}
|
|
|
|
func newAddPipelines(_ context.Context, _ factory.ProviderSettings, _ Config) (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.NewCreateTable().
|
|
Model(&struct {
|
|
bun.BaseModel `bun:"table:pipelines"`
|
|
ID string `bun:"id,pk,type:text"`
|
|
OrderID int `bun:"order_id"`
|
|
Enabled bool `bun:"enabled"`
|
|
CreatedBy string `bun:"created_by,type:text"`
|
|
CreatedAt time.Time `bun:"created_at,default:current_timestamp"`
|
|
Name string `bun:"name,type:varchar(400),notnull"`
|
|
Alias string `bun:"alias,type:varchar(20),notnull"`
|
|
Description string `bun:"description,type:text"`
|
|
Filter string `bun:"filter,type:text,notnull"`
|
|
ConfigJSON string `bun:"config_json,type:text"`
|
|
}{}).
|
|
IfNotExists().
|
|
Exec(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (migration *addPipelines) Down(ctx context.Context, db *bun.DB) error {
|
|
return nil
|
|
}
|