kami_backend/internal/logic/card_redeem_account/group.go

package card_redeem_account

import (
	"context"
	"kami/internal/dao"
	"kami/internal/model"
	"kami/internal/model/do"
	"kami/internal/model/entity"
	"kami/utility/config"

	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/os/gmlock"
	"github.com/gogf/gf/v2/text/gstr"
)

func (a *sCardRedeemAccount) GroupAdd(ctx context.Context, input *model.RedeemAccountGroup, tx gdb.TX) (id int64, err error) {
	m := dao.V1CardRedeemAccountGroup.Ctx(ctx).DB(config.GetDatabaseV1())
	if tx != nil {
		m = m.TX(tx)
	}
	id, err = m.InsertAndGetId(do.V1CardRedeemAccountGroup{
		Name:          input.Name,
		CreatedUserId: input.CreatedUserId,
		Category:      input.Category,
		Notes:         input.Notes,
	})
	return
}

// GroupAddOrGetId 如果没有这个分组，就新建分组，否则返回这个分组的id
func (a *sCardRedeemAccount) GroupAddOrGetId(ctx context.Context, input *model.RedeemAccountGroup, tx gdb.TX) (id int64, err error) {
	gmlock.Lock("RedeemAccountGroupAddOrGetId")
	defer gmlock.Unlock("RedeemAccountGroupAddOrGetId")
	m := dao.V1CardRedeemAccountGroup.Ctx(ctx).DB(config.GetDatabaseV1())
	if tx != nil {
		m = m.TX(tx)
	}
	data := entity.V1CardRedeemAccountGroup{}
	_ = m.Where(dao.V1CardRedeemAccountGroup.Columns().Name, gstr.TrimAll(input.Name)).Scan(&data)
	if data.Id == 0 {
		id, err = a.GroupAdd(ctx, input, tx)
		if err != nil {
			return 0, err
		}
		return id, nil
	}
	return int64(data.Id), nil
}

// GroupCheck 判断当前用户上传的name分组是不是已经存在
func (a *sCardRedeemAccount) GroupCheck(ctx context.Context, name string, createdUserId string, tx gdb.TX) (isExist bool, err error) {
	m := dao.V1CardRedeemAccountGroup.Ctx(ctx).DB(config.GetDatabaseV1())
	if tx != nil {
		m = m.TX(tx)
	}
	isExist, err = m.Where(dao.V1CardRedeemAccountGroup.Columns().Name, name).Where(dao.V1CardRedeemAccountGroup.Columns().CreatedUserId, createdUserId).Exist()
	return
}

// GroupUpdate 添加修改函数
func (a *sCardRedeemAccount) GroupUpdate(ctx context.Context, input *model.RedeemAccountGroup, id uint, tx gdb.TX) (err error) {
	m := dao.V1CardRedeemAccountGroup.Ctx(ctx).DB(config.GetDatabaseV1())
	if tx != nil {
		m = m.TX(tx)
	}
	_, err = m.Where(dao.V1CardRedeemAccountGroup.Columns().Id, id).Update(do.V1CardRedeemAccountGroup{
		Name:  input.Name,
		Notes: input.Notes,
	})
	return
}

// GroupList 查询分组列表
func (a *sCardRedeemAccount) GroupList(ctx context.Context, input *model.RedeemAccountGroupList, tx gdb.TX) (total int, data []*entity.V1CardRedeemAccountGroup, err error) {
	data = []*entity.V1CardRedeemAccountGroup{}
	m := dao.V1CardRedeemAccountGroup.Ctx(ctx).DB(config.GetDatabaseV1())
	if tx != nil {
		m = m.TX(tx)
	}
	if input.CreatedUserId != "" {
		m = m.Where(dao.V1CardRedeemAccountGroup.Columns().CreatedUserId, input.CreatedUserId)
	}
	err = m.Where(dao.V1CardRedeemAccountGroup.Columns().Category, input.Category).
		Page(input.Current, input.PageSize).
		OrderDesc(dao.V1CardRedeemAccountGroup.Columns().CreatedAt).ScanAndCount(&data, &total, false)
	return
}

// GroupAll 查询分组列表
func (a *sCardRedeemAccount) GroupAll(ctx context.Context, input *model.RedeemAccountGroupAll, tx gdb.TX) (total int, data []*entity.V1CardRedeemAccountGroup, err error) {
	data = []*entity.V1CardRedeemAccountGroup{}
	m := dao.V1CardRedeemAccountGroup.Ctx(ctx).DB(config.GetDatabaseV1())
	if tx != nil {
		m = m.TX(tx)
	}
	if input.CreatedUserId != "" {
		m = m.Where(dao.V1CardRedeemAccountGroup.Columns().CreatedUserId, input.CreatedUserId)
	}
	if len(input.GroupIds) > 0 {
		m = m.WhereIn(dao.V1CardRedeemAccountGroup.Columns().Id, input.GroupIds)
	}
	err = m.Where(dao.V1CardRedeemAccountGroup.Columns().Category, input.Category).
		OrderDesc(dao.V1CardRedeemAccountGroup.Columns().CreatedAt).ScanAndCount(&data, &total, false)
	return
}

// GroupDelete 删除某个分组
func (a *sCardRedeemAccount) GroupDelete(ctx context.Context, id uint, createdUserId string, tx gdb.TX) (err error) {
	m := dao.V1CardRedeemAccountGroup.Ctx(ctx).DB(config.GetDatabaseV1()).
		Where(dao.V1CardRedeemAccountGroup.Columns().CreatedUserId, createdUserId)
	if tx != nil {
		m = m.TX(tx)
	}
	_, _ = m.Where(dao.V1CardRedeemAccountGroup.Columns().Id, id).Delete()
	//清空账号关联
	m2 := dao.V1CardRedeemAccountInfo.Ctx(ctx).DB(config.GetDatabaseV1())
	if tx != nil {
		m2 = m2.TX(tx)
	}
	_, err = m2.Where(dao.V1CardRedeemAccountInfo.Columns().GroupId, id).Where(dao.V1CardRedeemAccountInfo.Columns().CreateUserId, createdUserId).
		Update(g.Map{
			dao.V1CardRedeemAccountInfo.Columns().GroupId: nil,
		})
	return
}