签入脚本
This commit is contained in:
小蚂蚁云
@@ -0,0 +1,227 @@
|
||||
package logic
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
|
||||
"easygoadmin/api/v1"
|
||||
"easygoadmin/internal/consts"
|
||||
"easygoadmin/internal/dao"
|
||||
"easygoadmin/internal/service"
|
||||
"easygoadmin/utility/excel"
|
||||
"easygoadmin/utility/upload"
|
||||
)
|
||||
|
||||
// ============================== 职级业务实现 ==============================
|
||||
|
||||
// sLevel 职级业务实现
|
||||
// 实现 service.ILevel 接口，接口定义见 api/v1/level.go
|
||||
type sLevel struct{}
|
||||
|
||||
func init() {
|
||||
service.RegisterLevel(&sLevel{})
|
||||
}
|
||||
|
||||
// levelSerializeMaps 枚举字段到字典编码的映射
|
||||
// {field}Text 由基座自动填充
|
||||
var levelSerializeMaps = map[string]string{
|
||||
"status": "level_status",
|
||||
}
|
||||
|
||||
// levelSpec 职级模块声明
|
||||
//
|
||||
// 采用声明式实现：查询、输出与规则均在声明中定义，方法仅做委托
|
||||
var levelSpec = ModuleSpec[v1.LevelItem, v1.LevelListItem, v1.LevelDetailRes]{
|
||||
Model: dao.Level.Ctx,
|
||||
LikeFields: []string{"name"}, // 名称模糊（querySchemas 输入）
|
||||
EqFields: []string{"status"}, // status=0 视为未筛选（零值跳过）
|
||||
OrderBy: "sort, id",
|
||||
SerializeMaps: levelSerializeMaps,
|
||||
ToData: levelToData,
|
||||
BeforeAdd: levelDefaultStatus, // status 缺省置 1
|
||||
// Delete 守卫（存在性 + 用户引用检查）
|
||||
BeforeDelete: levelDeleteGuard,
|
||||
}
|
||||
|
||||
// levelToData 全量表单语义转换
|
||||
// 新增与修改均提交全部字段，兼容两种入参类型
|
||||
func levelToData(req any) map[string]any {
|
||||
switch r := req.(type) {
|
||||
case *v1.LevelAddReq:
|
||||
return map[string]any{"name": r.Name, "status": r.Status, "sort": r.Sort}
|
||||
case *v1.LevelUpdateReq:
|
||||
return map[string]any{"name": r.Name, "status": r.Status, "sort": r.Sort}
|
||||
}
|
||||
panic(fmt.Sprintf("levelToData: 不支持的入参类型 %T", req))
|
||||
}
|
||||
|
||||
// levelDefaultStatus 状态缺省处理
|
||||
// 状态为零值时置为 1（正常）
|
||||
func levelDefaultStatus(ctx context.Context, data map[string]any, req any) error {
|
||||
if v, ok := data["status"].(int); ok && v == 0 {
|
||||
data["status"] = 1
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// levelDeleteGuard 删除守卫
|
||||
// 校验记录存在性，存在用户引用时禁止删除
|
||||
func levelDeleteGuard(ctx context.Context, id int64) error {
|
||||
if n, err := dao.Level.Ctx(ctx).Where("id", id).Count(); err != nil {
|
||||
return err
|
||||
} else if n == 0 {
|
||||
return gerror.NewCode(consts.CodeRecordNotFound)
|
||||
}
|
||||
if n, err := dao.User.Ctx(ctx).Where("level_id", id).Count(); err != nil {
|
||||
return err
|
||||
} else if n > 0 {
|
||||
return gerror.NewCode(consts.CodeInUse)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// ---------- 标准方法：3行委托 ----------
|
||||
|
||||
// Page 职级分页
|
||||
// 分页返回职级列表，并补充关联用户数与状态文本
|
||||
func (s *sLevel) Page(ctx context.Context, req *v1.LevelPageReq) (res *v1.LevelPageRes, err error) {
|
||||
list, total, err := levelSpec.Page(ctx, req, req.GetPageNo(), req.GetPageSize())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
fillUserCount(ctx, "level_id", &list)
|
||||
return v1.NewPageData(req.GetPageNo(), req.GetPageSize(), total, derefSlice(list)), nil
|
||||
}
|
||||
|
||||
// List 职级全量列表
|
||||
// 返回精简结构的职级列表
|
||||
func (s *sLevel) List(ctx context.Context, req *v1.LevelListReq) (res v1.LevelListRes, err error) {
|
||||
list, err := levelSpec.List(ctx, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return derefSlice(list), nil
|
||||
}
|
||||
|
||||
// Detail 职级详情
|
||||
// 按职级ID查询详情
|
||||
func (s *sLevel) Detail(ctx context.Context, req *v1.LevelDetailReq) (res *v1.LevelDetailRes, err error) {
|
||||
return levelSpec.Detail(ctx, req.Id, req)
|
||||
}
|
||||
|
||||
// Add 新增职级
|
||||
// 新增成功后返回新记录ID
|
||||
func (s *sLevel) Add(ctx context.Context, req *v1.LevelAddReq) (res *v1.IdRes, err error) {
|
||||
id, err := levelSpec.Add(ctx, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &v1.IdRes{Id: id}, nil
|
||||
}
|
||||
|
||||
// Update 修改职级
|
||||
// 按职级ID更新记录
|
||||
func (s *sLevel) Update(ctx context.Context, req *v1.LevelUpdateReq) (res *v1.LevelUpdateRes, err error) {
|
||||
err = levelSpec.Update(ctx, req.Id, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &v1.LevelUpdateRes{}, nil
|
||||
}
|
||||
|
||||
// Status 职级状态
|
||||
// 非标准方法，手写实现，设置职级的正常或禁用状态
|
||||
func (s *sLevel) Status(ctx context.Context, req *v1.LevelStatusReq) (res *v1.LevelStatusRes, err error) {
|
||||
if _, err = dao.Level.Ctx(ctx).Where("id", req.Id).Data(map[string]any{"status": req.Status}).Update(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &v1.LevelStatusRes{}, nil
|
||||
}
|
||||
|
||||
// Delete 删除职级
|
||||
// 按职级ID删除单条记录
|
||||
func (s *sLevel) Delete(ctx context.Context, req *v1.LevelDeleteReq) (res *v1.LevelDeleteRes, err error) {
|
||||
if err = levelSpec.Delete(ctx, req.Id); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &v1.LevelDeleteRes{}, nil
|
||||
}
|
||||
|
||||
// BatchDelete 批量删除职级
|
||||
// 按职级ID列表批量删除记录
|
||||
func (s *sLevel) BatchDelete(ctx context.Context, req *v1.LevelBatchDeleteReq) (res *v1.LevelBatchDeleteRes, err error) {
|
||||
if err = levelSpec.BatchDelete(ctx, req.Ids); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &v1.LevelBatchDeleteRes{}, nil
|
||||
}
|
||||
|
||||
// ---------- 非标准方法（手写实现）（Import/Export）----------
|
||||
|
||||
// Import 导入职级
|
||||
//
|
||||
// 表格列依次为名称、状态、排序，名称重复时跳过并记录原因
|
||||
func (s *sLevel) Import(ctx context.Context, req *v1.LevelImportReq) (res *v1.LevelImportRes, err error) {
|
||||
_, rows, err := excel.ReadSheet(req.File)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
res = &v1.LevelImportRes{Total: len(rows)}
|
||||
tenantId := consts.GetTenantID(ctx)
|
||||
for i, row := range rows {
|
||||
name := cellAt(row, 0)
|
||||
if name == "" {
|
||||
res.Failed++
|
||||
res.Reasons = append(res.Reasons, fmt.Sprintf("第%d行：名称为空", i+2))
|
||||
continue
|
||||
}
|
||||
count, e := dao.Level.Ctx(ctx).Where("name", name).Count()
|
||||
if e != nil {
|
||||
return nil, e
|
||||
}
|
||||
if count > 0 {
|
||||
res.Failed++
|
||||
res.Reasons = append(res.Reasons, fmt.Sprintf("第%d行：名称已存在", i+2))
|
||||
continue
|
||||
}
|
||||
status := cellInt(row, 1, 1)
|
||||
if status <= 0 {
|
||||
status = 1
|
||||
}
|
||||
if _, e = dao.Level.Ctx(ctx).Data(map[string]any{
|
||||
"name": name,
|
||||
"status": status,
|
||||
"sort": cellInt(row, 2, 0),
|
||||
"tenant_id": tenantId,
|
||||
}).Insert(); e != nil {
|
||||
res.Failed++
|
||||
res.Reasons = append(res.Reasons, fmt.Sprintf("第%d行：写入失败 %v", i+2, e))
|
||||
continue
|
||||
}
|
||||
res.Success++
|
||||
}
|
||||
return res, nil
|
||||
}
|
||||
|
||||
// Export 导出职级
|
||||
// 按查询条件导出职级数据并返回文件访问地址
|
||||
func (s *sLevel) Export(ctx context.Context, req *v1.LevelExportReq) (res *v1.LevelExportRes, err error) {
|
||||
m := levelSpec.applyFilters(ctx, levelSpec.Model(ctx), req)
|
||||
rows, err := m.Order("sort, id").All()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
headers := []string{"名称", "状态", "排序"}
|
||||
data := make([][]any, 0, len(rows))
|
||||
for _, r := range rows {
|
||||
data = append(data, []any{r["name"].String(), userStatusText(r["status"].Int()), r["sort"].Int()})
|
||||
}
|
||||
fileName, _, err := excel.WriteSheet(upload.BaseDir(ctx), "职级数据", headers, data)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
url := upload.GetFileUrl(ctx, fileName)
|
||||
return &url, nil
|
||||
}
|
||||
Reference in New Issue
Block a user