package controller
|
|
import (
|
"net/http"
|
"sync"
|
"time"
|
|
"gat1400Exchange/config"
|
"gat1400Exchange/pkg/auth"
|
"gat1400Exchange/pkg/logger"
|
"gat1400Exchange/repository"
|
"gat1400Exchange/vo"
|
|
"github.com/gin-gonic/gin"
|
)
|
|
type SystemController struct {
|
Auth *auth.DigestAuth
|
ApeRepo repository.ApeRepository
|
DevCache *sync.Map // 缓存设备类型, id为key, value为 ape 采集设备, subPlatform 下级, cascade上级
|
}
|
|
// 构造函数
|
func NewSystemController() SystemController {
|
svr := repository.NewApeRepository()
|
controller := SystemController{ApeRepo: svr}
|
|
realm := "Basic Realm"
|
if config.ServeConf.Realm != "" {
|
realm = config.ServeConf.Realm
|
}
|
|
controller.Auth = auth.NewDigestAuthenticator(realm, func(user, realm string) string {
|
if user == config.ServeConf.Username {
|
return config.ServeConf.Password
|
}
|
|
return config.ServeConf.Password
|
})
|
|
controller.Auth.PlainTextSecrets = true
|
|
for k, v := range svr.CollectDeviceType() {
|
controller.DevCache.Store(k, v)
|
}
|
|
return controller
|
}
|
|
// 设备注册
|
func (s SystemController) Register(c *gin.Context) {
|
user, _ := s.Auth.CheckAuth(c.Request)
|
if user == "" {
|
s.Auth.RequireAuth(c.Writer, c.Request)
|
c.AbortWithStatus(http.StatusUnauthorized)
|
return
|
}
|
|
var req vo.RequestRegister
|
if err := c.BindJSON(&req); err != nil {
|
c.AbortWithStatus(http.StatusBadRequest)
|
return
|
}
|
|
rspMsg := vo.ResponseStatus{
|
RequestURL: c.FullPath(),
|
StatusCode: vo.StatusSuccess,
|
StatusString: vo.StatusString[vo.StatusSuccess],
|
Id: req.RegisterObject.DeviceID,
|
LocalTime: time.Now().Format("20060102150405"),
|
}
|
|
if user == config.ServeConf.Username {
|
s.DevCache.Store(req.RegisterObject.DeviceID, "ape")
|
|
if err := s.ApeRepo.Create(req.RegisterObject.DeviceID); err != nil {
|
logger.Warn("Create ape failure,%s", err.Error())
|
c.AbortWithStatus(http.StatusInternalServerError)
|
return
|
}
|
} else {
|
// 未缓存的id, 可能是新添加的上下级, 更新缓存
|
_, ok := s.DevCache.Load(req.RegisterObject.DeviceID)
|
if ok {
|
s.ApeRepo.CollectDeviceType()
|
}
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ResponseStatusObject": rspMsg})
|
}
|
|
// 设备保活
|
func (s SystemController) Keepalive(c *gin.Context) {
|
var req vo.RequestKeepalive
|
if err := c.BindJSON(&req); err != nil {
|
c.AbortWithStatus(http.StatusBadRequest)
|
return
|
}
|
|
if devType, ok := s.DevCache.Load(req.KeepaliveObject.DeviceID); ok {
|
s.ApeRepo.Keepalive(req.KeepaliveObject.DeviceID, devType.(int))
|
}
|
|
rspMsg := vo.ResponseStatus{
|
RequestURL: c.FullPath(),
|
StatusCode: vo.StatusSuccess,
|
StatusString: vo.StatusString[vo.StatusSuccess],
|
Id: req.KeepaliveObject.DeviceID,
|
LocalTime: time.Now().Format("20060102150405"),
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ResponseStatusObject": rspMsg})
|
}
|
|
// 设备退出
|
func (s SystemController) UnRegister(c *gin.Context) {
|
var req vo.RequestUnRegister
|
if err := c.BindJSON(&req); err != nil {
|
c.AbortWithStatus(http.StatusBadRequest)
|
return
|
}
|
|
// 删库
|
rspMsg := vo.ResponseStatus{
|
RequestURL: c.FullPath(),
|
StatusCode: vo.StatusSuccess,
|
StatusString: vo.StatusString[vo.StatusSuccess],
|
Id: req.UnRegisterObject.DeviceID,
|
LocalTime: time.Now().Format("20060102150405"),
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ResponseStatusObject": rspMsg})
|
}
|
|
// 时间校准
|
func (s SystemController) Time(c *gin.Context) {
|
//iTime, _ := strconv.ParseInt(time.Now().Format("20060102150405"), 10, 64)
|
rspMsg := vo.ResponseSystemTime{
|
VIIDServerID: config.ServeConf.ID,
|
LocalTime: time.Now().Format("20060102150405"),
|
}
|
|
c.JSON(http.StatusOK, gin.H{"SystemTimeObject": rspMsg})
|
}
|
|
// 设备列表
|
func (s SystemController) ApeList(c *gin.Context) {
|
apeList, err := s.ApeRepo.List()
|
if err != nil {
|
logger.Error(err.Error())
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ApeList": apeList})
|
}
|
|
// 修改设备
|
func (s SystemController) ApeUpdate(c *gin.Context) {
|
var req vo.Ape
|
if err := c.BindJSON(&req); err != nil {
|
c.JSON(http.StatusBadRequest, gin.H{"msg": err.Error()})
|
return
|
}
|
|
err := s.ApeRepo.Update(&req)
|
if err != nil {
|
c.JSON(http.StatusBadRequest, gin.H{"msg": err.Error()})
|
return
|
}
|
|
c.JSON(http.StatusOK, gin.H{"msg": "ok"})
|
}
|
