liuxiaolong
2020-06-05 7c811247ecf143e08c576986a884bedadc57dd66
controllers/syssetcont.go
@@ -1,11 +1,16 @@
package controllers
import (
   "basic.com/valib/logger.git"
   "net/http"
   "regexp"
   "strconv"
   "webserver/extend/code"
   "webserver/extend/config"
   "webserver/extend/sys"
   "webserver/extend/util"
   "webserver/middlewares/auth"
   "webserver/service"
   "basic.com/dbapi.git"
   "github.com/gin-gonic/gin"
@@ -16,6 +21,7 @@
type LocalConfigVo struct {
   Id                 string `json:"id"`
   AlarmIp            string `json:"alarm_ip,omitempty" example:"192.168.1.182"`
   AlarmPort          int    `json:"alarm_port,omitempty" example:"22122"`
   AlarmThresholdType int    `json:"alarm_threshold_type"` //报警阈值类型设置:1:按最高分,2:按阈值以上
@@ -31,6 +37,13 @@
   ServerType         int    `json:"server_type,omitempty" example:"1"`
   Reserved           string `json:"reserved,omitempty" example:"1"`
   RealMax            int    `json:"real_max"` //实时处理的最大路数
}
type ResourceConfigVo struct {
   IpType       int      `json:"ipType"` //0:本机ip,1:漂移ip
   ServiceIp   string    `json:"serviceIp"`  //外部服务ip
   Domain       string  `json:"domain"` //域名
   FilePort    int    `json:"filePort"`
}
type Gb28181ConfigVo struct {
@@ -78,8 +91,8 @@
// @Produce json
// @Tags sysset
// @Param config body controllers.LocalConfigVo true "报警和存储信息数据"
// @Success 200 {string} json "{"code":200, msg:"目录结构数据", success:true}"
// @Failure 500 {string} json "{"code":500,  msg:"返回错误信息", success:false}"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sysset/alarmEdit [POST]
func (sset SysSetController) AlarmEdit(c *gin.Context) {
   var args LocalConfigVo
@@ -101,7 +114,6 @@
// @Security ApiKeyAuth
// @Summary 存储信息修改
// @Description 存储信息修改
// @Accept  json
// @Produce json
// @Tags sysset
// @Success 200 {string} json "{"code":200, msg:"目录结构数据", success:true}"
@@ -159,12 +171,14 @@
func (sset SysSetController) GetDevInfo(c *gin.Context) {
   var api dbapi.SysSetApi
   b, sysconf := api.GetServerInfo()
   resData := make(map[string]string, 0)
   resData := make(map[string]interface{}, 0)
   if !b { // 查询是否存在
      util.ResponseFormat(c, code.ServiceInsideError, "查询失败")
   } else {
      resData["server_id"] = sysconf.GetServerId()
      resData["server_name"] = sysconf.GetServerName()
      resData["server_port"] = sys.GetNginxListenPort()
      ipv4, mask, _ := sys.GetLocalIP(config.Server.NetworkAdapter)
      gateway, _ := sys.GetDefaultRoute(config.Server.NetworkAdapter)
      dns, _ := sys.GetDnsServer()
@@ -172,6 +186,14 @@
      resData["subMask"] = mask
      resData["gateway"] = gateway
      resData["dns"] = dns
      resData["min_video_len"] = sysconf.CutMinDuration
      resData["max_video_len"] = sysconf.CutMaxDuration
      resData["alarm_ip"] = sysconf.AlarmIp
      resData["alarm_port"] = sysconf.AlarmPort
      resData["web_pic_ip"] = sysconf.WebPicIp
      resData["web_pic_port"] = sysconf.WebPicPort
      resData["es_pic_ip"] = sysconf.EsPicIp
      resData["es_pic_port"] = sysconf.EsPicPort
      resData["deviceNum"] = config.Server.DeviceNum
      resData["deviceType"] = config.Server.DeviceType
@@ -197,19 +219,17 @@
// @Failure 500 {string} json "{"code":500,  msg:"返回错误信息", success:false}"
// @Router /data/api-v/sysset/saveDevInfo [POST]
func (sset SysSetController) SaveDevInfo(c *gin.Context) {
   dev_name := c.PostForm("server_name")
   var api dbapi.SysSetApi
   paramBody := make(map[string]interface{}, 0)
   paramBody["serverName"] = dev_name
   // 设置ip
   isOk, msg := sys.SetNetworkConfig(c.PostForm("ip"), c.PostForm("subMask"), c.PostForm("gateway"), c.PostForm("dns"))
   if !isOk {
      util.ResponseFormat(c, code.ComError, msg)
      return
   paramBody := map[string]interface{}{
      "serverName": c.PostForm("server_name"),
      "ip": c.PostForm("ip"),
      "subMask": c.PostForm("subMask"),
      "gateway": c.PostForm("gateway"),
      "dns": c.PostForm("dns"),
   }
   b, data := api.SaveServerInfo(paramBody)
   logger.Debug("isOk b:",b, "data:", data)
   if b { // 查询是否存在
      util.ResponseFormat(c, code.Success, data)
   } else { // 存在及修改 不存在 即添加
@@ -283,7 +303,6 @@
// @Security ApiKeyAuth
// @Summary 测试同步时间
// @Description 测试同步时间服务器是否可用
// @Accept  json
// @Produce json
// @Tags sysset
// @Param server query string true "时间服务器ip"
@@ -404,6 +423,7 @@
// @Summary 设置定时重启
// @Description 设置定时重启任务的配置规则
// @Accept x-www-form-urlencoded
// @Produce json
// @Tags sysset
// @Success 200 {string} json "{"code":200, msg:"目录结构数据", success:true}"
@@ -430,3 +450,188 @@
   util.ResponseFormat(c, code.Success, "配置成功")
}
// @Security ApiKeyAuth
// @Summary 获取存储对外服务信息
// @Description 获取存储对外服务信息
// @Produce json
// @Tags sysset
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sysset/getResourceConfig [get]
func (sset SysSetController) GetResourceConfig(c *gin.Context) {
   var sysApi dbapi.SysSetApi
   b,d := sysApi.GetResourceConfig()
   if b {
      util.ResponseFormat(c,code.Success,map[string]interface{}{
         "ipType":d.IpType,
         "serviceIp": d.ServiceIp,
         "domain": d.Domain,
         "filePort": d.FilePort,
      })
   } else {
      util.ResponseFormat(c,code.ComError,"")
   }
}
// @Security ApiKeyAuth
// @Summary 存储对外服务信息
// @Description 存储对外服务信息
// @Accept  json
// @Produce json
// @Tags sysset
// @Param reqBoby body controllers.ResourceConfigVo true "post请求,json参数"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sysset/saveResourceConfig [post]
func (sset SysSetController) SaveResourceConfig(c *gin.Context) {
   var reqBody ResourceConfigVo
   c.BindJSON(&reqBody)
   if reqBody.ServiceIp == "" && reqBody.Domain == "" {
      util.ResponseFormat(c, code.RequestParamError, "")
      return
   }
   var sysApi dbapi.SysSetApi
   paramBody := util.Struct2Map(reqBody)
   if b, _ := sysApi.SaveResourceConfig(paramBody); b {
      util.ResponseFormat(c, code.Success, "")
   } else {
      util.ResponseFormat(c, code.ComError, "")
   }
}
// @Summary 系统更新包上传(分块检查)
// @Description 系统更新包上传(分块检查)
// @Accept multipart/form-data
// @Produce json
// @Tags sysset
// @Param chunkNumber formData int true "当前是第几个分块"
// @Param chunkSize formData int true "每一块的大小"
// @Param currentChunkSize formData int true "当前块的大小"
// @Param identifier formData string true "整个文件的唯一标识,目前是md5"
// @Param filename formData string true "文件名称"
// @Param relativePath formData string true "文件在客户端电脑的路径"
// @Param totalChunks formData int true "文件总块数"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sysset/patchUpdate [get]
func (sset SysSetController) PatchUpdateCheck(c *gin.Context) {
   authDriver := auth.GenerateAuthDriver()
   userM := (*authDriver).User(c)
   if userM ==nil {
      util.ResponseFormat(c,code.TokenNotFound,"登录过期,请登录")
      return
   }
   chunkNumber, e1 := strconv.Atoi(c.Request.FormValue("chunkNumber"))
   chunkSize, e2 := strconv.Atoi(c.Request.FormValue("chunkSize"))
   currentChunkSize, e3 := strconv.Atoi(c.Request.FormValue("currentChunkSize"))
   identifier := c.Request.FormValue("identifier")
   filename := c.Request.FormValue("filename")
   relativePath := c.Request.FormValue("relativePath")
   totalChunks, e5 := strconv.Atoi(c.Request.FormValue("totalChunks"))
   if identifier == "" || e1 != nil || e2 != nil || e3 != nil || e5 !=nil {
      util.ResponseFormat(c,code.RequestParamError,"参数有误")
      return
   }
   var arg = service.FileChunkCheckVo {
      UserId: userM["id"].(string),
      FileName: filename,
      Identifier: identifier,
      RelativePath: relativePath,
      TotalChunks: totalChunks,
      ChunkNumber: chunkNumber,
      ChunkSize: chunkSize,
      CurrentChunkSize:currentChunkSize,
   }
   var sv service.SysService
   if b := sv.CheckUpdateFile(&arg);b {
      c.String(http.StatusOK,"found")
   } else {
      c.String(http.StatusNoContent,"")
   }
}
// @Security ApiKeyAuth
// @Summary 系统更新包上传
// @Description 系统更新包上传
// @Accept multipart/form-data
// @Produce json
// @Tags sysset
// @Param chunkNumber formData int true "当前是第几个分块"
// @Param chunkSize formData int true "每一块的大小"
// @Param currentChunkSize formData int true "当前块的大小"
// @Param totalSize formData string true "文件总大小"
// @Param identifier formData string true "整个文件的唯一标识,目前是md5"
// @Param filename formData string true "文件名称"
// @Param relativePath formData string true "文件在客户端电脑的路径"
// @Param totalChunks formData int true "文件总块数"
// @Param file formData file true "文件分片内容"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sysset/patchUpdate [post]
func (sset SysSetController) PatchUpdate(c *gin.Context) {
   authDriver := auth.GenerateAuthDriver()
   userM := (*authDriver).User(c)
   if userM == nil {
      util.ResponseFormat(c, code.TokenNotFound, "登录过期,请登录")
      return
   }
   chunkNumber, e1 := strconv.Atoi(c.Request.FormValue("chunkNumber"))
   chunkSize, e2 := strconv.Atoi(c.Request.FormValue("chunkSize"))
   currentChunkSize, e3 := strconv.Atoi(c.Request.FormValue("currentChunkSize"))
   totalSize, e4 := strconv.ParseInt(c.Request.FormValue("totalSize"), 10, 64)
   identifier := c.Request.FormValue("identifier")
   filename := c.Request.FormValue("filename")
   relativePath := c.Request.FormValue("relativePath")
   totalChunks, e5 := strconv.Atoi(c.Request.FormValue("totalChunks"))
   file, header, e6 := c.Request.FormFile("file")
   if identifier == "" || e1 != nil || e2 != nil || e3 != nil || e4 != nil || e5 != nil || e6 != nil {
      util.ResponseFormat(c, code.RequestParamError, "参数有误")
      return
   }
   var arg = service.FileUploadVo{
      UserId:           userM["id"].(string),
      FileName:         filename,
      Identifier:       identifier,
      RelativePath:     relativePath,
      TotalSize:        totalSize,
      TotalChunks:      totalChunks,
      ChunkNumber:      chunkNumber,
      ChunkSize:        chunkSize,
      CurrentChunkSize: currentChunkSize,
      File:             &file,
      Header:           header,
   }
   var sv service.SysService
   if b,_ := sv.PatchUpload(&arg); b {
      util.ResponseFormat(c, code.Success, "")
   } else {
      util.ResponseFormat(c, code.ComError, "")
   }
}
// @Security ApiKeyAuth
// @Summary 开始升级
// @Description 开始升级
// @Accept multipart/form-data
// @Produce json
// @Tags sysset
// @Param identifier formData string true "整个文件的唯一标识,目前是md5"
// @Param filename formData string true "文件名称"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sysset/upgrade [post]
func (sset SysSetController) Upgrade(c *gin.Context) {
   identifier := c.Request.FormValue("identifier")
   filename := c.Request.FormValue("filename")
   if identifier == "" || filename == "" {
      util.ResponseFormat(c,code.RequestParamError,"")
      return
   }
   var sv service.SysService
   if b,err := sv.Upgrade(identifier, filename);b {
      util.ResponseFormat(c,code.UpgradeSuccess,"升级成功")
   } else {
      util.ResponseFormat(c,code.UpgradeFail,err.Error())
   }
}