system_add_logic.go 2.0 KB
package department

import (
	"context"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/db/transaction"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/pkg/contextdata"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/pkg/xerr"

	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/api/internal/svc"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/api/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
)

type SystemAddLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewSystemAddLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SystemAddLogic {
	return &SystemAddLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *SystemAddLogic) SystemAdd(req *types.DepartmentAddRequest) (resp *types.DepartmentGetResponse, err error) {
	var userToken = contextdata.GetUserTokenFromCtx(l.ctx)
	var conn = l.svcCtx.DefaultDBConn()

	_, list, err := l.svcCtx.DepartmentRepository.Find(l.ctx, conn, domain.NewQueryOptions().
		WithOffsetLimit(1, 1).WithCountOnly().
		WithKV("companyId", userToken.CompanyId).
		WithKV("name", req.Name))
	if err != nil {
		return nil, err
	}
	if len(list) > 0 {
		return nil, xerr.NewErrMsg("该分组名称已存在(不能重复)")
	}

	insert := &domain.Department{
		Id:        0,
		ParentId:  0,
		CompanyId: userToken.CompanyId,
		Name:      req.Name,
	}
	err = transaction.UseTrans(l.ctx, conn.DB(), func(ctx context.Context, c transaction.Conn) error {
		_, err = l.svcCtx.DepartmentRepository.Insert(l.ctx, conn, insert)
		return err
	}, true)
	if err != nil {
		return nil, xerr.NewErrMsgErr("新增分组失败", err)
	}

	resp = &types.DepartmentGetResponse{
		Department: types.Department{
			Id:        insert.Id,
			CompanyId: insert.CompanyId,
			ParentId:  insert.ParentId,
			Name:      insert.Name,
		},
	}
	return resp, nil
}