菜单和安防

app/device/dao/cameraDao.go

@@ -102,3 +102,10 @@ func (c CameraDevice) GetDevicesByLampPole() []CameraDevice {
 		c.LampPoleId).Find(&devices)
 	return devices
 }
+
+func (c CameraDevice) GetDevicesByLampPoleGroup() []CameraDevice {
+	var devices []CameraDevice
+	Db.Debug().Model(&c).Where("group_id = ? and is_deleted = 0",
+		c.GroupId).Find(&devices)
+	return devices
+}

app/device/service/cameraService.go

@@ -185,3 +185,11 @@ func (s *cameraService) GetByLampPole(id int) []dao.CameraDevice {
 	}
 	return device.GetDevicesByLampPole()
 }
+
+func (s *cameraService) GetByLampPoleGroup(id int) []dao.CameraDevice {
+	// 创建查询实例
+	device := &dao.CameraDevice{
+		GroupId: id,
+	}
+	return device.GetDevicesByLampPoleGroup()
+}

app/security/controller/securityController.go

@@ -0,0 +1,26 @@
+package controller
+
+import (
+	"github.com/gin-gonic/gin"
+	"iot_manager_service/app/middleware"
+	"iot_manager_service/app/security/service"
+	"iot_manager_service/util"
+	"net/http"
+)
+
+// Security 安防
+var Security = new(securityCtl)
+
+type securityCtl struct{}
+
+func (c *securityCtl) GetCameraLiveList(ctx *gin.Context) {
+	value, _ := ctx.Get(middleware.Authorization)
+	claims := value.(*middleware.Claims)
+
+	list, err := service.SecurityService.GetCameraLiveList(claims.TenantId)
+	if err != nil {
+		ctx.JSON(http.StatusOK, err)
+		return
+	}
+	ctx.JSON(http.StatusOK, util.SuccessResponse(util.Success, list))
+}

app/security/model/security.go

@@ -0,0 +1,16 @@
+package model
+
+type CameraLiveDetail struct {
+	PoleGroupName string       `json:"poleGroupName"`
+	CameraList    []CameraLive `json:"cameraList"`
+}
+
+type CameraLive struct {
+	Id             int    `json:"id"`
+	DeviceName     string `json:"deviceName"`
+	DeviceSn       string `json:"deviceSn"`
+	PoleGroupName  string `json:"poleGroupName"`
+	MonitorAddress string `json:"monitorAddress"`
+	StreamId       string `json:"streamId"`
+	RunState       string `json:"runState"`
+}

app/security/service/securityService.go

@@ -0,0 +1,45 @@
+package service
+
+import (
+	"iot_manager_service/app/device/service"
+	"iot_manager_service/app/security/model"
+	"iot_manager_service/util"
+)
+
+var SecurityService = new(securityService)
+
+type securityService struct{}
+
+func (s *securityService) GetCameraLiveList(tenantId int) ([]model.CameraLiveDetail, *util.Errors) {
+	groups, err := service.LampPoleGroupService.GetList(tenantId)
+	if err != nil {
+		return nil, nil
+	}
+
+	var rsp []model.CameraLiveDetail
+	for _, group := range groups {
+		cameraDevices := service.CameraService.GetByLampPoleGroup(group.ID)
+		var cameraList []model.CameraLive
+		for _, cameraDevice := range cameraDevices {
+			_, state := util.GetDeviceState(cameraDevice.DeviceSN)
+			poleGroupName := cameraDevice.DeviceName + "(离线)"
+			if state == "1" {
+				poleGroupName = cameraDevice.DeviceName + "在线"
+			}
+			cameraList = append(cameraList, model.CameraLive{
+				Id:             cameraDevice.ID,
+				DeviceName:     cameraDevice.DeviceName,
+				DeviceSn:       cameraDevice.DeviceSN,
+				PoleGroupName:  poleGroupName,
+				MonitorAddress: cameraDevice.MonitorAddress,
+				StreamId:       cameraDevice.StreamId,
+				RunState:       state,
+			})
+		}
+		rsp = append(rsp, model.CameraLiveDetail{
+			PoleGroupName: group.PoleGroupName,
+			CameraList:    cameraList,
+		})
+	}
+	return rsp, nil
+}

app/system/controller/menu.go

