smart_streetlights/cloud/websvr/controllers/csensor.go

package controllers

import (
	"fmt"
	"strings"

	"github.com/astaxie/beego"
	validation "github.com/go-ozzo/ozzo-validation/v4"

	"lc/common/models"
)

type SensorController struct {
	BaseController
}

type ReqSensor struct {
	Code   string  `json:"code"`   //设备编号，禁止修改
	GID    string  `json:"gid"`    //网关ID
	Tenant string  `json:"tenant"` //租户ID
	Name   string  `json:"name"`   //设备名称
	Brand  int     `json:"brand"`  //品牌
	Model  int     `json:"model"`  //型号
	Height float32 `json:"height"` //安装高度
	State  int     `json:"state"`  //1启用，0禁用
}

func (a ReqSensor) Validate() error {
	return validation.ValidateStruct(&a,
		validation.Field(&a.Code, validation.Required, validation.Length(4, 32)),
		validation.Field(&a.GID, validation.Required, validation.Length(4, 32)),
		validation.Field(&a.Tenant, validation.Required, validation.Length(4, 8)),
		validation.Field(&a.State, validation.In(0, 1)),
	)
}

type Sensor struct {
	Code   string  `json:"code"`   //设备编号，禁止修改
	GID    string  `json:"gid"`    //网关ID
	Name   string  `json:"name"`   //设备名称
	Brand  int     `json:"brand"`  //品牌
	Model  int     `json:"model"`  //型号
	Height float32 `json:"height"` //安装高度
	State  int     `json:"state"`  //1启用，0禁用
}

func (a Sensor) Validate() error {
	return validation.ValidateStruct(&a,
		validation.Field(&a.Code, validation.Required, validation.Length(4, 32)),
		validation.Field(&a.GID, validation.Required, validation.Length(4, 32)),
		validation.Field(&a.State, validation.In(0, 1)),
	)
}

type ReqImportSensor struct {
	Tenant string   `json:"tenant"` //租户ID
	List   []Sensor `json:"list"`   //传感器列表
}

func (a ReqImportSensor) Validate() error {
	return validation.ValidateStruct(&a,
		validation.Field(&a.Tenant, validation.Required, validation.Length(4, 32)),
		validation.Field(&a.List, validation.Required),
	)
}

// CreateSensor @Title 创建传感器设备
// @Description 创建传感器设备
// @Param   body controllers.ReqSensor true "数据"
// @Success 0 {int} BaseResponse.Code "成功"
// @Failure 1 {int} BaseResponse.Code "失败"
// @router /v1/create [post]
func (o *SensorController) CreateSensor() {
	var obj ReqSensor
	if err := json.Unmarshal(o.Ctx.Input.RequestBody, &obj); err != nil {
		beego.Debug(string(o.Ctx.Input.RequestBody))
		o.Response(Failure, fmt.Sprintf("数据包解析错误:%s", err.Error()), nil)
		return
	}
	if err := obj.Validate(); err != nil {
		beego.Error(fmt.Sprintf("请求参数验证失败:%s", err.Error()))
		o.Response(Failure, fmt.Sprintf("请求参数验证失败:%s", err.Error()), nil)
		return
	}
	oo := models.DeviceSensor{
		ID:     obj.Code,
		GID:    obj.GID,
		Name:   obj.Name,
		Tenant: obj.Tenant,
		Brand:  obj.Brand,
		Model:  obj.Model,
		Height: obj.Height,
		State:  obj.State,
	}
	if err := oo.SaveFromWeb(); err != nil {
		o.Response(Failure, fmt.Sprintf("数据插入失败:%s", err.Error()), nil)
		return
	}
	o.Response(Success, "成功", oo.ID)
}

