mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-06-04 11:25:52 +08:00

* feat(sqlmigration): add sqlmigrations * feat(sqlmigration): test sqlmigrations * feat(sqlmigration): add remaining factories * feat(sqlmigration): consolidate into single package * fix(telemetrystore): remove existing env variables * fix(telemetrystore): fix DSN
68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package sqlmigration
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/uptrace/bun"
|
|
"github.com/uptrace/bun/migrate"
|
|
"go.signoz.io/signoz/pkg/factory"
|
|
)
|
|
|
|
type addIntegrations struct{}
|
|
|
|
func NewAddIntegrationsFactory() factory.ProviderFactory[SQLMigration, Config] {
|
|
return factory.NewProviderFactory(factory.MustNewName("add_integrations"), newAddIntegrations)
|
|
}
|
|
|
|
func newAddIntegrations(_ context.Context, _ factory.ProviderSettings, _ Config) (SQLMigration, error) {
|
|
return &addIntegrations{}, nil
|
|
}
|
|
|
|
func (migration *addIntegrations) Register(migrations *migrate.Migrations) error {
|
|
if err := migrations.Register(migration.Up, migration.Down); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (migration *addIntegrations) Up(ctx context.Context, db *bun.DB) error {
|
|
if _, err := db.ExecContext(ctx, `CREATE TABLE IF NOT EXISTS integrations_installed(
|
|
integration_id TEXT PRIMARY KEY,
|
|
config_json TEXT,
|
|
installed_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
);`); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := db.ExecContext(ctx, `CREATE TABLE IF NOT EXISTS cloud_integrations_accounts(
|
|
cloud_provider TEXT NOT NULL,
|
|
id TEXT NOT NULL,
|
|
config_json TEXT,
|
|
cloud_account_id TEXT,
|
|
last_agent_report_json TEXT,
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL,
|
|
removed_at TIMESTAMP,
|
|
UNIQUE(cloud_provider, id)
|
|
)`); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := db.ExecContext(ctx, `CREATE TABLE IF NOT EXISTS cloud_integrations_service_configs(
|
|
cloud_provider TEXT NOT NULL,
|
|
cloud_account_id TEXT NOT NULL,
|
|
service_id TEXT NOT NULL,
|
|
config_json TEXT,
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL,
|
|
UNIQUE(cloud_provider, cloud_account_id, service_id)
|
|
)`); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (migration *addIntegrations) Down(ctx context.Context, db *bun.DB) error {
|
|
return nil
|
|
}
|