opms_backend/opms_admin/app/service/base_product_auth.go

package service

import (
	"context"
	"dashoo.cn/micro/app/dao"
	"dashoo.cn/micro/app/model"
	"database/sql"
	"fmt"
	"github.com/gogf/gf/errors/gerror"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/os/gtime"
)

type lineService struct {
	*contextService

	Dao *dao.BaseProductAuthDao
}

func NewLineService(ctx context.Context) (svc *lineService, err error) {
	svc = new(lineService)
	if svc.contextService, err = svc.Init(ctx); err != nil {
		return nil, err
	}
	svc.Dao = dao.NewBaseProductAuthDao(svc.Tenant)
	svc.Table = svc.Dao.Table
	return svc, nil
}

// GetList 用户产品线列表
func (s *lineService) GetList(req *model.BaseProductAuthSearchReq) (total int, lines []*model.BaseProductAuth, err error) {
	lineModel := s.Dao.M
	if req.UserId != 0 {
		lineModel = lineModel.Where("user_id", req.UserId)
	}
	if req.UserName != "" {
		lineModel = lineModel.Where("user_name", req.UserName)
	}
	if req.UserPost != "" {
		lineModel = lineModel.Where("user_post", req.UserPost)
	}
	if req.UserPhone != "" {
		lineModel = lineModel.Where("user_phone", req.UserPhone)
	}
	if req.ProductCode != "" {
		lineModel = lineModel.Where("product_code", req.ProductCode)
	}
	if req.ProductName != "" {
		lineModel = lineModel.Where("product_name", req.ProductName)
	}

	total, err = lineModel.Count()
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("获取总行数失败")
	}

	err = lineModel.Order("id ASC").Scan(&lines)
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("获取数据失败")
	}
	return
}

// Save 创建或更新数据
func (s *lineService) Save(req *model.SaveBaseProductAuthReq) (err error) {
	var newLines []*model.BaseProductAuth

	// 标记删除旧数据
	_, err = s.Dao.Where(fmt.Sprintf("user_id='%v'", req.UserId)).Update(fmt.Sprintf("deleted_time='%v'", gtime.Now()))
	if err != nil {
		return err
	}

	// 填入新数据
	for _, line := range req.Lines {
		data := &model.BaseProductAuth{
			UserId:      req.UserId,
			UserName:    req.UserName,
			UserPhone:   req.UserPhone,
			UserPost:    req.UserPost,
			ProductCode: line.ProductCode,
			ProductName: line.ProductName,
			Remark:      line.Remark,
		}
		SetCreatedInfo(data, s.GetCxtUserId(), s.GetCxtUserName())
		SetUpdatedInfo(data, s.GetCxtUserId(), s.GetCxtUserName())

		newLines = append(newLines, data)
	}
	if len(newLines) > 0 {
		_, err = s.Dao.Save(newLines)
	}

	return
}

func (s *lineService) GetProductLineUsers(req *model.SysUserSearchReq) (total int, lines []*model.BaseProductAuth, err error) {
	userService, err := NewUserService(s.Ctx)
	if err != nil {
		return 0, nil, err
	}

	total, users, err := userService.GetUsersByDeptEx(req)
	if err != nil {
		return 0, nil, err
	}

	var userIds []int
	for _, user := range users {
		userIds = append(userIds, user.Id)
	}

	allLines, err := s.Dao.Where("user_id IN (?)", userIds).Order("user_id ASC, id ASC").FindAll()
	if err != nil && err != sql.ErrNoRows {
		return 0, nil, err
	}
	for _, user := range users {
		var data model.BaseProductAuth
		data.UserId = user.Id
		data.UserName = user.NickName
		data.UserPost = "" // TODO 暂时为空
		data.UserPhone = user.Phone
		for _, line := range allLines {
			if user.Id == line.UserId {
				if data.ProductCode == "" {
					data.ProductCode = line.ProductCode
					data.ProductName = line.ProductName
				} else {
					data.ProductCode += "," + line.ProductCode
					data.ProductName += "," + line.ProductName
				}
			}
		}

		lines = append(lines, &data)
	}

	return
}