添加webapi, 优化模块启动

zhangzengfei

2024-05-17 9ee887fce2f87f7a79d0b94640cf1d341a254319

 repository/subscribeRepo.go

@@ -1,9 +1,15 @@
package repository

import (
   "encoding/json"
   "errors"
   "fmt"
   "gat1400Exchange/client"
   "gat1400Exchange/models"
   "gat1400Exchange/pkg/snowflake"
   "gat1400Exchange/service"
   "gat1400Exchange/vo"
   "time"
)

type SubscribeRepository struct {
@@ -13,7 +19,42 @@
   return SubscribeRepository{}
}

func (s *SubscribeRepository) CreateSubscribe(fromId string, subscribe *vo.Subscribe) error {
func (s *SubscribeRepository) CreateSubscribe(sid string, req *vo.Subscribe) error {
   triggerTime := time.Now().Format("20060102150405")
   req.SubscribeID = triggerTime + snowflake.GenerateIdStr()

   var subscribeMsg vo.RequestSubscribe
   subscribeMsg.SubscribeListObject.SubscribeObject = []vo.Subscribe{*req}

   // 查找下级
   var platform models.SubPlatform
   err := platform.FindById(sid)
   if err != nil {
      return err
   }

   uri := fmt.Sprintf("http://%s:%d/VIID/Subscribes", platform.RemoteIP, platform.RemotePort)
   body, _ := json.Marshal(subscribeMsg)
   if client.Subscribe(uri, body) != vo.StatusSuccess {
      return errors.New("发送订阅消息失败")
   }

   var sub = models.Subscribe{
      Id:     req.SubscribeID,
      Status: req.SubscribeStatus,
      FromId: sid,
      Ext:    *req,
   }

   err = sub.Save()
   if err != nil {
      return err
   }

   return err
}

func (s *SubscribeRepository) SaveReceiveSubscribe(fromId string, subscribe *vo.Subscribe) error {
   var sub = models.Subscribe{
      Id:     subscribe.SubscribeID,
      Status: subscribe.SubscribeStatus,
@@ -31,7 +72,7 @@
   return err
}

func (s *SubscribeRepository) UpdateSubscribe(subscribe *vo.Subscribe) error {
func (s *SubscribeRepository) UpdateReceiveSubscribe(subscribe *vo.Subscribe) error {
   var sub = models.Subscribe{}
   err := sub.FindById(subscribe.SubscribeID)
   if err != nil {
@@ -46,7 +87,7 @@
   return sub.Save()
}

func (s *SubscribeRepository) DeleteSubscribe(id string) error {
func (s *SubscribeRepository) DeleteReceiveSubscribe(id string) error {
   var sub = models.Subscribe{}
   err := sub.DeleteById(id)
   if err != nil {
@@ -58,8 +99,8 @@
   return err
}

func (s *SubscribeRepository) List() ([]models.Subscribe, error) {
func (s *SubscribeRepository) ListByFromId(id string) ([]models.Subscribe, error) {
   var sub models.Subscribe

   return sub.FindAll()
   return sub.FindByFromId(id)
}