1.4.5.2

EdgeAPI/internal/rpc/services/service_message_receiver_plus.go

@@ -0,0 +1,229 @@
+//go:build plus
+
+package services
+
+import (
+	"context"
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	"github.com/TeaOSLab/EdgeAPI/internal/utils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/maps"
+)
+
+// MessageReceiverService 消息对象接收人
+type MessageReceiverService struct {
+	BaseService
+}
+
+// UpdateMessageReceivers 创建接收者
+func (this *MessageReceiverService) UpdateMessageReceivers(ctx context.Context, req *pb.UpdateMessageReceiversRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	if len(req.Role) == 0 {
+		req.Role = nodeconfigs.NodeRoleNode
+	}
+
+	params := maps.Map{}
+	if len(req.ParamsJSON) > 0 {
+		err = json.Unmarshal(req.ParamsJSON, &params)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	err = this.RunTx(func(tx *dbs.Tx) error {
+		err = models.SharedMessageReceiverDAO.DisableReceivers(tx, req.NodeClusterId, req.NodeId, req.ServerId)
+		if err != nil {
+			return err
+		}
+
+		for messageType, options := range req.RecipientOptions {
+			for _, option := range options.RecipientOptions {
+				_, err := models.SharedMessageReceiverDAO.CreateReceiver(tx, req.Role, req.NodeClusterId, req.NodeId, req.ServerId, messageType, params, option.MessageRecipientId, option.MessageRecipientGroupId)
+				if err != nil {
+					return err
+				}
+			}
+		}
+		return nil
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return this.Success()
+}
+
+// FindAllEnabledMessageReceivers 查找接收者
+func (this *MessageReceiverService) FindAllEnabledMessageReceivers(ctx context.Context, req *pb.FindAllEnabledMessageReceiversRequest) (*pb.FindAllEnabledMessageReceiversResponse, error) {
+	_, err := this.ValidateAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	if len(req.Role) == 0 {
+		req.Role = nodeconfigs.NodeRoleNode
+	}
+
+	var tx = this.NullTx()
+	var cacheMap = utils.NewCacheMap()
+	receivers, err := models.SharedMessageReceiverDAO.FindAllEnabledReceivers(tx, req.Role, req.NodeClusterId, req.NodeId, req.ServerId, "")
+	if err != nil {
+		return nil, err
+	}
+	var pbReceivers = []*pb.MessageReceiver{}
+	for _, receiver := range receivers {
+		var pbRecipient *pb.MessageRecipient = nil
+
+		// 接收人
+		if receiver.RecipientId > 0 {
+			recipient, err := models.SharedMessageRecipientDAO.FindEnabledMessageRecipient(tx, int64(receiver.RecipientId), cacheMap)
+			if err != nil {
+				return nil, err
+			}
+			if recipient == nil {
+				continue
+			}
+
+			// 管理员
+			admin, err := models.SharedAdminDAO.FindEnabledAdmin(tx, int64(recipient.AdminId))
+			if err != nil {
+				return nil, err
+			}
+			if admin == nil {
+				continue
+			}
+
+			// 接收人
+			instance, err := models.SharedMessageMediaInstanceDAO.FindEnabledMessageMediaInstance(tx, int64(recipient.InstanceId), cacheMap)
+			if err != nil {
+				return nil, err
+			}
+			if instance == nil {
+				continue
+			}
+
+			pbRecipient = &pb.MessageRecipient{
+				Id: int64(recipient.Id),
+				Admin: &pb.Admin{
+					Id:       int64(admin.Id),
+					Fullname: admin.Fullname,
+					Username: admin.Username,
+					IsOn:     admin.IsOn,
+				},
+				MessageMediaInstance: &pb.MessageMediaInstance{
+					Id:   int64(instance.Id),
+					Name: instance.Name,
+					IsOn: instance.IsOn,
+				},
+				IsOn:                   recipient.IsOn,
+				MessageRecipientGroups: nil,
+				Description:            "",
+				User:                   "",
+			}
+		}
+
+		// 接收人分组
+		var pbRecipientGroup *pb.MessageRecipientGroup = nil
+		if receiver.RecipientGroupId > 0 {
+			group, err := models.SharedMessageRecipientGroupDAO.FindEnabledMessageRecipientGroup(tx, int64(receiver.RecipientGroupId))
+			if err != nil {
+				return nil, err
+			}
+			if group == nil {
+				continue
+			}
+			pbRecipientGroup = &pb.MessageRecipientGroup{
+				Id:   int64(group.Id),
+				Name: group.Name,
+				IsOn: group.IsOn,
+			}
+		}
+
+		pbReceivers = append(pbReceivers, &pb.MessageReceiver{
+			Id:                    int64(receiver.Id),
+			ClusterId:             int64(receiver.ClusterId),
+			NodeId:                int64(receiver.NodeId),
+			ServerId:              int64(receiver.ServerId),
+			Type:                  receiver.Type,
+			ParamsJSON:            receiver.Params,
+			MessageRecipient:      pbRecipient,
+			MessageRecipientGroup: pbRecipientGroup,
+			Role:                  receiver.Role,
+		})
+	}
+	return &pb.FindAllEnabledMessageReceiversResponse{MessageReceivers: pbReceivers}, nil
+}
+
+// FindAllEnabledMessageReceiversWithMessageRecipientId 根据接收人查找关联的接收者
+func (this *MessageReceiverService) FindAllEnabledMessageReceiversWithMessageRecipientId(ctx context.Context, req *pb.FindAllEnabledMessageReceiversWithMessageRecipientIdRequest) (*pb.FindAllEnabledMessageReceiversWithMessageRecipientIdResponse, error) {
+	_, err := this.ValidateAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	receivers, err := models.SharedMessageReceiverDAO.FindAllEnabledReceiversWithRecipientId(tx, req.MessageRecipientId)
+	if err != nil {
+		return nil, err
+	}
+
+	var pbReceivers = []*pb.MessageReceiver{}
+	for _, receiver := range receivers {
+		pbReceivers = append(pbReceivers, &pb.MessageReceiver{
+			Id:                    int64(receiver.Id),
+			ClusterId:             int64(receiver.ClusterId),
+			NodeId:                int64(receiver.NodeId),
+			ServerId:              int64(receiver.ServerId),
+			Type:                  receiver.Type,
+			ParamsJSON:            nil,
+			MessageRecipient:      nil,
+			MessageRecipientGroup: nil,
+			Role:                  receiver.Role,
+		})
+	}
+
+	return &pb.FindAllEnabledMessageReceiversWithMessageRecipientIdResponse{
+		MessageReceivers: pbReceivers,
+	}, nil
+}
+
+// DeleteMessageReceiver 删除接收者
+func (this *MessageReceiverService) DeleteMessageReceiver(ctx context.Context, req *pb.DeleteMessageReceiverRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	err = models.SharedMessageReceiverDAO.DisableMessageReceiver(tx, req.MessageReceiverId)
+	if err != nil {
+		return nil, err
+	}
+	return this.Success()
+}
+
+// CountAllEnabledMessageReceivers 计算接收者数量
+func (this *MessageReceiverService) CountAllEnabledMessageReceivers(ctx context.Context, req *pb.CountAllEnabledMessageReceiversRequest) (*pb.RPCCountResponse, error) {
+	_, err := this.ValidateAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	if len(req.Role) == 0 {
+		req.Role = nodeconfigs.NodeRoleNode
+	}
+
+	var tx = this.NullTx()
+	count, err := models.SharedMessageReceiverDAO.CountAllEnabledReceivers(tx, req.Role, req.NodeClusterId, req.NodeId, req.ServerId, "")
+	if err != nil {
+		return nil, err
+	}
+	return this.SuccessCount(count)
+}