refactor: separate backend domains and storage adapters

2026-08-14 01:37:56 +08:00
parent e6775f0966
commit 8c4c6ba661
102 changed files with 6644 additions and 5735 deletions
@@ -0,0 +1,127 @@
+package data
+
+import (
+	"context"
+	"errors"
+	"time"
+
+	"kra/internal/biz"
+
+	"gorm.io/gorm"
+)
+
+type positionPO struct {
+	ID        uint `gorm:"primaryKey"`
+	CreatedAt time.Time
+	UpdatedAt time.Time
+	DeletedAt gorm.DeletedAt `gorm:"index"`
+	Name      string         `gorm:"index"`
+	Code      string
+	Sort      int
+	Status    bool `gorm:"default:true"`
+	Remark    string
+}
+
+func (positionPO) TableName() string { return "sys_positions" }
+
+type userPositionPO struct {
+	UserID     uint `gorm:"primaryKey;column:sys_user_id"`
+	PositionID uint `gorm:"primaryKey;column:sys_position_id"`
+}
+
+func (userPositionPO) TableName() string { return "sys_user_positions" }
+
+func posFromPO(po positionPO) *biz.Position {
+	return &biz.Position{ID: po.ID, CreatedAt: po.CreatedAt, UpdatedAt: po.UpdatedAt, Name: po.Name, Code: po.Code, Sort: po.Sort, Status: po.Status, Remark: po.Remark}
+}
+func (r *organizationRepo) CreatePosition(ctx context.Context, v *biz.Position) error {
+	return r.data.gormDB.WithContext(ctx).Create(&positionPO{Name: v.Name, Code: v.Code, Sort: v.Sort, Status: v.Status, Remark: v.Remark}).Error
+}
+func (r *organizationRepo) UpdatePosition(ctx context.Context, v *biz.Position) error {
+	return r.data.gormDB.WithContext(ctx).Model(&positionPO{}).Where("id = ?", v.ID).Updates(map[string]any{"name": v.Name, "code": v.Code, "sort": v.Sort, "status": v.Status, "remark": v.Remark}).Error
+}
+func (r *organizationRepo) DeletePosition(ctx context.Context, id uint) error {
+	if id == 0 {
+		return errors.New("岗位ID不能为空")
+	}
+	var count int64
+	if err := r.data.gormDB.WithContext(ctx).Model(&userPositionPO{}).Where("sys_position_id = ?", id).Count(&count).Error; err != nil {
+		return err
+	}
+	if count > 0 {
+		return errors.New("该岗位下存在用户,不允许删除")
+	}
+	return r.data.gormDB.WithContext(ctx).Delete(&positionPO{}, id).Error
+}
+func (r *organizationRepo) FindPosition(ctx context.Context, id uint) (*biz.Position, error) {
+	var po positionPO
+	if err := r.data.gormDB.WithContext(ctx).First(&po, id).Error; err != nil {
+		return nil, err
+	}
+	return posFromPO(po), nil
+}
+func (r *organizationRepo) ListPositions(ctx context.Context, page, size int, q *biz.Position) ([]*biz.Position, int64, error) {
+	if page < 1 {
+		page = 1
+	}
+	if size < 1 {
+		size = 10
+	}
+	db := r.data.gormDB.WithContext(ctx).Model(&positionPO{})
+	if q != nil {
+		if q.Name != "" {
+			db = db.Where("name LIKE ?", "%"+q.Name+"%")
+		}
+		if q.Code != "" {
+			db = db.Where("code LIKE ?", "%"+q.Code+"%")
+		}
+	}
+	var total int64
+	if err := db.Count(&total).Error; err != nil {
+		return nil, 0, err
+	}
+	var pos []positionPO
+	if err := db.Order("sort,id").Offset((page - 1) * size).Limit(size).Find(&pos).Error; err != nil {
+		return nil, 0, err
+	}
+	out := make([]*biz.Position, 0, len(pos))
+	for _, po := range pos {
+		out = append(out, posFromPO(po))
+	}
+	return out, total, nil
+}
+func (r *organizationRepo) PositionUserIDs(ctx context.Context, id uint) ([]uint, error) {
+	var ids []uint
+	err := r.data.gormDB.WithContext(ctx).Model(&userPositionPO{}).Where("sys_position_id = ?", id).Pluck("sys_user_id", &ids).Error
+	return ids, err
+}
+func (r *organizationRepo) SetPositionUsers(ctx context.Context, id uint, ids []uint) error {
+	return r.data.gormDB.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		if err := tx.Where("sys_position_id = ?", id).Delete(&userPositionPO{}).Error; err != nil {
+			return err
+		}
+		links := make([]userPositionPO, 0, len(ids))
+		for _, uid := range ids {
+			links = append(links, userPositionPO{UserID: uid, PositionID: id})
+		}
+		if len(links) > 0 {
+			return tx.Create(&links).Error
+		}
+		return nil
+	})
+}
+func (r *organizationRepo) SetUserPositions(ctx context.Context, uid uint, ids []uint) error {
+	return r.data.gormDB.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		if err := tx.Where("sys_user_id = ?", uid).Delete(&userPositionPO{}).Error; err != nil {
+			return err
+		}
+		links := make([]userPositionPO, 0, len(ids))
+		for _, id := range ids {
+			links = append(links, userPositionPO{UserID: uid, PositionID: id})
+		}
+		if len(links) > 0 {
+			return tx.Create(&links).Error
+		}
+		return nil
+	})
+}