106 lines
3.6 KiB
Go
Executable File
106 lines
3.6 KiB
Go
Executable File
// Code generated by goctl. DO NOT EDIT.
|
||
// versions:
|
||
// goctl version: 1.7.3
|
||
|
||
package model
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"fmt"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/zeromicro/go-zero/core/stores/builder"
|
||
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
||
"github.com/zeromicro/go-zero/core/stringx"
|
||
)
|
||
|
||
var (
|
||
nhTaskAmbassadorFieldNames = builder.RawFieldNames(&NhTaskAmbassador{})
|
||
nhTaskAmbassadorRows = strings.Join(nhTaskAmbassadorFieldNames, ",")
|
||
nhTaskAmbassadorRowsExpectAutoSet = strings.Join(stringx.Remove(nhTaskAmbassadorFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",")
|
||
nhTaskAmbassadorRowsWithPlaceHolder = strings.Join(stringx.Remove(nhTaskAmbassadorFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?"
|
||
)
|
||
|
||
type (
|
||
nhTaskAmbassadorModel interface {
|
||
Insert(ctx context.Context, data *NhTaskAmbassador) (sql.Result, error)
|
||
FindOne(ctx context.Context, id uint) (*NhTaskAmbassador, error)
|
||
FindOneByEmailStage(ctx context.Context, email string, stage int8) (*NhTaskAmbassador, error)
|
||
Update(ctx context.Context, data *NhTaskAmbassador) error
|
||
Delete(ctx context.Context, id uint) error
|
||
}
|
||
|
||
defaultNhTaskAmbassadorModel struct {
|
||
conn sqlx.SqlConn
|
||
table string
|
||
}
|
||
|
||
NhTaskAmbassador struct {
|
||
Id uint `db:"id"`
|
||
Uid uint `db:"uid"` // 用户ID
|
||
Email string `db:"email"` // 大使邮箱
|
||
Stage int8 `db:"stage"` // 0=一期，1=二期
|
||
CreatedAt time.Time `db:"created_at"` // 创建时间
|
||
UpdatedAt time.Time `db:"updated_at"` // 修改时间
|
||
}
|
||
)
|
||
|
||
func newNhTaskAmbassadorModel(conn sqlx.SqlConn) *defaultNhTaskAmbassadorModel {
|
||
return &defaultNhTaskAmbassadorModel{
|
||
conn: conn,
|
||
table: "`nh_task_ambassador`",
|
||
}
|
||
}
|
||
|
||
func (m *defaultNhTaskAmbassadorModel) Delete(ctx context.Context, id uint) error {
|
||
query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
|
||
_, err := m.conn.ExecCtx(ctx, query, id)
|
||
return err
|
||
}
|
||
|
||
func (m *defaultNhTaskAmbassadorModel) FindOne(ctx context.Context, id uint) (*NhTaskAmbassador, error) {
|
||
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", nhTaskAmbassadorRows, m.table)
|
||
var resp NhTaskAmbassador
|
||
err := m.conn.QueryRowCtx(ctx, &resp, query, id)
|
||
switch err {
|
||
case nil:
|
||
return &resp, nil
|
||
case sqlx.ErrNotFound:
|
||
return nil, ErrNotFound
|
||
default:
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
func (m *defaultNhTaskAmbassadorModel) FindOneByEmailStage(ctx context.Context, email string, stage int8) (*NhTaskAmbassador, error) {
|
||
var resp NhTaskAmbassador
|
||
query := fmt.Sprintf("select %s from %s where `email` = ? and `stage` = ? limit 1", nhTaskAmbassadorRows, m.table)
|
||
err := m.conn.QueryRowCtx(ctx, &resp, query, email, stage)
|
||
switch err {
|
||
case nil:
|
||
return &resp, nil
|
||
case sqlx.ErrNotFound:
|
||
return nil, ErrNotFound
|
||
default:
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
func (m *defaultNhTaskAmbassadorModel) Insert(ctx context.Context, data *NhTaskAmbassador) (sql.Result, error) {
|
||
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?)", m.table, nhTaskAmbassadorRowsExpectAutoSet)
|
||
ret, err := m.conn.ExecCtx(ctx, query, data.Uid, data.Email, data.Stage)
|
||
return ret, err
|
||
}
|
||
|
||
func (m *defaultNhTaskAmbassadorModel) Update(ctx context.Context, newData *NhTaskAmbassador) error {
|
||
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, nhTaskAmbassadorRowsWithPlaceHolder)
|
||
_, err := m.conn.ExecCtx(ctx, query, newData.Uid, newData.Email, newData.Stage, newData.Id)
|
||
return err
|
||
}
|
||
|
||
func (m *defaultNhTaskAmbassadorModel) tableName() string {
|
||
return m.table
|
||
}
|