EdgeAPI/internal/rpc/services/service_message_recipient.go

package services

import (
	"context"
	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
)

// 消息接收人服务
type MessageRecipientService struct {
	BaseService
}

// 创建接收人
func (this *MessageRecipientService) CreateMessageRecipient(ctx context.Context, req *pb.CreateMessageRecipientRequest) (*pb.CreateMessageRecipientResponse, error) {
	_, err := this.ValidateAdmin(ctx, 0)
	if err != nil {
		return nil, err
	}

	var tx = this.NullTx()
	recipientId, err := models.SharedMessageRecipientDAO.CreateRecipient(tx, req.AdminId, req.InstanceId, req.User, req.GroupIds, req.Description)
	if err != nil {
		return nil, err
	}

	return &pb.CreateMessageRecipientResponse{MessageRecipientId: recipientId}, nil
}

// 修改接收人
func (this *MessageRecipientService) UpdateMessageRecipient(ctx context.Context, req *pb.UpdateMessageRecipientRequest) (*pb.RPCSuccess, error) {
	_, err := this.ValidateAdmin(ctx, 0)
	if err != nil {
		return nil, err
	}

	var tx = this.NullTx()
	err = models.SharedMessageRecipientDAO.UpdateRecipient(tx, req.MessageRecipientId, req.AdminId, req.InstanceId, req.User, req.GroupIds, req.Description, req.IsOn)
	if err != nil {
		return nil, err
	}

	return this.Success()
}

// 删除接收人
func (this *MessageRecipientService) DeleteMessageRecipient(ctx context.Context, req *pb.DeleteMessageRecipientRequest) (*pb.RPCSuccess, error) {
	_, err := this.ValidateAdmin(ctx, 0)
	if err != nil {
		return nil, err
	}

	var tx = this.NullTx()
	err = models.SharedMessageRecipientDAO.DisableMessageRecipient(tx, req.MessageRecipientId)
	if err != nil {
		return nil, err
	}
	return this.Success()
}

// 计算接收人数量
func (this *MessageRecipientService) CountAllEnabledMessageRecipients(ctx context.Context, req *pb.CountAllEnabledMessageRecipientsRequest) (*pb.RPCCountResponse, error) {
	_, err := this.ValidateAdmin(ctx, 0)
	if err != nil {
		return nil, err
	}

	var tx = this.NullTx()
	count, err := models.SharedMessageRecipientDAO.CountAllEnabledRecipients(tx, req.AdminId, req.GroupId, req.MediaType, req.Keyword)
	if err != nil {
		return nil, err
	}

	return this.SuccessCount(count)
}

// 列出单页接收人
func (this *MessageRecipientService) ListEnabledMessageRecipients(ctx context.Context, req *pb.ListEnabledMessageRecipientsRequest) (*pb.ListEnabledMessageRecipientsResponse, error) {
	_, err := this.ValidateAdmin(ctx, 0)
	if err != nil {
		return nil, err
	}

	var tx = this.NullTx()
	recipients, err := models.SharedMessageRecipientDAO.ListAllEnabledRecipients(tx, req.AdminId, req.GroupId, req.MediaType, req.Keyword, req.Offset, req.Size)
	if err != nil {
		return nil, err
	}
	pbRecipients := []*pb.MessageRecipient{}
	for _, recipient := range recipients {
		// admin
		admin, err := models.SharedAdminDAO.FindEnabledAdmin(tx, int64(recipient.AdminId))
		if err != nil {
			return nil, err
		}
		if admin == nil {
			continue
		}
		pbAdmin := &pb.Admin{
			Id:       int64(admin.Id),
			Fullname: admin.Fullname,
			Username: admin.Username,
			IsOn:     admin.IsOn == 1,
		}

		// 媒介实例
		instance, err := models.SharedMessageMediaInstanceDAO.FindEnabledMessageMediaInstance(tx, int64(recipient.InstanceId))
		if err != nil {
			return nil, err
		}
		if instance == nil {
			continue
		}
		pbInstance := &pb.MessageMediaInstance{
			Id:          int64(instance.Id),
			IsOn:        instance.IsOn == 1,
			Name:        instance.Name,
			Description: instance.Description,
		}

		// 分组
		pbGroups := []*pb.MessageRecipientGroup{}
		groupIds := recipient.DecodeGroupIds()
		if len(groupIds) > 0 {
			for _, groupId := range groupIds {
				group, err := models.SharedMessageRecipientGroupDAO.FindEnabledMessageRecipientGroup(tx, groupId)
				if err != nil {
					return nil, err
				}
				if group != nil {
					pbGroups = append(pbGroups, &pb.MessageRecipientGroup{
						Id:   int64(group.Id),
						Name: group.Name,
						IsOn: group.IsOn == 1,
					})
				}
			}
		}

		pbRecipients = append(pbRecipients, &pb.MessageRecipient{
			Id:                     int64(recipient.Id),
			Admin:                  pbAdmin,
			User:                   recipient.User,
			MessageMediaInstance:   pbInstance,
			IsOn:                   recipient.IsOn == 1,
			MessageRecipientGroups: pbGroups,
			Description:            recipient.Description,
		})
	}

	return &pb.ListEnabledMessageRecipientsResponse{MessageRecipients: pbRecipients}, nil
}

