mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-23 02:48:34 +08:00
255 lines
7.8 KiB
Go
255 lines
7.8 KiB
Go
package services
|
||
|
||
import (
|
||
"context"
|
||
"github.com/PandaXGO/PandaKit/biz"
|
||
"pandax/apps/device/entity"
|
||
"pandax/pkg/global"
|
||
"pandax/pkg/global_model"
|
||
"time"
|
||
)
|
||
|
||
type (
|
||
DeviceModel interface {
|
||
Insert(data entity.Device) *entity.Device
|
||
FindOneByToken(token string) (*entity.DeviceRes, error)
|
||
FindOneByName(name string) (*entity.DeviceRes, error)
|
||
FindOne(id string) *entity.DeviceRes
|
||
FindListPage(page, pageSize int, data entity.Device) (*[]entity.DeviceRes, int64)
|
||
FindList(data entity.Device) *[]entity.DeviceRes
|
||
Update(data entity.Device) *entity.Device
|
||
UpdateStatus(id, linkStatus string) error
|
||
Delete(ids []string)
|
||
FindDeviceCount() entity.DeviceCount
|
||
FindDeviceCountGroupByLinkStatus() []entity.DeviceCountLinkStatus
|
||
FindDeviceCountGroupByType() []entity.DeviceCountType
|
||
}
|
||
|
||
deviceModelImpl struct {
|
||
table string
|
||
}
|
||
)
|
||
|
||
var DeviceModelDao DeviceModel = &deviceModelImpl{
|
||
table: `devices`,
|
||
}
|
||
|
||
func (m *deviceModelImpl) Insert(data entity.Device) *entity.Device {
|
||
tx := global.Db.Begin()
|
||
//1 检查设备名称是否存在
|
||
list := m.FindList(entity.Device{Name: data.Name})
|
||
biz.IsTrue(list != nil && len(*list) == 0, "设备名称已经存在")
|
||
//2 创建认证TOKEN IOTHUB使用
|
||
token := GetDeviceToken(&data)
|
||
// 子网关不需要设置token
|
||
if data.DeviceType == global.GATEWAYS {
|
||
data.Token = ""
|
||
err := global.RedisDb.Set(data.Name, token.GetMarshal(), time.Hour*24*365)
|
||
biz.ErrIsNil(err, "设备缓存失败")
|
||
} else {
|
||
err := global.RedisDb.Set(data.Token, token.GetMarshal(), time.Hour*24*365)
|
||
biz.ErrIsNil(err, "设备缓存失败")
|
||
}
|
||
//3 添加设备
|
||
err := tx.Table(m.table).Create(&data).Error
|
||
biz.ErrIsNil(err, "添加设备失败")
|
||
// 创建超级表 失败就
|
||
if data.Pid != "" {
|
||
err = createDeviceTable(data.Pid, data.Name)
|
||
if err != nil {
|
||
tx.Rollback()
|
||
biz.ErrIsNil(err, "添加设备失败，设备表创建失败")
|
||
}
|
||
}
|
||
tx.Commit()
|
||
return &data
|
||
}
|
||
|
||
func (m *deviceModelImpl) FindOne(id string) *entity.DeviceRes {
|
||
resData := new(entity.DeviceRes)
|
||
db := global.Db.Table(m.table).Where("id = ?", id)
|
||
err := db.Preload("Product").Preload("DeviceGroup").First(resData).Error
|
||
biz.ErrIsNil(err, "查询设备失败")
|
||
return resData
|
||
}
|
||
|
||
func (m *deviceModelImpl) FindOneByName(name string) (*entity.DeviceRes, error) {
|
||
resData := new(entity.DeviceRes)
|
||
db := global.Db.Table(m.table).Where("name = ?", name)
|
||
err := db.First(resData).Error
|
||
return resData, err
|
||
}
|
||
|
||
func (m *deviceModelImpl) FindOneByToken(token string) (*entity.DeviceRes, error) {
|
||
resData := new(entity.DeviceRes)
|
||
db := global.Db.Table(m.table).Preload("Product").Where("token = ?", token)
|
||
err := db.First(resData).Error
|
||
return resData, err
|
||
}
|
||
|
||
func (m *deviceModelImpl) FindListPage(page, pageSize int, data entity.Device) (*[]entity.DeviceRes, int64) {
|
||
list := make([]entity.DeviceRes, 0)
|
||
var total int64 = 0
|
||
offset := pageSize * (page - 1)
|
||
db := global.Db.Table(m.table)
|
||
// 此处填写 where参数判断
|
||
if data.Alias != "" {
|
||
db = db.Where("alias = ?", data.Alias)
|
||
}
|
||
if data.Gid != "" {
|
||
db = db.Where("gid = ?", data.Gid)
|
||
}
|
||
if data.Name != "" {
|
||
db = db.Where("name like ?", "%"+data.Name+"%")
|
||
}
|
||
if data.Status != "" {
|
||
db = db.Where("status = ?", data.Status)
|
||
}
|
||
if data.LinkStatus != "" {
|
||
db = db.Where("Link_status = ?", data.LinkStatus)
|
||
}
|
||
if data.Pid != "" {
|
||
db = db.Where("pid = ?", data.Pid)
|
||
}
|
||
if data.ParentId != "" {
|
||
db = db.Where("parent_id = ?", data.ParentId)
|
||
}
|
||
// 组织数据访问权限
|
||
global_model.OrgAuthSet(db, data.RoleId, data.Owner)
|
||
|
||
err := db.Count(&total).Error
|
||
err = db.Order("create_time").Preload("Product").Preload("DeviceGroup").Limit(pageSize).Offset(offset).Find(&list).Error
|
||
biz.ErrIsNil(err, "查询设备分页列表失败")
|
||
return &list, total
|
||
}
|
||
|
||
func (m *deviceModelImpl) FindList(data entity.Device) *[]entity.DeviceRes {
|
||
list := make([]entity.DeviceRes, 0)
|
||
db := global.Db.Table(m.table)
|
||
// 此处填写 where参数判断
|
||
if data.Alias != "" {
|
||
db = db.Where("alias = ?", data.Alias)
|
||
}
|
||
if data.Gid != "" {
|
||
db = db.Where("gid = ?", data.Gid)
|
||
}
|
||
if data.Name != "" {
|
||
db = db.Where("name like ?", "%"+data.Name+"%")
|
||
}
|
||
if data.DeviceType != "" {
|
||
db = db.Where("device_type = ?", data.DeviceType)
|
||
}
|
||
if data.Status != "" {
|
||
db = db.Where("status = ?", data.Status)
|
||
}
|
||
if data.LinkStatus != "" {
|
||
db = db.Where("Link_status = ?", data.LinkStatus)
|
||
}
|
||
if data.Pid != "" {
|
||
db = db.Where("pid = ?", data.Pid)
|
||
}
|
||
if data.ParentId != "" {
|
||
db = db.Where("parent_id = ?", data.ParentId)
|
||
}
|
||
global_model.OrgAuthSet(db, data.RoleId, data.Owner)
|
||
db.Preload("Product").Preload("DeviceGroup")
|
||
biz.ErrIsNil(db.Order("create_time").Find(&list).Error, "查询设备列表失败")
|
||
return &list
|
||
}
|
||
|
||
// TODO 如果更改的是产品，tdengine的设备表也要更改
|
||
func (m *deviceModelImpl) Update(data entity.Device) *entity.Device {
|
||
if data.DeviceType == global.GATEWAYS {
|
||
data.Token = ""
|
||
} else {
|
||
token := GetDeviceToken(&data)
|
||
err := global.RedisDb.Set(data.Token, token.GetMarshal(), time.Hour*24*365)
|
||
biz.ErrIsNil(err, "设备更改缓存失败")
|
||
}
|
||
biz.ErrIsNil(global.Db.Table(m.table).Updates(&data).Error, "修改设备失败")
|
||
return &data
|
||
}
|
||
func (m *deviceModelImpl) UpdateStatus(id, linkStatus string) error {
|
||
return global.Db.Table(m.table).Where("id", id).Update("link_status", linkStatus).Update("last_time", time.Now()).Error
|
||
}
|
||
|
||
func (m *deviceModelImpl) Delete(ids []string) {
|
||
biz.ErrIsNil(global.Db.Table(m.table).Delete(&entity.Device{}, "id in (?)", ids).Error, "删除设备失败")
|
||
for _, id := range ids {
|
||
list := m.FindOne(id)
|
||
// 删除表
|
||
err := deleteDeviceTable(list.Name)
|
||
global.Log.Error("设备时序表删除失败", err)
|
||
// 删除所有缓存
|
||
global.RedisDb.Del(context.Background(), id)
|
||
}
|
||
}
|
||
|
||
// 获取Tdengine时序数据
|
||
func createDeviceTable(productId, device string) error {
|
||
err := global.TdDb.CreateTable(productId+"_"+entity.ATTRIBUTES_TSL, device+"_"+entity.ATTRIBUTES_TSL)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
err = global.TdDb.CreateTable(productId+"_"+entity.TELEMETRY_TSL, device+"_"+entity.TELEMETRY_TSL)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 删除Tdengine时序数据
|
||
func deleteDeviceTable(device string) error {
|
||
err := global.TdDb.DropTable(device + "_" + entity.ATTRIBUTES_TSL)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
err = global.TdDb.DropTable(device + "_" + entity.TELEMETRY_TSL)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func GetDeviceToken(data *entity.Device) *global_model.DeviceAuth {
|
||
now := time.Now()
|
||
etoken := &global_model.DeviceAuth{
|
||
DeviceId: data.Id,
|
||
OrgId: data.OrgId,
|
||
Owner: data.Owner,
|
||
Name: data.Name,
|
||
DeviceType: data.DeviceType,
|
||
ProductId: data.Pid,
|
||
}
|
||
//设备有效期360天
|
||
etoken.CreatedAt = now.Unix()
|
||
etoken.ExpiredAt = now.Add(time.Hour * 24 * 365).Unix()
|
||
if data.Token == "" {
|
||
data.Token = etoken.MD5ID()
|
||
}
|
||
return etoken
|
||
}
|
||
|
||
// 获取设备数量统计
|
||
func (m *deviceModelImpl) FindDeviceCount() (result entity.DeviceCount) {
|
||
sql := `SELECT COUNT(*) AS total, (SELECT COUNT(*) FROM devices WHERE DATE(create_time) = CURDATE()) AS today FROM devices`
|
||
err := global.Db.Raw(sql).Scan(&result).Error
|
||
biz.ErrIsNil(err, "获取设备统计总数失败")
|
||
return result
|
||
}
|
||
|
||
// 获取设备类型数量统计
|
||
func (m *deviceModelImpl) FindDeviceCountGroupByLinkStatus() (count []entity.DeviceCountLinkStatus) {
|
||
sql := `SELECT link_status, COUNT(*) AS total FROM devices GROUP BY link_status`
|
||
err := global.Db.Raw(sql, m.table).Scan(&count).Error
|
||
biz.ErrIsNil(err, "获取通过设备在线状态的设备统计总数失败")
|
||
return
|
||
}
|
||
|
||
func (m *deviceModelImpl) FindDeviceCountGroupByType() (count []entity.DeviceCountType) {
|
||
sql := `SELECT device_type, COUNT(*) AS total FROM devices GROUP BY device_type`
|
||
err := global.Db.Raw(sql, m.table).Scan(&count).Error
|
||
biz.ErrIsNil(err, "获取通过设备类型的设备统计总数失败")
|
||
return
|
||
}
|