添加取消订阅接口

zhangzengfei

2024-05-17 9e8d8b630affc4a4be428c775264ef4b0e5104b3

 repository/subscribeRepo.go

@@ -53,6 +53,43 @@

   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}