// 查找单个接收人信息
func (this *MessageRecipientService) FindEnabledMessageRecipient(ctx context.Context, req *pb.FindEnabledMessageRecipientRequest) (*pb.FindEnabledMessageRecipientResponse, error) {
	_, err := this.ValidateAdmin(ctx, 0)
	if err != nil {
		return nil, err
	}

	var tx = this.NullTx()
	recipient, err := models.SharedMessageRecipientDAO.FindEnabledMessageRecipient(tx, req.MessageRecipientId)
	if err != nil {
		return nil, err
	}
	if recipient == nil {
		return &pb.FindEnabledMessageRecipientResponse{MessageRecipient: nil}, nil
	}

	// admin
	admin, err := models.SharedAdminDAO.FindEnabledAdmin(tx, int64(recipient.AdminId))
	if err != nil {
		return nil, err
	}
	if admin == nil {
		return &pb.FindEnabledMessageRecipientResponse{MessageRecipient: nil}, nil
	}
	pbAdmin := &pb.Admin{
		Id:       int64(admin.Id),
		Fullname: admin.Fullname,
		Username: admin.Username,
		IsOn:     admin.IsOn == 1,
	}

	// 媒介实例
	instance, err := models.SharedMessageMediaInstanceDAO.FindEnabledMessageMediaInstance(tx, int64(recipient.InstanceId))
	if err != nil {
		return nil, err
	}
	if instance == nil {
		return &pb.FindEnabledMessageRecipientResponse{MessageRecipient: nil}, nil
	}
	pbInstance := &pb.MessageMediaInstance{
		Id:          int64(instance.Id),
		IsOn:        instance.IsOn == 1,
		Name:        instance.Name,
		Description: instance.Description,
	}

	// 分组
	pbGroups := []*pb.MessageRecipientGroup{}
	groupIds := recipient.DecodeGroupIds()
	if len(groupIds) > 0 {
		for _, groupId := range groupIds {
			group, err := models.SharedMessageRecipientGroupDAO.FindEnabledMessageRecipientGroup(tx, groupId)
			if err != nil {
				return nil, err
			}
			if group != nil {
				pbGroups = append(pbGroups, &pb.MessageRecipientGroup{
					Id:   int64(group.Id),
					Name: group.Name,
					IsOn: group.IsOn == 1,
				})
			}
		}
	}

	return &pb.FindEnabledMessageRecipientResponse{MessageRecipient: &pb.MessageRecipient{
		Id:                     int64(recipient.Id),
		User:                   recipient.User,
		Admin:                  pbAdmin,
		MessageMediaInstance:   pbInstance,
		IsOn:                   recipient.IsOn == 1,
		MessageRecipientGroups: pbGroups,
		Description:            recipient.Description,
	}}, nil
}