实现基础的通知媒介管理

2025-12-07 18:10:24 +08:00 · 2021-04-05 20:48:33 +08:00
parent 975b307ac1
commit b0990961b8
41 changed files with 2467 additions and 3 deletions
--- a/internal/rpc/services/service_message_recipient.go
+++ b/internal/rpc/services/service_message_recipient.go
@@ -0,0 +1,228 @@
+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
+}