smart-enterprise-management-gateway/controller/v1/device.go

// Copyright 2019 github.com. All rights reserved.
// Use of this source code is governed by github.com.

package v1

import (
	"github.com/gin-gonic/gin"
	"github.com/jaryhe/gopkgs/logger"
	"github.com/jaryhe/gopkgs/tasker/httptasker"
	"github.com/jaryhe/gopkgs/util"
	"go.uber.org/zap"
	"net/http"
	"smart-enterprise-management-gateway/consts"
	"smart-enterprise-management-gateway/errors"
	param_v1 "smart-enterprise-management-gateway/param/v1"
	"smart-enterprise-management-gateway/pb"
	"smart-enterprise-management-gateway/pb/v1"
	"smart-enterprise-management-gateway/utils"
	"strconv"
	"strings"
)

// 设备类型总列表
// @Summary 设备类型总列表
// @Description 设备类型总列表
// @Tags device
// @Accept  json
// @Produce  json
// @Success 200 {object} v1.DeviceAllTypeListResponse
// @Failure 500 {object} base.HTTPError
// @Router /api/v1/device_type_all [get]
func (c *Controller) DeviceTypeAll(ctx *gin.Context) {

	// 业务处理
	handleServiceTask := func() error {
		// 响应数据

		resp := param_v1.DeviceAllTypeListResponse{}
		rpcReq := &v1.DeviceAllTypeListRequest{}
		reply, err := pb.Enterprise.DeviceAllTypeList(ctx, rpcReq)
		if err != nil {

			logger.Error("func",
				zap.String("call", "DeviceAllTypeList"),
				zap.String("error", err.Error()))

			return errors.ErrorTransForm(err)
		}
		resp.Data = *reply
		if resp.Data.List == nil {
			resp.Data.List= make([]*v1.DeviceAllTypeItem, 0)
		}
		ctx.JSON(http.StatusOK, resp)
		return nil
	}

	// 执行任务
	httptasker.Exec(ctx, handleServiceTask)
}

// 已通过审核的设备类型列表
// @Summary 已通过审核的设备类型列表
// @Description 已通过审核的设备类型列表
// @Tags device
// @Accept  json
// @Produce  json
// @Param token header string true " "
// @Param filter query string false " "
// @Param page query int true " "
// @Success 200 {object} v1.DeviceTypeListResponse
// @Failure 500 {object} base.HTTPError
// @Router /api/v1/device/type_list [get]
func (c *Controller) DeviceTypeList(ctx *gin.Context) {
	// 解析参数
	req := &param_v1.DeviceTypeListRequest{}
	//var loginUid int64
	//var userName string
	parseParamTask := func() error {
		err := util.ShouldBind(ctx, &req.Header, nil, &req.DeviceTypeListQuery, nil)
		if err != nil {
			logger.Error("func",
				zap.String("call", "util.ShouldBind"),
				zap.String("error", err.Error()))
			return errors.ParamsError
		}
		//loginUid, userName, err = utils.GetJwtIdFromCtx(ctx)
		return nil
	}

	// 业务处理
	handleServiceTask := func() error {
		// 响应数据

		resp := param_v1.DeviceTypeListResponse{}
		rpcReq := &v1.DeviceTypeListRequest{}
		rpcReq.Page = req.Page
		if rpcReq.Page < 1 {
			rpcReq.Page = 1
		}
		rpcReq.Filter = req.Filter
		req.StatusFilters = "3"
		if req.StatusFilters != "" {
			array := strings.Split(req.StatusFilters, ",")
			for _, v := range array {
				value, _ := strconv.Atoi(v)
				rpcReq.FilterStatus = append(rpcReq.FilterStatus, int32(value))
			}
		}
		reply, err := pb.Enterprise.DeviceTypeList(ctx, rpcReq)
		if err != nil {
			s, _ := json.MarshalToString(req)
			logger.Error("func",
				zap.String("call", "DeviceTypeList"),
				zap.String("params", s),
				zap.String("error", err.Error()))

			return errors.ErrorTransForm(err)
		}
		resp.Data = *reply
		if resp.Data.List == nil {
			resp.Data.List = make([]*v1.DeviceTypeItem, 0)
		}
		ctx.JSON(http.StatusOK, resp)
		return nil
	}

	// 执行任务
	httptasker.Exec(ctx, parseParamTask, handleServiceTask)

}


