实现基础的通知媒介管理

2025-12-05 08:30:26 +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_media_instance.go
+++ b/internal/rpc/services/service_message_media_instance.go
@@ -0,0 +1,181 @@
+package services
+
+import (
+	"context"
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/maps"
+)
+
+// 消息媒介实例服务
+type MessageMediaInstanceService struct {
+	BaseService
+}
+
+// 创建消息媒介实例
+func (this *MessageMediaInstanceService) CreateMessageMediaInstance(ctx context.Context, req *pb.CreateMessageMediaInstanceRequest) (*pb.CreateMessageMediaInstanceResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+
+	params := maps.Map{}
+	if len(req.ParamsJSON) > 0 {
+		err = json.Unmarshal(req.ParamsJSON, &params)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	instanceId, err := models.SharedMessageMediaInstanceDAO.CreateMediaInstance(tx, req.Name, req.MediaType, params, req.Description)
+	if err != nil {
+		return nil, err
+	}
+
+	return &pb.CreateMessageMediaInstanceResponse{MessageMediaInstanceId: instanceId}, nil
+}
+
+// 修改消息实例
+func (this *MessageMediaInstanceService) UpdateMessageMediaInstance(ctx context.Context, req *pb.UpdateMessageMediaInstanceRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	params := maps.Map{}
+	if len(req.ParamsJSON) > 0 {
+		err = json.Unmarshal(req.ParamsJSON, &params)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	var tx = this.NullTx()
+	err = models.SharedMessageMediaInstanceDAO.UpdateMediaInstance(tx, req.MessageMediaInstanceId, req.Name, req.MediaType, params, req.Description, req.IsOn)
+	if err != nil {
+		return nil, err
+	}
+
+	return this.Success()
+}
+
+// 删除媒介实例
+func (this *MessageMediaInstanceService) DeleteMessageMediaInstance(ctx context.Context, req *pb.DeleteMessageMediaInstanceRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	err = models.SharedMessageMediaInstanceDAO.DisableMessageMediaInstance(tx, req.MessageMediaInstanceId)
+	if err != nil {
+		return nil, err
+	}
+	return this.Success()
+}
+
+// 计算媒介实例数量
+func (this *MessageMediaInstanceService) CountAllEnabledMessageMediaInstances(ctx context.Context, req *pb.CountAllEnabledMessageMediaInstancesRequest) (*pb.RPCCountResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	count, err := models.SharedMessageMediaInstanceDAO.CountAllEnabledMediaInstances(tx, req.MediaType, req.Keyword)
+	if err != nil {
+		return nil, err
+	}
+
+	return this.SuccessCount(count)
+}
+
+// 列出单页媒介实例
+func (this *MessageMediaInstanceService) ListEnabledMessageMediaInstances(ctx context.Context, req *pb.ListEnabledMessageMediaInstancesRequest) (*pb.ListEnabledMessageMediaInstancesResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	instances, err := models.SharedMessageMediaInstanceDAO.ListAllEnabledMediaInstances(tx, req.MediaType, req.Keyword, req.Offset, req.Size)
+	if err != nil {
+		return nil, err
+	}
+	pbInstances := []*pb.MessageMediaInstance{}
+	for _, instance := range instances {
+		// 媒介
+		media, err := models.SharedMessageMediaDAO.FindEnabledMediaWithType(tx, instance.MediaType)
+		if err != nil {
+			return nil, err
+		}
+		if media == nil {
+			continue
+		}
+		pbMedia := &pb.MessageMedia{
+			Id:              int64(media.Id),
+			Type:            media.Type,
+			Name:            media.Name,
+			Description:     media.Description,
+			UserDescription: media.UserDescription,
+			IsOn:            media.IsOn == 1,
+		}
+
+		pbInstances = append(pbInstances, &pb.MessageMediaInstance{
+			Id:           int64(instance.Id),
+			Name:         instance.Name,
+			IsOn:         instance.IsOn == 1,
+			MessageMedia: pbMedia,
+			ParamsJSON:   []byte(instance.Params),
+			Description:  instance.Description,
+		})
+	}
+
+	return &pb.ListEnabledMessageMediaInstancesResponse{MessageMediaInstances: pbInstances}, nil
+}
+
+// 查找单个媒介实例信息
+func (this *MessageMediaInstanceService) FindEnabledMessageMediaInstance(ctx context.Context, req *pb.FindEnabledMessageMediaInstanceRequest) (*pb.FindEnabledMessageMediaInstanceResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	instance, err := models.SharedMessageMediaInstanceDAO.FindEnabledMessageMediaInstance(tx, req.MessageMediaInstanceId)
+	if err != nil {
+		return nil, err
+	}
+	if instance == nil {
+		return &pb.FindEnabledMessageMediaInstanceResponse{MessageMediaInstance: nil}, nil
+	}
+
+	// 媒介
+	media, err := models.SharedMessageMediaDAO.FindEnabledMediaWithType(tx, instance.MediaType)
+	if err != nil {
+		return nil, err
+	}
+	if media == nil {
+		return &pb.FindEnabledMessageMediaInstanceResponse{MessageMediaInstance: nil}, nil
+	}
+	pbMedia := &pb.MessageMedia{
+		Id:              int64(media.Id),
+		Type:            media.Type,
+		Name:            media.Name,
+		Description:     media.Description,
+		UserDescription: media.UserDescription,
+		IsOn:            media.IsOn == 1,
+	}
+
+	return &pb.FindEnabledMessageMediaInstanceResponse{MessageMediaInstance: &pb.MessageMediaInstance{
+		Id:           int64(instance.Id),
+		Name:         instance.Name,
+		IsOn:         instance.IsOn == 1,
+		MessageMedia: pbMedia,
+		ParamsJSON:   []byte(instance.Params),
+		Description:  instance.Description,
+	}}, nil
+}