db

type Config struct {
	Type            string `json:"type,omitempty"`
	Host            string `json:"host,omitempty"`
	Port            int    `json:"port,omitempty"`
	Database        string `json:"database,omitempty"`
	DbName          string `json:"dbName,omitempty"`
	Username        string `json:"username,omitempty"`
	Password        string `json:"password,omitempty"`
	OdbcDsn         string `json:"odbcDsn,omitempty"`
	OdbcDialectName string `json:"odbcDialectName,omitempty"`

	Schema       string      `json:"schema,omitempty"`
	Sid          string      `json:"sid,omitempty"`
	MaxIdleConn  int         `json:"maxIdleConn,omitempty"`
	MaxOpenConn  int         `json:"maxOpenConn,omitempty"`
	DatabasePath string      `json:"databasePath,omitempty"`
	SSHClient    *ssh.Client `json:"-"`
	Dsn          string      `json:"dsn,omitempty"`

	TlsConfig     string `json:"tlsConfig,omitempty"`
	TlsRootCert   string `json:"tlsRootCert,omitempty"`
	TlsClientCert string `json:"tlsClientCert,omitempty"`
	TlsClientKey  string `json:"tlsClientKey,omitempty"`
}

type DataListResult struct {
	Sql      string                   `json:"sql"`
	Total    int                      `json:"total"`
	Args     []interface{}            `json:"args"`
	DataList []map[string]interface{} `json:"dataList"`
}

type DatabaseType struct {
	DialectName string                                       `json:"dialectName"`
	NewDb       func(config *Config) (db *sql.DB, err error) `json:"-"`

	Matches []string `json:"-"`
	// contains filtered or unexported fields
}

type ExecuteOptions struct {
	SelectDataMax int  `json:"selectDataMax"`
	OpenProfiling bool `json:"openProfiling"`
}

type IService interface {
	// Close 关闭 db 客户端
	Close()
	// GetConfig 获取 数据库 配置
	GetConfig() Config
	// GetDialect 获取 方言
	GetDialect() dialect.Dialect
	// GetDb 获取 sql.DB
	GetDb() *sql.DB

	// Info 查询数据库信息
	Info() (res interface{}, err error)

	// Exec 执行 SQL
	Exec(sql string, args []interface{}) (rowsAffected int64, err error)
	// Execs 批量执行 SQL
	Execs(sqlList []string, argsList [][]interface{}) (rowsAffected int64, err error)
	// Count 统计查询 SQL
	Count(sql string, args []interface{}) (count int64, err error)
	// QueryOne 查询 单个 结构体 SQL  需要 传入 结构体 接收
	QueryOne(sql string, args []interface{}, one interface{}) (find bool, err error)
	// Query 查询 列表 结构体 SQL  需要 传入 结构体 接收
	Query(sql string, args []interface{}, list interface{}) (err error)
	// QueryMap 查询 列表 map SQL  返回 list < map >
	QueryMap(sql string, args []interface{}) (list []map[string]interface{}, err error)
	// QueryPage 分页查询 列表 结构体 SQL  需要 传入 结构体 接收
	QueryPage(sql string, args []interface{}, list interface{}, page *worker.Page) (err error)
	// QueryMapPage 分页查询 列表 map 列表 SQL  返回 list < map >
	QueryMapPage(sql string, args []interface{}, page *worker.Page) (list []map[string]interface{}, err error)

	// OwnerCreateSql 创建 库|模式|用户等 SQL 取决于哪种数据库
	OwnerCreateSql(param *Param, owner *dialect.OwnerModel) (sqlList []string, err error)
	// OwnersSelect 查询 库|模式|用户等 取决于哪种数据库
	OwnersSelect(param *Param) (owners []*dialect.OwnerModel, err error)

	// TablesSelect 查询 库|模式|用户 下所有表
	TablesSelect(param *Param, ownerName string) (tables []*dialect.TableModel, err error)
	// TableDetail 查询 某个表明细
	TableDetail(param *Param, ownerName string, tableName string) (tableDetail *dialect.TableModel, err error)

	// OwnerCreate 创建 库|模式|用户等 取决于哪种数据库
	OwnerCreate(param *Param, owner *dialect.OwnerModel) (created bool, err error)
	// OwnerDelete 删除 库|模式|用户等 取决于哪种数据库
	OwnerDelete(param *Param, ownerName string) (deleted bool, err error)
	// OwnerDataTrim 清空 库|模式|用户等 数据 取决于哪种数据库
	OwnerDataTrim(param *Param, ownerName string) (err error)
	// DDL 查看 库 或 表 SQL语句
	DDL(param *Param, ownerName string, tableName string) (sqlList []string, err error)
	// Model 将 表 转换某些语言的模型
	Model(param *Param, ownerName string, tableName string) (content string, err error)

	// TableCreate 创建表
	TableCreate(param *Param, ownerName string, table *dialect.TableModel) (err error)
	// TableDelete 删除表
	TableDelete(param *Param, ownerName string, tableName string) (err error)
	// TableDataTrim 表数据清理
	TableDataTrim(param *Param, ownerName string, tableName string) (err error)
	// TableCreateSql 获取建表 SQL
	TableCreateSql(param *Param, ownerName string, table *dialect.TableModel) (sqlList []string, err error)
	// TableUpdateSql 获取更新表的 SQL
	TableUpdateSql(param *Param, ownerName string, tableName string, updateTableParam *UpdateTableParam) (sqlList []string, err error)
	// TableUpdate 更新表
	TableUpdate(param *Param, ownerName string, tableName string, updateTableParam *UpdateTableParam) (err error)

	// DataListSql 将 需要 新增、修改、删除的 数据 转为 SQL
	DataListSql(param *Param, ownerName string, tableName string, columnList []*dialect.ColumnModel,
		insertDataList []map[string]interface{},
		updateDataList []map[string]interface{}, updateWhereDataList []map[string]interface{},
		deleteDataList []map[string]interface{},
	) (sqlList []string, err error)
	// DataListExec 执行 需要 新增、修改、删除的 数据
	DataListExec(param *Param, ownerName string, tableName string, columnList []*dialect.ColumnModel,
		insertDataList []map[string]interface{},
		updateDataList []map[string]interface{}, updateWhereDataList []map[string]interface{},
		deleteDataList []map[string]interface{},
	) (err error)

	// TableData 根据 一些 条件 查询 表数据
	TableData(param *Param, ownerName string, tableName string, columnList []*dialect.ColumnModel, whereList []*dialect.Where, orderList []*dialect.Order, pageSize int, pageNo int) (dataListResult DataListResult, err error)

	// GetTargetDialect 获取 目标数据库方言
	GetTargetDialect(param *Param) (dia dialect.Dialect)

	// ExecuteSQL 执行某段 SQL
	ExecuteSQL(param *Param, ownerName string, sqlContent string, options *ExecuteOptions) (executeList []map[string]interface{}, errStr string, err error)

	// StartExport 开始 导出
	StartExport(param *Param, exportParam *worker.TaskExportParam) (task *worker.Task, err error)
	// StartImport 开始 导入
	StartImport(param *Param, importParam *worker.TaskImportParam) (task *worker.Task, err error)
	// StartSync 开始 同步
	StartSync(param *Param, syncParam *worker.TaskSyncParam) (task *worker.Task, err error)

	// NewTestExecutor 新建测试任务
	NewTestExecutor(options *TestTaskOptions) (testExecutor *TestExecutor, err error)
}

