219 lines
7.5 KiB
Go
219 lines
7.5 KiB
Go
// 功能：用户设备分组资料与本人设备归组，严格隔离账户并保持新增幂等；版本：1.0.0。
|
||
package user
|
||
|
||
import (
|
||
"errors"
|
||
"strings"
|
||
"unicode/utf8"
|
||
|
||
"git.apinb.com/bsm-sdk/core/errcode"
|
||
"git.apinb.com/bsm-sdk/core/infra"
|
||
"git.apinb.com/heqiapp/platforms/backend/api/internal/impl"
|
||
"git.apinb.com/heqiapp/platforms/backend/api/internal/logic/common"
|
||
"git.apinb.com/heqiapp/platforms/backend/api/internal/models"
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/google/uuid"
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/clause"
|
||
)
|
||
|
||
const maxDeviceGroups = 20
|
||
|
||
var (
|
||
errDeviceGroupLimit = errcode.NewError(2511, "最多可创建20个设备分组")
|
||
errDeviceGroupDuplicate = errcode.NewError(2512, "该分组名称已存在")
|
||
errDeviceGroupRequestChanged = errcode.NewError(2513, "该新增请求已处理，请关闭表单并刷新分组列表")
|
||
)
|
||
|
||
type deviceGroupInput struct {
|
||
Name string `json:"name"`
|
||
RequestNo string `json:"request_no"`
|
||
}
|
||
|
||
// validDeviceGroupInput 清理名称并校验新增幂等标识。
|
||
func validDeviceGroupInput(input *deviceGroupInput, creating bool) bool {
|
||
input.Name = strings.TrimSpace(input.Name)
|
||
if input.Name == "" || utf8.RuneCountInString(input.Name) > 32 {
|
||
return false
|
||
}
|
||
if !creating {
|
||
return true
|
||
}
|
||
parsed, err := uuid.Parse(input.RequestNo)
|
||
return err == nil && parsed != uuid.Nil && parsed.String() == input.RequestNo
|
||
}
|
||
|
||
// ListDeviceGroups 返回本人分组及每组当前仍归属本人的设备公开标识。
|
||
func ListDeviceGroups(ctx *gin.Context) {
|
||
account, ok := common.UserAccount(ctx)
|
||
if !ok {
|
||
return
|
||
}
|
||
var groups []models.UserDeviceGroup
|
||
if err := impl.DBService.Where("user_account_id = ? AND status = ?", account.ID, common.StatusEnable).
|
||
Order("sort_no, created_at, identity").Limit(maxDeviceGroups).Find(&groups).Error; err != nil {
|
||
infra.Response.Error(ctx, err)
|
||
return
|
||
}
|
||
groupIDs := make([]uint64, 0, len(groups))
|
||
for _, group := range groups {
|
||
groupIDs = append(groupIDs, group.ID)
|
||
}
|
||
devicesByGroup := map[uint64][]string{}
|
||
if len(groupIDs) > 0 {
|
||
var devices []models.ProductInfo
|
||
if err := ownedProductQuery(account.ID).Where("device_group_id IN ? AND device_kind IN ? AND product_status <> ?", groupIDs, []string{"valve", "alarm"}, common.StatusScrapped).
|
||
Order("created_at, identity").Find(&devices).Error; err != nil {
|
||
infra.Response.Error(ctx, err)
|
||
return
|
||
}
|
||
for _, device := range devices {
|
||
devicesByGroup[device.DeviceGroupID] = append(devicesByGroup[device.DeviceGroupID], device.Identity)
|
||
}
|
||
}
|
||
items := make([]gin.H, 0, len(groups))
|
||
for _, group := range groups {
|
||
identities := devicesByGroup[group.ID]
|
||
if identities == nil {
|
||
identities = []string{}
|
||
}
|
||
items = append(items, gin.H{"identity": group.Identity, "name": group.Name, "sort_no": group.SortNo, "device_identities": identities})
|
||
}
|
||
infra.Response.Success(ctx, items)
|
||
}
|
||
|
||
// SaveDeviceGroup 新建或改名；账户行锁保证上限、幂等和同名约束串行执行。
|
||
func SaveDeviceGroup(ctx *gin.Context) {
|
||
account, ok := common.UserAccount(ctx)
|
||
if !ok {
|
||
return
|
||
}
|
||
creating := ctx.Param("identity") == ""
|
||
var input deviceGroupInput
|
||
if ctx.ShouldBindJSON(&input) != nil || !validDeviceGroupInput(&input, creating) {
|
||
infra.Response.Error(ctx, errcode.ErrInvalidArgument)
|
||
return
|
||
}
|
||
var group models.UserDeviceGroup
|
||
err := impl.DBService.Transaction(func(tx *gorm.DB) error {
|
||
if err := lockAddressOwner(tx, account.ID); err != nil {
|
||
return err
|
||
}
|
||
if creating {
|
||
err := tx.Where("user_account_id = ? AND request_no = ?", account.ID, input.RequestNo).First(&group).Error
|
||
if err == nil {
|
||
if group.Status != common.StatusEnable || group.Name != input.Name {
|
||
return errDeviceGroupRequestChanged
|
||
}
|
||
return nil
|
||
}
|
||
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return err
|
||
}
|
||
var count int64
|
||
if err := tx.Model(&models.UserDeviceGroup{}).Where("user_account_id = ? AND status = ?", account.ID, common.StatusEnable).Count(&count).Error; err != nil {
|
||
return err
|
||
}
|
||
if count >= maxDeviceGroups {
|
||
return errDeviceGroupLimit
|
||
}
|
||
group = models.UserDeviceGroup{Entity: common.NewEntity(common.StatusEnable), UserAccountID: account.ID, Name: input.Name, RequestNo: input.RequestNo, SortNo: int(count)}
|
||
} else if err := tx.Where("identity = ? AND user_account_id = ? AND status = ?", ctx.Param("identity"), account.ID, common.StatusEnable).First(&group).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return errcode.ErrRecordNotFound
|
||
}
|
||
return err
|
||
}
|
||
var duplicates int64
|
||
if err := tx.Model(&models.UserDeviceGroup{}).Where("user_account_id = ? AND status = ? AND name = ? AND identity <> ?", account.ID, common.StatusEnable, input.Name, group.Identity).Count(&duplicates).Error; err != nil {
|
||
return err
|
||
}
|
||
if duplicates > 0 {
|
||
return errDeviceGroupDuplicate
|
||
}
|
||
group.Name = input.Name
|
||
return tx.Save(&group).Error
|
||
})
|
||
if err != nil {
|
||
infra.Response.Error(ctx, err)
|
||
return
|
||
}
|
||
infra.Response.Success(ctx, gin.H{"identity": group.Identity, "name": group.Name, "sort_no": group.SortNo})
|
||
}
|
||
|
||
// DeleteDeviceGroup 先解除本人设备归组再归档分组；重复删除保持成功。
|
||
func DeleteDeviceGroup(ctx *gin.Context) {
|
||
account, ok := common.UserAccount(ctx)
|
||
if !ok {
|
||
return
|
||
}
|
||
err := impl.DBService.Transaction(func(tx *gorm.DB) error {
|
||
if err := lockAddressOwner(tx, account.ID); err != nil {
|
||
return err
|
||
}
|
||
var group models.UserDeviceGroup
|
||
if err := tx.Where("identity = ? AND user_account_id = ?", ctx.Param("identity"), account.ID).First(&group).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return errcode.ErrRecordNotFound
|
||
}
|
||
return err
|
||
}
|
||
if group.Status == common.StatusArchived {
|
||
return nil
|
||
}
|
||
if err := tx.Model(&models.ProductInfo{}).Where("user_account_id = ? AND device_group_id = ?", account.ID, group.ID).Update("device_group_id", 0).Error; err != nil {
|
||
return err
|
||
}
|
||
return tx.Model(&group).Update("status", common.StatusArchived).Error
|
||
})
|
||
if err != nil {
|
||
infra.Response.Error(ctx, err)
|
||
return
|
||
}
|
||
infra.Response.Success(ctx, gin.H{"deleted": true})
|
||
}
|
||
|
||
// AssignDeviceGroup 将本人启用智能设备移入本人分组；空标识表示移出分组。
|
||
func AssignDeviceGroup(ctx *gin.Context) {
|
||
account, ok := common.UserAccount(ctx)
|
||
if !ok {
|
||
return
|
||
}
|
||
var request struct {
|
||
GroupIdentity string `json:"group_identity"`
|
||
}
|
||
if ctx.ShouldBindJSON(&request) != nil {
|
||
infra.Response.Error(ctx, errcode.ErrInvalidArgument)
|
||
return
|
||
}
|
||
err := impl.DBService.Transaction(func(tx *gorm.DB) error {
|
||
var device models.ProductInfo
|
||
if err := tx.Clauses(clause.Locking{Strength: "UPDATE"}).Where(
|
||
"identity = ? AND user_account_id = ? AND status = ? AND device_kind IN ? AND product_status <> ?",
|
||
ctx.Param("identity"), account.ID, common.StatusEnable, []string{"valve", "alarm"}, common.StatusScrapped,
|
||
).First(&device).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return errcode.ErrRecordNotFound
|
||
}
|
||
return err
|
||
}
|
||
groupID := uint64(0)
|
||
if strings.TrimSpace(request.GroupIdentity) != "" {
|
||
var group models.UserDeviceGroup
|
||
if err := tx.Where("identity = ? AND user_account_id = ? AND status = ?", strings.TrimSpace(request.GroupIdentity), account.ID, common.StatusEnable).First(&group).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return errcode.ErrRecordNotFound
|
||
}
|
||
return err
|
||
}
|
||
groupID = group.ID
|
||
}
|
||
return tx.Model(&device).Update("device_group_id", groupID).Error
|
||
})
|
||
if err != nil {
|
||
infra.Response.Error(ctx, err)
|
||
return
|
||
}
|
||
infra.Response.Success(ctx, gin.H{"updated": true})
|
||
}
|