package model

import (
	"fmt"

	"git.x2erp.com/qdy/go-base/ctx"
	"git.x2erp.com/qdy/go-base/model/bson"
)

// 租户配置信息
type ETLConfig struct {
	bson.BsonStringModel `bson:",inline"`
	AgentDatabaseId      string `bson:"agent_database_id,omitempty"`
	AgentChangeCountSQL  string `bson:"agent_change_count_sql,omitempty"`
	AgentChangeQuerySQL  string `bson:"agent_change_query_sql,omitempty"`
	AgentFullCountSQL    string `bson:"agent_full_count_sql,omitempty"`
	AgentFullQuerySQL    string `bson:"agent_full_query_sql,omitempty"`
	DorisDatabaseName    string `bson:"doris_database_name,omitempty"`
	DorisTabelNsme       string `bson:"doris_tabel_nsme,omitempty"`
}

// 构造函数，设置默认值
func NewETLConfig(ctx *ctx.RequestContext, agentDatabaseId, dorisDatabaseName, dorisTableName string) *ETLConfig {
	id := fmt.Sprintf("%s.%s.%s", ctx.TenantID, dorisDatabaseName, dorisTableName)
	return &ETLConfig{
		BsonStringModel:   bson.NewStringIDModelWithID(ctx, id),
		AgentDatabaseId:   agentDatabaseId,
		DorisDatabaseName: dorisDatabaseName,
		DorisTabelNsme:    dorisTableName,
	}
}