101 lines
3.9 KiB
Go
101 lines
3.9 KiB
Go
// Code generated by goctl. DO NOT EDIT.
|
||
|
||
package model
|
||
|
||
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 (
|
||
taskFieldNames = builder.RawFieldNames(&Task{})
|
||
taskRows = strings.Join(taskFieldNames, ",")
|
||
taskRowsExpectAutoSet = strings.Join(stringx.Remove(taskFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",")
|
||
taskRowsWithPlaceHolder = strings.Join(stringx.Remove(taskFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?"
|
||
)
|
||
|
||
type (
|
||
taskModel interface {
|
||
Insert(ctx context.Context, data *Task) (sql.Result, error)
|
||
FindOne(ctx context.Context, id int64) (*Task, error)
|
||
Update(ctx context.Context, data *Task) error
|
||
Delete(ctx context.Context, id int64) error
|
||
}
|
||
|
||
defaultTaskModel struct {
|
||
conn sqlx.SqlConn
|
||
table string
|
||
}
|
||
|
||
Task struct {
|
||
Id int64 `db:"id"` // id
|
||
Name string `db:"name"` // 作业名称
|
||
Description string `db:"description"` // 作业描述
|
||
Status string `db:"status"` // 作业状态
|
||
Strategy int64 `db:"strategy"` // 策略
|
||
SynergyStatus int64 `db:"synergy_status"` // 协同状态（0-未协同、1-已协同）
|
||
StartTime time.Time `db:"start_time"` // 开始运行时间
|
||
EndTime string `db:"end_time"` // 结束运行时间
|
||
RunningTime int64 `db:"running_time"` // 已运行时间（单位秒）
|
||
YamlString string `db:"yaml_string"`
|
||
Result string `db:"result"` // 作业结果
|
||
CreatedBy int64 `db:"created_by"` // 创建人
|
||
CreatedTime time.Time `db:"created_time"` // 创建时间
|
||
UpdatedBy int64 `db:"updated_by"` // 更新人
|
||
UpdatedTime time.Time `db:"updated_time"` // 更新时间
|
||
DeletedFlag int64 `db:"deleted_flag"` // 是否删除（0-否，1-是）
|
||
ServiceName string `db:"service_name"`
|
||
}
|
||
)
|
||
|
||
func newTaskModel(conn sqlx.SqlConn) *defaultTaskModel {
|
||
return &defaultTaskModel{
|
||
conn: conn,
|
||
table: "`task`",
|
||
}
|
||
}
|
||
|
||
func (m *defaultTaskModel) 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 *defaultTaskModel) FindOne(ctx context.Context, id int64) (*Task, error) {
|
||
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", taskRows, m.table)
|
||
var resp Task
|
||
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 *defaultTaskModel) Insert(ctx context.Context, data *Task) (sql.Result, error) {
|
||
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, taskRowsExpectAutoSet)
|
||
ret, err := m.conn.ExecCtx(ctx, query, data.Name, data.Description, data.Status, data.Strategy, data.SynergyStatus, data.StartTime, data.EndTime, data.RunningTime, data.YamlString, data.Result, data.CreatedBy, data.CreatedTime, data.UpdatedBy, data.UpdatedTime, data.DeletedFlag, data.ServiceName)
|
||
return ret, err
|
||
}
|
||
|
||
func (m *defaultTaskModel) Update(ctx context.Context, data *Task) error {
|
||
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, taskRowsWithPlaceHolder)
|
||
_, err := m.conn.ExecCtx(ctx, query, data.Name, data.Description, data.Status, data.Strategy, data.SynergyStatus, data.StartTime, data.EndTime, data.RunningTime, data.YamlString, data.Result, data.CreatedBy, data.CreatedTime, data.UpdatedBy, data.UpdatedTime, data.DeletedFlag, data.ServiceName, data.Id)
|
||
return err
|
||
}
|
||
|
||
func (m *defaultTaskModel) tableName() string {
|
||
return m.table
|
||
}
|