// 扬尘设备列表
// @Summary 扬尘设备列表
// @Description 扬尘设备列表
// @Tags device
// @Accept  json
// @Produce  json
// @Param token header string true " "
// @Param page query int false " "
// @Param filter query string false " "
// @Param project_id query int64 false " "
// @Param state query int32 false " -1 0 1"
// @Success 200 {object} v1.DeviceListResponse
// @Failure 500 {object} base.HTTPError
// @Router /api/v1/device/dust_list [get]
func (c *Controller) DeviceDustList(ctx *gin.Context) {
	// 解析参数
	req := &param_v1.DeviceListRequest{}
	var loginUid int64
	//var userName string
	parseParamTask := func() error {
		err := util.ShouldBind(ctx, &req.Header, nil, &req.DeviceListQuery, nil)
		if err != nil {
			logger.Error("func",
				zap.String("call", "util.ShouldBind"),
				zap.String("error", err.Error()))
			return errors.ParamsError
		}
		loginUid, _, err = utils.GetJwtIdFromCtx(ctx)
		return nil
	}

	// 业务处理
	handleServiceTask := func() error {
		// 响应数据

		resp := param_v1.DeviceListResponse{}
		rpcReq := &v1.DeviceListRequest{}
		rpcReq.Page = req.Page
		if rpcReq.Page < 1 {
			rpcReq.Page = 1
		}
		rpcReq.Cid = loginUid
		rpcReq.Filter = req.Filter
		rpcReq.TypeCode = consts.DeviceTypeDust
		req.FilterStatus = "1"
		rpcReq.State = req.State
		rpcReq.ProjectId = req.ProjectId
		if req.FilterStatus != "" {
			array := strings.Split(req.FilterStatus, ",")
			for _, v := range array {
				value, _ := strconv.Atoi(v)
				rpcReq.StatusFilters = append(rpcReq.StatusFilters, int32(value))
			}
		}
		reply, err := pb.Enterprise.DeviceList(ctx, rpcReq)
		if err != nil {
			s, _ := json.MarshalToString(req)
			logger.Error("func",
				zap.String("call", "DeviceList"),
				zap.String("params", s),
				zap.String("error", err.Error()))

			return errors.ErrorTransForm(err)
		}
		resp.Data = *reply
		if resp.Data.List == nil {
			resp.Data.List = make([]*v1.DeviceItem, 0)
		}
		ctx.JSON(http.StatusOK, resp)
		return nil
	}

	// 执行任务
	httptasker.Exec(ctx, parseParamTask, handleServiceTask)
}

// 塔机设备列表
// @Summary 塔机设备列表
// @Description 塔机设备列表
// @Tags device
// @Accept  json
// @Produce  json
// @Param token header string true " "
// @Param page query int false " "
// @Param filter query string false " "
// @Param project_id query int64 false " "
// @Param state query int32 false " -1 0 1"
// @Success 200 {object} v1.DeviceListResponse
// @Failure 500 {object} base.HTTPError
// @Router /api/v1/device/tower_list [get]
func (c *Controller) DeviceTowerList(ctx *gin.Context) {
	// 解析参数
	req := &param_v1.DeviceListRequest{}
	var loginUid int64
	//var userName string
	parseParamTask := func() error {
		err := util.ShouldBind(ctx, &req.Header, nil, &req.DeviceListQuery, nil)
		if err != nil {
			logger.Error("func",
				zap.String("call", "util.ShouldBind"),
				zap.String("error", err.Error()))
			return errors.ParamsError
		}
		loginUid, _, err = utils.GetJwtIdFromCtx(ctx)
		return nil
	}

	// 业务处理
	handleServiceTask := func() error {
		// 响应数据

		resp := param_v1.DeviceListResponse{}
		rpcReq := &v1.DeviceListRequest{}
		rpcReq.Page = req.Page
		if rpcReq.Page < 1 {
			rpcReq.Page = 1
		}
		rpcReq.Cid = loginUid
		rpcReq.Filter = req.Filter
		rpcReq.TypeCode = consts.DeviceTypeTower
		req.FilterStatus = "1"
		rpcReq.State = req.State
		rpcReq.ProjectId = req.ProjectId
		if req.FilterStatus != "" {
			array := strings.Split(req.FilterStatus, ",")
			for _, v := range array {
				value, _ := strconv.Atoi(v)
				rpcReq.StatusFilters = append(rpcReq.StatusFilters, int32(value))
			}
		}
		reply, err := pb.Enterprise.DeviceList(ctx, rpcReq)
		if err != nil {
			s, _ := json.MarshalToString(req)
			logger.Error("func",
				zap.String("call", "DeviceList"),
				zap.String("params", s),
				zap.String("error", err.Error()))

			return errors.ErrorTransForm(err)
		}
		resp.Data = *reply
		if resp.Data.List == nil {
			resp.Data.List = make([]*v1.DeviceItem, 0)
		}
		ctx.JSON(http.StatusOK, resp)
		return nil
	}

	// 执行任务
	httptasker.Exec(ctx, parseParamTask, handleServiceTask)
}

