opms_backend/opms_parent/app/service/partner/partner_company.go

package partner

import (
	"context"
	"database/sql"
	"fmt"

	dao "dashoo.cn/micro/app/dao/partner"
	model "dashoo.cn/micro/app/model/partner"

	"dashoo.cn/opms_libary/micro_srv"
	"dashoo.cn/opms_libary/myerrors"
	"dashoo.cn/opms_libary/request"
	"github.com/gogf/gf/database/gdb"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/os/gtime"
	"github.com/gogf/gf/util/gvalid"
)

type PartnerCompanyService struct {
	Dao        *dao.PartnerCompanyDao
	ContactDao *dao.PartnerCompanyContactDao

	Tenant    string
	userInfo  request.UserInfo
	DataScope g.Map `json:"dataScope"`
}

func NewPartnerCompanyService(ctx context.Context) (*PartnerCompanyService, error) {
	tenant, err := micro_srv.GetTenant(ctx)
	if err != nil {
		err = myerrors.TipsError(fmt.Sprintf("获取租户码异常：%s", err.Error()))
		return nil, err
	}
	// 获取用户信息
	userInfo, err := micro_srv.GetUserInfo(ctx)
	if err != nil {
		return nil, fmt.Errorf("获取用户信息异常：%s", err.Error())
	}
	return &PartnerCompanyService{
		Dao:        dao.NewPartnerCompanyDao(tenant),
		ContactDao: dao.NewPartnerCompanyContactDao(tenant),
		Tenant:     tenant,
		userInfo:   userInfo,
		DataScope:  userInfo.DataScope,
	}, nil
}

func (s PartnerCompanyService) Get(ctx context.Context, id int) (*model.PartnerCompanyGetRsp, error) {
	ent, err := s.Dao.Where("Id = ?", id).One()
	if err != nil {
		return nil, err
	}
	if ent == nil {
		return nil, myerrors.TipsError("合作伙伴不存在")
	}
	contact, err := s.ContactDao.Where("partner_id = ?", id).All()
	if err != nil {
		return nil, err
	}
	if contact == nil {
		contact = []*model.PartnerCompanyContact{}
	}
	return &model.PartnerCompanyGetRsp{
		PartnerCompany: *ent,
		Contact:        contact,
	}, nil
}

func (s PartnerCompanyService) List(ctx context.Context, req *model.PartnerCompanyListReq) (int, []*model.PartnerCompany, error) {
	dao := &s.Dao.PartnerCompanyDao
	if req.PartnerName != "" {
		likestr := fmt.Sprintf("%%%s%%", req.PartnerName)
		dao = dao.Where("partner_name like ?", likestr)
	}

	total, err := dao.Count()
	if err != nil {
		return 0, nil, err
	}
	if req.PageNum != 0 {
		dao = dao.Page(req.GetPage())
	}
	orderby := "created_time desc"
	if req.OrderBy != "" {
		orderby = req.OrderBy
	}
	dao = dao.Order(orderby)

	ents := []*model.PartnerCompany{}
	err = dao.Structs(&ents)
	if err != nil && err != sql.ErrNoRows {
		return 0, nil, err
	}
	return total, ents, err
}

func (s PartnerCompanyService) Add(ctx context.Context, req *model.PartnerCompanyAddReq) (int, error) {
	validErr := gvalid.CheckStruct(ctx, req, nil)
	if validErr != nil {
		return 0, myerrors.TipsError(validErr.Current().Error())
	}

	var id int
	txerr := s.Dao.DB.Transaction(ctx, func(ctx context.Context, tx *gdb.TX) error {
		var err error
		cid, err := tx.InsertAndGetId("partner_company", model.PartnerCompany{
			PartnerName: req.PartnerName,
			Remark:      req.Remark,
			CreatedBy:   int(s.userInfo.Id),
			CreatedName: s.userInfo.NickName,
			CreatedTime: gtime.Now(),
			UpdatedBy:   int(s.userInfo.Id),
			UpdatedName: s.userInfo.NickName,
			UpdatedTime: gtime.Now(),
		})
		if err != nil {
			return err
		}
		id = int(cid)
		return nil
	})
	return id, txerr
}

func (s PartnerCompanyService) Update(ctx context.Context, req *model.PartnerCompanyUpdateReq) error {
	validErr := gvalid.CheckStruct(ctx, req, nil)
	if validErr != nil {
		return myerrors.TipsError(validErr.Current().Error())
	}

	ent, err := s.Dao.Where("id = ?", req.Id).One()
	if err != nil {
		return err
	}
	if ent == nil {
		return myerrors.TipsError(fmt.Sprintf("合作伙伴不存在: %d", req.Id))
	}

	toupdate := map[string]interface{}{}
	if req.PartnerName != "" {
		toupdate["partner_name"] = req.PartnerName
	}
	if req.Remark != nil {
		toupdate["remark"] = *req.Remark
	}
	if len(toupdate) != 0 {
		toupdate["updated_by"] = int(s.userInfo.Id)
		toupdate["updated_name"] = s.userInfo.NickName
		toupdate["updated_time"] = gtime.Now()
		txerr := s.Dao.DB.Transaction(ctx, func(ctx context.Context, tx *gdb.TX) error {
			_, err = tx.Update("partner_company", toupdate, "id = ?", req.Id)
			if err != nil {
				return err
			}
			return nil
		})
		if txerr != nil {
			return txerr
		}
	}
	return nil
}

func (s PartnerCompanyService) Delete(ctx context.Context, id []int) error {
	if len(id) == 0 {
		return nil
	}
	_, err := s.Dao.Where("Id  IN (?)", id).Delete()
	return err
}