// Code generated by goctl. DO NOT EDIT. // versions: // goctl version: 1.10.0 package dept 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 ( sysDeptFieldNames = builder.RawFieldNames(&SysDept{}) sysDeptRows = strings.Join(sysDeptFieldNames, ",") sysDeptRowsExpectAutoSet = strings.Join(stringx.Remove(sysDeptFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") sysDeptRowsWithPlaceHolder = strings.Join(stringx.Remove(sysDeptFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" cacheSysDeptIdPrefix = "cache:sysDept:id:" ) type ( sysDeptModel interface { Insert(ctx context.Context, data *SysDept) (sql.Result, error) InsertWithTx(ctx context.Context, session sqlx.Session, data *SysDept) (sql.Result, error) BatchInsert(ctx context.Context, dataList []*SysDept) error BatchInsertWithTx(ctx context.Context, session sqlx.Session, dataList []*SysDept) error TransactCtx(ctx context.Context, fn func(context.Context, sqlx.Session) error) error FindOne(ctx context.Context, id int64) (*SysDept, error) FindOneWithTx(ctx context.Context, session sqlx.Session, id int64) (*SysDept, error) Update(ctx context.Context, data *SysDept) error UpdateWithTx(ctx context.Context, session sqlx.Session, data *SysDept) error BatchUpdate(ctx context.Context, dataList []*SysDept) error BatchUpdateWithTx(ctx context.Context, session sqlx.Session, dataList []*SysDept) 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 } defaultSysDeptModel struct { sqlc.CachedConn table string } SysDept struct { Id int64 `db:"id"` // ID ParentId int64 `db:"parentId"` // 父部门ID,0表示顶级部门 Name string `db:"name"` // 部门名称 Path string `db:"path"` // 层级路径,如 /1/3/7/ 用于快速判断祖先关系 Sort int64 `db:"sort"` // 排序值 DeptType string `db:"deptType"` // 部门类型 NORMAL普通 DEV研发 Remark string `db:"remark"` // 备注 Status int64 `db:"status"` // 状态 1启用 2禁用 CreateTime int64 `db:"createTime"` // 创建时间 UpdateTime int64 `db:"updateTime"` // 修改时间 } ) func newSysDeptModel(conn sqlx.SqlConn, c cache.CacheConf, cachePrefix string, opts ...cache.Option) *defaultSysDeptModel { if cachePrefix != "" { cacheSysDeptIdPrefix = cachePrefix + ":" + "cache:sysDept:id:" } return &defaultSysDeptModel{ CachedConn: sqlc.NewConn(conn, c, opts...), table: "`sys_dept`", } } func (m *defaultSysDeptModel) Delete(ctx context.Context, id int64) error { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, 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` = ?", m.table) return conn.ExecCtx(ctx, query, id) }, sysDeptIdKey) return err } func (m *defaultSysDeptModel) DeleteWithTx(ctx context.Context, session sqlx.Session, id int64) error { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, 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` = ?", m.table) return session.ExecCtx(ctx, query, id) }, sysDeptIdKey) return err } func (m *defaultSysDeptModel) BatchDelete(ctx context.Context, ids []int64) error { if len(ids) == 0 { return nil } keys := make([]string, 0) for _, id := range ids { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, id) keys = append(keys, sysDeptIdKey) } 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 *defaultSysDeptModel) BatchDeleteWithTx(ctx context.Context, session sqlx.Session, ids []int64) error { if len(ids) == 0 { return nil } keys := make([]string, 0) for _, id := range ids { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, id) keys = append(keys, sysDeptIdKey) } 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 *defaultSysDeptModel) FindOne(ctx context.Context, id int64) (*SysDept, error) { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, id) var resp SysDept err := m.QueryRowCtx(ctx, &resp, sysDeptIdKey, func(ctx context.Context, conn sqlx.SqlConn, v any) error { query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysDeptRows, 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 *defaultSysDeptModel) FindOneWithTx(ctx context.Context, session sqlx.Session, id int64) (*SysDept, error) { var resp SysDept query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysDeptRows, 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 *defaultSysDeptModel) Insert(ctx context.Context, data *SysDept) (sql.Result, error) { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, data.Id) 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, sysDeptRowsExpectAutoSet) return conn.ExecCtx(ctx, query, data.ParentId, data.Name, data.Path, data.Sort, data.DeptType, data.Remark, data.Status, data.CreateTime, data.UpdateTime) }, sysDeptIdKey) return ret, err } func (m *defaultSysDeptModel) InsertWithTx(ctx context.Context, session sqlx.Session, data *SysDept) (sql.Result, error) { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, data.Id) 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, sysDeptRowsExpectAutoSet) return session.ExecCtx(ctx, query, data.ParentId, data.Name, data.Path, data.Sort, data.DeptType, data.Remark, data.Status, data.CreateTime, data.UpdateTime) }, sysDeptIdKey) return ret, err } func (m *defaultSysDeptModel) BatchInsert(ctx context.Context, dataList []*SysDept) 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.ParentId, data.Name, data.Path, data.Sort, data.DeptType, data.Remark, data.Status, data.CreateTime, data.UpdateTime) sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, data.Id) keys = append(keys, sysDeptIdKey) } query := fmt.Sprintf("INSERT INTO %s (%s) VALUES %s", m.table, sysDeptRowsExpectAutoSet, 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 *defaultSysDeptModel) BatchInsertWithTx(ctx context.Context, session sqlx.Session, dataList []*SysDept) 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.ParentId, data.Name, data.Path, data.Sort, data.DeptType, data.Remark, data.Status, data.CreateTime, data.UpdateTime) sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, data.Id) keys = append(keys, sysDeptIdKey) } query := fmt.Sprintf("INSERT INTO %s (%s) VALUES %s", m.table, sysDeptRowsExpectAutoSet, 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 *defaultSysDeptModel) Update(ctx context.Context, data *SysDept) error { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, data.Id) _, 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, sysDeptRowsWithPlaceHolder) return conn.ExecCtx(ctx, query, data.ParentId, data.Name, data.Path, data.Sort, data.DeptType, data.Remark, data.Status, data.CreateTime, data.UpdateTime, data.Id) }, sysDeptIdKey) return err } func (m *defaultSysDeptModel) UpdateWithTx(ctx context.Context, session sqlx.Session, data *SysDept) error { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, data.Id) _, 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, sysDeptRowsWithPlaceHolder) return session.ExecCtx(ctx, query, data.ParentId, data.Name, data.Path, data.Sort, data.DeptType, data.Remark, data.Status, data.CreateTime, data.UpdateTime, data.Id) }, sysDeptIdKey) return err } func (m *defaultSysDeptModel) BatchUpdate(ctx context.Context, dataList []*SysDept) error { if len(dataList) == 0 { return nil } keys := make([]string, 0) for _, data := range dataList { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, data.Id) keys = append(keys, sysDeptIdKey) } 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 *defaultSysDeptModel) BatchUpdateWithTx(ctx context.Context, session sqlx.Session, dataList []*SysDept) error { if len(dataList) == 0 { return nil } keys := make([]string, 0) for _, data := range dataList { sysDeptIdKey := fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, data.Id) keys = append(keys, sysDeptIdKey) } 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 *defaultSysDeptModel) buildBatchUpdateQuery(dataList []*SysDept) (string, []interface{}) { fields := strings.Split(strings.ReplaceAll(sysDeptRowsWithPlaceHolder, "=?", ""), ",") listValues := make([][]interface{}, 0, len(dataList)) for _, data := range dataList { values := make([]interface{}, 0, len(fields)+1) values = append(values, data.ParentId, data.Name, data.Path, data.Sort, data.DeptType, data.Remark, data.Status, data.CreateTime, data.UpdateTime, data.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 *defaultSysDeptModel) formatPrimary(primary any) string { return fmt.Sprintf("%s%v", cacheSysDeptIdPrefix, primary) } func (m *defaultSysDeptModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary any) error { query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` = ? LIMIT 1", sysDeptRows, m.table) return conn.QueryRowCtx(ctx, v, query, primary) } func (m *defaultSysDeptModel) TableName() string { return m.table } func (m *defaultSysDeptModel) findListByPrimaryKeys(ctx context.Context, ids []interface{}) ([]*SysDept, error) { if len(ids) == 0 { return []*SysDept{}, nil } placeholders := make([]string, len(ids)) args := make([]interface{}, len(ids)) for i, id := range ids { placeholders[i] = "?" args[i] = id } var resp []*SysDept query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` IN (%s)", sysDeptRows, m.table, strings.Join(placeholders, ",")) err := m.QueryRowsNoCacheCtx(ctx, &resp, query, args...) if err != nil { return nil, err } return resp, nil } func (m *defaultSysDeptModel) getPrimaryKeyValue(data *SysDept) interface{} { return data.Id }