mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-24 03:18:35 +08:00
94 lines
2.4 KiB
Go
94 lines
2.4 KiB
Go
package model
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/base64"
|
||
"encoding/json"
|
||
"errors"
|
||
"github.com/google/uuid"
|
||
"gorm.io/gorm"
|
||
"math/rand"
|
||
"pandax/apps/system/entity"
|
||
"pandax/apps/system/services"
|
||
"pandax/pkg/cache"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
type DeviceAuth struct {
|
||
Owner string `json:"owner"`
|
||
OrgId int64 `json:"orgId"`
|
||
DeviceId string `json:"deviceId"`
|
||
DeviceType string `json:"deviceType"`
|
||
DeviceProtocol string `json:"deviceProtocol"`
|
||
ProductId string `json:"productId"`
|
||
RuleChainId string `json:"ruleChainId"`
|
||
Name string `json:"name"`
|
||
CreatedAt int64 `json:"created_at"`
|
||
ExpiredAt int64 `json:"expired_at"`
|
||
}
|
||
|
||
func (entity *DeviceAuth) GetDeviceToken(key string) error {
|
||
if err := cache.GetDeviceEtoken(key, entity); err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (token *DeviceAuth) MD5ID() string {
|
||
buf := bytes.NewBufferString(token.DeviceId)
|
||
buf.WriteString(token.DeviceType)
|
||
buf.WriteString(strconv.FormatInt(token.CreatedAt, 10))
|
||
access := base64.URLEncoding.EncodeToString([]byte(uuid.NewMD5(uuid.Must(uuid.NewRandom()), buf.Bytes()).String()))
|
||
access = strings.TrimRight(access, "=")
|
||
return access
|
||
}
|
||
func (token *DeviceAuth) GetMarshal() string {
|
||
marshal, _ := json.Marshal(*token)
|
||
return string(marshal)
|
||
}
|
||
|
||
func (token *DeviceAuth) GetUnMarshal(data []byte) error {
|
||
return json.Unmarshal(data, token)
|
||
}
|
||
|
||
// 序列化
|
||
func (m *DeviceAuth) MarshalBinary() (data []byte, err error) {
|
||
return json.Marshal(m)
|
||
}
|
||
|
||
// 反序列化
|
||
func (m *DeviceAuth) UnmarshalBinary(data []byte) error {
|
||
return json.Unmarshal(data, m)
|
||
}
|
||
|
||
func OrgAuthSet(tx *gorm.DB, roleId int64, owner string) error {
|
||
if roleId == 0 {
|
||
return errors.New("角色Id不能为空")
|
||
}
|
||
role, err := services.SysRoleModelDao.FindOrganizationsByRoleId(roleId)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if role.DataScope != entity.SELFDATASCOPE {
|
||
if len(role.Org) <= 0 {
|
||
return errors.New("该角色下未分配组织权限")
|
||
}
|
||
tx.Where("org_id in (?)", strings.Split(role.Org, ","))
|
||
} else {
|
||
tx.Where("owner = ?", owner)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
func GenerateID() string {
|
||
rand.Seed(time.Now().UnixNano())
|
||
id := make([]byte, 7) // 由于base64编码会增加字符数，这里使用7个字节生成10位ID
|
||
_, err := rand.Read(id)
|
||
if err != nil {
|
||
panic(err) // 错误处理，根据实际情况进行处理
|
||
}
|
||
return base64.URLEncoding.EncodeToString(id)[:10]
|
||
}
|