// UpdateSensor @Title 更新传感器设备
// @Description 更新传感器设备
// @Param   body body controllers.ReqSensor true "数据"
// @Success 0 {int} BaseResponse.Code "成功"
// @Failure 1 {int} BaseResponse.Code "失败"
// @router /v1/update [post]
func (o *SensorController) UpdateSensor() {
	var obj ReqSensor
	if err := json.Unmarshal(o.Ctx.Input.RequestBody, &obj); err != nil {
		beego.Debug(string(o.Ctx.Input.RequestBody))
		o.Response(Failure, fmt.Sprintf("数据包解析错误:%s", err.Error()), nil)
		return
	}
	if err := obj.Validate(); err != nil {
		beego.Error(fmt.Sprintf("请求参数验证失败:%s", err.Error()))
		o.Response(Failure, fmt.Sprintf("请求参数验证失败:%s", err.Error()), nil)
		return
	}
	oo := models.DeviceSensor{
		ID:     obj.Code,
		GID:    obj.GID,
		Name:   obj.Name,
		Tenant: obj.Tenant,
		Brand:  obj.Brand,
		Model:  obj.Model,
		Height: obj.Height,
		State:  obj.State,
	}
	if err := oo.SaveFromWeb(); err != nil {
		o.Response(Failure, fmt.Sprintf("数据更新失败:%s", err.Error()), nil)
		return
	}
	o.Response(Success, "成功", oo.ID)
}

// DeleteSensor @Title 删除传感器设备
// @Description 删除传感器设备
// @Param   code    query   string     true "设备ID"
// @Success 0 {int} BaseResponse.Code "成功"
// @Failure 1 {int} BaseResponse.Code "失败"
// @router /v1/delete [post]
func (o *SensorController) DeleteSensor() {
	code := strings.Trim(o.GetString("code"), " ")
	if err := validation.Validate(code, validation.Required, validation.Length(4, 100)); err != nil {
		beego.Error(fmt.Sprintf("请求参数验证失败:%s", err.Error()))
		o.Response(Failure, fmt.Sprintf("请求参数验证失败:%s", err.Error()), nil)
		return
	}
	c := models.DeviceSensor{
		ID: code,
	}
	if err := c.Delete(); err != nil {
		beego.Error(fmt.Sprintf("删除失败:%s", err.Error()))
		o.Response(Failure, fmt.Sprintf("数据删除失败:%s", err.Error()), nil)
		return
	}
	o.Response(Success, "成功", c.ID)
}

// ImportSensor @Title 导入环境监测设备
// @Description 导入环境监测设备
// @Param   body body controllers.ReqImportEnvironment true "数据"
// @Success 0 {int} BaseResponse.Code "成功"
// @Failure 1 {int} BaseResponse.Code "失败"
// @router /v1/import [post]
func (o *SensorController) ImportSensor() {
	var obj ReqImportSensor
	if err := json.Unmarshal(o.Ctx.Input.RequestBody, &obj); err != nil {
		beego.Debug(string(o.Ctx.Input.RequestBody))
		o.Response(Failure, fmt.Sprintf("数据包解析错误:%s", err.Error()), nil)
		return
	}
	if err := obj.Validate(); err != nil {
		beego.Error(fmt.Sprintf("请求参数验证失败:%s", err.Error()))
		o.Response(Failure, fmt.Sprintf("请求参数验证失败:%s", err.Error()), nil)
		return
	}
	var resp []RespImport
	for _, v := range obj.List {
		var aresp RespImport
		aresp.Code = v.Code
		oo := models.DeviceSensor{
			ID:     v.Code,
			GID:    v.GID,
			Name:   v.Name,
			Tenant: obj.Tenant,
			Brand:  v.Brand,
			Model:  v.Model,
			Height: v.Height,
			State:  v.State,
		}
		err := oo.SaveFromWeb()
		if err != nil {
			aresp.Error = err.Error()
			beego.Error(fmt.Sprintf("传感器数据导入失败,code=%s,失败原因:%s", v.Code, err.Error()))
		}
		resp = append(resp, aresp)
	}
	o.Response(Success, "成功", resp)
}