zhangzengfei
2024-10-11 539a78196da60eb97cf7057c5c85dfaa9b240741
repository/subscribeRepo.go
@@ -54,6 +54,87 @@
   return err
}
func (s *SubscribeRepository) CancelSubscribe(sid string, req *vo.Subscribe) error {
   var subscribeMsg vo.RequestCancelSubscribe
   subscribeMsg.SubscribeObject = *req
   if req.SubscribeID == "" {
      return errors.New("订阅消息主题为空")
   }
   // 查找下级
   var platform models.SubPlatform
   err := platform.FindById(sid)
   if err != nil {
      return err
   }
   uri := fmt.Sprintf("http://%s:%d/VIID/Subscribes/%s", platform.RemoteIP, platform.RemotePort, req.SubscribeID)
   body, _ := json.Marshal(subscribeMsg)
   if client.UpdateSubscribe(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) UpdateSubscribe(sid string, req *vo.Subscribe) error {
   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.UpdateSubscribe(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) RemoteList(sid string) ([]byte, error) {
   // 查找下级
   var platform models.SubPlatform
   err := platform.FindById(sid)
   if err != nil {
      return nil, err
   }
   uri := fmt.Sprintf("http://%s:%d/VIID/Subscribes", platform.RemoteIP, platform.RemotePort)
   return client.GetSubscribes(uri)
}
func (s *SubscribeRepository) SaveReceiveSubscribe(fromId string, subscribe *vo.Subscribe) error {
   var sub = models.Subscribe{
      Id:     subscribe.SubscribeID,
@@ -67,7 +148,7 @@
      return err
   }
   service.UpdateTaskProcs(sub.Id, vo.Msg_Type_Create_Subscribe, &sub)
   service.UpdateNotificationTask(sub.Id, vo.Msg_Type_Create_Subscribe, &sub)
   return err
}
@@ -82,7 +163,7 @@
   sub.Status = subscribe.SubscribeStatus
   sub.Ext = *subscribe
   service.UpdateTaskProcs(subscribe.SubscribeID, vo.Msg_Type_Update_Subscribe, nil)
   service.UpdateNotificationTask(subscribe.SubscribeID, vo.Msg_Type_Update_Subscribe, &sub)
   return sub.Save()
}
@@ -94,7 +175,7 @@
      return err
   }
   service.UpdateTaskProcs(id, vo.Msg_Type_Delete_Subscribe, nil)
   service.UpdateNotificationTask(id, vo.Msg_Type_Delete_Subscribe, nil)
   return err
}