// 升降机设备列表
// @Summary 升降机设备列表
// @Description 升降机设备列表
// @Tags device
// @Accept  json
// @Produce  json
// @Param token header string true " "
// @Param page query int false " "
// @Param filter query string false " "
// @Param project_id query int64 false " "
// @Param state query int32 false " -1 0 1"
// @Success 200 {object} v1.DeviceListResponse
// @Failure 500 {object} base.HTTPError
// @Router /api/v1/device/lift_list [get]
func (c *Controller) DeviceLiftList(ctx *gin.Context) {
	// 解析参数
	req := &param_v1.DeviceListRequest{}
	var loginUid int64
	//var userName string
	parseParamTask := func() error {
		err := util.ShouldBind(ctx, &req.Header, nil, &req.DeviceListQuery, nil)
		if err != nil {
			logger.Error("func",
				zap.String("call", "util.ShouldBind"),
				zap.String("error", err.Error()))
			return errors.ParamsError
		}
		loginUid, _, err = utils.GetJwtIdFromCtx(ctx)
		return nil
	}

	// 业务处理
	handleServiceTask := func() error {
		// 响应数据

		resp := param_v1.DeviceListResponse{}
		rpcReq := &v1.DeviceListRequest{}
		rpcReq.Page = req.Page
		if rpcReq.Page < 1 {
			rpcReq.Page = 1
		}
		rpcReq.Cid = loginUid
		rpcReq.Filter = req.Filter
		rpcReq.TypeCode = consts.DeviceTypeLift
		req.FilterStatus = "1"
		rpcReq.State = req.State
		rpcReq.ProjectId = req.ProjectId
		if req.FilterStatus != "" {
			array := strings.Split(req.FilterStatus, ",")
			for _, v := range array {
				value, _ := strconv.Atoi(v)
				rpcReq.StatusFilters = append(rpcReq.StatusFilters, int32(value))
			}
		}
		reply, err := pb.Enterprise.DeviceList(ctx, rpcReq)
		if err != nil {
			s, _ := json.MarshalToString(req)
			logger.Error("func",
				zap.String("call", "DeviceList"),
				zap.String("params", s),
				zap.String("error", err.Error()))

			return errors.ErrorTransForm(err)
		}
		resp.Data = *reply
		if resp.Data.List == nil {
			resp.Data.List = make([]*v1.DeviceItem, 0)
		}
		ctx.JSON(http.StatusOK, resp)
		return nil
	}

	// 执行任务
	httptasker.Exec(ctx, parseParamTask, handleServiceTask)
}

// 实名制设备列表
// @Summary 实名制设备列表
// @Description 实名制设备列表
// @Tags device
// @Accept  json
// @Produce  json
// @Param token header string true " "
// @Param page query int false " "
// @Param filter query string false " "
// @Param project_id query int64 false " "
// @Param state query int32 false " -1 0 1"
// @Success 200 {object} v1.DeviceListResponse
// @Failure 500 {object} base.HTTPError
// @Router /api/v1/device/attendance_list [get]
func (c *Controller) DeviceAttendanceList(ctx *gin.Context) {
	// 解析参数
	req := &param_v1.DeviceListRequest{}
	var loginUid int64
	//var userName string
	parseParamTask := func() error {
		err := util.ShouldBind(ctx, &req.Header, nil, &req.DeviceListQuery, nil)
		if err != nil {
			logger.Error("func",
				zap.String("call", "util.ShouldBind"),
				zap.String("error", err.Error()))
			return errors.ParamsError
		}
		loginUid, _, err = utils.GetJwtIdFromCtx(ctx)
		return nil
	}

	// 业务处理
	handleServiceTask := func() error {
		// 响应数据

		resp := param_v1.DeviceListResponse{}
		rpcReq := &v1.DeviceListRequest{}
		rpcReq.Page = req.Page
		if rpcReq.Page < 1 {
			rpcReq.Page = 1
		}
		rpcReq.Cid = loginUid
		rpcReq.Filter = req.Filter
		rpcReq.TypeCode = consts.DeviceTypeAttendance
		req.FilterStatus = "1"
		rpcReq.State = req.State
		rpcReq.ProjectId = req.ProjectId
		if req.FilterStatus != "" {
			array := strings.Split(req.FilterStatus, ",")
			for _, v := range array {
				value, _ := strconv.Atoi(v)
				rpcReq.StatusFilters = append(rpcReq.StatusFilters, int32(value))
			}
		}
		reply, err := pb.Enterprise.DeviceList(ctx, rpcReq)
		if err != nil {
			s, _ := json.MarshalToString(req)
			logger.Error("func",
				zap.String("call", "DeviceList"),
				zap.String("params", s),
				zap.String("error", err.Error()))

			return errors.ErrorTransForm(err)
		}
		resp.Data = *reply
		if resp.Data.List == nil {
			resp.Data.List = make([]*v1.DeviceItem, 0)
		}
		ctx.JSON(http.StatusOK, resp)
		return nil
	}

	// 执行任务
	httptasker.Exec(ctx, parseParamTask, handleServiceTask)
}

