diff --git a/pkg/models/taskaimodel_gen.go b/pkg/models/taskaimodel_gen.go index ab0c5502..34ecd67b 100644 --- a/pkg/models/taskaimodel_gen.go +++ b/pkg/models/taskaimodel_gen.go @@ -49,6 +49,7 @@ type ( CommitTime time.Time `db:"commit_time"` // 提交时间 StartTime string `db:"start_time"` // 开始时间 EndTime string `db:"end_time"` // 结束时间 + TaskType string `db:"task_type"` } ) @@ -87,14 +88,14 @@ func (m *defaultTaskAiModel) FindOne(ctx context.Context, id int64) (*TaskAi, er } 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.JobId, data.Strategy, data.Status, data.Msg, data.CommitTime, data.StartTime, data.EndTime) + 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.JobId, data.Strategy, data.Status, data.Msg, data.CommitTime, data.StartTime, data.EndTime, data.TaskType) 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.JobId, data.Strategy, data.Status, data.Msg, data.CommitTime, data.StartTime, data.EndTime, data.Id) + _, err := m.conn.ExecCtx(ctx, query, data.TaskId, data.AdapterId, data.ClusterId, data.Name, data.Replica, data.JobId, data.Strategy, data.Status, data.Msg, data.CommitTime, data.StartTime, data.EndTime, data.TaskType, data.Id) return err } diff --git a/pkg/models/tclustertaskqueuemodel.go b/pkg/models/tclustertaskqueuemodel.go new file mode 100644 index 00000000..35e52522 --- /dev/null +++ b/pkg/models/tclustertaskqueuemodel.go @@ -0,0 +1,24 @@ +package models + +import "github.com/zeromicro/go-zero/core/stores/sqlx" + +var _ TClusterTaskQueueModel = (*customTClusterTaskQueueModel)(nil) + +type ( + // TClusterTaskQueueModel is an interface to be customized, add more methods here, + // and implement the added methods in customTClusterTaskQueueModel. + TClusterTaskQueueModel interface { + tClusterTaskQueueModel + } + + customTClusterTaskQueueModel struct { + *defaultTClusterTaskQueueModel + } +) + +// NewTClusterTaskQueueModel returns a model for the database table. +func NewTClusterTaskQueueModel(conn sqlx.SqlConn) TClusterTaskQueueModel { + return &customTClusterTaskQueueModel{ + defaultTClusterTaskQueueModel: newTClusterTaskQueueModel(conn), + } +} diff --git a/pkg/models/tclustertaskqueuemodel_gen.go b/pkg/models/tclustertaskqueuemodel_gen.go new file mode 100644 index 00000000..9113c7dc --- /dev/null +++ b/pkg/models/tclustertaskqueuemodel_gen.go @@ -0,0 +1,95 @@ +// 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 ( + tClusterTaskQueueFieldNames = builder.RawFieldNames(&TClusterTaskQueue{}) + tClusterTaskQueueRows = strings.Join(tClusterTaskQueueFieldNames, ",") + tClusterTaskQueueRowsExpectAutoSet = strings.Join(stringx.Remove(tClusterTaskQueueFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") + tClusterTaskQueueRowsWithPlaceHolder = strings.Join(stringx.Remove(tClusterTaskQueueFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" +) + +type ( + tClusterTaskQueueModel interface { + Insert(ctx context.Context, data *TClusterTaskQueue) (sql.Result, error) + FindOne(ctx context.Context, id int64) (*TClusterTaskQueue, error) + Update(ctx context.Context, data *TClusterTaskQueue) error + Delete(ctx context.Context, id int64) error + } + + defaultTClusterTaskQueueModel struct { + conn sqlx.SqlConn + table string + } + + TClusterTaskQueue struct { + Id int64 `db:"id"` // id + AdapterId int64 `db:"adapter_id"` // 适配器id + ClusterId int64 `db:"cluster_id"` // 集群id + QueueNum sql.NullInt64 `db:"queue_num"` // 任务排队数量 + Date time.Time `db:"date"` + } +) + +func newTClusterTaskQueueModel(conn sqlx.SqlConn) *defaultTClusterTaskQueueModel { + return &defaultTClusterTaskQueueModel{ + conn: conn, + table: "`t_cluster_task_queue`", + } +} + +func (m *defaultTClusterTaskQueueModel) withSession(session sqlx.Session) *defaultTClusterTaskQueueModel { + return &defaultTClusterTaskQueueModel{ + conn: sqlx.NewSqlConnFromSession(session), + table: "`t_cluster_task_queue`", + } +} + +func (m *defaultTClusterTaskQueueModel) 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 *defaultTClusterTaskQueueModel) FindOne(ctx context.Context, id int64) (*TClusterTaskQueue, error) { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", tClusterTaskQueueRows, m.table) + var resp TClusterTaskQueue + 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 *defaultTClusterTaskQueueModel) Insert(ctx context.Context, data *TClusterTaskQueue) (sql.Result, error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?)", m.table, tClusterTaskQueueRowsExpectAutoSet) + ret, err := m.conn.ExecCtx(ctx, query, data.AdapterId, data.ClusterId, data.QueueNum, data.Date) + return ret, err +} + +func (m *defaultTClusterTaskQueueModel) Update(ctx context.Context, data *TClusterTaskQueue) error { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, tClusterTaskQueueRowsWithPlaceHolder) + _, err := m.conn.ExecCtx(ctx, query, data.AdapterId, data.ClusterId, data.QueueNum, data.Date, data.Id) + return err +} + +func (m *defaultTClusterTaskQueueModel) tableName() string { + return m.table +}