// Code generated by goctl. DO NOT EDIT. // versions: // goctl version: 1.10.0 package role import ( "context" "database/sql" "fmt" "strings" "github.com/zeromicro/go-zero/core/stores/builder" "github.com/zeromicro/go-zero/core/stores/cache" "github.com/zeromicro/go-zero/core/stores/sqlc" "github.com/zeromicro/go-zero/core/stores/sqlx" "github.com/zeromicro/go-zero/core/stringx" ) var ( sysRoleFieldNames = builder.RawFieldNames(&SysRole{}) sysRoleRows = strings.Join(sysRoleFieldNames, ",") sysRoleRowsExpectAutoSet = strings.Join(stringx.Remove(sysRoleFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") sysRoleRowsWithPlaceHolder = strings.Join(stringx.Remove(sysRoleFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" cacheSysRoleIdPrefix = "cache:sysRole:id:" cacheSysRoleProductCodeNamePrefix = "cache:sysRole:productCode:name:" ) type ( sysRoleModel interface { Insert(ctx context.Context, data *SysRole) (sql.Result, error) InsertWithTx(ctx context.Context, session sqlx.Session, data *SysRole) (sql.Result, error) BatchInsert(ctx context.Context, dataList []*SysRole) error BatchInsertWithTx(ctx context.Context, session sqlx.Session, dataList []*SysRole) error TransactCtx(ctx context.Context, fn func(context.Context, sqlx.Session) error) error FindOne(ctx context.Context, id int64) (*SysRole, error) FindOneWithTx(ctx context.Context, session sqlx.Session, id int64) (*SysRole, error) FindOneByProductCodeName(ctx context.Context, productCode string, name string) (*SysRole, error) FindOneByProductCodeNameWithTx(ctx context.Context, session sqlx.Session, productCode string, name string) (*SysRole, error) Update(ctx context.Context, data *SysRole) error UpdateWithTx(ctx context.Context, session sqlx.Session, data *SysRole) error BatchUpdate(ctx context.Context, dataList []*SysRole) error BatchUpdateWithTx(ctx context.Context, session sqlx.Session, dataList []*SysRole) error Delete(ctx context.Context, id int64) error DeleteWithTx(ctx context.Context, session sqlx.Session, id int64) error BatchDelete(ctx context.Context, ids []int64) error BatchDeleteWithTx(ctx context.Context, session sqlx.Session, ids []int64) error TableName() string } defaultSysRoleModel struct { sqlc.CachedConn table string } SysRole struct { Id int64 `db:"id"` // ID ProductCode string `db:"productCode"` // 所属产品编码 Name string `db:"name"` // 角色名 Remark string `db:"remark"` // 备注 Status int64 `db:"status"` // 状态 1启用 2禁用 PermsLevel int64 `db:"permsLevel"` // 权限等级,值越大权限越小,用于上下级管控 CreateTime int64 `db:"createTime"` // 创建时间 UpdateTime int64 `db:"updateTime"` // 修改时间 } ) func newSysRoleModel(conn sqlx.SqlConn, c cache.CacheConf, cachePrefix string, opts ...cache.Option) *defaultSysRoleModel { if cachePrefix != "" { cacheSysRoleIdPrefix = cachePrefix + ":" + "cache:sysRole:id:" cacheSysRoleProductCodeNamePrefix = cachePrefix + ":" + "cache:sysRole:productCode:name:" } return &defaultSysRoleModel{ CachedConn: sqlc.NewConn(conn, c, opts...), table: "`sys_role`", } } func (m *defaultSysRoleModel) Delete(ctx context.Context, id int64) error { data, err := m.FindOne(ctx, id) if err != nil { return err } sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { query := fmt.Sprintf("DELETE FROM %s WHERE `id` = ?", m.table) return conn.ExecCtx(ctx, query, id) }, sysRoleIdKey, sysRoleProductCodeNameKey) return err } func (m *defaultSysRoleModel) DeleteWithTx(ctx context.Context, session sqlx.Session, id int64) error { data, err := m.FindOne(ctx, id) if err != nil { return err } sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { query := fmt.Sprintf("DELETE FROM %s WHERE `id` = ?", m.table) return session.ExecCtx(ctx, query, id) }, sysRoleIdKey, sysRoleProductCodeNameKey) return err } func (m *defaultSysRoleModel) BatchDelete(ctx context.Context, ids []int64) error { if len(ids) == 0 { return nil } primaryKeys := make([]interface{}, 0, len(ids)) for _, key := range ids { primaryKeys = append(primaryKeys, key) } oldDataList, err := m.findListByPrimaryKeys(ctx, primaryKeys) if err != nil { return err } keys := make([]string, 0) for _, data := range oldDataList { id := m.getPrimaryKeyValue(data) sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) keys = append(keys, sysRoleIdKey, sysRoleProductCodeNameKey) } placeholders := make([]string, 0, len(ids)) args := make([]interface{}, 0, len(ids)) for _, id := range ids { placeholders = append(placeholders, "?") args = append(args, id) } _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { query := fmt.Sprintf("DELETE FROM %s WHERE `id` IN (%s)", m.table, strings.Join(placeholders, ",")) return conn.ExecCtx(ctx, query, args...) }, keys...) return err } func (m *defaultSysRoleModel) BatchDeleteWithTx(ctx context.Context, session sqlx.Session, ids []int64) error { if len(ids) == 0 { return nil } primaryKeys := make([]interface{}, 0, len(ids)) for _, key := range ids { primaryKeys = append(primaryKeys, key) } oldDataList, err := m.findListByPrimaryKeys(ctx, primaryKeys) if err != nil { return err } keys := make([]string, 0) for _, data := range oldDataList { id := m.getPrimaryKeyValue(data) sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) keys = append(keys, sysRoleIdKey, sysRoleProductCodeNameKey) } placeholders := make([]string, 0, len(ids)) args := make([]interface{}, 0, len(ids)) for _, id := range ids { placeholders = append(placeholders, "?") args = append(args, id) } query := fmt.Sprintf("DELETE FROM %s WHERE `id` IN (%s)", m.table, strings.Join(placeholders, ",")) _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { return session.ExecCtx(ctx, query, args...) }, keys...) return err } func (m *defaultSysRoleModel) FindOne(ctx context.Context, id int64) (*SysRole, error) { sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, id) var resp SysRole err := m.QueryRowCtx(ctx, &resp, sysRoleIdKey, func(ctx context.Context, conn sqlx.SqlConn, v any) error { query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysRoleRows, m.table) return conn.QueryRowCtx(ctx, v, query, id) }) switch err { case nil: return &resp, nil case sqlc.ErrNotFound: return nil, ErrNotFound default: return nil, err } } func (m *defaultSysRoleModel) FindOneWithTx(ctx context.Context, session sqlx.Session, id int64) (*SysRole, error) { var resp SysRole query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysRoleRows, m.table) err := session.QueryRowCtx(ctx, &resp, query, id) switch err { case nil: return &resp, nil case sqlx.ErrNotFound: return nil, ErrNotFound default: return nil, err } } func (m *defaultSysRoleModel) FindOneByProductCodeName(ctx context.Context, productCode string, name string) (*SysRole, error) { sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, productCode, name) var resp SysRole err := m.QueryRowIndexCtx(ctx, &resp, sysRoleProductCodeNameKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v any) (i any, e error) { query := fmt.Sprintf("SELECT %s FROM %s WHERE `productCode` = ? and `name` = ? LIMIT 1", sysRoleRows, m.table) if err := conn.QueryRowCtx(ctx, &resp, query, productCode, name); err != nil { return nil, err } return resp.Id, nil }, m.queryPrimary) switch err { case nil: return &resp, nil case sqlc.ErrNotFound: return nil, ErrNotFound default: return nil, err } } func (m *defaultSysRoleModel) FindOneByProductCodeNameWithTx(ctx context.Context, session sqlx.Session, productCode string, name string) (*SysRole, error) { var resp SysRole query := fmt.Sprintf("SELECT %s FROM %s WHERE `productCode` = ? and `name` = ? LIMIT 1", sysRoleRows, m.table) err := session.QueryRowCtx(ctx, &resp, query, productCode, name) switch err { case nil: return &resp, nil case sqlx.ErrNotFound: return nil, ErrNotFound default: return nil, err } } func (m *defaultSysRoleModel) Insert(ctx context.Context, data *SysRole) (sql.Result, error) { sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, data.Id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { query := fmt.Sprintf("INSERT INTO %s (%s) VALUES (?, ?, ?, ?, ?, ?, ?)", m.table, sysRoleRowsExpectAutoSet) return conn.ExecCtx(ctx, query, data.ProductCode, data.Name, data.Remark, data.Status, data.PermsLevel, data.CreateTime, data.UpdateTime) }, sysRoleIdKey, sysRoleProductCodeNameKey) return ret, err } func (m *defaultSysRoleModel) InsertWithTx(ctx context.Context, session sqlx.Session, data *SysRole) (sql.Result, error) { sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, data.Id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { query := fmt.Sprintf("INSERT INTO %s (%s) VALUES (?, ?, ?, ?, ?, ?, ?)", m.table, sysRoleRowsExpectAutoSet) return session.ExecCtx(ctx, query, data.ProductCode, data.Name, data.Remark, data.Status, data.PermsLevel, data.CreateTime, data.UpdateTime) }, sysRoleIdKey, sysRoleProductCodeNameKey) return ret, err } func (m *defaultSysRoleModel) BatchInsert(ctx context.Context, dataList []*SysRole) error { if len(dataList) == 0 { return nil } keys := make([]string, 0, len(dataList)) valueSets := make([]string, 0, len(dataList)) args := make([]interface{}, 0) for _, data := range dataList { valueSets = append(valueSets, "(?, ?, ?, ?, ?, ?, ?)") args = append(args, data.ProductCode, data.Name, data.Remark, data.Status, data.PermsLevel, data.CreateTime, data.UpdateTime) sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, data.Id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) keys = append(keys, sysRoleIdKey, sysRoleProductCodeNameKey) } query := fmt.Sprintf("INSERT INTO %s (%s) VALUES %s", m.table, sysRoleRowsExpectAutoSet, strings.Join(valueSets, ",")) _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { return conn.ExecCtx(ctx, query, args...) }, keys...) return err } func (m *defaultSysRoleModel) BatchInsertWithTx(ctx context.Context, session sqlx.Session, dataList []*SysRole) error { if len(dataList) == 0 { return nil } keys := make([]string, 0, len(dataList)) valueSets := make([]string, 0, len(dataList)) args := make([]interface{}, 0) for _, data := range dataList { valueSets = append(valueSets, "(?, ?, ?, ?, ?, ?, ?)") args = append(args, data.ProductCode, data.Name, data.Remark, data.Status, data.PermsLevel, data.CreateTime, data.UpdateTime) sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, data.Id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) keys = append(keys, sysRoleIdKey, sysRoleProductCodeNameKey) } query := fmt.Sprintf("INSERT INTO %s (%s) VALUES %s", m.table, sysRoleRowsExpectAutoSet, strings.Join(valueSets, ",")) _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { return session.ExecCtx(ctx, query, args...) }, keys...) return err } func (m *defaultSysRoleModel) Update(ctx context.Context, newData *SysRole) error { data, err := m.FindOne(ctx, newData.Id) if err != nil { return err } sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, data.Id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { query := fmt.Sprintf("UPDATE %s SET %s WHERE `id` = ?", m.table, sysRoleRowsWithPlaceHolder) return conn.ExecCtx(ctx, query, newData.ProductCode, newData.Name, newData.Remark, newData.Status, newData.PermsLevel, newData.CreateTime, newData.UpdateTime, newData.Id) }, sysRoleIdKey, sysRoleProductCodeNameKey) return err } func (m *defaultSysRoleModel) UpdateWithTx(ctx context.Context, session sqlx.Session, newData *SysRole) error { data, err := m.FindOne(ctx, newData.Id) if err != nil { return err } sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, data.Id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { query := fmt.Sprintf("UPDATE %s SET %s WHERE `id` = ?", m.table, sysRoleRowsWithPlaceHolder) return session.ExecCtx(ctx, query, newData.ProductCode, newData.Name, newData.Remark, newData.Status, newData.PermsLevel, newData.CreateTime, newData.UpdateTime, newData.Id) }, sysRoleIdKey, sysRoleProductCodeNameKey) return err } func (m *defaultSysRoleModel) BatchUpdate(ctx context.Context, dataList []*SysRole) error { if len(dataList) == 0 { return nil } keys := make([]string, 0) primaryKeys := make([]interface{}, 0, len(dataList)) for _, item := range dataList { primaryKeys = append(primaryKeys, item.Id) } oldList, err := m.findListByPrimaryKeys(ctx, primaryKeys) if err != nil { return err } for _, data := range oldList { sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, data.Id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) keys = append(keys, sysRoleIdKey, sysRoleProductCodeNameKey) } query, vals := m.buildBatchUpdateQuery(dataList) _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { return conn.ExecCtx(ctx, query, vals...) }, keys...) return err } func (m *defaultSysRoleModel) BatchUpdateWithTx(ctx context.Context, session sqlx.Session, dataList []*SysRole) error { if len(dataList) == 0 { return nil } keys := make([]string, 0) primaryKeys := make([]interface{}, 0, len(dataList)) for _, item := range dataList { primaryKeys = append(primaryKeys, item.Id) } oldList, err := m.findListByPrimaryKeys(ctx, primaryKeys) if err != nil { return err } for _, data := range oldList { sysRoleIdKey := fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, data.Id) sysRoleProductCodeNameKey := fmt.Sprintf("%s%v:%v", cacheSysRoleProductCodeNamePrefix, data.ProductCode, data.Name) keys = append(keys, sysRoleIdKey, sysRoleProductCodeNameKey) } query, vals := m.buildBatchUpdateQuery(dataList) _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { return session.ExecCtx(ctx, query, vals...) }, keys...) return err } func (m *defaultSysRoleModel) buildBatchUpdateQuery(dataList []*SysRole) (string, []interface{}) { fields := strings.Split(strings.ReplaceAll(sysRoleRowsWithPlaceHolder, "=?", ""), ",") listValues := make([][]interface{}, 0, len(dataList)) for _, newData := range dataList { values := make([]interface{}, 0, len(fields)+1) values = append(values, newData.ProductCode, newData.Name, newData.Remark, newData.Status, newData.PermsLevel, newData.CreateTime, newData.UpdateTime, newData.Id) listValues = append(listValues, values) } vals := make([]interface{}, 0, len(dataList)*(len(fields)*2+1)) setClauses := make([]string, len(fields)) for i, field := range fields { caseClauses := make([]string, len(dataList)) for j, item := range dataList { caseClauses[j] = "WHEN `id` = ? THEN ?" vals = append(vals, item.Id, listValues[j][i]) } setClauses[i] = fmt.Sprintf("%s = CASE %s ELSE %s END", field, strings.Join(caseClauses, " "), field) } wherePlaceholders := make([]string, len(dataList)) for i, item := range dataList { wherePlaceholders[i] = "?" vals = append(vals, item.Id) } query := fmt.Sprintf("UPDATE %s SET %s WHERE `id` IN (%s)", m.table, strings.Join(setClauses, ", "), strings.Join(wherePlaceholders, ",")) return query, vals } func (m *defaultSysRoleModel) formatPrimary(primary any) string { return fmt.Sprintf("%s%v", cacheSysRoleIdPrefix, primary) } func (m *defaultSysRoleModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary any) error { query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysRoleRows, m.table) return conn.QueryRowCtx(ctx, v, query, primary) } func (m *defaultSysRoleModel) TableName() string { return m.table } func (m *defaultSysRoleModel) findListByPrimaryKeys(ctx context.Context, ids []interface{}) ([]*SysRole, error) { if len(ids) == 0 { return []*SysRole{}, nil } placeholders := make([]string, len(ids)) args := make([]interface{}, len(ids)) for i, id := range ids { placeholders[i] = "?" args[i] = id } var resp []*SysRole query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` IN (%s)", sysRoleRows, m.table, strings.Join(placeholders, ",")) err := m.QueryRowsNoCacheCtx(ctx, &resp, query, args...) if err != nil { return nil, err } return resp, nil } func (m *defaultSysRoleModel) getPrimaryKeyValue(data *SysRole) interface{} { return data.Id }