// 视频设备列表
// @Summary 视频设备列表
// @Description 视频设备列表
// @Tags device
// @Accept  json
// @Produce  json
// @Param token header string true " "
// @Param filter query string false "过滤字段"
// @Param project_id query int64 false " "
// @Param state query int32 false "-1,0,1"
// @Param page query int false " "
// @Success 200 {object} v1.VedioListResponse
// @Failure 500 {object} base.HTTPError
// @Router /api/v1/device/vedio_list [get]
func (c *Controller) DeviceVedioList(ctx *gin.Context) {
	// 解析参数
	req := &param_v1.VedioListRequest{}
	var loginUid int64
	//var userName string
	parseParamTask := func() error {
		err := util.ShouldBind(ctx, &req.Header, nil, &req.VedioListQuery, nil)
		if err != nil {
			logger.Error("func",
				zap.String("call", "util.ShouldBind"),
				zap.String("error", err.Error()))
			return errors.ParamsError
		}
		loginUid, _, err = utils.GetJwtIdFromCtx(ctx)
		return nil
	}

	// 业务处理
	handleServiceTask := func() error {
		// 响应数据

		resp := param_v1.VedioListResponse{}
		rpcReq := &v1.VedioListRequest{}
		rpcReq.Page = req.Page
		rpcReq.Cid = loginUid
		rpcReq.Filter = req.Filter
		rpcReq.Cid = loginUid
		rpcReq.ProjectId = req.ProjectId
		rpcReq.State = req.State
		rpcReq.StatusFilters = append(rpcReq.StatusFilters, 1)
		/*
			if req.FilterStatus != "" {
				array := strings.Split(req.FilterStatus, ",")
				for _, v := range array {
					value, _ := strconv.Atoi(v)
					rpcReq.StatusFilters = append(rpcReq.StatusFilters, int32(value))
				}
			}*/
		reply, err := pb.Enterprise.VedioList(ctx, rpcReq)
		if err != nil {
			s, _ := json.MarshalToString(req)
			logger.Error("func",
				zap.String("call", "VedioList"),
				zap.String("params", s),
				zap.String("error", err.Error()))

			return errors.ErrorTransForm(err)
		}
		resp.Data = *reply
		if resp.Data.List == nil {
			resp.Data.List = make([]*v1.VedioItem, 0)
		}
		ctx.JSON(http.StatusOK, resp)
		return nil
	}

	// 执行任务
	httptasker.Exec(ctx, parseParamTask, handleServiceTask)
}


// 通道列表
// @Summary 通道列表
// @Description 通道列表
// @Tags device
// @Accept  json
// @Produce  json
// @Param token header string true " "
// @Param sn query string true " "
// @Param page query int false " "
// @Param channel_no query string false " "
// @Success 200 {object} v1.ChannelListResponse
// @Failure 500 {object} base.HTTPError
// @Router /api/v1/device/vedio_channel_list [get]
func (c *Controller) DeviceVedioChannelList(ctx *gin.Context) {
	// 解析参数
	req := &param_v1.ChannelListRequest{}
	//var loginUid int64
	//var userName string
	parseParamTask := func() error {
		err := util.ShouldBind(ctx, &req.Header, nil, &req.ChannelListQuery, nil)
		if err != nil {
			logger.Error("func",
				zap.String("call", "util.ShouldBind"),
				zap.String("error", err.Error()))
			return errors.ParamsError
		}
		//loginUid, userName, err = utils.GetJwtIdFromCtx(ctx)
		return nil
	}

	// 业务处理
	handleServiceTask := func() error {
		// 响应数据

		resp := param_v1.ChannelListResponse{}
		rpcReq := &v1.ChannelListRequest{}
		rpcReq.Page = req.Page
		rpcReq.Sn = req.Sn
		rpcReq.ChannelNo = req.ChannelNo
		/*
			if req.FilterStatus != "" {
				array := strings.Split(req.FilterStatus, ",")
				for _, v := range array {
					value, _ := strconv.Atoi(v)
					rpcReq.StatusFilters = append(rpcReq.StatusFilters, int32(value))
				}
			}*/
		reply, err := pb.Enterprise.ChannelList(ctx, rpcReq)
		if err != nil {
			s, _ := json.MarshalToString(req)
			logger.Error("func",
				zap.String("call", "ChannelList"),
				zap.String("params", s),
				zap.String("error", err.Error()))

			return errors.ErrorTransForm(err)
		}
		resp.Data = *reply
		if resp.Data.List == nil {
			resp.Data.List = make([]*v1.ChannelItem, 0)
		}
		ctx.JSON(http.StatusOK, resp)
		return nil
	}

	// 执行任务
	httptasker.Exec(ctx, parseParamTask, handleServiceTask)
}