@@ -34,7 +34,21 @@ func (c *menu) List(ctx *gin.Context) {
 }
 
 func (c *menu) LazyList(ctx *gin.Context) {
+	id := ctx.Query("parentId")
+	name := ctx.Query("name")
+	code := ctx.Query("code")
+	alias := ctx.Query("alias")
+	var parentId int64 = -1
+	if id != "" {
+		parentId, _ = strconv.ParseInt(id, 10, 64)
+	}
 
+	menus, err := service.MenuService.LazyList(parentId, name, code, alias)
+	if err != nil {
+		ctx.JSON(http.StatusOK, util.FailResponse(err.Error(), nil))
+		return
+	}
+	ctx.JSON(http.StatusOK, util.SuccessResponse(util.Succeeded, menus))
 }
 
 func (c *menu) MenuList(ctx *gin.Context) {
@@ -60,16 +74,15 @@ func (c *menu) Remove(ctx *gin.Context) {
 }
 
 func (c *menu) Routes(ctx *gin.Context) {
-	value, isExist := ctx.Get(middleware.Authorization)
-	if !isExist || value == nil {
-		ctx.JSON(http.StatusUnauthorized, util.NormalResponse(http.StatusUnauthorized, "", nil))
-	}
-	menus, err := service.MenuService.Routes(value.(*middleware.Claims).RoleId)
+	value, _ := ctx.Get(middleware.Authorization)
+	claims := value.(*middleware.Claims)
+
+	menus, err := service.MenuService.Routes(claims.RoleId)
 	if err != nil {
 		ctx.JSON(http.StatusOK, util.FailResponse(err.Error(), nil))
 		return
 	}
-	ctx.JSON(http.StatusOK, menus)
+	ctx.JSON(http.StatusOK, util.SuccessResponse(util.Succeeded, menus))
 }
 
 func (c *menu) RoutesExt(cxt *gin.Context) {
@@ -81,7 +94,12 @@ func (c *menu) Buttons(ctx *gin.Context) {
 }
 
 func (c *menu) Tree(ctx *gin.Context) {
-
+	menus, err := service.MenuService.Tree()
+	if err != nil {
+		ctx.JSON(http.StatusOK, util.FailResponse(err.Error(), nil))
+		return
+	}
+	ctx.JSON(http.StatusOK, util.SuccessResponse(util.Succeeded, menus))
 }
 
 func (c *menu) GrantTree(ctx *gin.Context) {

app/system/dao/menu.go

@@ -38,3 +38,37 @@ func (c *Menu) GetAll() ([]Menu, error) {
 	err := Db.Debug().Model(&c).Where(" is_deleted = 0 and category = 1").Order("sort").Find(&menus).Error
 	return menus, err
 }
+
+func (c *Menu) GetByParentId() ([]Menu, error) {
+	var menus []Menu
+	db := Db.Debug().Model(&c).Where(" is_deleted = 0")
+	if c.ParentId >= 0 {
+		db = db.Where("parent_id = ?", c.ParentId)
+	}
+	if c.Name != "" {
+		db = db.Where("name like ?", "%"+c.Name+"%")
+	}
+	if c.Code != "" {
+		db = db.Where("code like ?", "%"+c.Code+"%")
+	}
+	if c.Alias != "" {
+		db = db.Where("alias like ?", "%"+c.Alias+"%")
+	}
+
+	err := db.Order("sort").Find(&menus).Error
+	return menus, err
+}
+
+func (c *Menu) HasChildren(parentIds []int64) (map[int64]bool, error) {
+	type hasChildren struct {
+		ParentId int64 `gorm:"type:bigint"`
+		Count    int   `gorm:"type:int"`
+	}
+	var result []hasChildren
+	err := Db.Debug().Table("menu").Select("parent_id, count(*) count").Where(" is_deleted = 0 and parent_id in (?)", parentIds).Group("parent_id").Find(&result).Error
+	m := make(map[int64]bool)
+	for _, item := range result {
+		m[item.ParentId] = item.Count > 0
+	}
+	return m, err
+}

app/system/model/menu.go

@@ -10,12 +10,17 @@ type RsqMenuList struct {
 	Total   int        `json:"total"`    //总数
 }
 
-type RsqMenuRoutes struct {
-	MenuRouteDetails []MenuRouteDetail `json:"data"`
-}
-
 type MenuRouteDetail struct {
 	dao.Menu
 	HasChildren bool              `json:"hasChildren"`
 	Children    []MenuRouteDetail `json:"children"`
 }
+
+type RsqMenuLazyList struct {
+	List []MenuLazyListDetail `json:"data"`
+}
+
+type MenuLazyListDetail struct {
+	dao.Menu
+	HasChildren bool `json:"hasChildren"`
+}

app/system/service/menuService.go

@@ -46,8 +46,33 @@ func (s *menuService) GetAll() []dao.Menu {
 	return menus
 }
 
-func (s *menuService) Routes(roleId int64) (*model.RsqMenuRoutes, error) {
-	allMenus := MenuService.GetAll()
+func (s *menuService) Tree() ([]model.MenuRouteDetail, error) {
+	allMenus := s.GetAll()
+
+	var rsp []model.MenuRouteDetail
+	//上级角色——多个下级角色数组
+	tmp := make(map[int64][]dao.Menu)
+	for _, menu := range allMenus {
+		childes, isExist := tmp[menu.ParentId]
+		if isExist {
+			childes = append(childes, menu)
+			tmp[menu.ParentId] = childes
+		} else {
+			tmp[menu.ParentId] = []dao.Menu{menu}
+		}
+	}
+
+	if len(tmp) == 0 {
+		return nil, nil
+	}
+	roots, _ := tmp[0]
+	rsp = menuDfs(rsp, roots, tmp)
+
+	return rsp, nil
+}
+
+func (s *menuService) Routes(roleId int64) ([]model.MenuRouteDetail, error) {
+	allMenus := s.GetAll()
 	menuMap := make(map[int64]dao.Menu)
 	for _, menu := range allMenus {
 		menuMap[menu.ID] = menu
@@ -55,7 +80,7 @@ func (s *menuService) Routes(roleId int64) (*model.RsqMenuRoutes, error) {
 
 	roleMenus := RoleMenuService.GetMenuByRole(roleId)
 
-	rsp := &model.RsqMenuRoutes{}
+	var rsp []model.MenuRouteDetail
 	//上级角色——多个下级角色数组
 	tmp := make(map[int64][]dao.Menu)
 	for _, roleMenu := range roleMenus {
@@ -75,7 +100,7 @@ func (s *menuService) Routes(roleId int64) (*model.RsqMenuRoutes, error) {
 		return nil, nil
 	}
 	roots, _ := tmp[0]
-	rsp.MenuRouteDetails = menuDfs(rsp.MenuRouteDetails, roots, tmp)
+	rsp = menuDfs(rsp, roots, tmp)
 
 	return rsp, nil
 }
@@ -99,3 +124,33 @@ func menuDfs(result []model.MenuRouteDetail, menus []dao.Menu, tmp map[int64][]d
 	}
 	return result
 }
+
+func (s *menuService) LazyList(parentId int64, name, code, alias string) ([]model.MenuLazyListDetail, error) {
+	menu := &dao.Menu{
+		ParentId: parentId,
+		Name:     name,
+		Code:     code,
+		Alias:    alias,
+	}
+	menus, err := menu.GetByParentId()
+	if err != nil {
+		return nil, err
+	}
+	var parentIds []int64
+	for _, m := range menus {
+		parentIds = append(parentIds, m.ID)
+	}
+	hasChildrenMap, err := menu.HasChildren(parentIds)
+	if err != nil {
+		return nil, err
+	}
+	var rsp []model.MenuLazyListDetail
+	for _, m := range menus {
+		hasChildren := hasChildrenMap[m.ID]
+		rsp = append(rsp, model.MenuLazyListDetail{
+			Menu:        m,
+			HasChildren: hasChildren,
+		})
+	}
+	return rsp, nil
+}

router/router.go

@@ -5,6 +5,7 @@ import (
 	device "iot_manager_service/app/device/controller"
 	"iot_manager_service/app/middleware"
 	record "iot_manager_service/app/record/controller"
+	security "iot_manager_service/app/security/controller"
 	system "iot_manager_service/app/system/controller"
 )
 
@@ -406,7 +407,7 @@ func InitRouter(engine *gin.Engine) {
 	{
 		menu.GET("/detail", system.Menu.GetDetail)
 		menu.GET("list", system.Menu.List)
-		menu.GET("lazyList", system.Menu.LazyList)
+		menu.GET("lazy-list", system.Menu.LazyList)
 		menu.GET("menuList", system.Menu.MenuList)
 		menu.GET("lazyMenuList", system.Menu.LazyMenuList)
 		menu.POST("submit", system.Menu.Submit)
@@ -445,4 +446,10 @@ func InitRouter(engine *gin.Engine) {
 	{
 		lightRecord.GET("/list", record.LightRecord.List)
 	}
+
+	//安防
+	securityGroup := engine.Group("/api/longchi/security")
+	{
+		securityGroup.GET("/getCameraLiveList", security.Security.GetCameraLiveList)
+	}
 }

util/errors.go

@@ -18,51 +18,44 @@ const (
 const (
 	Succeeded = "操作成功"
 	Success   = "success"
-	Failed    = "操作失败"
-	Fail      = "fail"
 )
 
 func SuccessResponse(msg string, data interface{}) *Errors {
 	return &Errors{
-		Code:    CodeSucceed,
-		Success: true,
-		Msg:     msg,
-		Data:    data,
+		Code: CodeSucceed,
+		Msg:  msg,
+		Data: data,
 	}
 }
 
 func FailResponse(msg string, data interface{}) *Errors {
 	return &Errors{
-		Code:    CodeInternal,
-		Success: false,
-		Msg:     msg,
-		Data:    data,
+		Code: CodeInternal,
+		Msg:  msg,
+		Data: data,
 	}
 }
 
 func ParamsInvalidResponse(msg string, data interface{}) *Errors {
 	return &Errors{
-		Code:    CodeParamsInvalid,
-		Success: false,
-		Msg:     msg,
-		Data:    data,
+		Code: CodeParamsInvalid,
+		Msg:  msg,
+		Data: data,
 	}
 }
 
 func OperationInvalidResponse(msg string, data interface{}) *Errors {
 	return &Errors{
-		Code:    CodeOperationInvalid,
-		Success: false,
-		Msg:     msg,
-		Data:    data,
+		Code: CodeOperationInvalid,
+		Msg:  msg,
+		Data: data,
 	}
 }
 
 func NormalResponse(code int, msg string, data interface{}) *Errors {
 	return &Errors{
-		Code:    code,
-		Success: true,
-		Msg:     msg,
-		Data:    data,
+		Code: code,
+		Msg:  msg,
+		Data: data,
 	}
 }