// Code generated by goctl. DO NOT EDIT. package model import ( "context" "database/sql" "fmt" "strings" "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 ( roleFieldNames = builder.RawFieldNames(&Role{}) roleRows = strings.Join(roleFieldNames, ",") roleRowsExpectAutoSet = strings.Join(stringx.Remove(roleFieldNames, "`id`"), ",") roleRowsWithPlaceHolder = strings.Join(stringx.Remove(roleFieldNames, "`id`"), "=?,") + "=?" ) type ( roleModel interface { Insert(ctx context.Context, data *Role) (sql.Result, error) FindOne(ctx context.Context, id int64) (*Role, error) FindOneByDisplayName(ctx context.Context, displayName string) (*Role, error) FindOneByRoleId(ctx context.Context, roleId string) (*Role, error) Update(ctx context.Context, data *Role) error Delete(ctx context.Context, id int64) error } defaultRoleModel struct { conn sqlx.SqlConn table string } Role struct { Id int64 `db:"id"` // PK RoleId string `db:"role_id"` DisplayName string `db:"display_name"` // 名稱 Status int64 `db:"status"` // 狀態 1: 啟用, 2: 禁用 CreateTime int64 `db:"create_time"` // 創建時間 UpdateTime int64 `db:"update_time"` // 更新時間 } ) func newRoleModel(conn sqlx.SqlConn) *defaultRoleModel { return &defaultRoleModel{ conn: conn, table: "`role`", } } func (m *defaultRoleModel) withSession(session sqlx.Session) *defaultRoleModel { return &defaultRoleModel{ conn: sqlx.NewSqlConnFromSession(session), table: "`role`", } } func (m *defaultRoleModel) 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 *defaultRoleModel) FindOne(ctx context.Context, id int64) (*Role, error) { query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", roleRows, m.table) var resp Role 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 *defaultRoleModel) FindOneByDisplayName(ctx context.Context, displayName string) (*Role, error) { var resp Role query := fmt.Sprintf("select %s from %s where `display_name` = ? limit 1", roleRows, m.table) err := m.conn.QueryRowCtx(ctx, &resp, query, displayName) switch err { case nil: return &resp, nil case sqlc.ErrNotFound: return nil, ErrNotFound default: return nil, err } } func (m *defaultRoleModel) FindOneByRoleId(ctx context.Context, roleId string) (*Role, error) { var resp Role query := fmt.Sprintf("select %s from %s where `role_id` = ? limit 1", roleRows, m.table) err := m.conn.QueryRowCtx(ctx, &resp, query, roleId) switch err { case nil: return &resp, nil case sqlc.ErrNotFound: return nil, ErrNotFound default: return nil, err } } func (m *defaultRoleModel) Insert(ctx context.Context, data *Role) (sql.Result, error) { query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?)", m.table, roleRowsExpectAutoSet) ret, err := m.conn.ExecCtx(ctx, query, data.RoleId, data.DisplayName, data.Status, data.CreateTime, data.UpdateTime) return ret, err } func (m *defaultRoleModel) Update(ctx context.Context, newData *Role) error { query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, roleRowsWithPlaceHolder) _, err := m.conn.ExecCtx(ctx, query, newData.RoleId, newData.DisplayName, newData.Status, newData.CreateTime, newData.UpdateTime, newData.Id) return err } func (m *defaultRoleModel) tableName() string { return m.table }