package controller
|
|
import (
|
"net/http"
|
"strings"
|
"time"
|
|
"gat1400Exchange/config"
|
"gat1400Exchange/pkg/logger"
|
"gat1400Exchange/repository"
|
"gat1400Exchange/vo"
|
|
"github.com/gin-gonic/gin"
|
)
|
|
type SubscribeController struct {
|
Repository repository.SubscribeRepository
|
Capture repository.CaptureRepository
|
Ape repository.ApeRepository
|
}
|
|
// 构造函数
|
func NewSubscribeController() SubscribeController {
|
svr := repository.NewSubscribeRepository()
|
svr1 := repository.NewCaptureRepository()
|
svr2 := repository.NewApeRepository()
|
controller := SubscribeController{svr, svr1, svr2}
|
|
return controller
|
}
|
|
func (s SubscribeController) VIIDSubscribesList(c *gin.Context) {
|
fromId := c.GetHeader("User-Identify")
|
subList, _ := s.Repository.ListByFromId(fromId)
|
|
c.Header("Content-Type", "application/VIID+json;charset=UTF-8")
|
c.JSON(http.StatusOK, gin.H{"SubscribeList": subList})
|
}
|
|
func (s SubscribeController) VIIDSubscribes(c *gin.Context) {
|
var req vo.RequestSubscribe
|
if err := c.BindJSON(&req); err != nil {
|
c.AbortWithStatus(http.StatusBadRequest)
|
return
|
}
|
|
fromId := c.GetHeader("User-Identify")
|
|
var rsp vo.ResponseStatusList
|
for idx, sub := range req.SubscribeListObject.SubscribeObject {
|
if err := s.Repository.SaveReceiveSubscribe(fromId, &req.SubscribeListObject.SubscribeObject[idx]); err == nil {
|
rsp.ResponseStatusObject = append(rsp.ResponseStatusObject, vo.ResponseStatus{
|
RequestURL: c.FullPath(),
|
StatusCode: vo.StatusSuccess,
|
StatusString: vo.StatusString[vo.StatusSuccess],
|
Id: sub.SubscribeID,
|
LocalTime: time.Now().Format("20060102150405"),
|
})
|
}
|
}
|
|
c.Header("Content-Type", "application/VIID+json;charset=UTF-8")
|
c.JSON(http.StatusOK, gin.H{"ResponseStatusListObject": rsp})
|
}
|
|
func (s SubscribeController) VIIDUpdateSubscribes(c *gin.Context) {
|
var req vo.RequestSubscribe
|
if err := c.BindJSON(&req); err != nil {
|
c.AbortWithStatus(http.StatusBadRequest)
|
return
|
}
|
|
var rsp vo.ResponseStatusList
|
for idx, sub := range req.SubscribeListObject.SubscribeObject {
|
if err := s.Repository.UpdateReceiveSubscribe(&req.SubscribeListObject.SubscribeObject[idx]); err == nil {
|
rsp.ResponseStatusObject = append(rsp.ResponseStatusObject, vo.ResponseStatus{
|
RequestURL: c.FullPath(),
|
StatusCode: vo.StatusSuccess,
|
StatusString: vo.StatusString[vo.StatusSuccess],
|
Id: sub.SubscribeID,
|
LocalTime: time.Now().Format("20060102150405"),
|
})
|
} else {
|
logger.Error("Update receive subscribe failure, %s", err.Error())
|
}
|
}
|
|
c.Header("Content-Type", "application/VIID+json;charset=UTF-8")
|
c.JSON(http.StatusOK, gin.H{"ResponseStatusListObject": rsp})
|
}
|
|
func (s SubscribeController) VIIDDeleteSubscribe(c *gin.Context) {
|
idList := c.Query("IDList")
|
var rsp vo.ResponseStatusList
|
|
for _, id := range strings.Split(idList, ",") {
|
if err := s.Repository.DeleteReceiveSubscribe(id); err == nil {
|
rsp.ResponseStatusObject = append(rsp.ResponseStatusObject, vo.ResponseStatus{
|
RequestURL: c.FullPath(),
|
StatusCode: vo.StatusSuccess,
|
StatusString: vo.StatusString[vo.StatusSuccess],
|
Id: id,
|
LocalTime: time.Now().Format("20060102150405"),
|
})
|
}
|
}
|
|
c.Header("Content-Type", "application/VIID+json;charset=UTF-8")
|
c.JSON(http.StatusOK, gin.H{"ResponseStatusListObject": rsp})
|
}
|
|
func (s SubscribeController) VIIDNotifications(c *gin.Context) {
|
var rsp vo.ResponseStatusList
|
var req vo.RequestSubscribeNotificationBind
|
if err := c.BindJSON(&req); err != nil {
|
c.JSON(http.StatusBadRequest, gin.H{"ResponseStatusListObject": rsp})
|
return
|
}
|
|
logger.Debug("Receive notifications: %+v", req)
|
|
for _, msg := range req.SubscribeNotificationListObject.SubscribeNotificationObject {
|
rsp.ResponseStatusObject = append(rsp.ResponseStatusObject, vo.ResponseStatus{
|
RequestURL: c.FullPath(),
|
StatusCode: vo.StatusSuccess,
|
StatusString: vo.StatusString[vo.StatusSuccess],
|
Id: msg.SubscribeID,
|
LocalTime: time.Now().Format("20060102150405"),
|
})
|
|
// 转发
|
if config.ForwardConf.SyncServer != "" && len(msg.FaceObjectList.FaceObject) > 0 {
|
go s.Capture.FaceForward(msg.FaceObjectList.FaceObject)
|
}
|
|
if config.ForwardConf.SyncServer != "" && len(msg.PersonObjectList.PersonObject) > 0 {
|
go s.Capture.PersonForward(msg.PersonObjectList.PersonObject)
|
}
|
|
if len(msg.DeviceList.APEObject) > 0 {
|
fromId := c.GetHeader("User-Identify")
|
go s.Ape.HandleNotification(fromId, msg.DeviceList.APEObject)
|
}
|
}
|
|
c.Header("Content-Type", "application/VIID+json;charset=UTF-8")
|
c.JSON(http.StatusOK, gin.H{"ResponseStatusListObject": rsp})
|
}
|
|
// CreateSubscribes 添加下级平台的订阅消息
|
func (s SubscribeController) CreateSubscribes(c *gin.Context) {
|
var req vo.Subscribe
|
if err := c.BindJSON(&req); err != nil {
|
c.JSON(http.StatusBadRequest, gin.H{"msg": err.Error()})
|
return
|
}
|
|
if c.Param("id") == "" {
|
c.JSON(http.StatusBadRequest, gin.H{"msg": "下级id为空"})
|
}
|
|
if err := s.Repository.CreateSubscribe(c.Param("id"), &req); err != nil {
|
c.JSON(http.StatusInternalServerError, gin.H{"msg": err.Error()})
|
return
|
}
|
|
c.JSON(http.StatusOK, gin.H{"msg": "ok"})
|
}
|
|
// UpdateSubscribes 更新下级平台的订阅消息
|
func (s SubscribeController) UpdateSubscribes(c *gin.Context) {
|
var req vo.Subscribe
|
if err := c.BindJSON(&req); err != nil {
|
c.JSON(http.StatusBadRequest, gin.H{"msg": err.Error()})
|
return
|
}
|
|
if c.Param("id") == "" {
|
c.JSON(http.StatusBadRequest, gin.H{"msg": "下级id为空"})
|
}
|
|
if err := s.Repository.UpdateSubscribe(c.Param("id"), &req); err != nil {
|
c.JSON(http.StatusInternalServerError, gin.H{"msg": err.Error()})
|
return
|
}
|
|
c.JSON(http.StatusOK, gin.H{"msg": "ok"})
|
}
|
|
func (s SubscribeController) CancelSubscribes(c *gin.Context) {
|
var req vo.Subscribe
|
if err := c.BindJSON(&req); err != nil {
|
c.JSON(http.StatusBadRequest, gin.H{"msg": err.Error()})
|
return
|
}
|
|
if c.Param("id") == "" {
|
c.JSON(http.StatusBadRequest, gin.H{"msg": "下级id为空"})
|
}
|
|
if err := s.Repository.CancelSubscribe(c.Param("id"), &req); err != nil {
|
c.JSON(http.StatusInternalServerError, gin.H{"msg": err.Error()})
|
return
|
}
|
|
c.JSON(http.StatusOK, gin.H{"msg": "ok"})
|
}
|
|
func (s SubscribeController) List(c *gin.Context) {
|
if c.Param("id") == "" {
|
c.JSON(http.StatusBadRequest, gin.H{"msg": "下级id为空"})
|
}
|
|
list, err := s.Repository.ListByFromId(c.Param("id"))
|
if err != nil {
|
c.JSON(http.StatusInternalServerError, gin.H{"msg": err.Error()})
|
return
|
}
|
|
c.JSON(http.StatusOK, gin.H{"msg": "ok", "data": list})
|
}
|
|
func (s SubscribeController) RemoteList(c *gin.Context) {
|
if c.Param("id") == "" {
|
c.JSON(http.StatusBadRequest, gin.H{"msg": "下级id为空"})
|
}
|
|
data, err := s.Repository.RemoteList(c.Param("id"))
|
if err != nil {
|
c.JSON(http.StatusInternalServerError, gin.H{"msg": err.Error()})
|
return
|
}
|
|
c.JSON(http.StatusOK, gin.H{"msg": "ok", "data": data})
|
}
|