type Param struct {
	*dialect.ParamModel
	TargetDatabaseType   string  `json:"targetDatabaseType"`
	AppendOwnerCreateSql bool    `json:"appendOwnerCreateSql"`
	AppendOwnerName      bool    `json:"appendOwnerName"`
	OpenTransaction      bool    `json:"openTransaction"`
	ErrorContinue        bool    `json:"errorContinue"`
	ExecUsername         string  `json:"execUsername"`
	ExecPassword         string  `json:"execPassword"`
	ExportType           string  `json:"exportType"`
	ImportType           string  `json:"importType"`
	TargetDatabaseConfig *Config `json:"targetDatabaseConfig"`
	ModelType            string  `json:"modelType"`
	Charset              string  `json:"charset"`

	FormatIndexName bool `json:"formatIndexName"`
}

type ScriptVar struct {
	Name  string      `json:"name,omitempty"`
	Value interface{} `json:"value,omitempty"`
}

type Service struct {
	dialect.Dialect
	// contains filtered or unexported fields
}

type SqlConditionalOperation struct {
	Value string `json:"value,omitempty"`
	Text  string `json:"text,omitempty"`
}

type TestExecutor struct {
	*TestTaskOptions
	// contains filtered or unexported fields
}

type TestTask struct {
}

type TestTaskOptions struct {
	*Param
	OwnerName string `json:"ownerName"`

	Username      string                                                `json:"username,omitempty"`
	Password      string                                                `json:"password,omitempty"`
	IsBatch       bool                                                  `json:"isBatch,omitempty"`
	BatchSize     int                                                   `json:"batchSize,omitempty"`
	TestSql       string                                                `json:"testSql,omitempty"`
	FormatSqlList func(sqlList *[]string, sqlArgsList *[][]interface{}) `json:"-"`
	OnExec        func(sqlList *[]string, sqlArgsList *[][]interface{}) `json:"-"`
	ScriptVars    []*ScriptVar                                          `json:"scriptVars,omitempty"`
	MaxIdleConn   int                                                   `json:"maxIdleConn,omitempty"`
	MaxOpenConn   int                                                   `json:"maxOpenConn,omitempty"`
}

type TestWorkerParam struct {
	*TestExecutor
	SqlList     []string
	SqlArgsList [][]interface{}
	// contains filtered or unexported fields
}

type UpdateTableColumn struct {
	*dialect.ColumnModel
	OldColumn *dialect.ColumnModel `json:"oldColumn"`
	Deleted   bool                 `json:"deleted"`
}

type UpdateTableIndex struct {
	*dialect.IndexModel
	OldIndex *dialect.IndexModel `json:"oldIndex"`
	Deleted  bool                `json:"deleted"`
}

type UpdateTableParam struct {
	TableComment    string               `json:"tableComment"`
	OldTableComment string               `json:"oldTableComment"`
	ColumnList      []*UpdateTableColumn `json:"columnList"`
	IndexList       []*UpdateTableIndex  `json:"indexList"`
}