// Code generated by goctl. DO NOT EDIT. // versions: // goctl version: 1.10.0 package perm 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 ( sysPermFieldNames = builder.RawFieldNames(&SysPerm{}) sysPermRows = strings.Join(sysPermFieldNames, ",") sysPermRowsExpectAutoSet = strings.Join(stringx.Remove(sysPermFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") sysPermRowsWithPlaceHolder = strings.Join(stringx.Remove(sysPermFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" cacheSysPermIdPrefix = "cache:sysPerm:id:" cacheSysPermProductCodeCodePrefix = "cache:sysPerm:productCode:code:" ) type ( sysPermModel interface { Insert(ctx context.Context, data *SysPerm) (sql.Result, error) InsertWithTx(ctx context.Context, session sqlx.Session, data *SysPerm) (sql.Result, error) BatchInsert(ctx context.Context, dataList []*SysPerm) error BatchInsertWithTx(ctx context.Context, session sqlx.Session, dataList []*SysPerm) error TransactCtx(ctx context.Context, fn func(context.Context, sqlx.Session) error) error FindOne(ctx context.Context, id int64) (*SysPerm, error) FindOneWithTx(ctx context.Context, session sqlx.Session, id int64) (*SysPerm, error) FindOneByProductCodeCode(ctx context.Context, productCode string, code string) (*SysPerm, error) FindOneByProductCodeCodeWithTx(ctx context.Context, session sqlx.Session, productCode string, code string) (*SysPerm, error) Update(ctx context.Context, data *SysPerm) error UpdateWithTx(ctx context.Context, session sqlx.Session, data *SysPerm) error BatchUpdate(ctx context.Context, dataList []*SysPerm) error BatchUpdateWithTx(ctx context.Context, session sqlx.Session, dataList []*SysPerm) 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 } defaultSysPermModel struct { sqlc.CachedConn table string } SysPerm struct { Id int64 `db:"id"` // ID ProductCode string `db:"productCode"` // 所属产品编码 Name string `db:"name"` // 权限名 Code string `db:"code"` // 权限code Remark string `db:"remark"` // 备注 Status int64 `db:"status"` // 状态 1启用 2禁用 CreateTime int64 `db:"createTime"` // 创建时间 UpdateTime int64 `db:"updateTime"` // 修改时间 } ) func newSysPermModel(conn sqlx.SqlConn, c cache.CacheConf, cachePrefix string, opts ...cache.Option) *defaultSysPermModel { if cachePrefix != "" { cacheSysPermIdPrefix = cachePrefix + ":" + "cache:sysPerm:id:" cacheSysPermProductCodeCodePrefix = cachePrefix + ":" + "cache:sysPerm:productCode:code:" } return &defaultSysPermModel{ CachedConn: sqlc.NewConn(conn, c, opts...), table: "`sys_perm`", } } func (m *defaultSysPermModel) Delete(ctx context.Context, id int64) error { data, err := m.FindOne(ctx, id) if err != nil { return err } sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) _, 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) }, sysPermIdKey, sysPermProductCodeCodeKey) return err } func (m *defaultSysPermModel) DeleteWithTx(ctx context.Context, session sqlx.Session, id int64) error { data, err := m.FindOne(ctx, id) if err != nil { return err } sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) _, 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) }, sysPermIdKey, sysPermProductCodeCodeKey) return err } func (m *defaultSysPermModel) 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) sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) keys = append(keys, sysPermIdKey, sysPermProductCodeCodeKey) } 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 *defaultSysPermModel) 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) sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) keys = append(keys, sysPermIdKey, sysPermProductCodeCodeKey) } 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 *defaultSysPermModel) FindOne(ctx context.Context, id int64) (*SysPerm, error) { sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, id) var resp SysPerm err := m.QueryRowCtx(ctx, &resp, sysPermIdKey, func(ctx context.Context, conn sqlx.SqlConn, v any) error { query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysPermRows, 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 *defaultSysPermModel) FindOneWithTx(ctx context.Context, session sqlx.Session, id int64) (*SysPerm, error) { var resp SysPerm query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysPermRows, 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 *defaultSysPermModel) FindOneByProductCodeCode(ctx context.Context, productCode string, code string) (*SysPerm, error) { sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, productCode, code) var resp SysPerm err := m.QueryRowIndexCtx(ctx, &resp, sysPermProductCodeCodeKey, 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 `code` = ? LIMIT 1", sysPermRows, m.table) if err := conn.QueryRowCtx(ctx, &resp, query, productCode, code); 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 *defaultSysPermModel) FindOneByProductCodeCodeWithTx(ctx context.Context, session sqlx.Session, productCode string, code string) (*SysPerm, error) { var resp SysPerm query := fmt.Sprintf("SELECT %s FROM %s WHERE `productCode` = ? and `code` = ? LIMIT 1", sysPermRows, m.table) err := session.QueryRowCtx(ctx, &resp, query, productCode, code) switch err { case nil: return &resp, nil case sqlx.ErrNotFound: return nil, ErrNotFound default: return nil, err } } func (m *defaultSysPermModel) Insert(ctx context.Context, data *SysPerm) (sql.Result, error) { sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, data.Id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) 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, sysPermRowsExpectAutoSet) return conn.ExecCtx(ctx, query, data.ProductCode, data.Name, data.Code, data.Remark, data.Status, data.CreateTime, data.UpdateTime) }, sysPermIdKey, sysPermProductCodeCodeKey) return ret, err } func (m *defaultSysPermModel) InsertWithTx(ctx context.Context, session sqlx.Session, data *SysPerm) (sql.Result, error) { sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, data.Id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) 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, sysPermRowsExpectAutoSet) return session.ExecCtx(ctx, query, data.ProductCode, data.Name, data.Code, data.Remark, data.Status, data.CreateTime, data.UpdateTime) }, sysPermIdKey, sysPermProductCodeCodeKey) return ret, err } func (m *defaultSysPermModel) BatchInsert(ctx context.Context, dataList []*SysPerm) 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.Code, data.Remark, data.Status, data.CreateTime, data.UpdateTime) sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, data.Id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) keys = append(keys, sysPermIdKey, sysPermProductCodeCodeKey) } query := fmt.Sprintf("INSERT INTO %s (%s) VALUES %s", m.table, sysPermRowsExpectAutoSet, 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 *defaultSysPermModel) BatchInsertWithTx(ctx context.Context, session sqlx.Session, dataList []*SysPerm) 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.Code, data.Remark, data.Status, data.CreateTime, data.UpdateTime) sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, data.Id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) keys = append(keys, sysPermIdKey, sysPermProductCodeCodeKey) } query := fmt.Sprintf("INSERT INTO %s (%s) VALUES %s", m.table, sysPermRowsExpectAutoSet, 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 *defaultSysPermModel) Update(ctx context.Context, newData *SysPerm) error { data, err := m.FindOne(ctx, newData.Id) if err != nil { return err } sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, data.Id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) _, 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, sysPermRowsWithPlaceHolder) return conn.ExecCtx(ctx, query, newData.ProductCode, newData.Name, newData.Code, newData.Remark, newData.Status, newData.CreateTime, newData.UpdateTime, newData.Id) }, sysPermIdKey, sysPermProductCodeCodeKey) return err } func (m *defaultSysPermModel) UpdateWithTx(ctx context.Context, session sqlx.Session, newData *SysPerm) error { data, err := m.FindOne(ctx, newData.Id) if err != nil { return err } sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, data.Id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) _, 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, sysPermRowsWithPlaceHolder) return session.ExecCtx(ctx, query, newData.ProductCode, newData.Name, newData.Code, newData.Remark, newData.Status, newData.CreateTime, newData.UpdateTime, newData.Id) }, sysPermIdKey, sysPermProductCodeCodeKey) return err } func (m *defaultSysPermModel) BatchUpdate(ctx context.Context, dataList []*SysPerm) 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 { sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, data.Id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) keys = append(keys, sysPermIdKey, sysPermProductCodeCodeKey) } 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 *defaultSysPermModel) BatchUpdateWithTx(ctx context.Context, session sqlx.Session, dataList []*SysPerm) 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 { sysPermIdKey := fmt.Sprintf("%s%v", cacheSysPermIdPrefix, data.Id) sysPermProductCodeCodeKey := fmt.Sprintf("%s%v:%v", cacheSysPermProductCodeCodePrefix, data.ProductCode, data.Code) keys = append(keys, sysPermIdKey, sysPermProductCodeCodeKey) } 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 *defaultSysPermModel) buildBatchUpdateQuery(dataList []*SysPerm) (string, []interface{}) { fields := strings.Split(strings.ReplaceAll(sysPermRowsWithPlaceHolder, "=?", ""), ",") 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.Code, newData.Remark, newData.Status, 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 *defaultSysPermModel) formatPrimary(primary any) string { return fmt.Sprintf("%s%v", cacheSysPermIdPrefix, primary) } func (m *defaultSysPermModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary any) error { query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysPermRows, m.table) return conn.QueryRowCtx(ctx, v, query, primary) } func (m *defaultSysPermModel) TableName() string { return m.table } func (m *defaultSysPermModel) findListByPrimaryKeys(ctx context.Context, ids []interface{}) ([]*SysPerm, error) { if len(ids) == 0 { return []*SysPerm{}, nil } placeholders := make([]string, len(ids)) args := make([]interface{}, len(ids)) for i, id := range ids { placeholders[i] = "?" args[i] = id } var resp []*SysPerm query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` IN (%s)", sysPermRows, m.table, strings.Join(placeholders, ",")) err := m.QueryRowsNoCacheCtx(ctx, &resp, query, args...) if err != nil { return nil, err } return resp, nil } func (m *defaultSysPermModel) getPrimaryKeyValue(data *SysPerm) interface{} { return data.Id }