// Code generated by goctl. DO NOT EDIT. // versions: // goctl version: 1.10.0 package roleperm 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 ( sysRolePermFieldNames = builder.RawFieldNames(&SysRolePerm{}) sysRolePermRows = strings.Join(sysRolePermFieldNames, ",") sysRolePermRowsExpectAutoSet = strings.Join(stringx.Remove(sysRolePermFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") sysRolePermRowsWithPlaceHolder = strings.Join(stringx.Remove(sysRolePermFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" cacheSysRolePermIdPrefix = "cache:sysRolePerm:id:" cacheSysRolePermRoleIdPermIdPrefix = "cache:sysRolePerm:roleId:permId:" ) type ( sysRolePermModel interface { Insert(ctx context.Context, data *SysRolePerm) (sql.Result, error) InsertWithTx(ctx context.Context, session sqlx.Session, data *SysRolePerm) (sql.Result, error) BatchInsert(ctx context.Context, dataList []*SysRolePerm) error BatchInsertWithTx(ctx context.Context, session sqlx.Session, dataList []*SysRolePerm) error TransactCtx(ctx context.Context, fn func(context.Context, sqlx.Session) error) error FindOne(ctx context.Context, id int64) (*SysRolePerm, error) FindOneWithTx(ctx context.Context, session sqlx.Session, id int64) (*SysRolePerm, error) FindOneByRoleIdPermId(ctx context.Context, roleId int64, permId int64) (*SysRolePerm, error) FindOneByRoleIdPermIdWithTx(ctx context.Context, session sqlx.Session, roleId int64, permId int64) (*SysRolePerm, error) Update(ctx context.Context, data *SysRolePerm) error UpdateWithTx(ctx context.Context, session sqlx.Session, data *SysRolePerm) error BatchUpdate(ctx context.Context, dataList []*SysRolePerm) error BatchUpdateWithTx(ctx context.Context, session sqlx.Session, dataList []*SysRolePerm) 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 } defaultSysRolePermModel struct { sqlc.CachedConn table string } SysRolePerm struct { Id int64 `db:"id"` // ID RoleId int64 `db:"roleId"` // 角色ID PermId int64 `db:"permId"` // 权限ID CreateTime int64 `db:"createTime"` // 创建时间 UpdateTime int64 `db:"updateTime"` // 修改时间 } ) func newSysRolePermModel(conn sqlx.SqlConn, c cache.CacheConf, cachePrefix string, opts ...cache.Option) *defaultSysRolePermModel { if cachePrefix != "" { cacheSysRolePermIdPrefix = cachePrefix + ":" + "cache:sysRolePerm:id:" cacheSysRolePermRoleIdPermIdPrefix = cachePrefix + ":" + "cache:sysRolePerm:roleId:permId:" } return &defaultSysRolePermModel{ CachedConn: sqlc.NewConn(conn, c, opts...), table: "`sys_role_perm`", } } func (m *defaultSysRolePermModel) Delete(ctx context.Context, id int64) error { data, err := m.FindOne(ctx, id) if err != nil { return err } sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) _, 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) }, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) return err } func (m *defaultSysRolePermModel) DeleteWithTx(ctx context.Context, session sqlx.Session, id int64) error { data, err := m.FindOne(ctx, id) if err != nil { return err } sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) _, 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) }, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) return err } func (m *defaultSysRolePermModel) 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) sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) keys = append(keys, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) } 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 *defaultSysRolePermModel) 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) sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) keys = append(keys, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) } 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 *defaultSysRolePermModel) FindOne(ctx context.Context, id int64) (*SysRolePerm, error) { sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, id) var resp SysRolePerm err := m.QueryRowCtx(ctx, &resp, sysRolePermIdKey, func(ctx context.Context, conn sqlx.SqlConn, v any) error { query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysRolePermRows, 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 *defaultSysRolePermModel) FindOneWithTx(ctx context.Context, session sqlx.Session, id int64) (*SysRolePerm, error) { var resp SysRolePerm query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysRolePermRows, 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 *defaultSysRolePermModel) FindOneByRoleIdPermId(ctx context.Context, roleId int64, permId int64) (*SysRolePerm, error) { sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, roleId, permId) var resp SysRolePerm err := m.QueryRowIndexCtx(ctx, &resp, sysRolePermRoleIdPermIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v any) (i any, e error) { query := fmt.Sprintf("SELECT %s FROM %s WHERE `roleId` = ? and `permId` = ? LIMIT 1", sysRolePermRows, m.table) if err := conn.QueryRowCtx(ctx, &resp, query, roleId, permId); 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 *defaultSysRolePermModel) FindOneByRoleIdPermIdWithTx(ctx context.Context, session sqlx.Session, roleId int64, permId int64) (*SysRolePerm, error) { var resp SysRolePerm query := fmt.Sprintf("SELECT %s FROM %s WHERE `roleId` = ? and `permId` = ? LIMIT 1", sysRolePermRows, m.table) err := session.QueryRowCtx(ctx, &resp, query, roleId, permId) switch err { case nil: return &resp, nil case sqlx.ErrNotFound: return nil, ErrNotFound default: return nil, err } } func (m *defaultSysRolePermModel) Insert(ctx context.Context, data *SysRolePerm) (sql.Result, error) { sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, data.Id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) 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, sysRolePermRowsExpectAutoSet) return conn.ExecCtx(ctx, query, data.RoleId, data.PermId, data.CreateTime, data.UpdateTime) }, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) return ret, err } func (m *defaultSysRolePermModel) InsertWithTx(ctx context.Context, session sqlx.Session, data *SysRolePerm) (sql.Result, error) { sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, data.Id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) 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, sysRolePermRowsExpectAutoSet) return session.ExecCtx(ctx, query, data.RoleId, data.PermId, data.CreateTime, data.UpdateTime) }, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) return ret, err } func (m *defaultSysRolePermModel) BatchInsert(ctx context.Context, dataList []*SysRolePerm) 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.RoleId, data.PermId, data.CreateTime, data.UpdateTime) sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, data.Id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) keys = append(keys, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) } query := fmt.Sprintf("INSERT INTO %s (%s) VALUES %s", m.table, sysRolePermRowsExpectAutoSet, 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 *defaultSysRolePermModel) BatchInsertWithTx(ctx context.Context, session sqlx.Session, dataList []*SysRolePerm) 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.RoleId, data.PermId, data.CreateTime, data.UpdateTime) sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, data.Id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) keys = append(keys, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) } query := fmt.Sprintf("INSERT INTO %s (%s) VALUES %s", m.table, sysRolePermRowsExpectAutoSet, 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 *defaultSysRolePermModel) Update(ctx context.Context, newData *SysRolePerm) error { data, err := m.FindOne(ctx, newData.Id) if err != nil { return err } sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, data.Id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) _, 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, sysRolePermRowsWithPlaceHolder) return conn.ExecCtx(ctx, query, newData.RoleId, newData.PermId, newData.CreateTime, newData.UpdateTime, newData.Id) }, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) return err } func (m *defaultSysRolePermModel) UpdateWithTx(ctx context.Context, session sqlx.Session, newData *SysRolePerm) error { data, err := m.FindOne(ctx, newData.Id) if err != nil { return err } sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, data.Id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) _, 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, sysRolePermRowsWithPlaceHolder) return session.ExecCtx(ctx, query, newData.RoleId, newData.PermId, newData.CreateTime, newData.UpdateTime, newData.Id) }, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) return err } func (m *defaultSysRolePermModel) BatchUpdate(ctx context.Context, dataList []*SysRolePerm) 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 { sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, data.Id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) keys = append(keys, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) } 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 *defaultSysRolePermModel) BatchUpdateWithTx(ctx context.Context, session sqlx.Session, dataList []*SysRolePerm) 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 { sysRolePermIdKey := fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, data.Id) sysRolePermRoleIdPermIdKey := fmt.Sprintf("%s%v:%v", cacheSysRolePermRoleIdPermIdPrefix, data.RoleId, data.PermId) keys = append(keys, sysRolePermIdKey, sysRolePermRoleIdPermIdKey) } 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 *defaultSysRolePermModel) buildBatchUpdateQuery(dataList []*SysRolePerm) (string, []interface{}) { fields := strings.Split(strings.ReplaceAll(sysRolePermRowsWithPlaceHolder, "=?", ""), ",") listValues := make([][]interface{}, 0, len(dataList)) for _, newData := range dataList { values := make([]interface{}, 0, len(fields)+1) values = append(values, newData.RoleId, newData.PermId, 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 *defaultSysRolePermModel) formatPrimary(primary any) string { return fmt.Sprintf("%s%v", cacheSysRolePermIdPrefix, primary) } func (m *defaultSysRolePermModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary any) error { query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysRolePermRows, m.table) return conn.QueryRowCtx(ctx, v, query, primary) } func (m *defaultSysRolePermModel) TableName() string { return m.table } func (m *defaultSysRolePermModel) findListByPrimaryKeys(ctx context.Context, ids []interface{}) ([]*SysRolePerm, error) { if len(ids) == 0 { return []*SysRolePerm{}, nil } placeholders := make([]string, len(ids)) args := make([]interface{}, len(ids)) for i, id := range ids { placeholders[i] = "?" args[i] = id } var resp []*SysRolePerm query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` IN (%s)", sysRolePermRows, m.table, strings.Join(placeholders, ",")) err := m.QueryRowsNoCacheCtx(ctx, &resp, query, args...) if err != nil { return nil, err } return resp, nil } func (m *defaultSysRolePermModel) getPrimaryKeyValue(data *SysRolePerm) interface{} { return data.Id }