feat: 优化web

internal/iam/repository/dept_repository.go

@@ -0,0 +1,93 @@
+package repository
+
+import (
+	"context"
+
+	"giter.top/smart/internal/iam/entity"
+	"gorm.io/gorm"
+)
+
+// DeptRepository 部门数据访问
+type DeptRepository interface {
+	Create(ctx context.Context, d *entity.Dept) error
+	Update(ctx context.Context, d *entity.Dept) error
+	Delete(ctx context.Context, id string) error
+	GetByID(ctx context.Context, id string) (*entity.Dept, error)
+	ListByTenant(ctx context.Context, tenantID string) ([]entity.Dept, error)
+	CountChildren(ctx context.Context, id string) (int64, error)
+	ExistsSiblingName(ctx context.Context, tenantID, parentID, name string, excludeID string) (bool, error)
+	FindRoot(ctx context.Context, tenantID string) (*entity.Dept, error)
+	UpdatePath(ctx context.Context, id string, path string) error
+}
+
+type deptRepository struct {
+	db *gorm.DB
+}
+
+func NewDeptRepository(db *gorm.DB) DeptRepository {
+	return &deptRepository{db: db}
+}
+
+func (r *deptRepository) Create(ctx context.Context, d *entity.Dept) error {
+	return r.db.WithContext(ctx).Create(d).Error
+}
+
+func (r *deptRepository) Update(ctx context.Context, d *entity.Dept) error {
+	return r.db.WithContext(ctx).Save(d).Error
+}
+
+func (r *deptRepository) Delete(ctx context.Context, id string) error {
+	return r.db.WithContext(ctx).Delete(&entity.Dept{}, "id = ?", id).Error
+}
+
+func (r *deptRepository) GetByID(ctx context.Context, id string) (*entity.Dept, error) {
+	var out entity.Dept
+	err := r.db.WithContext(ctx).Where("id = ?", id).First(&out).Error
+	if err != nil {
+		if err == gorm.ErrRecordNotFound {
+			return nil, ErrNotFound
+		}
+		return nil, err
+	}
+	return &out, nil
+}
+
+func (r *deptRepository) ListByTenant(ctx context.Context, tenantID string) ([]entity.Dept, error) {
+	var rows []entity.Dept
+	err := r.db.WithContext(ctx).Where("tenant_id = ?", tenantID).Order("sort_order ASC, created_at ASC").Find(&rows).Error
+	return rows, err
+}
+
+func (r *deptRepository) CountChildren(ctx context.Context, id string) (int64, error) {
+	var n int64
+	err := r.db.WithContext(ctx).Model(&entity.Dept{}).Where("parent_id = ?", id).Count(&n).Error
+	return n, err
+}
+
+func (r *deptRepository) ExistsSiblingName(ctx context.Context, tenantID, parentID, name string, excludeID string) (bool, error) {
+	q := r.db.WithContext(ctx).Model(&entity.Dept{}).Where("tenant_id = ? AND parent_id = ? AND dept_name = ?", tenantID, parentID, name)
+	if excludeID != "" {
+		q = q.Where("id <> ?", excludeID)
+	}
+	var n int64
+	err := q.Count(&n).Error
+	return n > 0, err
+}
+
+func (r *deptRepository) FindRoot(ctx context.Context, tenantID string) (*entity.Dept, error) {
+	var out entity.Dept
+	err := r.db.WithContext(ctx).
+		Where("tenant_id = ? AND (parent_id = '' OR parent_id = '0')", tenantID).
+		First(&out).Error
+	if err != nil {
+		if err == gorm.ErrRecordNotFound {
+			return nil, ErrNotFound
+		}
+		return nil, err
+	}
+	return &out, nil
+}
+
+func (r *deptRepository) UpdatePath(ctx context.Context, id string, path string) error {
+	return r.db.WithContext(ctx).Model(&entity.Dept{}).Where("id = ?", id).Update("dept_path", path).Error
+}