diff --git a/pkg/models/taskaimodel.go b/pkg/models/taskaimodel.go new file mode 100644 index 000000000..8db14d5bd --- /dev/null +++ b/pkg/models/taskaimodel.go @@ -0,0 +1,24 @@ +package models + +import "github.com/zeromicro/go-zero/core/stores/sqlx" + +var _ TaskAiModel = (*customTaskAiModel)(nil) + +type ( + // TaskAiModel is an interface to be customized, add more methods here, + // and implement the added methods in customTaskAiModel. + TaskAiModel interface { + taskAiModel + } + + customTaskAiModel struct { + *defaultTaskAiModel + } +) + +// NewTaskAiModel returns a model for the database table. +func NewTaskAiModel(conn sqlx.SqlConn) TaskAiModel { + return &customTaskAiModel{ + defaultTaskAiModel: newTaskAiModel(conn), + } +} diff --git a/pkg/models/taskaimodel_gen.go b/pkg/models/taskaimodel_gen.go new file mode 100644 index 000000000..026eea1d8 --- /dev/null +++ b/pkg/models/taskaimodel_gen.go @@ -0,0 +1,103 @@ +// Code generated by goctl. DO NOT EDIT. + +package models + +import ( + "context" + "database/sql" + "fmt" + "strings" + "time" + + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + taskAiFieldNames = builder.RawFieldNames(&TaskAi{}) + taskAiRows = strings.Join(taskAiFieldNames, ",") + taskAiRowsExpectAutoSet = strings.Join(stringx.Remove(taskAiFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") + taskAiRowsWithPlaceHolder = strings.Join(stringx.Remove(taskAiFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" +) + +type ( + taskAiModel interface { + Insert(ctx context.Context, data *TaskAi) (sql.Result, error) + FindOne(ctx context.Context, id int64) (*TaskAi, error) + Update(ctx context.Context, data *TaskAi) error + Delete(ctx context.Context, id int64) error + } + + defaultTaskAiModel struct { + conn sqlx.SqlConn + table string + } + + TaskAi struct { + Id int64 `db:"id"` // id + TaskId int64 `db:"task_id"` // 任务id + AdapterId int64 `db:"adapter_id"` // 设配器id + ClusterId int64 `db:"cluster_id"` // 集群id + Name string `db:"name"` // 任务名 + Replica int64 `db:"replica"` // 执行数 + CTaskId string `db:"c_task_id"` // 集群返回任务id + Strategy string `db:"strategy"` // 主任务使用策略 + Status string `db:"status"` // 任务状态 + Msg sql.NullString `db:"msg"` // 集群返回任务信息 + CommitTime time.Time `db:"commit_time"` // 提交时间 + StartTime time.Time `db:"start_time"` // 开始时间 + EndTime time.Time `db:"end_time"` // 结束时间 + } +) + +func newTaskAiModel(conn sqlx.SqlConn) *defaultTaskAiModel { + return &defaultTaskAiModel{ + conn: conn, + table: "`task_ai`", + } +} + +func (m *defaultTaskAiModel) withSession(session sqlx.Session) *defaultTaskAiModel { + return &defaultTaskAiModel{ + conn: sqlx.NewSqlConnFromSession(session), + table: "`task_ai`", + } +} + +func (m *defaultTaskAiModel) Delete(ctx context.Context, id int64) error { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + _, err := m.conn.ExecCtx(ctx, query, id) + return err +} + +func (m *defaultTaskAiModel) FindOne(ctx context.Context, id int64) (*TaskAi, error) { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", taskAiRows, m.table) + var resp TaskAi + err := m.conn.QueryRowCtx(ctx, &resp, query, id) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultTaskAiModel) Insert(ctx context.Context, data *TaskAi) (sql.Result, error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, taskAiRowsExpectAutoSet) + ret, err := m.conn.ExecCtx(ctx, query, data.TaskId, data.AdapterId, data.ClusterId, data.Name, data.Replica, data.CTaskId, data.Strategy, data.Status, data.Msg, data.CommitTime, data.StartTime, data.EndTime) + return ret, err +} + +func (m *defaultTaskAiModel) Update(ctx context.Context, data *TaskAi) error { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, taskAiRowsWithPlaceHolder) + _, err := m.conn.ExecCtx(ctx, query, data.TaskId, data.AdapterId, data.ClusterId, data.Name, data.Replica, data.CTaskId, data.Strategy, data.Status, data.Msg, data.CommitTime, data.StartTime, data.EndTime, data.Id) + return err +} + +func (m *defaultTaskAiModel) tableName() string { + return m.table +}