Browse Source
Add the backend application, MQTT device lifecycle, task execution tracking, media control, and supporting persistence.master
110 changed files with 13797 additions and 0 deletions
File diff suppressed because it is too large
@ -0,0 +1,10 @@ |
|||
.PHONY: build run clean |
|||
|
|||
build: |
|||
go build -o laic-backend . |
|||
|
|||
run: |
|||
go run . |
|||
|
|||
clean: |
|||
rm -f laic-backend |
|||
@ -0,0 +1,82 @@ |
|||
package cache |
|||
|
|||
import "fmt" |
|||
|
|||
const ( |
|||
UserTokenKey = "laic:token:%d:%s" // 用户 access token 存证（userId, shortId）
|
|||
RefreshTokenKey = "laic:refresh:%d:%s" // 用户 refresh token 存证（userId, shortId）
|
|||
SmsCodeKey = "laic:sms:%s" // 短信验证码（phone）
|
|||
|
|||
DockStatusKey = "device:dock:%s:status" // 机巢实时状态 Hash（dockId）
|
|||
DockAlarmKey = "device:dock:%s:alarm_codes" // 机巢告警码 Set（dockId）
|
|||
DroneTelemetryKey = "device:drone:%s:telemetry" // 无人机遥测 Hash（dockId）
|
|||
DroneAlarmKey = "device:drone:%s:alarm_codes" // 无人机告警码 Set（dockId）
|
|||
OtaStatusKey = "device:dock:%s:ota" // 机巢 OTA 升级进度 Hash（dockId）
|
|||
WorkflowKey = "device:workflow:%s" // 当前工作流状态 Hash（dockId）
|
|||
VideoKey = "device:video:%s" // 当前视频推流状态 Hash（dockId）
|
|||
OnlineDockSetKey = "laic:devices:online:dock" // 在线机巢 Set
|
|||
OnlineDroneSetKey = "laic:devices:online:drone" // 在线无人机 Set
|
|||
DockHeartbeatKey = "laic:devices:heartbeat:dock:%s" |
|||
DroneHeartbeatKey = "laic:devices:heartbeat:drone:%s" |
|||
|
|||
TrafficKey = "laic:traffic:%d" // 用户云媒体流量余额（字节，Redis 为权威）
|
|||
|
|||
TrafficKeyPrefix = "laic:traffic:" // 流量余额 key 前缀（快照刷新扫描用）
|
|||
LiveBilledKey = "laic:live:billed:%s" // 直播会话上次计费时间戳（sessionId）
|
|||
) |
|||
|
|||
func UserTokenKeyOf(userID int64, shortID string) string { |
|||
return fmt.Sprintf(UserTokenKey, userID, shortID) |
|||
} |
|||
|
|||
func RefreshTokenKeyOf(userID int64, shortID string) string { |
|||
return fmt.Sprintf(RefreshTokenKey, userID, shortID) |
|||
} |
|||
|
|||
func SmsCodeKeyOf(phone string) string { |
|||
return fmt.Sprintf(SmsCodeKey, phone) |
|||
} |
|||
|
|||
func DockStatusKeyOf(dockID string) string { |
|||
return fmt.Sprintf(DockStatusKey, dockID) |
|||
} |
|||
|
|||
func DockAlarmKeyOf(dockID string) string { |
|||
return fmt.Sprintf(DockAlarmKey, dockID) |
|||
} |
|||
|
|||
func DroneTelemetryKeyOf(dockID string) string { |
|||
return fmt.Sprintf(DroneTelemetryKey, dockID) |
|||
} |
|||
|
|||
func DroneAlarmKeyOf(dockID string) string { |
|||
return fmt.Sprintf(DroneAlarmKey, dockID) |
|||
} |
|||
|
|||
func OtaStatusKeyOf(dockID string) string { |
|||
return fmt.Sprintf(OtaStatusKey, dockID) |
|||
} |
|||
|
|||
func WorkflowKeyOf(dockID string) string { |
|||
return fmt.Sprintf(WorkflowKey, dockID) |
|||
} |
|||
|
|||
func VideoKeyOf(dockID string) string { |
|||
return fmt.Sprintf(VideoKey, dockID) |
|||
} |
|||
|
|||
func DockHeartbeatKeyOf(dockID string) string { |
|||
return fmt.Sprintf(DockHeartbeatKey, dockID) |
|||
} |
|||
|
|||
func DroneHeartbeatKeyOf(droneSN string) string { |
|||
return fmt.Sprintf(DroneHeartbeatKey, droneSN) |
|||
} |
|||
|
|||
func TrafficKeyOf(userID int64) string { |
|||
return fmt.Sprintf(TrafficKey, userID) |
|||
} |
|||
|
|||
func LiveBilledKeyOf(sessionID string) string { |
|||
return fmt.Sprintf(LiveBilledKey, sessionID) |
|||
} |
|||
@ -0,0 +1,17 @@ |
|||
package cache |
|||
|
|||
import ( |
|||
"laic-backend/common" |
|||
"laic-backend/model" |
|||
) |
|||
|
|||
// WarmUp 预热缓存：将用户流量余额从 MySQL 快照 SetNX 到 Redis（仅缺 key 时写入）
|
|||
func WarmUp() { |
|||
var users []model.User |
|||
if err := common.DB.Find(&users).Error; err != nil { |
|||
return |
|||
} |
|||
for _, u := range users { |
|||
_, _ = common.SetNX(TrafficKeyOf(u.ID), u.TrafficBalance) |
|||
} |
|||
} |
|||
@ -0,0 +1,14 @@ |
|||
[request_definition] |
|||
r = sub, obj, act |
|||
|
|||
[policy_definition] |
|||
p = sub, obj, act |
|||
|
|||
[role_definition] |
|||
g = _, _ |
|||
|
|||
[policy_effect] |
|||
e = some(where (p.eft == allow)) |
|||
|
|||
[matchers] |
|||
m = g(r.sub, p.sub) && keyMatch2(r.obj, p.obj) && (r.act == p.act || p.act == "*") |
|||
@ -0,0 +1,46 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
"gorm.io/gorm" |
|||
) |
|||
|
|||
const ( |
|||
RoleAdmin = "admin" |
|||
RoleUser = "user" |
|||
) |
|||
|
|||
// IsAdmin 判断当前用户是否为 admin（全量数据）
|
|||
func IsAdmin(c *gin.Context) bool { |
|||
roles := c.MustGet("roles").([]string) |
|||
for _, role := range roles { |
|||
if role == RoleAdmin { |
|||
return true |
|||
} |
|||
} |
|||
return false |
|||
} |
|||
|
|||
// GetUserId 从上下文获取当前用户 ID
|
|||
func GetUserId(c *gin.Context) int64 { |
|||
return c.GetInt64("user_id") |
|||
} |
|||
|
|||
// GetAuthFilter admin 返回空 filter（全量），user 返回 user_id 过滤
|
|||
func GetAuthFilter(c *gin.Context) map[string]interface{} { |
|||
filter := make(map[string]interface{}) |
|||
if IsAdmin(c) { |
|||
return filter |
|||
} |
|||
filter["user_id"] = GetUserId(c) |
|||
return filter |
|||
} |
|||
|
|||
// WithUserFilter admin 不过滤，user 按 user_id 过滤（GORM scope）
|
|||
func WithUserFilter(c *gin.Context) func(db *gorm.DB) *gorm.DB { |
|||
if IsAdmin(c) { |
|||
return func(db *gorm.DB) *gorm.DB { return db } |
|||
} |
|||
uid := GetUserId(c) |
|||
return func(db *gorm.DB) *gorm.DB { return db.Where("user_id = ?", uid) } |
|||
} |
|||
@ -0,0 +1,128 @@ |
|||
package common |
|||
|
|||
// BusiError 业务错误类型
|
|||
type BusiError struct { |
|||
Code int `json:"code"` |
|||
Msg string `json:"msg"` |
|||
} |
|||
|
|||
func (e *BusiError) Error() string { |
|||
return e.Msg |
|||
} |
|||
|
|||
// 通用错误码
|
|||
const ( |
|||
Success = 200 |
|||
ParamError = 400 |
|||
Unauthorized = 401 |
|||
Forbidden = 403 |
|||
NotFound = 404 |
|||
InternalError = 500 |
|||
) |
|||
|
|||
// 业务错误码（按模块分段）
|
|||
const ( |
|||
// 认证/用户 51xxx
|
|||
UserPhoneExists = 51001 |
|||
UserNotFound = 51002 |
|||
PasswordError = 51003 |
|||
TokenInvalid = 51004 |
|||
TokenExpired = 51005 |
|||
SmsCodeError = 51006 |
|||
|
|||
// 机巢 52xxx
|
|||
DockNotFound = 52001 |
|||
DockOffline = 52002 |
|||
DockPending = 52003 |
|||
DockExists = 52004 |
|||
CommandNotFound = 52005 |
|||
|
|||
// 无人机 53xxx
|
|||
DroneNotFound = 53001 |
|||
|
|||
// 告警 54xxx
|
|||
AlarmNotFound = 54001 |
|||
|
|||
// 固件 55xxx
|
|||
FirmwareNotFound = 55001 |
|||
FirmwareVersionExist = 55002 |
|||
FirmwareNotReleased = 55003 |
|||
|
|||
// 任务/航线 56xxx
|
|||
TaskNotFound = 56001 |
|||
RouteNotFound = 56002 |
|||
TaskStateErr = 56003 |
|||
ExecutionNotFound = 56004 |
|||
|
|||
// 媒体 57xxx
|
|||
LiveNotFound = 57001 |
|||
VideoNotFound = 57002 |
|||
VideoUploading = 57003 |
|||
LiveProviderDisabled = 57004 |
|||
LiveProviderInvalid = 57005 |
|||
LiveLeaseNotFound = 57006 |
|||
|
|||
// 计费 58xxx
|
|||
TrafficNotEnough = 58001 |
|||
OrderPaid = 58002 |
|||
SimCardNotFound = 58003 |
|||
OrderNotFound = 58004 |
|||
) |
|||
|
|||
// 预定义的业务错误
|
|||
var ( |
|||
ErrSuccess = &BusiError{Code: Success, Msg: "成功"} |
|||
ErrParam = &BusiError{Code: ParamError, Msg: "参数错误"} |
|||
ErrUnauthorized = &BusiError{Code: Unauthorized, Msg: "未登录"} |
|||
ErrToken = &BusiError{Code: Unauthorized, Msg: "Token异常"} |
|||
SignOutToken = &BusiError{Code: Unauthorized, Msg: "Token已失效"} |
|||
ErrForbidden = &BusiError{Code: Forbidden, Msg: "无权限访问"} |
|||
ErrNotFound = &BusiError{Code: NotFound, Msg: "资源不存在"} |
|||
ErrInternal = &BusiError{Code: InternalError, Msg: "系统内部错误"} |
|||
|
|||
ErrUserPhoneExists = &BusiError{Code: UserPhoneExists, Msg: "该手机号已注册"} |
|||
ErrUserNotFound = &BusiError{Code: UserNotFound, Msg: "用户不存在"} |
|||
ErrPasswordError = &BusiError{Code: PasswordError, Msg: "用户名或密码错误"} |
|||
ErrTokenInvalid = &BusiError{Code: TokenInvalid, Msg: "Token无效"} |
|||
ErrTokenExpired = &BusiError{Code: TokenExpired, Msg: "Token已过期"} |
|||
ErrSmsCodeError = &BusiError{Code: SmsCodeError, Msg: "短信验证码错误"} |
|||
|
|||
ErrDockNotFound = &BusiError{Code: DockNotFound, Msg: "机巢不存在"} |
|||
ErrDockOffline = &BusiError{Code: DockOffline, Msg: "机巢离线"} |
|||
ErrDockPending = &BusiError{Code: DockPending, Msg: "机巢待登记"} |
|||
ErrDockExists = &BusiError{Code: DockExists, Msg: "该机巢编号已存在"} |
|||
ErrCommandNotFound = &BusiError{Code: CommandNotFound, Msg: "指令记录不存在"} |
|||
|
|||
ErrDroneNotFound = &BusiError{Code: DroneNotFound, Msg: "无人机不存在"} |
|||
|
|||
ErrAlarmNotFound = &BusiError{Code: AlarmNotFound, Msg: "告警不存在"} |
|||
|
|||
ErrFirmwareNotFound = &BusiError{Code: FirmwareNotFound, Msg: "固件版本不存在"} |
|||
ErrFirmwareVersionExist = &BusiError{Code: FirmwareVersionExist, Msg: "该固件版本已存在"} |
|||
ErrFirmwareNotReleased = &BusiError{Code: FirmwareNotReleased, Msg: "固件未发布，无法下发升级"} |
|||
|
|||
ErrTaskNotFound = &BusiError{Code: TaskNotFound, Msg: "任务不存在"} |
|||
ErrRouteNotFound = &BusiError{Code: RouteNotFound, Msg: "航线不存在"} |
|||
ErrTaskStateErr = &BusiError{Code: TaskStateErr, Msg: "任务状态不允许此操作"} |
|||
ErrExecutionNotFound = &BusiError{Code: ExecutionNotFound, Msg: "执行记录不存在"} |
|||
|
|||
ErrLiveNotFound = &BusiError{Code: LiveNotFound, Msg: "直播会话不存在"} |
|||
ErrLiveProviderDisabled = &BusiError{Code: LiveProviderDisabled, Msg: "直播服务未配置"} |
|||
ErrLiveProviderInvalid = &BusiError{Code: LiveProviderInvalid, Msg: "直播服务配置无效"} |
|||
ErrLiveLeaseNotFound = &BusiError{Code: LiveLeaseNotFound, Msg: "直播观看租约不存在或已过期"} |
|||
ErrVideoNotFound = &BusiError{Code: VideoNotFound, Msg: "视频不存在"} |
|||
ErrVideoUploading = &BusiError{Code: VideoUploading, Msg: "视频上传中"} |
|||
|
|||
ErrTrafficNotEnough = &BusiError{Code: TrafficNotEnough, Msg: "流量不足，请充值"} |
|||
ErrOrderPaid = &BusiError{Code: OrderPaid, Msg: "该订单已支付"} |
|||
ErrSimCardNotFound = &BusiError{Code: SimCardNotFound, Msg: "SIM卡不存在"} |
|||
ErrOrderNotFound = &BusiError{Code: OrderNotFound, Msg: "订单不存在"} |
|||
) |
|||
|
|||
// NewBusiError 创建自定义业务错误
|
|||
func NewBusiError(code int, msg string) *BusiError { |
|||
return &BusiError{ |
|||
Code: code, |
|||
Msg: msg, |
|||
} |
|||
} |
|||
@ -0,0 +1,54 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"fmt" |
|||
"strconv" |
|||
|
|||
"laic-backend/logger" |
|||
|
|||
"github.com/casbin/casbin/v3" |
|||
gormadapter "github.com/casbin/gorm-adapter/v3" |
|||
) |
|||
|
|||
var CasbinEnforcer *casbin.Enforcer |
|||
|
|||
// SetUserRole 同步用户在 Casbin 中的角色映射（g 策略），role 为空则清空
|
|||
func SetUserRole(userID int64, role string) error { |
|||
if CasbinEnforcer == nil { |
|||
return nil |
|||
} |
|||
uid := strconv.FormatInt(userID, 10) |
|||
if _, err := CasbinEnforcer.DeleteRolesForUser(uid); err != nil { |
|||
return err |
|||
} |
|||
if role == "" { |
|||
return nil |
|||
} |
|||
_, err := CasbinEnforcer.AddRoleForUser(uid, role) |
|||
return err |
|||
} |
|||
|
|||
func InitCasbinEnforcer() { |
|||
e, err := mysqlCasbin() |
|||
if err != nil { |
|||
logger.ERROR("初始化Casbin失败", err) |
|||
panic(fmt.Sprintf("初始化Casbin失败: %v", err)) |
|||
} |
|||
CasbinEnforcer = e |
|||
logger.INFO("初始化Casbin完成") |
|||
} |
|||
|
|||
func mysqlCasbin() (*casbin.Enforcer, error) { |
|||
a, err := gormadapter.NewAdapterByDB(DB) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
e, err := casbin.NewEnforcer("casbin.model", a) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
if err = e.LoadPolicy(); err != nil { |
|||
return nil, err |
|||
} |
|||
return e, nil |
|||
} |
|||
@ -0,0 +1,99 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"github.com/spf13/viper" |
|||
) |
|||
|
|||
type AppConfig struct { |
|||
ApiPort int `mapstructure:"api-port"` |
|||
Mysql Mysql `mapstructure:"mysql"` |
|||
TDengine TDengine `mapstructure:"tdengine"` |
|||
Redis string `mapstructure:"redis"` |
|||
MQTT MQTT `mapstructure:"mqtt"` |
|||
JWT JWT `mapstructure:"jwt"` |
|||
Log Log `mapstructure:"log"` |
|||
OSS OSS `mapstructure:"oss"` |
|||
Live Live `mapstructure:"live"` |
|||
Heartbeat Heartbeat `mapstructure:"heartbeat"` |
|||
} |
|||
|
|||
// AppConf 全局配置（LoadConfig 后可用）
|
|||
var AppConf *AppConfig |
|||
|
|||
type Mysql struct { |
|||
Url string `mapstructure:"url"` |
|||
Idle int `mapstructure:"idle"` |
|||
MaxConn int `mapstructure:"max-conn"` |
|||
MaxWait int `mapstructure:"max-wait"` |
|||
} |
|||
|
|||
type TDengine struct { |
|||
Dsn string `mapstructure:"dsn"` |
|||
} |
|||
|
|||
type MQTT struct { |
|||
Broker string `mapstructure:"broker"` |
|||
ClientId string `mapstructure:"client-id"` |
|||
Username string `mapstructure:"username"` |
|||
Password string `mapstructure:"password"` |
|||
} |
|||
|
|||
type Log struct { |
|||
Level string `mapstructure:"level"` |
|||
Path string `mapstructure:"path"` |
|||
} |
|||
|
|||
type JWT struct { |
|||
Secret string `mapstructure:"secret"` |
|||
AccessExpireH int `mapstructure:"access-expire-hours"` |
|||
RefreshExpireH int `mapstructure:"refresh-expire-hours"` |
|||
} |
|||
|
|||
// OSS 对象存储（S3 兼容，MinIO / 阿里云 OSS）
|
|||
type OSS struct { |
|||
Endpoint string `mapstructure:"endpoint"` // http://127.0.0.1:9000 或 https://oss-cn-hangzhou.aliyuncs.com
|
|||
AccessKey string `mapstructure:"access-key"` |
|||
SecretKey string `mapstructure:"secret-key"` |
|||
Bucket string `mapstructure:"bucket"` |
|||
Region string `mapstructure:"region"` |
|||
UseSSL bool `mapstructure:"use-ssl"` |
|||
} |
|||
|
|||
// Live 直播（阿里云直播）
|
|||
type Live struct { |
|||
Mode string `mapstructure:"mode"` // disabled / fake / aliyun
|
|||
Provider string `mapstructure:"provider"` |
|||
PushDomain string `mapstructure:"push-domain"` |
|||
PlayDomain string `mapstructure:"play-domain"` |
|||
AppName string `mapstructure:"app-name"` |
|||
AuthKey string `mapstructure:"auth-key"` |
|||
PlayAuthKey string `mapstructure:"play-auth-key"` |
|||
AuthExpireSeconds int64 `mapstructure:"auth-expire-seconds"` |
|||
PlayURLTTLSeconds int64 `mapstructure:"play-url-ttl-seconds"` |
|||
Protocol string `mapstructure:"protocol"` |
|||
SRTPort int `mapstructure:"srt-port"` |
|||
MaxSessionSeconds int64 `mapstructure:"max-session-seconds"` |
|||
ViewerLeaseSeconds int `mapstructure:"viewer-lease-seconds"` |
|||
StopGraceSeconds int `mapstructure:"stop-grace-seconds"` |
|||
ReconcileIntervalSeconds int `mapstructure:"reconcile-interval-seconds"` |
|||
RecordPolicy string `mapstructure:"record-policy"` |
|||
AllowRealCloud bool `mapstructure:"allow-real-cloud"` |
|||
} |
|||
|
|||
type Heartbeat struct { |
|||
TimeoutSeconds int `mapstructure:"timeout-seconds"` |
|||
ScanSeconds int `mapstructure:"scan-seconds"` |
|||
} |
|||
|
|||
// LoadConfig 使用 Viper 加载本地 config.yaml
|
|||
func LoadConfig(path string, conf *AppConfig) { |
|||
viper.SetConfigFile(path) |
|||
viper.SetConfigType("yaml") |
|||
if err := viper.ReadInConfig(); err != nil { |
|||
panic("加载配置文件失败: " + err.Error()) |
|||
} |
|||
if err := viper.Unmarshal(conf); err != nil { |
|||
panic("解析配置文件失败: " + err.Error()) |
|||
} |
|||
AppConf = conf |
|||
} |
|||
@ -0,0 +1,128 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"fmt" |
|||
"io" |
|||
"net/http" |
|||
"strings" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
"github.com/go-playground/validator/v10" |
|||
) |
|||
|
|||
var _engine *gin.Engine |
|||
|
|||
type R struct { |
|||
Code int `json:"code"` |
|||
Msg string `json:"msg"` |
|||
Data any `json:"data"` |
|||
} |
|||
|
|||
func Server(debug bool) { |
|||
if !debug { |
|||
gin.SetMode(gin.ReleaseMode) |
|||
} |
|||
_engine = gin.New() |
|||
_engine.Use(gin.Recovery()) |
|||
} |
|||
|
|||
func GetEngine() *gin.Engine { |
|||
return _engine |
|||
} |
|||
|
|||
func SetWriter(w io.Writer) { |
|||
gin.DefaultWriter = w |
|||
gin.DefaultErrorWriter = w |
|||
} |
|||
|
|||
func RouteGroup(path string) *gin.RouterGroup { |
|||
return _engine.Group(path) |
|||
} |
|||
|
|||
func ServerRun(port int32) { |
|||
_engine.Run(fmt.Sprintf(":%d", port)) |
|||
} |
|||
|
|||
func OK(c *gin.Context) { |
|||
c.JSON(http.StatusOK, R{Code: http.StatusOK, Msg: "ok"}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func OKNoData(c *gin.Context) { |
|||
c.JSON(http.StatusNoContent, R{Code: http.StatusOK, Msg: "ok"}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func OKWithData(c *gin.Context, obj interface{}) { |
|||
c.JSON(http.StatusOK, R{Code: http.StatusOK, Msg: "ok", Data: obj}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithCodeAndMsg(c *gin.Context, code int, message string) { |
|||
c.JSON(http.StatusBadRequest, R{Code: code, Msg: message, Data: nil}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithBusiError(c *gin.Context, busiError *BusiError) { |
|||
c.JSON(http.StatusBadRequest, R{Code: busiError.Code, Msg: busiError.Msg, Data: nil}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithBusiErrorWithHttpStatus(c *gin.Context, status int, busiError *BusiError) { |
|||
c.JSON(status, R{Code: busiError.Code, Msg: busiError.Msg, Data: nil}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithMsgWithHttpStatus(c *gin.Context, status int, msg string) { |
|||
c.JSON(status, R{Code: status, Msg: msg, Data: nil}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithBusiErrorByServer(c *gin.Context, busiError *BusiError) { |
|||
c.JSON(http.StatusInternalServerError, R{Code: busiError.Code, Msg: busiError.Msg, Data: nil}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithMsgByServer(c *gin.Context, message string) { |
|||
c.JSON(http.StatusInternalServerError, R{Code: http.StatusInternalServerError, Msg: message, Data: nil}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithMsgAndErrByServer(c *gin.Context, message string, err error) { |
|||
c.JSON(http.StatusInternalServerError, R{Code: http.StatusInternalServerError, Msg: message, Data: err}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithBindError(c *gin.Context, busiError *BusiError, err error) { |
|||
var msg string |
|||
if validationErrors, ok := err.(validator.ValidationErrors); ok { |
|||
errorMessages := validationErrors.Translate(GetTrans()) |
|||
messages := make([]string, 0) |
|||
for _, m := range errorMessages { |
|||
messages = append(messages, m) |
|||
} |
|||
msg = strings.Join(messages, "; ") |
|||
} |
|||
c.JSON(http.StatusBadRequest, R{Code: busiError.Code, Msg: msg, Data: msg}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithBusiErrorAndData(c *gin.Context, busiError *BusiError, data any) { |
|||
c.JSON(http.StatusBadRequest, R{Code: busiError.Code, Msg: busiError.Msg, Data: data}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithError(c *gin.Context, err error) { |
|||
c.JSON(http.StatusBadRequest, R{Code: http.StatusBadRequest, Msg: err.Error(), Data: nil}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithMsg(c *gin.Context, message string) { |
|||
c.JSON(http.StatusBadRequest, R{Code: http.StatusInternalServerError, Msg: message, Data: nil}) |
|||
c.Abort() |
|||
} |
|||
|
|||
func FailWithMsgAndError(c *gin.Context, msg string, err error) { |
|||
c.JSON(http.StatusBadRequest, R{Code: http.StatusBadRequest, Msg: msg, Data: nil}) |
|||
c.Abort() |
|||
} |
|||
@ -0,0 +1,87 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"errors" |
|||
"strings" |
|||
"time" |
|||
|
|||
"laic-backend/logger" |
|||
|
|||
mysql_ "github.com/go-sql-driver/mysql" |
|||
"gorm.io/driver/mysql" |
|||
"gorm.io/gorm" |
|||
) |
|||
|
|||
var DB *gorm.DB |
|||
var err error |
|||
|
|||
func InitConnection(config *Mysql) { |
|||
url := config.Url |
|||
if !strings.Contains(url, "loc=") && !strings.Contains(url, "time_zone=") { |
|||
if strings.Contains(url, "?") { |
|||
url += "&loc=Local" |
|||
} else { |
|||
url += "?loc=Local" |
|||
} |
|||
} |
|||
idle := config.Idle |
|||
maxConn := config.MaxConn |
|||
maxWait := time.Duration(config.MaxWait) |
|||
logger.DEBUG("mysql config:", config) |
|||
|
|||
gConfig := &gorm.Config{ |
|||
Logger: logger.NewGormLogger(), |
|||
} |
|||
DB, err = gorm.Open(mysql.Open(url), gConfig) |
|||
if err != nil { |
|||
logger.ERROR("mysql connect error", err) |
|||
panic(err) |
|||
} |
|||
DB = DB.Debug() |
|||
|
|||
sqlDB, err := DB.DB() |
|||
if err != nil { |
|||
logger.ERROR("init mysql data pool error", err) |
|||
panic(err) |
|||
} |
|||
sqlDB.SetMaxIdleConns(idle) |
|||
sqlDB.SetMaxOpenConns(maxConn) |
|||
sqlDB.SetConnMaxLifetime(maxWait * time.Second) |
|||
|
|||
logger.INFO("mysql ready") |
|||
} |
|||
|
|||
type Pagination struct { |
|||
PageNum int64 `json:"pageNum" form:"pageNum" binding:"required" minimum:"1"` |
|||
PageSize int64 `json:"pageSize" form:"pageSize" binding:"required" minimum:"1"` |
|||
} |
|||
|
|||
func (p *Pagination) Paginate(db *gorm.DB) *gorm.DB { |
|||
offset := int((p.PageNum - 1) * p.PageSize) |
|||
return db.Offset(offset).Limit(int(p.PageSize)) |
|||
} |
|||
|
|||
// ParseError 解析 MySQL 唯一键冲突等错误，返回错误号和约束名
|
|||
func ParseError(err error) (uint16, string) { |
|||
if err != nil { |
|||
var mysqlErr *mysql_.MySQLError |
|||
if errors.As(err, &mysqlErr) { |
|||
if mysqlErr.Number == 1062 { |
|||
return mysqlErr.Number, extractConstraintName(mysqlErr.Message) |
|||
} |
|||
return mysqlErr.Number, "" |
|||
} else { |
|||
logger.ERROR("未知错误，gorm error:", err) |
|||
} |
|||
} |
|||
return 0, "" |
|||
} |
|||
|
|||
func extractConstraintName(errMsg string) string { |
|||
start := strings.LastIndex(errMsg, "for key '") + len("for key '") |
|||
end := strings.LastIndex(errMsg, "'") |
|||
if start < end { |
|||
return errMsg[start:end] |
|||
} |
|||
return "" |
|||
} |
|||
@ -0,0 +1,101 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"crypto/hmac" |
|||
"crypto/sha256" |
|||
"encoding/hex" |
|||
"net/url" |
|||
"strconv" |
|||
"strings" |
|||
"time" |
|||
) |
|||
|
|||
// PresignOSS 生成 S3 兼容的预签名 URL（MinIO / 阿里云 OSS），method 为 HTTP 方法（GET/PUT）
|
|||
// objectKey 为对象 key（不含 bucket），expires 为有效期秒数。
|
|||
func PresignOSS(method, objectKey string, expires int64) (string, error) { |
|||
oss := AppConf.OSS |
|||
if oss.Endpoint == "" { |
|||
return "", NewBusiError(InternalError, "OSS 未配置") |
|||
} |
|||
u, err := url.Parse(oss.Endpoint) |
|||
if err != nil { |
|||
return "", err |
|||
} |
|||
host := u.Host |
|||
scheme := u.Scheme |
|||
if scheme == "" { |
|||
scheme = "https" |
|||
} |
|||
region := oss.Region |
|||
if region == "" { |
|||
region = "us-east-1" |
|||
} |
|||
const service = "s3" |
|||
|
|||
now := time.Now().UTC() |
|||
amzDate := now.Format("20060102T150405Z") |
|||
dateStamp := now.Format("20060102") |
|||
credentialScope := dateStamp + "/" + region + "/" + service + "/aws4_request" |
|||
|
|||
canonicalURI := encodeObjectPath(oss.Bucket + "/" + objectKey) |
|||
|
|||
query := url.Values{} |
|||
query.Set("X-Amz-Algorithm", "AWS4-HMAC-SHA256") |
|||
query.Set("X-Amz-Credential", oss.AccessKey+"/"+credentialScope) |
|||
query.Set("X-Amz-Date", amzDate) |
|||
query.Set("X-Amz-Expires", strconv.FormatInt(expires, 10)) |
|||
query.Set("X-Amz-SignedHeaders", "host") |
|||
|
|||
canonicalHeaders := "host:" + host + "\n" |
|||
const signedHeaders = "host" |
|||
const payloadHash = "UNSIGNED-PAYLOAD" |
|||
|
|||
canonicalRequest := strings.Join([]string{ |
|||
method, |
|||
canonicalURI, |
|||
query.Encode(), |
|||
canonicalHeaders, |
|||
signedHeaders, |
|||
payloadHash, |
|||
}, "\n") |
|||
|
|||
stringToSign := strings.Join([]string{ |
|||
"AWS4-HMAC-SHA256", |
|||
amzDate, |
|||
credentialScope, |
|||
sha256Hex([]byte(canonicalRequest)), |
|||
}, "\n") |
|||
|
|||
signingKey := deriveSigningKey(oss.SecretKey, dateStamp, region, service) |
|||
signature := hex.EncodeToString(hmacSHA256(signingKey, []byte(stringToSign))) |
|||
|
|||
query.Set("X-Amz-Signature", signature) |
|||
return scheme + "://" + host + canonicalURI + "?" + query.Encode(), nil |
|||
} |
|||
|
|||
// encodeObjectPath 对对象路径做分段 URL 编码（保留 "/" 分隔符），并以 "/" 开头
|
|||
func encodeObjectPath(key string) string { |
|||
segs := strings.Split(key, "/") |
|||
for i, s := range segs { |
|||
segs[i] = url.PathEscape(s) |
|||
} |
|||
return "/" + strings.Join(segs, "/") |
|||
} |
|||
|
|||
func sha256Hex(data []byte) string { |
|||
sum := sha256.Sum256(data) |
|||
return hex.EncodeToString(sum[:]) |
|||
} |
|||
|
|||
func hmacSHA256(key, data []byte) []byte { |
|||
h := hmac.New(sha256.New, key) |
|||
h.Write(data) |
|||
return h.Sum(nil) |
|||
} |
|||
|
|||
func deriveSigningKey(secret, date, region, service string) []byte { |
|||
kDate := hmacSHA256([]byte("AWS4"+secret), []byte(date)) |
|||
kRegion := hmacSHA256(kDate, []byte(region)) |
|||
kService := hmacSHA256(kRegion, []byte(service)) |
|||
return hmacSHA256(kService, []byte("aws4_request")) |
|||
} |
|||
@ -0,0 +1,28 @@ |
|||
package common |
|||
|
|||
import "math" |
|||
|
|||
type PageRequest[T any] struct { |
|||
Page Pagination `json:"page"` |
|||
OrderBy string `json:"order_by"` |
|||
Desc bool `json:"desc"` |
|||
Query T `json:"query"` |
|||
} |
|||
|
|||
type PageResponse[T any] struct { |
|||
PageNum int64 `json:"pageNum"` |
|||
PageSize int64 `json:"pageSize"` |
|||
Pages int `json:"pages"` |
|||
Total int `json:"total"` |
|||
Records []T `json:"records"` |
|||
} |
|||
|
|||
func Page[T any](page Pagination, total int64, records []T) *PageResponse[T] { |
|||
return &PageResponse[T]{ |
|||
PageNum: page.PageNum, |
|||
PageSize: page.PageSize, |
|||
Pages: int(math.Ceil(float64(total) / float64(page.PageSize))), |
|||
Total: int(total), |
|||
Records: records, |
|||
} |
|||
} |
|||
@ -0,0 +1,264 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"time" |
|||
|
|||
"laic-backend/logger" |
|||
|
|||
"github.com/gomodule/redigo/redis" |
|||
) |
|||
|
|||
var _redis *redis.Pool |
|||
|
|||
func InitRedis(source string) { |
|||
_redis = &redis.Pool{ |
|||
MaxIdle: 3, |
|||
IdleTimeout: 240 * time.Second, |
|||
Dial: func() (redis.Conn, error) { return redis.DialURL(source) }, |
|||
TestOnBorrow: func(c redis.Conn, t time.Time) error { |
|||
if time.Since(t) < time.Minute { |
|||
return nil |
|||
} |
|||
_, err := c.Do("PING") |
|||
return err |
|||
}, |
|||
} |
|||
logger.INFO("redis ready") |
|||
} |
|||
|
|||
func GetRedisPool() *redis.Pool { |
|||
return _redis |
|||
} |
|||
|
|||
func GetString(key string) (string, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
return redis.String(c.Do("GET", key)) |
|||
} |
|||
|
|||
func GetBytes(key string) ([]byte, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
return redis.Bytes(c.Do("GET", key)) |
|||
} |
|||
|
|||
func GetInt64(key string) (int64, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
return redis.Int64(c.Do("GET", key)) |
|||
} |
|||
|
|||
func GetJSON[T any](key string) (*T, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
r, err := redis.Bytes(c.Do("GET", key)) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
var v T |
|||
err = json.Unmarshal(r, &v) |
|||
return &v, err |
|||
} |
|||
|
|||
func SetValue(key string, n any) error { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
_, err := c.Do("SET", key, n) |
|||
return err |
|||
} |
|||
|
|||
func SetJSON(key string, n any) error { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
data, err := json.Marshal(n) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
_, err = c.Do("SET", key, data) |
|||
return err |
|||
} |
|||
|
|||
func SetJSONWithExpired(key string, n any, second uint) error { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
data, err := json.Marshal(n) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
_, err = c.Do("SET", key, data, "EX", second) |
|||
return err |
|||
} |
|||
|
|||
func SetValueWithExpired(key string, n any, second uint) error { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
_, err := c.Do("SET", key, n, "EX", second) |
|||
return err |
|||
} |
|||
|
|||
func SetRemoveIfKeyMissing(setKey, presenceKey, member string) (bool, error) { |
|||
const script = `if redis.call('EXISTS', KEYS[2]) == 0 then return redis.call('SREM', KEYS[1], ARGV[1]) end return 0` |
|||
n, err := redis.Int(exec_lua(script, []string{setKey, presenceKey}, member)) |
|||
return n == 1, err |
|||
} |
|||
|
|||
// SetNX 仅当 key 不存在时写入，返回是否写入成功
|
|||
func SetNX(key string, value any) (bool, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
n, err := redis.Int(c.Do("SET", key, value, "NX")) |
|||
return n == 1, err |
|||
} |
|||
|
|||
// IncrBy 原子自增，返回自增后的值（计费充值用）
|
|||
func IncrBy(key string, delta int64) (int64, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
return redis.Int64(c.Do("INCRBY", key, delta)) |
|||
} |
|||
|
|||
// DecrBy 原子自减，返回自减后的值（计费扣减用）
|
|||
func DecrBy(key string, delta int64) (int64, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
return redis.Int64(c.Do("DECRBY", key, delta)) |
|||
} |
|||
|
|||
func HasKey(key string) bool { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
exists, err := redis.Int(c.Do("EXISTS", key)) |
|||
if err != nil { |
|||
return false |
|||
} |
|||
return exists == 1 |
|||
} |
|||
|
|||
func HashGet(key string, field string) (string, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
return redis.String(c.Do("HGET", key, field)) |
|||
} |
|||
|
|||
func HashGetAll(key string) (map[string]string, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
return redis.StringMap(c.Do("HGETALL", key)) |
|||
} |
|||
|
|||
func HashSetValue(key string, field string, n any) error { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
_, err := c.Do("HSET", key, field, n) |
|||
return err |
|||
} |
|||
|
|||
func HashSetValues(key string, values map[string]any) error { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
args := make([]any, 0, len(values)*2+1) |
|||
args = append(args, key) |
|||
for k, v := range values { |
|||
args = append(args, k, v) |
|||
} |
|||
_, err := c.Do("HMSET", args...) |
|||
return err |
|||
} |
|||
|
|||
func SetMembers(key string) ([]string, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
return redis.Strings(c.Do("SMEMBERS", key)) |
|||
} |
|||
|
|||
func SetAdd(key string, n any) error { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
_, err := c.Do("SADD", key, n) |
|||
return err |
|||
} |
|||
|
|||
func SetRemove(key string, n any) error { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
_, err := c.Do("SREM", key, n) |
|||
return err |
|||
} |
|||
|
|||
func SetMemberExists(key string, value string) (bool, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
exists, err := redis.Int(c.Do("SISMEMBER", key, value)) |
|||
if err != nil { |
|||
return false, err |
|||
} |
|||
return exists == 1, nil |
|||
} |
|||
|
|||
func exec_lua(lua string, keys []string, args ...any) (any, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
keylen := 0 |
|||
if keys != nil { |
|||
keylen = len(keys) |
|||
} |
|||
arr := make([]any, 0, keylen+len(args)+2) |
|||
arr = append(arr, lua) |
|||
arr = append(arr, keylen) |
|||
for _, k := range keys { |
|||
arr = append(arr, k) |
|||
} |
|||
arr = append(arr, args...) |
|||
return c.Do("EVAL", arr...) |
|||
} |
|||
|
|||
func ExecLua(lua string, keys []string, args ...any) error { |
|||
_, err := exec_lua(lua, keys, args...) |
|||
return err |
|||
} |
|||
|
|||
func GetLuaInt64(lua string, keys []string, args ...any) (int64, error) { |
|||
return redis.Int64(exec_lua(lua, keys, args...)) |
|||
} |
|||
|
|||
func GetLuaStrings(lua string, keys []string, args ...any) ([]string, error) { |
|||
return redis.Strings(exec_lua(lua, keys, args...)) |
|||
} |
|||
|
|||
func GetLuaInt64s(lua string, keys []string, args ...any) ([]int64, error) { |
|||
return redis.Int64s(exec_lua(lua, keys, args...)) |
|||
} |
|||
|
|||
func Delete(key ...string) error { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
param := make([]any, len(key)) |
|||
for i, k := range key { |
|||
param[i] = k |
|||
} |
|||
_, err := c.Do("DEL", param...) |
|||
return err |
|||
} |
|||
|
|||
func GetKeysWithPrefix(prefix string) ([]string, error) { |
|||
c := _redis.Get() |
|||
defer c.Close() |
|||
cursor := 0 |
|||
var keys []string |
|||
for { |
|||
values, err := redis.Values(c.Do("SCAN", cursor, "MATCH", prefix+"*")) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
var items []string |
|||
if _, err := redis.Scan(values, &cursor, &items); err != nil { |
|||
return nil, err |
|||
} |
|||
keys = append(keys, items...) |
|||
if cursor == 0 { |
|||
break |
|||
} |
|||
} |
|||
return keys, nil |
|||
} |
|||
@ -0,0 +1,34 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"database/sql" |
|||
"time" |
|||
|
|||
"laic-backend/logger" |
|||
|
|||
_ "github.com/taosdata/driver-go/v3/taosWS" |
|||
) |
|||
|
|||
var TD *sql.DB |
|||
|
|||
// InitTDengine 初始化 TDengine 连接池。Dsn 为空时跳过；连接失败仅告警不 panic（部署期确认）
|
|||
func InitTDengine(config *TDengine) { |
|||
if config == nil || config.Dsn == "" { |
|||
logger.WARN("TDengine dsn 为空，跳过初始化") |
|||
return |
|||
} |
|||
db, err := sql.Open("taosWS", config.Dsn) |
|||
if err != nil { |
|||
logger.ERROR("TDengine connect error", err) |
|||
return |
|||
} |
|||
db.SetMaxOpenConns(10) |
|||
db.SetMaxIdleConns(5) |
|||
db.SetConnMaxLifetime(time.Hour) |
|||
if err := db.Ping(); err != nil { |
|||
logger.ERROR("TDengine ping error", err) |
|||
return |
|||
} |
|||
TD = db |
|||
logger.INFO("TDengine ready") |
|||
} |
|||
@ -0,0 +1,81 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"regexp" |
|||
|
|||
"github.com/gin-gonic/gin/binding" |
|||
"github.com/go-playground/locales/zh" |
|||
ut "github.com/go-playground/universal-translator" |
|||
"github.com/go-playground/validator/v10" |
|||
zh_translations "github.com/go-playground/validator/v10/translations/zh" |
|||
) |
|||
|
|||
var ( |
|||
phoneRegexp = regexp.MustCompile(`^1[3456789]\d{9}$`) |
|||
emailRegexp = regexp.MustCompile(`^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$`) |
|||
) |
|||
|
|||
// IsPhoneValid 验证手机号格式
|
|||
func IsPhoneValid(fl validator.FieldLevel) bool { |
|||
return phoneRegexp.MatchString(fl.Field().String()) |
|||
} |
|||
|
|||
// IsEmailValid 验证邮箱格式
|
|||
func IsEmailValid(fl validator.FieldLevel) bool { |
|||
return emailRegexp.MatchString(fl.Field().String()) |
|||
} |
|||
|
|||
func ValidatePassword(fl validator.FieldLevel) bool { |
|||
password := fl.Field().String() |
|||
if password == "" { |
|||
return true |
|||
} |
|||
if len(password) < 8 || len(password) > 16 { |
|||
return false |
|||
} |
|||
if !regexp.MustCompile(`[0-9]`).MatchString(password) || |
|||
!regexp.MustCompile(`[a-zA-Z]`).MatchString(password) { |
|||
return false |
|||
} |
|||
return true |
|||
} |
|||
|
|||
var trans ut.Translator |
|||
|
|||
func init() { |
|||
if v, ok := binding.Validator.Engine().(*validator.Validate); ok { |
|||
v.RegisterValidation("password", ValidatePassword) |
|||
v.RegisterValidation("email", IsEmailValid) |
|||
v.RegisterValidation("phone", IsPhoneValid) |
|||
|
|||
zhCn := zh.New() |
|||
uni := ut.New(zhCn, zhCn) |
|||
trans, _ = uni.GetTranslator("zh") |
|||
zh_translations.RegisterDefaultTranslations(v, trans) |
|||
|
|||
v.RegisterTranslation("password", trans, func(ut ut.Translator) error { |
|||
return ut.Add("password", "{0}必须包含数字、字母，长度在8-16位之间", true) |
|||
}, func(ut ut.Translator, fe validator.FieldError) string { |
|||
t, _ := ut.T("password", fe.Field()) |
|||
return t |
|||
}) |
|||
|
|||
v.RegisterTranslation("phone", trans, func(ut ut.Translator) error { |
|||
return ut.Add("phone", "{0}格式不正确", true) |
|||
}, func(ut ut.Translator, fe validator.FieldError) string { |
|||
t, _ := ut.T("phone", fe.Field()) |
|||
return t |
|||
}) |
|||
|
|||
v.RegisterTranslation("email", trans, func(ut ut.Translator) error { |
|||
return ut.Add("email", "{0}格式不正确", true) |
|||
}, func(ut ut.Translator, fe validator.FieldError) string { |
|||
t, _ := ut.T("email", fe.Field()) |
|||
return t |
|||
}) |
|||
} |
|||
} |
|||
|
|||
func GetTrans() ut.Translator { |
|||
return trans |
|||
} |
|||
@ -0,0 +1,47 @@ |
|||
api-port: 8080 |
|||
mysql: |
|||
url: root:123456@tcp(127.0.0.1:3306)/laic?charset=utf8mb4&parseTime=True&loc=Local |
|||
idle: 10 |
|||
max-conn: 50 |
|||
max-wait: 3600 |
|||
tdengine: |
|||
dsn: root:taosdata@ws(127.0.0.1:6041)/laic |
|||
redis: redis://127.0.0.1:6379/0 |
|||
mqtt: |
|||
broker: tcp://127.0.0.1:1883 |
|||
client-id: laic-backend |
|||
username: laic |
|||
password: "" |
|||
jwt: |
|||
secret: laic-jwt-secret-change-me |
|||
access-expire-hours: 168 # 7 天 |
|||
refresh-expire-hours: 720 # 30 天 |
|||
oss: |
|||
endpoint: http://127.0.0.1:9000 |
|||
access-key: minioadmin |
|||
secret-key: minioadmin |
|||
bucket: laic-media |
|||
region: us-east-1 |
|||
use-ssl: false |
|||
live: |
|||
mode: fake |
|||
provider: fake |
|||
push-domain: local-push |
|||
play-domain: local-play |
|||
app-name: dock-live |
|||
auth-expire-seconds: 7200 |
|||
play-url-ttl-seconds: 300 |
|||
protocol: srt |
|||
srt-port: 1105 |
|||
max-session-seconds: 7200 |
|||
viewer-lease-seconds: 30 |
|||
stop-grace-seconds: 15 |
|||
reconcile-interval-seconds: 10 |
|||
record-policy: disabled |
|||
allow-real-cloud: false |
|||
heartbeat: |
|||
timeout-seconds: 30 |
|||
scan-seconds: 5 |
|||
log: |
|||
level: debug |
|||
path: ./logs/ |
|||
@ -0,0 +1,94 @@ |
|||
module laic-backend |
|||
|
|||
go 1.26.4 |
|||
|
|||
require ( |
|||
github.com/bluenviron/gomavlib/v3 v3.3.5 |
|||
github.com/casbin/casbin/v3 v3.8.1 |
|||
github.com/casbin/gorm-adapter/v3 v3.41.0 |
|||
github.com/eclipse/paho.mqtt.golang v1.5.1 |
|||
github.com/gin-gonic/gin v1.12.0 |
|||
github.com/go-playground/locales v0.14.1 |
|||
github.com/go-playground/universal-translator v0.18.1 |
|||
github.com/go-playground/validator/v10 v10.30.3 |
|||
github.com/go-sql-driver/mysql v1.10.0 |
|||
github.com/golang-jwt/jwt/v5 v5.3.1 |
|||
github.com/gomodule/redigo v1.9.3 |
|||
github.com/google/uuid v1.6.0 |
|||
github.com/gorilla/websocket v1.5.3 |
|||
github.com/robfig/cron/v3 v3.0.1 |
|||
github.com/spf13/viper v1.21.0 |
|||
github.com/taosdata/driver-go/v3 v3.3.1 |
|||
golang.org/x/crypto v0.52.0 |
|||
gopkg.in/natefinch/lumberjack.v2 v2.2.1 |
|||
gorm.io/driver/mysql v1.6.0 |
|||
gorm.io/gorm v1.31.2 |
|||
) |
|||
|
|||
require ( |
|||
filippo.io/edwards25519 v1.2.0 // indirect |
|||
github.com/bmatcuk/doublestar/v4 v4.9.1 // indirect |
|||
github.com/bytedance/gopkg v0.1.3 // indirect |
|||
github.com/bytedance/sonic v1.15.0 // indirect |
|||
github.com/bytedance/sonic/loader v0.5.0 // indirect |
|||
github.com/casbin/govaluate v1.10.0 // indirect |
|||
github.com/cloudwego/base64x v0.1.6 // indirect |
|||
github.com/dustin/go-humanize v1.0.1 // indirect |
|||
github.com/fsnotify/fsnotify v1.9.0 // indirect |
|||
github.com/gabriel-vasile/mimetype v1.4.13 // indirect |
|||
github.com/gin-contrib/sse v1.1.0 // indirect |
|||
github.com/glebarez/go-sqlite v1.22.0 // indirect |
|||
github.com/glebarez/sqlite v1.11.0 // indirect |
|||
github.com/go-viper/mapstructure/v2 v2.4.0 // indirect |
|||
github.com/goccy/go-json v0.10.5 // indirect |
|||
github.com/goccy/go-yaml v1.19.2 // indirect |
|||
github.com/golang-sql/civil v0.0.0-20220223132316-b832511892a9 // indirect |
|||
github.com/golang-sql/sqlexp v0.1.0 // indirect |
|||
github.com/jackc/pgpassfile v1.0.0 // indirect |
|||
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect |
|||
github.com/jackc/pgx/v5 v5.8.0 // indirect |
|||
github.com/jackc/puddle/v2 v2.2.2 // indirect |
|||
github.com/jinzhu/inflection v1.0.0 // indirect |
|||
github.com/jinzhu/now v1.1.5 // indirect |
|||
github.com/json-iterator/go v1.1.12 // indirect |
|||
github.com/klauspost/cpuid/v2 v2.3.0 // indirect |
|||
github.com/leodido/go-urn v1.4.0 // indirect |
|||
github.com/mattn/go-isatty v0.0.20 // indirect |
|||
github.com/microsoft/go-mssqldb v1.9.5 // indirect |
|||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect |
|||
github.com/modern-go/reflect2 v1.0.2 // indirect |
|||
github.com/ncruces/go-strftime v1.0.0 // indirect |
|||
github.com/pelletier/go-toml/v2 v2.2.4 // indirect |
|||
github.com/pion/logging v0.2.2 // indirect |
|||
github.com/pion/transport/v2 v2.2.10 // indirect |
|||
github.com/pkg/errors v0.9.1 // indirect |
|||
github.com/quic-go/qpack v0.6.0 // indirect |
|||
github.com/quic-go/quic-go v0.59.0 // indirect |
|||
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect |
|||
github.com/sagikazarmark/locafero v0.11.0 // indirect |
|||
github.com/shopspring/decimal v1.4.0 // indirect |
|||
github.com/sourcegraph/conc v0.3.1-0.20240121214520-5f936abd7ae8 // indirect |
|||
github.com/spf13/afero v1.15.0 // indirect |
|||
github.com/spf13/cast v1.10.0 // indirect |
|||
github.com/spf13/pflag v1.0.10 // indirect |
|||
github.com/subosito/gotenv v1.6.0 // indirect |
|||
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect |
|||
github.com/ugorji/go/codec v1.3.1 // indirect |
|||
go.bug.st/serial v1.7.1 // indirect |
|||
go.mongodb.org/mongo-driver/v2 v2.5.0 // indirect |
|||
go.yaml.in/yaml/v3 v3.0.4 // indirect |
|||
golang.org/x/arch v0.22.0 // indirect |
|||
golang.org/x/exp v0.0.0-20251219203646-944ab1f22d93 // indirect |
|||
golang.org/x/net v0.54.0 // indirect |
|||
golang.org/x/sync v0.20.0 // indirect |
|||
golang.org/x/sys v0.45.0 // indirect |
|||
golang.org/x/text v0.37.0 // indirect |
|||
google.golang.org/protobuf v1.36.10 // indirect |
|||
gorm.io/driver/postgres v1.6.0 // indirect |
|||
gorm.io/driver/sqlserver v1.6.3 // indirect |
|||
gorm.io/plugin/dbresolver v1.6.2 // indirect |
|||
modernc.org/libc v1.67.4 // indirect |
|||
modernc.org/mathutil v1.7.1 // indirect |
|||
modernc.org/memory v1.11.0 // indirect |
|||
modernc.org/sqlite v1.42.2 // indirect |
|||
) |
|||
@ -0,0 +1,409 @@ |
|||
filippo.io/edwards25519 v1.2.0 h1:crnVqOiS4jqYleHd9vaKZ+HKtHfllngJIiOpNpoJsjo= |
|||
filippo.io/edwards25519 v1.2.0/go.mod h1:xzAOLCNug/yB62zG1bQ8uziwrIqIuxhctzJT18Q77mc= |
|||
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.7.0/go.mod h1:bjGvMhVMb+EEm3VRNQawDMUyMMjo+S5ewNjflkep/0Q= |
|||
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.7.1/go.mod h1:bjGvMhVMb+EEm3VRNQawDMUyMMjo+S5ewNjflkep/0Q= |
|||
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.11.1/go.mod h1:a6xsAQUZg+VsS3TJ05SRp524Hs4pZ/AeFSr5ENf0Yjo= |
|||
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.18.0 h1:Gt0j3wceWMwPmiazCa8MzMA0MfhmPIz0Qp0FJ6qcM0U= |
|||
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.18.0/go.mod h1:Ot/6aikWnKWi4l9QB7qVSwa8iMphQNqkWALMoNT3rzM= |
|||
github.com/Azure/azure-sdk-for-go/sdk/azidentity v1.3.1/go.mod h1:uE9zaUfEQT/nbQjVi2IblCG9iaLtZsuYZ8ne+PuQ02M= |
|||
github.com/Azure/azure-sdk-for-go/sdk/azidentity v1.6.0/go.mod h1:9kIvujWAA58nmPmWB1m23fyWic1kYZMxD9CxaWn4Qpg= |
|||
github.com/Azure/azure-sdk-for-go/sdk/azidentity v1.10.1 h1:B+blDbyVIG3WaikNxPnhPiJ1MThR03b3vKGtER95TP4= |
|||
github.com/Azure/azure-sdk-for-go/sdk/azidentity v1.10.1/go.mod h1:JdM5psgjfBf5fo2uWOZhflPWyDBZ/O/CNAH9CtsuZE4= |
|||
github.com/Azure/azure-sdk-for-go/sdk/internal v1.3.0/go.mod h1:okt5dMMTOFjX/aovMlrjvvXoPMBVSPzk9185BT0+eZM= |
|||
github.com/Azure/azure-sdk-for-go/sdk/internal v1.5.2/go.mod h1:yInRyqWXAuaPrgI7p70+lDDgh3mlBohis29jGMISnmc= |
|||
github.com/Azure/azure-sdk-for-go/sdk/internal v1.8.0/go.mod h1:4OG6tQ9EOP/MT0NMjDlRzWoVFxfu9rN9B2X+tlSVktg= |
|||
github.com/Azure/azure-sdk-for-go/sdk/internal v1.11.1 h1:FPKJS1T+clwv+OLGt13a8UjqeRuh0O4SJ3lUriThc+4= |
|||
github.com/Azure/azure-sdk-for-go/sdk/internal v1.11.1/go.mod h1:j2chePtV91HrC22tGoRX3sGY42uF13WzmmV80/OdVAA= |
|||
github.com/Azure/azure-sdk-for-go/sdk/security/keyvault/azkeys v1.0.1/go.mod h1:GpPjLhVR9dnUoJMyHWSPy71xY9/lcmpzIPZXmF0FCVY= |
|||
github.com/Azure/azure-sdk-for-go/sdk/security/keyvault/azkeys v1.3.1 h1:Wgf5rZba3YZqeTNJPtvqZoBu1sBN/L4sry+u2U3Y75w= |
|||
github.com/Azure/azure-sdk-for-go/sdk/security/keyvault/azkeys v1.3.1/go.mod h1:xxCBG/f/4Vbmh2XQJBsOmNdxWUY5j/s27jujKPbQf14= |
|||
github.com/Azure/azure-sdk-for-go/sdk/security/keyvault/internal v1.0.0/go.mod h1:bTSOgj05NGRuHHhQwAdPnYr9TOdNmKlZTgGLL6nyAdI= |
|||
github.com/Azure/azure-sdk-for-go/sdk/security/keyvault/internal v1.1.1 h1:bFWuoEKg+gImo7pvkiQEFAc8ocibADgXeiLAxWhWmkI= |
|||
github.com/Azure/azure-sdk-for-go/sdk/security/keyvault/internal v1.1.1/go.mod h1:Vih/3yc6yac2JzU4hzpaDupBJP0Flaia9rXXrU8xyww= |
|||
github.com/AzureAD/microsoft-authentication-library-for-go v1.1.1/go.mod h1:wP83P5OoQ5p6ip3ScPr0BAq0BvuPAvacpEuSzyouqAI= |
|||
github.com/AzureAD/microsoft-authentication-library-for-go v1.2.2/go.mod h1:wP83P5OoQ5p6ip3ScPr0BAq0BvuPAvacpEuSzyouqAI= |
|||
github.com/AzureAD/microsoft-authentication-library-for-go v1.4.2 h1:oygO0locgZJe7PpYPXT5A29ZkwJaPqcva7BVeemZOZs= |
|||
github.com/AzureAD/microsoft-authentication-library-for-go v1.4.2/go.mod h1:wP83P5OoQ5p6ip3ScPr0BAq0BvuPAvacpEuSzyouqAI= |
|||
github.com/bluenviron/gomavlib/v3 v3.3.5 h1:4e0rxKqPH1ar1cJFkft0mjsdBDnVJKWUhDQQuuqhvyk= |
|||
github.com/bluenviron/gomavlib/v3 v3.3.5/go.mod h1:tm3CnJMWdWqDGtHwUpmM2B8VY7TF67hCrCNBrwjYJpc= |
|||
github.com/bmatcuk/doublestar/v4 v4.6.1/go.mod h1:xBQ8jztBU6kakFMg+8WGxn0c6z1fTSPVIjEY1Wr7jzc= |
|||
github.com/bmatcuk/doublestar/v4 v4.9.1 h1:X8jg9rRZmJd4yRy7ZeNDRnM+T3ZfHv15JiBJ/avrEXE= |
|||
github.com/bmatcuk/doublestar/v4 v4.9.1/go.mod h1:xBQ8jztBU6kakFMg+8WGxn0c6z1fTSPVIjEY1Wr7jzc= |
|||
github.com/bytedance/gopkg v0.1.3 h1:TPBSwH8RsouGCBcMBktLt1AymVo2TVsBVCY4b6TnZ/M= |
|||
github.com/bytedance/gopkg v0.1.3/go.mod h1:576VvJ+eJgyCzdjS+c4+77QF3p7ubbtiKARP3TxducM= |
|||
github.com/bytedance/sonic v1.15.0 h1:/PXeWFaR5ElNcVE84U0dOHjiMHQOwNIx3K4ymzh/uSE= |
|||
github.com/bytedance/sonic v1.15.0/go.mod h1:tFkWrPz0/CUCLEF4ri4UkHekCIcdnkqXw9VduqpJh0k= |
|||
github.com/bytedance/sonic/loader v0.5.0 h1:gXH3KVnatgY7loH5/TkeVyXPfESoqSBSBEiDd5VjlgE= |
|||
github.com/bytedance/sonic/loader v0.5.0/go.mod h1:AR4NYCk5DdzZizZ5djGqQ92eEhCCcdf5x77udYiSJRo= |
|||
github.com/casbin/casbin/v3 v3.8.1 h1:D4dEY4knePPR4YgNP5WZtWNaOxD0UK0LpPy9+zxtBwo= |
|||
github.com/casbin/casbin/v3 v3.8.1/go.mod h1:5rJbQr2e6AuuDDNxnPc5lQlC9nIgg6nS1zYwKXhpHC8= |
|||
github.com/casbin/gorm-adapter/v3 v3.41.0 h1:Xhpi0tfRP9aKPDWDf6dgBxHZ9UM6IophxxPIEGWqCNM= |
|||
github.com/casbin/gorm-adapter/v3 v3.41.0/go.mod h1:BQZRJhwUnwMpI+pT2m7/cUJwXxrHfzpBpPcNTyMGeGA= |
|||
github.com/casbin/govaluate v1.3.0/go.mod h1:G/UnbIjZk/0uMNaLwZZmFQrR72tYRZWQkO70si/iR7A= |
|||
github.com/casbin/govaluate v1.10.0 h1:ffGw51/hYH3w3rZcxO/KcaUIDOLP84w7nsidMVgaDG0= |
|||
github.com/casbin/govaluate v1.10.0/go.mod h1:G/UnbIjZk/0uMNaLwZZmFQrR72tYRZWQkO70si/iR7A= |
|||
github.com/cloudwego/base64x v0.1.6 h1:t11wG9AECkCDk5fMSoxmufanudBtJ+/HemLstXDLI2M= |
|||
github.com/cloudwego/base64x v0.1.6/go.mod h1:OFcloc187FXDaYHvrNIjxSe8ncn0OOM8gEHfghB2IPU= |
|||
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E= |
|||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= |
|||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= |
|||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= |
|||
github.com/dnaeon/go-vcr v1.1.0/go.mod h1:M7tiix8f0r6mKKJ3Yq/kqU1OYf3MnfmBWVbPx/yU9ko= |
|||
github.com/dnaeon/go-vcr v1.2.0/go.mod h1:R4UdLID7HZT3taECzJs4YgbbH6PIGXB6W/sc5OLb6RQ= |
|||
github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY= |
|||
github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto= |
|||
github.com/eclipse/paho.mqtt.golang v1.5.1 h1:/VSOv3oDLlpqR2Epjn1Q7b2bSTplJIeV2ISgCl2W7nE= |
|||
github.com/eclipse/paho.mqtt.golang v1.5.1/go.mod h1:1/yJCneuyOoCOzKSsOTUc0AJfpsItBGWvYpBLimhArU= |
|||
github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8= |
|||
github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0= |
|||
github.com/fsnotify/fsnotify v1.9.0 h1:2Ml+OJNzbYCTzsxtv8vKSFD9PbJjmhYF14k/jKC7S9k= |
|||
github.com/fsnotify/fsnotify v1.9.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0= |
|||
github.com/gabriel-vasile/mimetype v1.4.13 h1:46nXokslUBsAJE/wMsp5gtO500a4F3Nkz9Ufpk2AcUM= |
|||
github.com/gabriel-vasile/mimetype v1.4.13/go.mod h1:d+9Oxyo1wTzWdyVUPMmXFvp4F9tea18J8ufA774AB3s= |
|||
github.com/gin-contrib/sse v1.1.0 h1:n0w2GMuUpWDVp7qSpvze6fAu9iRxJY4Hmj6AmBOU05w= |
|||
github.com/gin-contrib/sse v1.1.0/go.mod h1:hxRZ5gVpWMT7Z0B0gSNYqqsSCNIJMjzvm6fqCz9vjwM= |
|||
github.com/gin-gonic/gin v1.12.0 h1:b3YAbrZtnf8N//yjKeU2+MQsh2mY5htkZidOM7O0wG8= |
|||
github.com/gin-gonic/gin v1.12.0/go.mod h1:VxccKfsSllpKshkBWgVgRniFFAzFb9csfngsqANjnLc= |
|||
github.com/glebarez/go-sqlite v1.22.0 h1:uAcMJhaA6r3LHMTFgP0SifzgXg46yJkgxqyuyec+ruQ= |
|||
github.com/glebarez/go-sqlite v1.22.0/go.mod h1:PlBIdHe0+aUEFn+r2/uthrWq4FxbzugL0L8Li6yQJbc= |
|||
github.com/glebarez/sqlite v1.11.0 h1:wSG0irqzP6VurnMEpFGer5Li19RpIRi2qvQz++w0GMw= |
|||
github.com/glebarez/sqlite v1.11.0/go.mod h1:h8/o8j5wiAsqSPoWELDUdJXhjAhsVliSn7bWZjOhrgQ= |
|||
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= |
|||
github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4= |
|||
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= |
|||
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= |
|||
github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= |
|||
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY= |
|||
github.com/go-playground/validator/v10 v10.30.3 h1:4MU6YkEwx7GbcPJOZxrtbu+QfF3pJLJuaYTeAH0DYy8= |
|||
github.com/go-playground/validator/v10 v10.30.3/go.mod h1:4Axh7oCNGcoGkqLoE4YWt6n20mcEIsPRlB7vPk3lpyc= |
|||
github.com/go-sql-driver/mysql v1.10.0 h1:Q+1LV8DkHJvSYAdR83XzuhDaTykuDx0l6fkXxoWCWfw= |
|||
github.com/go-sql-driver/mysql v1.10.0/go.mod h1:M+cqaI7+xxXGG9swrdeUIoPG3Y3KCkF0pZej+SK+nWk= |
|||
github.com/go-viper/mapstructure/v2 v2.4.0 h1:EBsztssimR/CONLSZZ04E8qAkxNYq4Qp9LvH92wZUgs= |
|||
github.com/go-viper/mapstructure/v2 v2.4.0/go.mod h1:oJDH3BJKyqBA2TXFhDsKDGDTlndYOZ6rGS0BRZIxGhM= |
|||
github.com/goccy/go-json v0.10.5 h1:Fq85nIqj+gXn/S5ahsiTlK3TmC85qgirsdTP/+DeaC4= |
|||
github.com/goccy/go-json v0.10.5/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M= |
|||
github.com/goccy/go-yaml v1.19.2 h1:PmFC1S6h8ljIz6gMRBopkjP1TVT7xuwrButHID66PoM= |
|||
github.com/goccy/go-yaml v1.19.2/go.mod h1:XBurs7gK8ATbW4ZPGKgcbrY1Br56PdM69F7LkFRi1kA= |
|||
github.com/golang-jwt/jwt/v5 v5.0.0/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk= |
|||
github.com/golang-jwt/jwt/v5 v5.2.1/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk= |
|||
github.com/golang-jwt/jwt/v5 v5.2.2/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk= |
|||
github.com/golang-jwt/jwt/v5 v5.3.1 h1:kYf81DTWFe7t+1VvL7eS+jKFVWaUnK9cB1qbwn63YCY= |
|||
github.com/golang-jwt/jwt/v5 v5.3.1/go.mod h1:fxCRLWMO43lRc8nhHWY6LGqRcf+1gQWArsqaEUEa5bE= |
|||
github.com/golang-sql/civil v0.0.0-20220223132316-b832511892a9 h1:au07oEsX2xN0ktxqI+Sida1w446QrXBRJ0nee3SNZlA= |
|||
github.com/golang-sql/civil v0.0.0-20220223132316-b832511892a9/go.mod h1:8vg3r2VgvsThLBIFL93Qb5yWzgyZWhEmBwUJWevAkK0= |
|||
github.com/golang-sql/sqlexp v0.1.0 h1:ZCD6MBpcuOVfGVqsEmY5/4FtYiKz6tSyUv9LPEDei6A= |
|||
github.com/golang-sql/sqlexp v0.1.0/go.mod h1:J4ad9Vo8ZCWQ2GMrC4UCQy1JpCbwU9m3EOqtpKwwwHI= |
|||
github.com/gomodule/redigo v1.9.3 h1:dNPSXeXv6HCq2jdyWfjgmhBdqnR6PRO3m/G05nvpPC8= |
|||
github.com/gomodule/redigo v1.9.3/go.mod h1:KsU3hiK/Ay8U42qpaJk+kuNa3C+spxapWpM+ywhcgtw= |
|||
github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= |
|||
github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8= |
|||
github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU= |
|||
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= |
|||
github.com/google/pprof v0.0.0-20250317173921-a4b03ec1a45e h1:ijClszYn+mADRFY17kjQEVQ1XRhq2/JR1M3sGqeJoxs= |
|||
github.com/google/pprof v0.0.0-20250317173921-a4b03ec1a45e/go.mod h1:boTsfXsheKC2y+lKOCMpSfarhxDeIzfZG1jqGcPl3cA= |
|||
github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= |
|||
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= |
|||
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= |
|||
github.com/gorilla/securecookie v1.1.1/go.mod h1:ra0sb63/xPlUeL+yeDciTfxMRAA+MP+HVt/4epWDjd4= |
|||
github.com/gorilla/sessions v1.2.1/go.mod h1:dk2InVEVJ0sfLlnXv9EAgkf6ecYs/i80K/zI+bUmuGM= |
|||
github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= |
|||
github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg= |
|||
github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= |
|||
github.com/hashicorp/go-uuid v1.0.2/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro= |
|||
github.com/hashicorp/go-uuid v1.0.3/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro= |
|||
github.com/hashicorp/golang-lru/v2 v2.0.7 h1:a+bsQ5rvGLjzHuww6tVxozPZFVghXaHOwFs4luLUK2k= |
|||
github.com/hashicorp/golang-lru/v2 v2.0.7/go.mod h1:QeFd9opnmA6QUJc5vARoKUSoFhyfM2/ZepoAG6RGpeM= |
|||
github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM= |
|||
github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg= |
|||
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo= |
|||
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM= |
|||
github.com/jackc/pgx/v5 v5.8.0 h1:TYPDoleBBme0xGSAX3/+NujXXtpZn9HBONkQC7IEZSo= |
|||
github.com/jackc/pgx/v5 v5.8.0/go.mod h1:QVeDInX2m9VyzvNeiCJVjCkNFqzsNb43204HshNSZKw= |
|||
github.com/jackc/puddle/v2 v2.2.2 h1:PR8nw+E/1w0GLuRFSmiioY6UooMp6KJv0/61nB7icHo= |
|||
github.com/jackc/puddle/v2 v2.2.2/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4= |
|||
github.com/jcmturner/aescts/v2 v2.0.0/go.mod h1:AiaICIRyfYg35RUkr8yESTqvSy7csK90qZ5xfvvsoNs= |
|||
github.com/jcmturner/dnsutils/v2 v2.0.0/go.mod h1:b0TnjGOvI/n42bZa+hmXL+kFJZsFT7G4t3HTlQ184QM= |
|||
github.com/jcmturner/gofork v1.7.6/go.mod h1:1622LH6i/EZqLloHfE7IeZ0uEJwMSUyQ/nDd82IeqRo= |
|||
github.com/jcmturner/goidentity/v6 v6.0.1/go.mod h1:X1YW3bgtvwAXju7V3LCIMpY0Gbxyjn/mY9zx4tFonSg= |
|||
github.com/jcmturner/gokrb5/v8 v8.4.4/go.mod h1:1btQEpgT6k+unzCwX1KdWMEwPPkkgBtP+F6aCACiMrs= |
|||
github.com/jcmturner/rpc/v2 v2.0.3/go.mod h1:VUJYCIDm3PVOEHw8sgt091/20OJjskO/YJki3ELg/Hc= |
|||
github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= |
|||
github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc= |
|||
github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ= |
|||
github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= |
|||
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= |
|||
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= |
|||
github.com/klauspost/cpuid/v2 v2.3.0 h1:S4CRMLnYUhGeDFDqkGriYKdfoFlDnMtqTiI/sFzhA9Y= |
|||
github.com/klauspost/cpuid/v2 v2.3.0/go.mod h1:hqwkgyIinND0mEev00jJYCxPNVRVXFQeu1XKlok6oO0= |
|||
github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= |
|||
github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= |
|||
github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= |
|||
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= |
|||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= |
|||
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= |
|||
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= |
|||
github.com/kylelemons/godebug v1.1.0 h1:RPNrshWIDI6G2gRW9EHilWtl7Z6Sb1BR0xunSBf0SNc= |
|||
github.com/kylelemons/godebug v1.1.0/go.mod h1:9/0rRGxNHcop5bhtWyNeEfOS8JIWk580+fNqagV/RAw= |
|||
github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ= |
|||
github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI= |
|||
github.com/lib/pq v1.10.2 h1:AqzbZs4ZoCBp+GtejcpCpcxM3zlSMx29dXbUSeVtJb8= |
|||
github.com/lib/pq v1.10.2/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o= |
|||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= |
|||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= |
|||
github.com/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU= |
|||
github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y= |
|||
github.com/microsoft/go-mssqldb v1.8.2/go.mod h1:vp38dT33FGfVotRiTmDo3bFyaHq+p3LektQrjTULowo= |
|||
github.com/microsoft/go-mssqldb v1.9.5 h1:orwya0X/5bsL1o+KasupTkk2eNTNFkTQG0BEe/HxCn0= |
|||
github.com/microsoft/go-mssqldb v1.9.5/go.mod h1:VCP2a0KEZZtGLRHd1PsLavLFYy/3xX2yJUPycv3Sr2Q= |
|||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= |
|||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= |
|||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= |
|||
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= |
|||
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= |
|||
github.com/modocache/gover v0.0.0-20171022184752-b58185e213c5/go.mod h1:caMODM3PzxT8aQXRPkAt8xlV/e7d7w8GM5g0fa5F0D8= |
|||
github.com/montanaflynn/stats v0.7.0/go.mod h1:etXPPgVO6n31NxCd9KQUMvCM+ve0ruNzt6R8Bnaayow= |
|||
github.com/ncruces/go-strftime v1.0.0 h1:HMFp8mLCTPp341M/ZnA4qaf7ZlsbTc+miZjCLOFAw7w= |
|||
github.com/ncruces/go-strftime v1.0.0/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls= |
|||
github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4= |
|||
github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY= |
|||
github.com/pion/logging v0.2.2 h1:M9+AIj/+pxNsDfAT64+MAVgJO0rsyLnoJKCqf//DoeY= |
|||
github.com/pion/logging v0.2.2/go.mod h1:k0/tDVsRCX2Mb2ZEmTqNa7CWsQPc+YYCB7Q+5pahoms= |
|||
github.com/pion/transport/v2 v2.2.10 h1:ucLBLE8nuxiHfvkFKnkDQRYWYfp8ejf4YBOPfaQpw6Q= |
|||
github.com/pion/transport/v2 v2.2.10/go.mod h1:sq1kSLWs+cHW9E+2fJP95QudkzbK7wscs8yYgQToO5E= |
|||
github.com/pion/transport/v3 v3.0.6 h1:k1mQU06bmmX143qSWgXFqSH1KUJceQvIUuVH/K5ELWw= |
|||
github.com/pion/transport/v3 v3.0.6/go.mod h1:HvJr2N/JwNJAfipsRleqwFoR3t/pWyHeZUs89v3+t5s= |
|||
github.com/pkg/browser v0.0.0-20210911075715-681adbf594b8/go.mod h1:HKlIX3XHQyzLZPlr7++PzdhaXEj94dEiJgZDTsxEqUI= |
|||
github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c h1:+mdjkGKdHQG3305AYmdv1U2eRNDiU2ErMBj1gwrq8eQ= |
|||
github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c/go.mod h1:7rwL4CYBLnjLxUqIJNnCWiEdr3bn6IUYi15bNlnbCCU= |
|||
github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA= |
|||
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4= |
|||
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= |
|||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= |
|||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= |
|||
github.com/quic-go/qpack v0.6.0 h1:g7W+BMYynC1LbYLSqRt8PBg5Tgwxn214ZZR34VIOjz8= |
|||
github.com/quic-go/qpack v0.6.0/go.mod h1:lUpLKChi8njB4ty2bFLX2x4gzDqXwUpaO1DP9qMDZII= |
|||
github.com/quic-go/quic-go v0.59.0 h1:OLJkp1Mlm/aS7dpKgTc6cnpynnD2Xg7C1pwL6vy/SAw= |
|||
github.com/quic-go/quic-go v0.59.0/go.mod h1:upnsH4Ju1YkqpLXC305eW3yDZ4NfnNbmQRCMWS58IKU= |
|||
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE= |
|||
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo= |
|||
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs= |
|||
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro= |
|||
github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs= |
|||
github.com/rogpeppe/go-internal v1.12.0 h1:exVL4IDcn6na9z1rAb56Vxr+CgyK3nn3O+epU5NdKM8= |
|||
github.com/rogpeppe/go-internal v1.12.0/go.mod h1:E+RYuTGaKKdloAfM02xzb0FW3Paa99yedzYV+kq4uf4= |
|||
github.com/sagikazarmark/locafero v0.11.0 h1:1iurJgmM9G3PA/I+wWYIOw/5SyBtxapeHDcg+AAIFXc= |
|||
github.com/sagikazarmark/locafero v0.11.0/go.mod h1:nVIGvgyzw595SUSUE6tvCp3YYTeHs15MvlmU87WwIik= |
|||
github.com/shopspring/decimal v1.4.0 h1:bxl37RwXBklmTi0C79JfXCEBD1cqqHt0bbgBAGFp81k= |
|||
github.com/shopspring/decimal v1.4.0/go.mod h1:gawqmDU56v4yIKSwfBSFip1HdCCXN8/+DMd9qYNcwME= |
|||
github.com/sourcegraph/conc v0.3.1-0.20240121214520-5f936abd7ae8 h1:+jumHNA0Wrelhe64i8F6HNlS8pkoyMv5sreGx2Ry5Rw= |
|||
github.com/sourcegraph/conc v0.3.1-0.20240121214520-5f936abd7ae8/go.mod h1:3n1Cwaq1E1/1lhQhtRK2ts/ZwZEhjcQeJQ1RuC6Q/8U= |
|||
github.com/spf13/afero v1.15.0 h1:b/YBCLWAJdFWJTN9cLhiXXcD7mzKn9Dm86dNnfyQw1I= |
|||
github.com/spf13/afero v1.15.0/go.mod h1:NC2ByUVxtQs4b3sIUphxK0NioZnmxgyCrfzeuq8lxMg= |
|||
github.com/spf13/cast v1.10.0 h1:h2x0u2shc1QuLHfxi+cTJvs30+ZAHOGRic8uyGTDWxY= |
|||
github.com/spf13/cast v1.10.0/go.mod h1:jNfB8QC9IA6ZuY2ZjDp0KtFO2LZZlg4S/7bzP6qqeHo= |
|||
github.com/spf13/pflag v1.0.10 h1:4EBh2KAYBwaONj6b2Ye1GiHfwjqyROoF4RwYO+vPwFk= |
|||
github.com/spf13/pflag v1.0.10/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= |
|||
github.com/spf13/viper v1.21.0 h1:x5S+0EU27Lbphp4UKm1C+1oQO+rKx36vfCoaVebLFSU= |
|||
github.com/spf13/viper v1.21.0/go.mod h1:P0lhsswPGWD/1lZJ9ny3fYnVqxiegrlNrEmgLjbTCAY= |
|||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= |
|||
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= |
|||
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= |
|||
github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA= |
|||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= |
|||
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4= |
|||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= |
|||
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= |
|||
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= |
|||
github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= |
|||
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= |
|||
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= |
|||
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= |
|||
github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U= |
|||
github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U= |
|||
github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8= |
|||
github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU= |
|||
github.com/taosdata/driver-go/v3 v3.3.1 h1:UXIoE+TbdbdRHtaNjIuMAQn+SjAAKUJHyeONAt5F1eI= |
|||
github.com/taosdata/driver-go/v3 v3.3.1/go.mod h1:H2vo/At+rOPY1aMzUV9P49SVX7NlXb3LAbKw+MCLrmU= |
|||
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= |
|||
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= |
|||
github.com/ugorji/go/codec v1.3.1 h1:waO7eEiFDwidsBN6agj1vJQ4AG7lh2yqXyOXqhgQuyY= |
|||
github.com/ugorji/go/codec v1.3.1/go.mod h1:pRBVtBSKl77K30Bv8R2P+cLSGaTtex6fsA2Wjqmfxj4= |
|||
github.com/wlynxg/anet v0.0.3/go.mod h1:eay5PRQr7fIVAMbTbchTnO9gG65Hg/uYGdc7mguHxoA= |
|||
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY= |
|||
go.bug.st/serial v1.7.1 h1:5aP8wYL0UjEYOVs3oPAGscjaSfRQLHtCvBFXNN/rwtc= |
|||
go.bug.st/serial v1.7.1/go.mod h1:d0MmS16Qt9b1m06yoYRNUXhRRTJV5Qg2S5EKqQtnayQ= |
|||
go.mongodb.org/mongo-driver/v2 v2.5.0 h1:yXUhImUjjAInNcpTcAlPHiT7bIXhshCTL3jVBkF3xaE= |
|||
go.mongodb.org/mongo-driver/v2 v2.5.0/go.mod h1:yOI9kBsufol30iFsl1slpdq1I0eHPzybRWdyYUs8K/0= |
|||
go.uber.org/mock v0.6.0 h1:hyF9dfmbgIX5EfOdasqLsWD6xqpNZlXblLB/Dbnwv3Y= |
|||
go.uber.org/mock v0.6.0/go.mod h1:KiVJ4BqZJaMj4svdfmHM0AUx4NJYO8ZNpPnZn1Z+BBU= |
|||
go.yaml.in/yaml/v3 v3.0.4 h1:tfq32ie2Jv2UxXFdLJdh3jXuOzWiL1fo0bu/FbuKpbc= |
|||
go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg= |
|||
golang.org/x/arch v0.22.0 h1:c/Zle32i5ttqRXjdLyyHZESLD/bB90DCU1g9l/0YBDI= |
|||
golang.org/x/arch v0.22.0/go.mod h1:dNHoOeKiyja7GTvF9NJS1l3Z2yntpQNzgrjh1cU103A= |
|||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= |
|||
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc= |
|||
golang.org/x/crypto v0.6.0/go.mod h1:OFC/31mSvZgRz0V1QTNCzfAI1aIRzbiufJtkMIlEp58= |
|||
golang.org/x/crypto v0.11.0/go.mod h1:xgJhtzW8F9jGdVFWZESrid1U1bjeNy4zgy5cRr/CIio= |
|||
golang.org/x/crypto v0.12.0/go.mod h1:NF0Gs7EO5K4qLn+Ylc+fih8BSTeIjAP05siRnAh98yw= |
|||
golang.org/x/crypto v0.13.0/go.mod h1:y6Z2r+Rw4iayiXXAIxJIDAJ1zMW4yaTpebo8fPOliYc= |
|||
golang.org/x/crypto v0.18.0/go.mod h1:R0j02AL6hcrfOiy9T4ZYp/rcWeMxM3L6QYxlOuEG1mg= |
|||
golang.org/x/crypto v0.19.0/go.mod h1:Iy9bg/ha4yyC70EfRS8jz+B6ybOBKMaSxLj6P6oBDfU= |
|||
golang.org/x/crypto v0.21.0/go.mod h1:0BP7YvVV9gBbVKyeTG0Gyn+gZm94bibOW5BjDEYAOMs= |
|||
golang.org/x/crypto v0.22.0/go.mod h1:vr6Su+7cTlO45qkww3VDJlzDn0ctJvRgYbC2NvXHt+M= |
|||
golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8= |
|||
golang.org/x/crypto v0.24.0/go.mod h1:Z1PMYSOR5nyMcyAVAIQSKCDwalqy85Aqn1x3Ws4L5DM= |
|||
golang.org/x/crypto v0.52.0 h1:RMs7fP2rXdep0CftQlK8Uf+kibLm7qkCcradZWYz988= |
|||
golang.org/x/crypto v0.52.0/go.mod h1:1QgfPxDqh0T2M/elOJtp9RvuR95kVjir0e6/BvEmGbc= |
|||
golang.org/x/exp v0.0.0-20251219203646-944ab1f22d93 h1:fQsdNF2N+/YewlRZiricy4P1iimyPKZ/xwniHj8Q2a0= |
|||
golang.org/x/exp v0.0.0-20251219203646-944ab1f22d93/go.mod h1:EPRbTFwzwjXj9NpYyyrvenVh9Y+GFeEvMNh7Xuz7xgU= |
|||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4= |
|||
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs= |
|||
golang.org/x/mod v0.9.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs= |
|||
golang.org/x/mod v0.12.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs= |
|||
golang.org/x/mod v0.15.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c= |
|||
golang.org/x/mod v0.17.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c= |
|||
golang.org/x/mod v0.35.0 h1:Ww1D637e6Pg+Zb2KrWfHQUnH2dQRLBQyAtpr/haaJeM= |
|||
golang.org/x/mod v0.35.0/go.mod h1:+GwiRhIInF8wPm+4AoT6L0FA1QWAad3OMdTRx4tFYlU= |
|||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= |
|||
golang.org/x/net v0.0.0-20200114155413-6afb5195e5aa/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= |
|||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= |
|||
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c= |
|||
golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= |
|||
golang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= |
|||
golang.org/x/net v0.8.0/go.mod h1:QVkue5JL9kW//ek3r6jTKnTFis1tRmNAW2P1shuFdJc= |
|||
golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg= |
|||
golang.org/x/net v0.13.0/go.mod h1:zEVYFnQC7m/vmpQFELhcD1EWkZlX69l4oqgmer6hfKA= |
|||
golang.org/x/net v0.14.0/go.mod h1:PpSgVXXLK0OxS0F31C1/tv6XNguvCrnXIDrFMspZIUI= |
|||
golang.org/x/net v0.15.0/go.mod h1:idbUs1IY1+zTqbi8yxTbhexhEEk5ur9LInksu6HrEpk= |
|||
golang.org/x/net v0.20.0/go.mod h1:z8BVo6PvndSri0LbOE3hAn0apkU+1YvI6E70E9jsnvY= |
|||
golang.org/x/net v0.21.0/go.mod h1:bIjVDfnllIU7BJ2DNgfnXvpSvtn8VRwhlsaeUTyUS44= |
|||
golang.org/x/net v0.22.0/go.mod h1:JKghWKKOSdJwpW2GEx0Ja7fmaKnMsbu+MWVZTokSYmg= |
|||
golang.org/x/net v0.24.0/go.mod h1:2Q7sJY5mzlzWjKtYUEXSlBWCdyaioyXzRB2RtU8KVE8= |
|||
golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM= |
|||
golang.org/x/net v0.26.0/go.mod h1:5YKkiSynbBIh3p6iOc/vibscux0x38BZDkn8sCUPxHE= |
|||
golang.org/x/net v0.54.0 h1:2zJIZAxAHV/OHCDTCOHAYehQzLfSXuf/5SoL/Dv6w/w= |
|||
golang.org/x/net v0.54.0/go.mod h1:Sj4oj8jK6XmHpBZU/zWHw3BV3abl4Kvi+Ut7cQcY+cQ= |
|||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= |
|||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= |
|||
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= |
|||
golang.org/x/sync v0.3.0/go.mod h1:FU7BRWz2tNW+3quACPkgCx/L+uEAv1htQ0V83Z9Rj+Y= |
|||
golang.org/x/sync v0.6.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= |
|||
golang.org/x/sync v0.7.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= |
|||
golang.org/x/sync v0.9.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= |
|||
golang.org/x/sync v0.20.0 h1:e0PTpb7pjO8GAtTs2dQ6jYa5BWYlMuX047Dco/pItO4= |
|||
golang.org/x/sync v0.20.0/go.mod h1:9xrNwdLfx4jkKbNva9FpL6vEN7evnE43NNNJQ2LF3+0= |
|||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= |
|||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= |
|||
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.0.0-20210616045830-e2b7044e8c71/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.10.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= |
|||
golang.org/x/sys v0.16.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= |
|||
golang.org/x/sys v0.17.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= |
|||
golang.org/x/sys v0.18.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= |
|||
golang.org/x/sys v0.19.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= |
|||
golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= |
|||
golang.org/x/sys v0.21.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= |
|||
golang.org/x/sys v0.45.0 h1:dO4czNzziLiiXplLQgBCEpCvXQ3dnkn0SdaZSYdQ+FY= |
|||
golang.org/x/sys v0.45.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw= |
|||
golang.org/x/telemetry v0.0.0-20240228155512-f48c80bd79b2/go.mod h1:TeRTkGYfJXctD9OcfyVLyj2J3IxLnKwHJR8f4D8a3YE= |
|||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= |
|||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= |
|||
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k= |
|||
golang.org/x/term v0.6.0/go.mod h1:m6U89DPEgQRMq3DNkDClhWw02AUbt2daBVO4cn4Hv9U= |
|||
golang.org/x/term v0.8.0/go.mod h1:xPskH00ivmX89bAKVGSKKtLOWNx2+17Eiy94tnKShWo= |
|||
golang.org/x/term v0.10.0/go.mod h1:lpqdcUyK/oCiQxvxVrppt5ggO2KCZ5QblwqPnfZ6d5o= |
|||
golang.org/x/term v0.11.0/go.mod h1:zC9APTIj3jG3FdV/Ons+XE1riIZXG4aZ4GTHiPZJPIU= |
|||
golang.org/x/term v0.12.0/go.mod h1:owVbMEjm3cBLCHdkQu9b1opXd4ETQWc3BhuQGKgXgvU= |
|||
golang.org/x/term v0.16.0/go.mod h1:yn7UURbUtPyrVJPGPq404EukNFxcm/foM+bV/bfcDsY= |
|||
golang.org/x/term v0.17.0/go.mod h1:lLRBjIVuehSbZlaOtGMbcMncT+aqLLLmKrsjNrUguwk= |
|||
golang.org/x/term v0.18.0/go.mod h1:ILwASektA3OnRv7amZ1xhE/KTR+u50pbXfZ03+6Nx58= |
|||
golang.org/x/term v0.19.0/go.mod h1:2CuTdWZ7KHSQwUzKva0cbMg6q2DMI3Mmxp+gKJbskEk= |
|||
golang.org/x/term v0.20.0/go.mod h1:8UkIAJTvZgivsXaD6/pH6U9ecQzZ45awqEOzuCvwpFY= |
|||
golang.org/x/term v0.21.0/go.mod h1:ooXLefLobQVslOqselCNF4SxFAaoS6KujMbsGzSDmX0= |
|||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= |
|||
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= |
|||
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= |
|||
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= |
|||
golang.org/x/text v0.8.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8= |
|||
golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8= |
|||
golang.org/x/text v0.11.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= |
|||
golang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= |
|||
golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= |
|||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU= |
|||
golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU= |
|||
golang.org/x/text v0.16.0/go.mod h1:GhwF1Be+LQoKShO3cGOHzqOgRrGaYc9AvblQOmPVHnI= |
|||
golang.org/x/text v0.20.0/go.mod h1:D4IsuqiFMhST5bX19pQ9ikHC2GsaKyk/oF+pn3ducp4= |
|||
golang.org/x/text v0.37.0 h1:Cqjiwd9eSg8e0QAkyCaQTNHFIIzWtidPahFWR83rTrc= |
|||
golang.org/x/text v0.37.0/go.mod h1:a5sjxXGs9hsn/AJVwuElvCAo9v8QYLzvavO5z2PiM38= |
|||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= |
|||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= |
|||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc= |
|||
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU= |
|||
golang.org/x/tools v0.13.0/go.mod h1:HvlwmtVNQAhOuCjW7xxvovg8wbNq7LwfXh/k7wXUl58= |
|||
golang.org/x/tools v0.21.1-0.20240508182429-e35e4ccd0d2d/go.mod h1:aiJjzUbINMkxbQROHiO6hDPo2LHcIPhhQsa9DLh0yGk= |
|||
golang.org/x/tools v0.44.0 h1:UP4ajHPIcuMjT1GqzDWRlalUEoY+uzoZKnhOjbIPD2c= |
|||
golang.org/x/tools v0.44.0/go.mod h1:KA0AfVErSdxRZIsOVipbv3rQhVXTnlU6UhKxHd1seDI= |
|||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= |
|||
google.golang.org/protobuf v1.36.10 h1:AYd7cD/uASjIL6Q9LiTjz8JLcrh/88q5UObnmY3aOOE= |
|||
google.golang.org/protobuf v1.36.10/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco= |
|||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= |
|||
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= |
|||
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= |
|||
gopkg.in/natefinch/lumberjack.v2 v2.2.1 h1:bBRl1b0OH9s/DuPhuXpNl+VtCaJXFZ5/uEFST95x9zc= |
|||
gopkg.in/natefinch/lumberjack.v2 v2.2.1/go.mod h1:YD8tP3GAjkrDg1eZH7EGmyESg/lsYskCTPBJVb9jqSc= |
|||
gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= |
|||
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= |
|||
gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= |
|||
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ= |
|||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= |
|||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= |
|||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= |
|||
gorm.io/driver/mysql v1.6.0 h1:eNbLmNTpPpTOVZi8MMxCi2aaIm0ZpInbORNXDwyLGvg= |
|||
gorm.io/driver/mysql v1.6.0/go.mod h1:D/oCC2GWK3M/dqoLxnOlaNKmXz8WNTfcS9y5ovaSqKo= |
|||
gorm.io/driver/postgres v1.6.0 h1:2dxzU8xJ+ivvqTRph34QX+WrRaJlmfyPqXmoGVjMBa4= |
|||
gorm.io/driver/postgres v1.6.0/go.mod h1:vUw0mrGgrTK+uPHEhAdV4sfFELrByKVGnaVRkXDhtWo= |
|||
gorm.io/driver/sqlite v1.6.0 h1:WHRRrIiulaPiPFmDcod6prc4l2VGVWHz80KspNsxSfQ= |
|||
gorm.io/driver/sqlite v1.6.0/go.mod h1:AO9V1qIQddBESngQUKWL9yoH93HIeA1X6V633rBwyT8= |
|||
gorm.io/driver/sqlserver v1.6.3 h1:UR+nWCuphPnq7UxnL57PSrlYjuvs+sf1N59GgFX7uAI= |
|||
gorm.io/driver/sqlserver v1.6.3/go.mod h1:VZeNn7hqX1aXoN5TPAFGWvxWG90xtA8erGn2gQmpc6U= |
|||
gorm.io/gorm v1.30.0/go.mod h1:8Z33v652h4//uMA76KjeDH8mJXPm1QNCYrMeatR0DOE= |
|||
gorm.io/gorm v1.31.2 h1:3o8FXNo9v9S858gil+3LlZA1LkCOzgb4g5BL64FgaCo= |
|||
gorm.io/gorm v1.31.2/go.mod h1:XyQVbO2k6YkOis7C2437jSit3SsDK72s7n7rsSHd+Gs= |
|||
gorm.io/plugin/dbresolver v1.6.2 h1:F4b85TenghUeITqe3+epPSUtHH7RIk3fXr5l83DF8Pc= |
|||
gorm.io/plugin/dbresolver v1.6.2/go.mod h1:tctw63jdrOezFR9HmrKnPkmig3m5Edem9fdxk9bQSzM= |
|||
modernc.org/cc/v4 v4.27.1 h1:9W30zRlYrefrDV2JE2O8VDtJ1yPGownxciz5rrbQZis= |
|||
modernc.org/cc/v4 v4.27.1/go.mod h1:uVtb5OGqUKpoLWhqwNQo/8LwvoiEBLvZXIQ/SmO6mL0= |
|||
modernc.org/ccgo/v4 v4.30.1 h1:4r4U1J6Fhj98NKfSjnPUN7Ze2c6MnAdL0hWw6+LrJpc= |
|||
modernc.org/ccgo/v4 v4.30.1/go.mod h1:bIOeI1JL54Utlxn+LwrFyjCx2n2RDiYEaJVSrgdrRfM= |
|||
modernc.org/fileutil v1.3.40 h1:ZGMswMNc9JOCrcrakF1HrvmergNLAmxOPjizirpfqBA= |
|||
modernc.org/fileutil v1.3.40/go.mod h1:HxmghZSZVAz/LXcMNwZPA/DRrQZEVP9VX0V4LQGQFOc= |
|||
modernc.org/gc/v2 v2.6.5 h1:nyqdV8q46KvTpZlsw66kWqwXRHdjIlJOhG6kxiV/9xI= |
|||
modernc.org/gc/v2 v2.6.5/go.mod h1:YgIahr1ypgfe7chRuJi2gD7DBQiKSLMPgBQe9oIiito= |
|||
modernc.org/gc/v3 v3.1.1 h1:k8T3gkXWY9sEiytKhcgyiZ2L0DTyCQ/nvX+LoCljoRE= |
|||
modernc.org/gc/v3 v3.1.1/go.mod h1:HFK/6AGESC7Ex+EZJhJ2Gni6cTaYpSMmU/cT9RmlfYY= |
|||
modernc.org/goabi0 v0.2.0 h1:HvEowk7LxcPd0eq6mVOAEMai46V+i7Jrj13t4AzuNks= |
|||
modernc.org/goabi0 v0.2.0/go.mod h1:CEFRnnJhKvWT1c1JTI3Avm+tgOWbkOu5oPA8eH8LnMI= |
|||
modernc.org/libc v1.67.4 h1:zZGmCMUVPORtKv95c2ReQN5VDjvkoRm9GWPTEPuvlWg= |
|||
modernc.org/libc v1.67.4/go.mod h1:QvvnnJ5P7aitu0ReNpVIEyesuhmDLQ8kaEoyMjIFZJA= |
|||
modernc.org/mathutil v1.7.1 h1:GCZVGXdaN8gTqB1Mf/usp1Y/hSqgI2vAGGP4jZMCxOU= |
|||
modernc.org/mathutil v1.7.1/go.mod h1:4p5IwJITfppl0G4sUEDtCr4DthTaT47/N3aT6MhfgJg= |
|||
modernc.org/memory v1.11.0 h1:o4QC8aMQzmcwCK3t3Ux/ZHmwFPzE6hf2Y5LbkRs+hbI= |
|||
modernc.org/memory v1.11.0/go.mod h1:/JP4VbVC+K5sU2wZi9bHoq2MAkCnrt2r98UGeSK7Mjw= |
|||
modernc.org/opt v0.1.4 h1:2kNGMRiUjrp4LcaPuLY2PzUfqM/w9N23quVwhKt5Qm8= |
|||
modernc.org/opt v0.1.4/go.mod h1:03fq9lsNfvkYSfxrfUhZCWPk1lm4cq4N+Bh//bEtgns= |
|||
modernc.org/sortutil v1.2.1 h1:+xyoGf15mM3NMlPDnFqrteY07klSFxLElE2PVuWIJ7w= |
|||
modernc.org/sortutil v1.2.1/go.mod h1:7ZI3a3REbai7gzCLcotuw9AC4VZVpYMjDzETGsSMqJE= |
|||
modernc.org/sqlite v1.42.2 h1:7hkZUNJvJFN2PgfUdjni9Kbvd4ef4mNLOu0B9FGxM74= |
|||
modernc.org/sqlite v1.42.2/go.mod h1:+VkC6v3pLOAE0A0uVucQEcbVW0I5nHCeDaBf+DpsQT8= |
|||
modernc.org/strutil v1.2.1 h1:UneZBkQA+DX2Rp35KcM69cSsNES9ly8mQWD71HKlOA0= |
|||
modernc.org/strutil v1.2.1/go.mod h1:EHkiggD70koQxjVdSBM3JKM7k6L0FbGE5eymy9i3B9A= |
|||
modernc.org/token v1.1.0 h1:Xl7Ap9dKaEs5kLoOQeQmPWevfnk/DM5qcLcYlA8ys6Y= |
|||
modernc.org/token v1.1.0/go.mod h1:UGzOrNV1mAFSEB63lOFHIpNRUVMvYTc6yu1SMY/XTDM= |
|||
@ -0,0 +1,166 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// GetProfile 查询个人资料
|
|||
func GetProfile(c *gin.Context) { |
|||
user, busiErr := service.DefaultAccountService.GetProfile(common.GetUserId(c)) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, user) |
|||
} |
|||
|
|||
// UpdateProfile 更新个人资料
|
|||
func UpdateProfile(c *gin.Context) { |
|||
var req vo.ProfileUpdateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
user, busiErr := service.DefaultAccountService.UpdateProfile(common.GetUserId(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, user) |
|||
} |
|||
|
|||
// GetTrafficBalance 查询流量余额
|
|||
func GetTrafficBalance(c *gin.Context) { |
|||
bal, busiErr := service.DefaultAccountService.GetTrafficBalance(common.GetUserId(c)) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, bal) |
|||
} |
|||
|
|||
// GetUsagePage 流量消费流水分页
|
|||
func GetUsagePage(c *gin.Context) { |
|||
var req vo.UsagePageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultAccountService.GetUsagePage(common.GetUserId(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// GetOrderPage 流量订单分页
|
|||
func GetOrderPage(c *gin.Context) { |
|||
var req vo.OrderPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultAccountService.GetOrderPage(common.GetUserId(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// CreateTrafficOrder 创建流量充值订单
|
|||
func CreateTrafficOrder(c *gin.Context) { |
|||
var req vo.TrafficOrderCreateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
order, busiErr := service.DefaultAccountService.CreateTrafficOrder(common.GetUserId(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, order) |
|||
} |
|||
|
|||
// PayTrafficOrder 订单支付入账（admin）
|
|||
func PayTrafficOrder(c *gin.Context) { |
|||
orderID, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
order, busiErr := service.DefaultAccountService.PayTrafficOrder(orderID) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, order) |
|||
} |
|||
|
|||
// ListSimCards SIM 卡列表
|
|||
func ListSimCards(c *gin.Context) { |
|||
var req vo.SimCardPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultAccountService.ListSimCards(common.GetUserId(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// GetSimRechargeLogPage SIM 卡充值记录分页
|
|||
func GetSimRechargeLogPage(c *gin.Context) { |
|||
var req vo.SimRechargeLogPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultAccountService.GetSimRechargeLogPage(common.GetUserId(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// RechargeSimCard SIM 卡充值
|
|||
func RechargeSimCard(c *gin.Context) { |
|||
simCardID, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
var req vo.SimRechargeReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
log, busiErr := service.DefaultAccountService.RechargeSimCard(common.GetUserId(c), simCardID, &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, log) |
|||
} |
|||
@ -0,0 +1,55 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// GetAlarmPage 告警分页列表
|
|||
func GetAlarmPage(c *gin.Context) { |
|||
var req vo.AlarmPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultAlarmService.GetPage(common.GetUserId(c), common.IsAdmin(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// AcknowledgeAlarm 确认告警
|
|||
func AcknowledgeAlarm(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultAlarmService.Acknowledge(common.GetUserId(c), common.IsAdmin(c), id); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
|
|||
// ResolveAlarm 关闭告警
|
|||
func ResolveAlarm(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultAlarmService.Resolve(common.GetUserId(c), common.IsAdmin(c), id); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
@ -0,0 +1,98 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"strings" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// Register 用户注册
|
|||
func Register(c *gin.Context) { |
|||
var req vo.RegisterReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
resp, busiErr := service.DefaultUserService.Register(req.Name, req.Phone, req.Email, req.Password) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, resp) |
|||
} |
|||
|
|||
// Login 用户登录
|
|||
func Login(c *gin.Context) { |
|||
var req vo.UserLoginReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
resp, busiErr := service.DefaultUserService.Login(req.Phone, req.Password) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, resp) |
|||
} |
|||
|
|||
// Refresh 刷新令牌
|
|||
func Refresh(c *gin.Context) { |
|||
var req vo.RefreshReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
resp, busiErr := service.DefaultUserService.Refresh(req.RefreshToken) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, resp) |
|||
} |
|||
|
|||
// Logout 登出
|
|||
func Logout(c *gin.Context) { |
|||
var req vo.LogoutReq |
|||
_ = c.ShouldBindJSON(&req) // refreshToken 可选，忽略绑定错误
|
|||
|
|||
accessToken := strings.TrimPrefix(c.GetHeader("Authorization"), "Bearer ") |
|||
userID := common.GetUserId(c) |
|||
if busiErr := service.DefaultUserService.Logout(userID, accessToken, req.RefreshToken); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
|
|||
// SmsCode 发送短信验证码
|
|||
func SmsCode(c *gin.Context) { |
|||
var req vo.SmsCodeReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
if _, busiErr := service.DefaultUserService.SendSmsCode(req.Phone); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
|
|||
// ResetPassword 重置密码
|
|||
func ResetPassword(c *gin.Context) { |
|||
var req vo.ResetPasswordReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultUserService.ResetPassword(req.Phone, req.SmsCode, req.NewPassword); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
@ -0,0 +1,62 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// DispatchCommand 下发设备指令
|
|||
func DispatchCommand(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
var req vo.CommandReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
cmd, busiErr := service.DefaultCommandService.Dispatch(common.GetUserId(c), common.IsAdmin(c), id, &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, cmd) |
|||
} |
|||
|
|||
// GetCommandPage 指令历史分页列表
|
|||
func GetCommandPage(c *gin.Context) { |
|||
var req vo.CommandPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultCommandService.GetPage(common.GetUserId(c), common.IsAdmin(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// GetCommand 指令详情
|
|||
func GetCommand(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
cmd, busiErr := service.DefaultCommandService.GetDetail(common.GetUserId(c), common.IsAdmin(c), id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, cmd) |
|||
} |
|||
@ -0,0 +1,106 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// GetDockPage 机巢分页列表
|
|||
func GetDockPage(c *gin.Context) { |
|||
var req vo.DockPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultDockService.GetPage(common.GetUserId(c), common.IsAdmin(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// GetDock 机巢详情
|
|||
func GetDock(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
dock, busiErr := service.DefaultDockService.GetDetail(common.GetUserId(c), common.IsAdmin(c), id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, dock) |
|||
} |
|||
|
|||
// GetDockStatus 机巢实时状态
|
|||
func GetDockStatus(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
status, busiErr := service.DefaultDockService.GetStatus(common.GetUserId(c), common.IsAdmin(c), id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, status) |
|||
} |
|||
|
|||
// CreateDock 手动登记机巢
|
|||
func CreateDock(c *gin.Context) { |
|||
var req vo.DockCreateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
dock, busiErr := service.DefaultDockService.Create(common.GetUserId(c), common.IsAdmin(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, dock) |
|||
} |
|||
|
|||
// UpdateDock 更新/认领机巢
|
|||
func UpdateDock(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
var req vo.DockUpdateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
dock, busiErr := service.DefaultDockService.Update(common.GetUserId(c), common.IsAdmin(c), id, &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, dock) |
|||
} |
|||
|
|||
// DeleteDock 删除机巢
|
|||
func DeleteDock(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultDockService.Delete(common.GetUserId(c), common.IsAdmin(c), id); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
@ -0,0 +1,91 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// GetDronePage 无人机分页列表
|
|||
func GetDronePage(c *gin.Context) { |
|||
var req vo.DronePageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultDroneService.GetPage(common.GetUserId(c), common.IsAdmin(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// GetDrone 无人机详情
|
|||
func GetDrone(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
drone, busiErr := service.DefaultDroneService.GetDetail(common.GetUserId(c), common.IsAdmin(c), id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, drone) |
|||
} |
|||
|
|||
// GetDroneTelemetry 无人机实时遥测
|
|||
func GetDroneTelemetry(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
telemetry, busiErr := service.DefaultDroneService.GetTelemetry(common.GetUserId(c), common.IsAdmin(c), id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, telemetry) |
|||
} |
|||
|
|||
// UpdateDrone 更新无人机
|
|||
func UpdateDrone(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
var req vo.DroneUpdateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
drone, busiErr := service.DefaultDroneService.Update(common.GetUserId(c), common.IsAdmin(c), id, &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, drone) |
|||
} |
|||
|
|||
// DeleteDrone 删除无人机
|
|||
func DeleteDrone(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultDroneService.Delete(common.GetUserId(c), common.IsAdmin(c), id); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
@ -0,0 +1,57 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// GetExecutionPage 执行记录分页列表
|
|||
func GetExecutionPage(c *gin.Context) { |
|||
var req vo.ExecutionPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultExecutionService.GetPage(common.GetUserId(c), common.IsAdmin(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// GetExecution 执行记录详情
|
|||
func GetExecution(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
exec, busiErr := service.DefaultExecutionService.GetDetail(common.GetUserId(c), common.IsAdmin(c), id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, exec) |
|||
} |
|||
|
|||
// GetTrajectory 执行轨迹
|
|||
func GetTrajectory(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
traj, busiErr := service.DefaultExecutionService.GetTrajectory(common.GetUserId(c), common.IsAdmin(c), id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, traj) |
|||
} |
|||
@ -0,0 +1,120 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// GetFirmwarePage 固件分页列表（admin）
|
|||
func GetFirmwarePage(c *gin.Context) { |
|||
var req vo.FirmwarePageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultFirmwareService.GetPage(&req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// GetReleasedFirmwares 已发布固件列表（普通用户选择升级）
|
|||
func GetReleasedFirmwares(c *gin.Context) { |
|||
list, busiErr := service.DefaultFirmwareService.GetReleasedList() |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, list) |
|||
} |
|||
|
|||
// GetFirmware 固件详情（admin）
|
|||
func GetFirmware(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
fw, busiErr := service.DefaultFirmwareService.GetDetail(id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, fw) |
|||
} |
|||
|
|||
// CreateFirmware 新增固件版本（admin）
|
|||
func CreateFirmware(c *gin.Context) { |
|||
var req vo.FirmwareCreateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
fw, busiErr := service.DefaultFirmwareService.Create(common.GetUserId(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, fw) |
|||
} |
|||
|
|||
// UpdateFirmware 编辑固件版本（admin）
|
|||
func UpdateFirmware(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
var req vo.FirmwareUpdateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultFirmwareService.Update(id, &req); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
|
|||
// DeleteFirmware 删除固件版本（admin）
|
|||
func DeleteFirmware(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultFirmwareService.Delete(id); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
|
|||
// UpgradeFirmware 下发 OTA 升级（对所属机巢）
|
|||
func UpgradeFirmware(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
var req vo.FirmwareUpgradeReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
result, busiErr := service.DefaultFirmwareService.Upgrade(common.GetUserId(c), common.IsAdmin(c), id, req.FirmwareID) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, result) |
|||
} |
|||
@ -0,0 +1,133 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// GetLivePage 直播会话分页列表
|
|||
func GetLivePage(c *gin.Context) { |
|||
var req vo.LivePageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultLiveService.GetPage(common.GetUserId(c), common.IsAdmin(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// CreateLiveSession 创建或加入机巢的唯一活动直播会话，并创建观看租约。
|
|||
func CreateLiveSession(c *gin.Context) { |
|||
dockID := c.Param("dockId") |
|||
if dockID == "" { |
|||
common.FailWithBusiError(c, common.ErrParam) |
|||
return |
|||
} |
|||
var req vo.LiveSessionCreateReq |
|||
_ = c.ShouldBindJSON(&req) |
|||
result, busiErr := service.DefaultLiveService.Join(common.GetUserId(c), common.IsAdmin(c), dockID, &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, result) |
|||
} |
|||
|
|||
// GetLiveSession 获取指定直播会话。
|
|||
func GetLiveSession(c *gin.Context) { |
|||
dockID, sessionID := c.Param("dockId"), c.Param("streamSessionId") |
|||
if dockID == "" || sessionID == "" { |
|||
common.FailWithBusiError(c, common.ErrParam) |
|||
return |
|||
} |
|||
session, busiErr := service.DefaultLiveService.GetSession(common.GetUserId(c), common.IsAdmin(c), dockID, sessionID) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, session) |
|||
} |
|||
|
|||
// HeartbeatLiveSession 续期当前用户的观看租约。
|
|||
func HeartbeatLiveSession(c *gin.Context) { |
|||
dockID, sessionID := c.Param("dockId"), c.Param("streamSessionId") |
|||
if dockID == "" || sessionID == "" { |
|||
common.FailWithBusiError(c, common.ErrParam) |
|||
return |
|||
} |
|||
result, busiErr := service.DefaultLiveService.Heartbeat(common.GetUserId(c), common.IsAdmin(c), dockID, sessionID) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, result) |
|||
} |
|||
|
|||
// LeaveLiveSession 释放当前用户的观看租约。
|
|||
func LeaveLiveSession(c *gin.Context) { |
|||
dockID, sessionID := c.Param("dockId"), c.Param("streamSessionId") |
|||
if dockID == "" || sessionID == "" { |
|||
common.FailWithBusiError(c, common.ErrParam) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultLiveService.Leave(common.GetUserId(c), common.IsAdmin(c), dockID, sessionID); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
|
|||
// StartLive 启动直播推流
|
|||
func StartLive(c *gin.Context) { |
|||
dockID := c.Param("dockId") |
|||
if dockID == "" { |
|||
common.FailWithBusiError(c, common.ErrParam) |
|||
return |
|||
} |
|||
var req vo.LiveStartReq |
|||
_ = c.ShouldBindJSON(&req) |
|||
session, busiErr := service.DefaultLiveService.Start(common.GetUserId(c), common.IsAdmin(c), dockID, &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, session) |
|||
} |
|||
|
|||
func StopLive(c *gin.Context) { |
|||
dockID := c.Param("dockId") |
|||
if dockID == "" { |
|||
common.FailWithBusiError(c, common.ErrParam) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultLiveService.Stop(common.GetUserId(c), common.IsAdmin(c), dockID); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
|
|||
// GetLivePlayURL 获取直播播放地址
|
|||
func GetLivePlayURL(c *gin.Context) { |
|||
dockID := c.Param("dockId") |
|||
if dockID == "" { |
|||
common.FailWithBusiError(c, common.ErrParam) |
|||
return |
|||
} |
|||
play, busiErr := service.DefaultLiveService.GetPlayURL(common.GetUserId(c), common.IsAdmin(c), dockID) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, play) |
|||
} |
|||
@ -0,0 +1,91 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// GetRoutePage 航线分页列表
|
|||
func GetRoutePage(c *gin.Context) { |
|||
var req vo.RoutePageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultRouteService.GetPage(common.GetUserId(c), common.IsAdmin(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// GetRoute 航线详情
|
|||
func GetRoute(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
route, busiErr := service.DefaultRouteService.GetDetail(common.GetUserId(c), common.IsAdmin(c), id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, route) |
|||
} |
|||
|
|||
// CreateRoute 新增航线
|
|||
func CreateRoute(c *gin.Context) { |
|||
var req vo.RouteCreateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
route, busiErr := service.DefaultRouteService.Create(common.GetUserId(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, route) |
|||
} |
|||
|
|||
// UpdateRoute 编辑航线
|
|||
func UpdateRoute(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
var req vo.RouteUpdateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
route, busiErr := service.DefaultRouteService.Update(common.GetUserId(c), common.IsAdmin(c), id, &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, route) |
|||
} |
|||
|
|||
// DeleteRoute 删除航线
|
|||
func DeleteRoute(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultRouteService.Delete(common.GetUserId(c), common.IsAdmin(c), id); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
@ -0,0 +1,102 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// GetUserPage 用户分页列表（admin）
|
|||
func GetUserPage(c *gin.Context) { |
|||
var req vo.UserPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultSystemService.GetUserPage(&req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// CreateUser 创建用户（admin）
|
|||
func CreateUser(c *gin.Context) { |
|||
var req vo.UserCreateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
user, busiErr := service.DefaultSystemService.CreateUser(&req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, user) |
|||
} |
|||
|
|||
// UpdateUser 编辑用户（admin）
|
|||
func UpdateUser(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
var req vo.UserUpdateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
user, busiErr := service.DefaultSystemService.UpdateUser(id, &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, user) |
|||
} |
|||
|
|||
// DeleteUser 删除用户（admin）
|
|||
func DeleteUser(c *gin.Context) { |
|||
id, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultSystemService.DeleteUser(id); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
|
|||
// GetRoles 角色列表（admin）
|
|||
func GetRoles(c *gin.Context) { |
|||
common.OKWithData(c, service.DefaultSystemService.GetRoles()) |
|||
} |
|||
|
|||
// GetOperationLogPage 操作日志分页（admin 全量，user 仅本人）
|
|||
func GetOperationLogPage(c *gin.Context) { |
|||
var req vo.OperationLogPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
if !common.IsAdmin(c) { |
|||
req.UserID = common.GetUserId(c) |
|||
} |
|||
page, busiErr := service.DefaultOperationLogService.GetPage(&req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
@ -0,0 +1,105 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// GetTaskPage 任务分页列表
|
|||
func GetTaskPage(c *gin.Context) { |
|||
var req vo.TaskPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultTaskService.GetPage(common.GetUserId(c), common.IsAdmin(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// GetTask 任务详情
|
|||
func GetTask(c *gin.Context) { |
|||
id, busiErr := parseStringID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
task, busiErr := service.DefaultTaskService.GetDetail(common.GetUserId(c), common.IsAdmin(c), id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, task) |
|||
} |
|||
|
|||
// CreateTask 新增任务
|
|||
func CreateTask(c *gin.Context) { |
|||
var req vo.TaskCreateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
task, busiErr := service.DefaultTaskService.Create(common.GetUserId(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, task) |
|||
} |
|||
|
|||
// UpdateTask 编辑任务
|
|||
func UpdateTask(c *gin.Context) { |
|||
id, busiErr := parseStringID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
var req vo.TaskUpdateReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultTaskService.Update(common.GetUserId(c), common.IsAdmin(c), id, &req); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
|
|||
// DeleteTask 删除任务
|
|||
func DeleteTask(c *gin.Context) { |
|||
id, busiErr := parseStringID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
if busiErr := service.DefaultTaskService.Delete(common.GetUserId(c), common.IsAdmin(c), id); busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OK(c) |
|||
} |
|||
|
|||
// ExecuteTask 立即执行任务
|
|||
func ExecuteTask(c *gin.Context) { |
|||
id, busiErr := parseStringID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
result, busiErr := service.DefaultTaskService.Execute(common.GetUserId(c), common.IsAdmin(c), id) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, result) |
|||
} |
|||
@ -0,0 +1,27 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"strconv" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
) |
|||
|
|||
// parseID 解析路径参数 :id 为 int64
|
|||
func parseID(c *gin.Context) (int64, *common.BusiError) { |
|||
id, err := strconv.ParseInt(c.Param("id"), 10, 64) |
|||
if err != nil || id <= 0 { |
|||
return 0, common.ErrParam |
|||
} |
|||
return id, nil |
|||
} |
|||
|
|||
// parseStringID 解析路径参数 :id 为字符串（如任务 ID）
|
|||
func parseStringID(c *gin.Context) (string, *common.BusiError) { |
|||
id := c.Param("id") |
|||
if id == "" { |
|||
return "", common.ErrParam |
|||
} |
|||
return id, nil |
|||
} |
|||
@ -0,0 +1,89 @@ |
|||
package handler |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/service" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// CreateVideoUpload 申请视频上传
|
|||
func CreateVideoUpload(c *gin.Context) { |
|||
var req vo.VideoUploadReq |
|||
if err := c.ShouldBindJSON(&req); err != nil { |
|||
common.FailWithBindError(c, common.ErrParam, err) |
|||
return |
|||
} |
|||
result, busiErr := service.DefaultVideoService.CreateUpload(common.GetUserId(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, result) |
|||
} |
|||
|
|||
// CompleteVideoUpload 上传完成确认
|
|||
func CompleteVideoUpload(c *gin.Context) { |
|||
videoID, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
var req vo.VideoCompleteReq |
|||
_ = c.ShouldBindJSON(&req) |
|||
video, busiErr := service.DefaultVideoService.Complete(common.GetUserId(c), videoID, &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, video) |
|||
} |
|||
|
|||
// GetVideoPage 视频分页列表
|
|||
func GetVideoPage(c *gin.Context) { |
|||
var req vo.VideoPageReq |
|||
_ = c.ShouldBindQuery(&req) |
|||
if req.PageNum <= 0 { |
|||
req.PageNum = 1 |
|||
} |
|||
if req.PageSize <= 0 { |
|||
req.PageSize = 10 |
|||
} |
|||
page, busiErr := service.DefaultVideoService.GetPage(common.GetUserId(c), common.IsAdmin(c), &req) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, page) |
|||
} |
|||
|
|||
// GetVideo 视频详情
|
|||
func GetVideo(c *gin.Context) { |
|||
videoID, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
video, busiErr := service.DefaultVideoService.GetDetail(common.GetUserId(c), common.IsAdmin(c), videoID) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, video) |
|||
} |
|||
|
|||
// DownloadVideo 下载视频
|
|||
func DownloadVideo(c *gin.Context) { |
|||
videoID, busiErr := parseID(c) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
result, busiErr := service.DefaultVideoService.Download(common.GetUserId(c), common.IsAdmin(c), videoID) |
|||
if busiErr != nil { |
|||
common.FailWithBusiError(c, busiErr) |
|||
return |
|||
} |
|||
common.OKWithData(c, result) |
|||
} |
|||
@ -0,0 +1,61 @@ |
|||
package live |
|||
|
|||
import ( |
|||
"errors" |
|||
"time" |
|||
) |
|||
|
|||
var ErrDisabled = errors.New("live provider disabled") |
|||
var ErrInvalidConfig = errors.New("live provider configuration invalid") |
|||
|
|||
type Config struct { |
|||
Mode string |
|||
Provider string |
|||
PushDomain string |
|||
PlayDomain string |
|||
AppName string |
|||
AuthKey string |
|||
PlayAuthKey string |
|||
AuthExpireSeconds int64 |
|||
PlayURLTTLSeconds int64 |
|||
Protocol string |
|||
SRTPort int |
|||
AllowRealCloud bool |
|||
} |
|||
|
|||
type StreamRequest struct { |
|||
StreamName string |
|||
ExpiresAt time.Time |
|||
MaxBitrateBps int64 |
|||
} |
|||
|
|||
type StreamCredentials struct { |
|||
Provider string |
|||
PushURL string |
|||
ExpiresAt time.Time |
|||
} |
|||
|
|||
type PlayRequest struct { |
|||
StreamName string |
|||
ExpiresAt time.Time |
|||
} |
|||
|
|||
type PlayURLs struct { |
|||
HLS string |
|||
FLV string |
|||
RTMP string |
|||
ExpiresAt time.Time |
|||
} |
|||
|
|||
type OnlineStatus struct { |
|||
Online bool |
|||
BitrateBps int64 |
|||
ClientCount int |
|||
} |
|||
|
|||
type Adapter interface { |
|||
Provider() string |
|||
CreateStream(req StreamRequest) (StreamCredentials, error) |
|||
CreatePlayURLs(req PlayRequest) (PlayURLs, error) |
|||
QueryOnline(streamName string) (OnlineStatus, error) |
|||
} |
|||
@ -0,0 +1,43 @@ |
|||
package live |
|||
|
|||
import ( |
|||
"strings" |
|||
|
|||
"laic-backend/common" |
|||
) |
|||
|
|||
func NewAdapter(conf common.Live) (Adapter, error) { |
|||
cfg := Config{ |
|||
Mode: strings.ToLower(conf.Mode), |
|||
Provider: strings.ToLower(conf.Provider), |
|||
PushDomain: conf.PushDomain, |
|||
PlayDomain: conf.PlayDomain, |
|||
AppName: conf.AppName, |
|||
AuthKey: conf.AuthKey, |
|||
PlayAuthKey: conf.PlayAuthKey, |
|||
AuthExpireSeconds: conf.AuthExpireSeconds, |
|||
PlayURLTTLSeconds: conf.PlayURLTTLSeconds, |
|||
Protocol: strings.ToLower(conf.Protocol), |
|||
SRTPort: conf.SRTPort, |
|||
AllowRealCloud: conf.AllowRealCloud, |
|||
} |
|||
switch cfg.Mode { |
|||
case "", "disabled": |
|||
return DisabledAdapter{}, nil |
|||
case "fake": |
|||
if cfg.PushDomain == "" { |
|||
cfg.PushDomain = "local-push" |
|||
} |
|||
if cfg.PlayDomain == "" { |
|||
cfg.PlayDomain = "local-play" |
|||
} |
|||
if cfg.AppName == "" { |
|||
cfg.AppName = "dock-live" |
|||
} |
|||
return FakeAdapter{PushDomain: cfg.PushDomain, PlayDomain: cfg.PlayDomain, AppName: cfg.AppName}, nil |
|||
case "aliyun": |
|||
return NewAliyunAdapter(cfg) |
|||
default: |
|||
return nil, ErrInvalidConfig |
|||
} |
|||
} |
|||
@ -0,0 +1,90 @@ |
|||
package live |
|||
|
|||
import ( |
|||
"crypto/md5" |
|||
"encoding/hex" |
|||
"errors" |
|||
"fmt" |
|||
"strconv" |
|||
"time" |
|||
) |
|||
|
|||
type DisabledAdapter struct{} |
|||
|
|||
func (DisabledAdapter) Provider() string { return "disabled" } |
|||
func (DisabledAdapter) CreateStream(StreamRequest) (StreamCredentials, error) { |
|||
return StreamCredentials{}, ErrDisabled |
|||
} |
|||
func (DisabledAdapter) CreatePlayURLs(PlayRequest) (PlayURLs, error) { return PlayURLs{}, ErrDisabled } |
|||
func (DisabledAdapter) QueryOnline(string) (OnlineStatus, error) { return OnlineStatus{}, ErrDisabled } |
|||
|
|||
// FakeAdapter is a local provider that never contacts a cloud service.
|
|||
type FakeAdapter struct { |
|||
PlayDomain string |
|||
PushDomain string |
|||
AppName string |
|||
} |
|||
|
|||
func (f FakeAdapter) Provider() string { return "fake" } |
|||
func (f FakeAdapter) CreateStream(req StreamRequest) (StreamCredentials, error) { |
|||
return StreamCredentials{ |
|||
Provider: f.Provider(), |
|||
PushURL: "fake://" + f.PushDomain + "/" + f.AppName + "/" + req.StreamName, |
|||
ExpiresAt: req.ExpiresAt, |
|||
}, nil |
|||
} |
|||
func (f FakeAdapter) CreatePlayURLs(req PlayRequest) (PlayURLs, error) { |
|||
return PlayURLs{ |
|||
HLS: "fake://" + f.PlayDomain + "/" + f.AppName + "/" + req.StreamName + ".m3u8", |
|||
ExpiresAt: req.ExpiresAt, |
|||
}, nil |
|||
} |
|||
func (f FakeAdapter) QueryOnline(string) (OnlineStatus, error) { |
|||
return OnlineStatus{Online: true}, nil |
|||
} |
|||
|
|||
type AliyunAdapter struct { |
|||
cfg Config |
|||
} |
|||
|
|||
func NewAliyunAdapter(cfg Config) (Adapter, error) { |
|||
if !cfg.AllowRealCloud || cfg.PushDomain == "" || cfg.PlayDomain == "" || cfg.AppName == "" || cfg.AuthKey == "" || cfg.PlayAuthKey == "" { |
|||
return nil, ErrInvalidConfig |
|||
} |
|||
if cfg.Protocol == "" { |
|||
cfg.Protocol = "srt" |
|||
} |
|||
if cfg.SRTPort <= 0 { |
|||
cfg.SRTPort = 1105 |
|||
} |
|||
return AliyunAdapter{cfg: cfg}, nil |
|||
} |
|||
|
|||
func (a AliyunAdapter) Provider() string { return "aliyun" } |
|||
func (a AliyunAdapter) CreateStream(req StreamRequest) (StreamCredentials, error) { |
|||
return StreamCredentials{Provider: a.Provider(), PushURL: a.pushURL(req.StreamName, req.ExpiresAt), ExpiresAt: req.ExpiresAt}, nil |
|||
} |
|||
func (a AliyunAdapter) CreatePlayURLs(req PlayRequest) (PlayURLs, error) { |
|||
expires := req.ExpiresAt |
|||
path := "/" + a.cfg.AppName + "/" + req.StreamName |
|||
signature := sign(path, expires.Unix(), a.cfg.PlayAuthKey) |
|||
base := "https://" + a.cfg.PlayDomain + "/" + a.cfg.AppName + "/" + req.StreamName |
|||
return PlayURLs{HLS: base + ".m3u8?auth_key=" + signature, FLV: base + ".flv?auth_key=" + signature, RTMP: "rtmp://" + a.cfg.PlayDomain + "/" + a.cfg.AppName + "/" + req.StreamName, ExpiresAt: expires}, nil |
|||
} |
|||
func (a AliyunAdapter) QueryOnline(string) (OnlineStatus, error) { |
|||
return OnlineStatus{}, errors.New("aliyun online query is not configured") |
|||
} |
|||
func sign(path string, expiresAt int64, key string) string { |
|||
rand := strconv.FormatInt(expiresAt%10000, 10) |
|||
sum := md5.Sum([]byte(fmt.Sprintf("%s-%d-%s-%s-%s", path, expiresAt, rand, "0", key))) |
|||
return fmt.Sprintf("%d-%s-0-%s", expiresAt, rand, hex.EncodeToString(sum[:])) |
|||
} |
|||
|
|||
func (a AliyunAdapter) pushURL(streamName string, expires time.Time) string { |
|||
path := "/" + a.cfg.AppName + "/" + streamName |
|||
auth := sign(path, expires.Unix(), a.cfg.AuthKey) |
|||
if a.cfg.Protocol == "srt" { |
|||
return "srt://" + a.cfg.PushDomain + ":" + strconv.Itoa(a.cfg.SRTPort) + "?streamid=#!::h=" + a.cfg.PushDomain + ",r=" + path + "?auth_key=" + auth + ",m=publish" |
|||
} |
|||
return "rtmps://" + a.cfg.PushDomain + "/" + a.cfg.AppName + "/" + streamName + "?auth_key=" + auth |
|||
} |
|||
@ -0,0 +1,80 @@ |
|||
package logger |
|||
|
|||
import ( |
|||
"context" |
|||
"errors" |
|||
"fmt" |
|||
"log" |
|||
"path/filepath" |
|||
"runtime" |
|||
"time" |
|||
|
|||
"gorm.io/gorm/logger" |
|||
) |
|||
|
|||
type GormLogger struct { |
|||
LogLevel logger.LogLevel |
|||
logger *log.Logger |
|||
} |
|||
|
|||
func NewGormLogger() logger.Interface { |
|||
return &GormLogger{ |
|||
LogLevel: logger.Info, |
|||
logger: log.New(DefaultWriter, "", log.LstdFlags|log.Lshortfile), |
|||
} |
|||
} |
|||
|
|||
func (gl *GormLogger) LogMode(level logger.LogLevel) logger.Interface { |
|||
newLogger := *gl |
|||
newLogger.LogLevel = level |
|||
return &newLogger |
|||
} |
|||
|
|||
func (gl *GormLogger) Info(ctx context.Context, msg string, data ...interface{}) { |
|||
if gl.LogLevel <= logger.Info { |
|||
DEBUG(fmt.Sprintf(msg, data...)) |
|||
} |
|||
} |
|||
|
|||
func (gl *GormLogger) Warn(ctx context.Context, msg string, data ...interface{}) { |
|||
if gl.LogLevel <= logger.Warn { |
|||
WARN(fmt.Sprintf(msg, data...)) |
|||
} |
|||
} |
|||
|
|||
func (gl *GormLogger) Error(ctx context.Context, msg string, data ...interface{}) { |
|||
if gl.LogLevel <= logger.Error { |
|||
ERROR(fmt.Sprintf(msg, data...), nil) |
|||
} |
|||
} |
|||
|
|||
func (gl *GormLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) { |
|||
if gl.LogLevel > logger.Silent { |
|||
elapsed := time.Since(begin) |
|||
sql, rows := fc() |
|||
|
|||
_, file, line, _ := runtime.Caller(3) |
|||
source := fmt.Sprintf("%s:%d", filepath.Base(file), line) |
|||
|
|||
logMsg := struct { |
|||
Source string |
|||
Duration string |
|||
Rows int64 |
|||
SQL string |
|||
}{ |
|||
Source: source, |
|||
Duration: fmt.Sprintf("%.2f:ms", float64(elapsed.Nanoseconds())/1e6), |
|||
Rows: rows, |
|||
SQL: sql, |
|||
} |
|||
|
|||
switch { |
|||
case err != nil && errors.Is(err, logger.ErrRecordNotFound): |
|||
ERROR(logMsg, err) |
|||
case err != nil: |
|||
ERROR(logMsg, err) |
|||
default: |
|||
DEBUG(logMsg) |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,318 @@ |
|||
package logger |
|||
|
|||
import ( |
|||
"bytes" |
|||
"fmt" |
|||
"io" |
|||
"log" |
|||
"os" |
|||
"path/filepath" |
|||
"runtime" |
|||
"strconv" |
|||
"strings" |
|||
"sync" |
|||
"time" |
|||
|
|||
"gopkg.in/natefinch/lumberjack.v2" |
|||
) |
|||
|
|||
const ( |
|||
Level_T Level = iota |
|||
Level_D |
|||
Level_I |
|||
Level_W |
|||
Level_E |
|||
) |
|||
|
|||
const ( |
|||
Reset = "\033[0m" |
|||
Red = "\033[31m" |
|||
Green = "\033[32m" |
|||
Yellow = "\033[33m" |
|||
Blue = "\033[34m" |
|||
Magenta = "\033[35m" |
|||
Cyan = "\033[36m" |
|||
White = "\033[37m" |
|||
) |
|||
|
|||
// LogConfig 日志配置
|
|||
type LogConfig struct { |
|||
LogPath string |
|||
MaxSize int |
|||
MaxBackups int |
|||
MaxAge int |
|||
Compress bool |
|||
} |
|||
|
|||
var defaultConfig = LogConfig{ |
|||
LogPath: "./log", |
|||
MaxSize: 50, |
|||
MaxBackups: 20, |
|||
MaxAge: 30, |
|||
Compress: true, |
|||
} |
|||
|
|||
type Level int |
|||
|
|||
// TraceContext 存储跟踪信息
|
|||
type TraceContext struct { |
|||
TraceID string |
|||
SpanID string |
|||
} |
|||
|
|||
// goroutineID 到 traceID 的映射
|
|||
var traceMap sync.Map |
|||
|
|||
// 获取当前 goroutine 的 ID
|
|||
func getGoroutineID() uint64 { |
|||
b := make([]byte, 64) |
|||
b = b[:runtime.Stack(b, false)] |
|||
b = bytes.TrimPrefix(b, []byte("goroutine ")) |
|||
b = b[:bytes.IndexByte(b, ' ')] |
|||
n, _ := strconv.ParseUint(string(b), 10, 64) |
|||
return n |
|||
} |
|||
|
|||
// SetTraceID 设置当前 goroutine 的 traceID
|
|||
func SetTraceID(traceID string) { |
|||
traceMap.Store(getGoroutineID(), traceID) |
|||
} |
|||
|
|||
// GetTraceID 获取当前 goroutine 的 traceID
|
|||
func GetTraceID() string { |
|||
if id, ok := traceMap.Load(getGoroutineID()); ok { |
|||
return id.(string) |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
// CleanTraceID 清理当前 goroutine 的 traceID
|
|||
func CleanTraceID() { |
|||
traceMap.Delete(getGoroutineID()) |
|||
} |
|||
|
|||
type customLogWriter struct { |
|||
prefix string |
|||
output io.Writer |
|||
} |
|||
|
|||
var DefaultWriter io.Writer |
|||
|
|||
var ( |
|||
loglevel Level |
|||
_trace *log.Logger |
|||
_debug *log.Logger |
|||
_info *log.Logger |
|||
_warn *log.Logger |
|||
_error *log.Logger |
|||
) |
|||
|
|||
func SetLevel(level Level) { |
|||
loglevel = level |
|||
} |
|||
|
|||
func GetLevel() string { |
|||
return loglevel.String() |
|||
} |
|||
|
|||
func formatLogWithTrace(data ...any) string { |
|||
logData := formatLogData(data...) |
|||
return fmt.Sprintf("===> %v", logData) |
|||
} |
|||
|
|||
func formatLogData(data ...any) string { |
|||
var formattedData []string |
|||
for _, item := range data { |
|||
switch v := item.(type) { |
|||
case string: |
|||
formattedData = append(formattedData, v) |
|||
case int, int64, float64, bool: |
|||
formattedData = append(formattedData, fmt.Sprintf("%v", v)) |
|||
case []interface{}: |
|||
for _, elem := range v { |
|||
formattedData = append(formattedData, fmt.Sprintf("%+v", elem)) |
|||
} |
|||
default: |
|||
formattedData = append(formattedData, fmt.Sprintf("%+v", v)) |
|||
} |
|||
} |
|||
return strings.Join(formattedData, " ") |
|||
} |
|||
|
|||
func TRACE(data ...any) { |
|||
if loglevel <= Level_T { |
|||
_trace.Println(formatLogWithTrace(data...)) |
|||
} |
|||
} |
|||
|
|||
func INFO(data ...any) { |
|||
if loglevel <= Level_I { |
|||
_info.Println(formatLogWithTrace(data...)) |
|||
} |
|||
} |
|||
|
|||
func DEBUG(data ...any) { |
|||
if loglevel <= Level_D { |
|||
_debug.Println(formatLogWithTrace(data...)) |
|||
} |
|||
} |
|||
|
|||
func ERROR(data any, e error) { |
|||
if loglevel <= Level_E { |
|||
_error.Println(formatLogWithTrace(data), e) |
|||
} |
|||
} |
|||
|
|||
func WARN(data ...any) { |
|||
if loglevel <= Level_W { |
|||
_warn.Println(formatLogWithTrace(data...)) |
|||
} |
|||
} |
|||
|
|||
// AsyncWriter 异步写入器
|
|||
type AsyncWriter struct { |
|||
output io.Writer |
|||
ch chan []byte |
|||
wg sync.WaitGroup |
|||
} |
|||
|
|||
func NewAsyncWriter(output io.Writer) *AsyncWriter { |
|||
w := &AsyncWriter{ |
|||
output: output, |
|||
ch: make(chan []byte, 1000), |
|||
} |
|||
w.wg.Add(1) |
|||
go w.run() |
|||
return w |
|||
} |
|||
|
|||
func NewAsyncWriterWithSize(output io.Writer, size int) *AsyncWriter { |
|||
w := &AsyncWriter{ |
|||
output: output, |
|||
ch: make(chan []byte, size), |
|||
} |
|||
w.wg.Add(1) |
|||
go w.run() |
|||
return w |
|||
} |
|||
|
|||
func (w *AsyncWriter) run() { |
|||
defer w.wg.Done() |
|||
for data := range w.ch { |
|||
w.output.Write(data) |
|||
} |
|||
} |
|||
|
|||
func (w *AsyncWriter) Write(p []byte) (n int, err error) { |
|||
w.ch <- append([]byte(nil), p...) |
|||
return len(p), nil |
|||
} |
|||
|
|||
func (w *AsyncWriter) Close() { |
|||
close(w.ch) |
|||
w.wg.Wait() |
|||
} |
|||
|
|||
func InitCustomLog(level Level, config *LogConfig, logName string) { |
|||
if config == nil { |
|||
config = &defaultConfig |
|||
} |
|||
|
|||
SetLevel(level) |
|||
|
|||
if err := os.MkdirAll(config.LogPath, 0o755); err != nil { |
|||
panic(err) |
|||
} |
|||
|
|||
infoLogger := newLevelLogger(fmt.Sprintf("%s-info", logName), config) |
|||
errorLogger := newLevelLogger(fmt.Sprintf("%s-error", logName), config) |
|||
env := os.Getenv("LAIC_ENV") |
|||
|
|||
var output io.Writer |
|||
if env != "prod" && loglevel <= Level_D { |
|||
output = io.MultiWriter(os.Stdout, infoLogger) |
|||
} else { |
|||
output = infoLogger |
|||
} |
|||
|
|||
DefaultWriter = NewAsyncWriter(output) |
|||
|
|||
var errorOutput io.Writer |
|||
if loglevel <= Level_D { |
|||
errorOutput = NewAsyncWriter(io.MultiWriter(os.Stdout, errorLogger, infoLogger)) |
|||
} else { |
|||
errorOutput = NewAsyncWriter(io.MultiWriter(errorLogger, infoLogger)) |
|||
} |
|||
|
|||
_trace = newLogger(Level_T, Magenta, DefaultWriter) |
|||
_debug = newLogger(Level_D, Blue, DefaultWriter) |
|||
_info = newLogger(Level_I, Green, DefaultWriter) |
|||
_warn = newLogger(Level_W, Yellow, DefaultWriter) |
|||
_error = newLogger(Level_E, Red, errorOutput) |
|||
|
|||
log.SetOutput(DefaultWriter) |
|||
log.SetFlags(0) |
|||
} |
|||
|
|||
func newLevelLogger(level string, config *LogConfig) io.Writer { |
|||
return &lumberjack.Logger{ |
|||
Filename: filepath.Join(config.LogPath, level+".log"), |
|||
MaxSize: config.MaxSize, |
|||
MaxBackups: config.MaxBackups, |
|||
MaxAge: config.MaxAge, |
|||
Compress: config.Compress, |
|||
} |
|||
} |
|||
|
|||
func newLogger(level Level, color string, output io.Writer) *log.Logger { |
|||
logger := log.New(output, "", 0) |
|||
logger.SetOutput(&customLogWriter{ |
|||
prefix: fmt.Sprintf("%s[%s]: %s", color, strings.ToUpper(level.String()), Reset), |
|||
output: output, |
|||
}) |
|||
return logger |
|||
} |
|||
|
|||
// String 方法
|
|||
func (l Level) String() string { |
|||
switch l { |
|||
case Level_T: |
|||
return "TRACE" |
|||
case Level_D: |
|||
return "DEBUG" |
|||
case Level_I: |
|||
return "INFO" |
|||
case Level_W: |
|||
return "WARN" |
|||
case Level_E: |
|||
return "ERROR" |
|||
default: |
|||
return "UNKNOWN" |
|||
} |
|||
} |
|||
|
|||
func (c *customLogWriter) Write(p []byte) (n int, err error) { |
|||
buf := &bytes.Buffer{} |
|||
|
|||
traceID := GetTraceID() |
|||
|
|||
_, file, line, ok := runtime.Caller(4) |
|||
if !ok { |
|||
file = "unknown" |
|||
line = 0 |
|||
} |
|||
file = file[strings.LastIndex(file, "/")+1:] |
|||
|
|||
fmt.Fprintf(buf, "[%s] ", time.Now().Format("2006-01-02 15:04:05.999")) |
|||
fmt.Fprintf(buf, "%s", c.prefix) |
|||
if traceID != "" { |
|||
fmt.Fprintf(buf, " (%s) ", traceID) |
|||
} |
|||
fmt.Fprintf(buf, "[%s:%d] %s", file, line, p) |
|||
|
|||
return c.output.Write(buf.Bytes()) |
|||
} |
|||
|
|||
func GetLogWriter() io.Writer { |
|||
return DefaultWriter |
|||
} |
|||
@ -0,0 +1,69 @@ |
|||
package main |
|||
|
|||
import ( |
|||
"os" |
|||
"os/signal" |
|||
"syscall" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/mqtt" |
|||
"laic-backend/route" |
|||
"laic-backend/service" |
|||
"laic-backend/token" |
|||
) |
|||
|
|||
func main() { |
|||
// 1. 初始化日志（先用默认配置，便于后续报错输出）
|
|||
logger.InitCustomLog(logger.Level_D, nil, "laic-backend") |
|||
|
|||
// 2. 加载 config.yaml
|
|||
conf := &common.AppConfig{} |
|||
common.LoadConfig("config.yaml", conf) |
|||
if err := service.InitLiveProvider(conf.Live); err != nil { |
|||
logger.ERROR("直播提供商初始化失败", err) |
|||
panic(err) |
|||
} |
|||
setLogLevel(conf.Log.Level) |
|||
token.Init(conf.JWT.Secret, conf.JWT.AccessExpireH, conf.JWT.RefreshExpireH) |
|||
|
|||
// 3. 初始化基础设施
|
|||
common.InitConnection(&conf.Mysql) |
|||
common.InitTDengine(&conf.TDengine) |
|||
common.InitRedis(conf.Redis) |
|||
common.InitCasbinEnforcer() |
|||
|
|||
// 4. 预热缓存 + 引导 Casbin 策略 + 启动遥测批量写入
|
|||
cache.WarmUp() |
|||
service.BootstrapCasbin() |
|||
service.BootstrapOperationLog() |
|||
service.DefaultTelemetryStore.Start() |
|||
|
|||
// 5. 启动 MQTT（订阅设备上行消息）与 HTTP
|
|||
mqtt.InitMQTT(&conf.MQTT) |
|||
service.InitMQTTSubscriber() |
|||
service.DefaultCommandService.StartRetryScanner() |
|||
service.DefaultScheduler.Start() |
|||
route.InitRouter(int32(conf.ApiPort)) |
|||
logger.INFO("laic-backend starting on port:", conf.ApiPort) |
|||
common.ServerRun(int32(conf.ApiPort)) |
|||
|
|||
// 5. 优雅退出
|
|||
quit := make(chan os.Signal, 1) |
|||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM) |
|||
<-quit |
|||
} |
|||
|
|||
func setLogLevel(level string) { |
|||
switch level { |
|||
case "info": |
|||
logger.SetLevel(logger.Level_I) |
|||
case "warn": |
|||
logger.SetLevel(logger.Level_W) |
|||
case "error": |
|||
logger.SetLevel(logger.Level_E) |
|||
default: |
|||
logger.SetLevel(logger.Level_D) |
|||
} |
|||
} |
|||
@ -0,0 +1,75 @@ |
|||
package middleware |
|||
|
|||
import ( |
|||
"net/http" |
|||
"slices" |
|||
"strconv" |
|||
"strings" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/token" |
|||
) |
|||
|
|||
// AuthMiddleware 鉴权中间件：解析 JWT + Redis 存证校验 + 注入上下文
|
|||
func AuthMiddleware() gin.HandlerFunc { |
|||
return func(c *gin.Context) { |
|||
tokenString := c.GetHeader("Authorization") |
|||
if tokenString == "" { |
|||
common.FailWithBusiErrorWithHttpStatus(c, http.StatusUnauthorized, common.ErrUnauthorized) |
|||
return |
|||
} |
|||
tokenString = strings.TrimPrefix(tokenString, "Bearer ") |
|||
if tokenString == "" { |
|||
common.FailWithBusiErrorWithHttpStatus(c, http.StatusUnauthorized, common.ErrUnauthorized) |
|||
return |
|||
} |
|||
|
|||
claims, err := token.ParseToken(tokenString) |
|||
if err != nil { |
|||
common.FailWithBusiErrorWithHttpStatus(c, http.StatusUnauthorized, common.ErrToken) |
|||
return |
|||
} |
|||
|
|||
// Redis 存证校验：登出会删除该 key，从而让已签发 token 失效
|
|||
storeKey := cache.UserTokenKeyOf(claims.UserID, token.GenerateShortID(tokenString)) |
|||
if !common.HasKey(storeKey) { |
|||
common.FailWithBusiErrorWithHttpStatus(c, http.StatusUnauthorized, common.SignOutToken) |
|||
return |
|||
} |
|||
|
|||
roles := claims.Roles |
|||
if roles == nil { |
|||
roles = []string{} |
|||
} |
|||
isAdmin := slices.Contains(roles, common.RoleAdmin) |
|||
c.Set("user_id", claims.UserID) |
|||
c.Set("username", claims.Username) |
|||
c.Set("roles", roles) |
|||
c.Set("is_admin", isAdmin) |
|||
|
|||
// Casbin 权限校验（admin 与普通用户均按 p/g 策略校验，admin 仅限平台管理职能）
|
|||
if common.CasbinEnforcer != nil { |
|||
ok, err := common.CasbinEnforcer.Enforce(strconv.FormatInt(claims.UserID, 10), c.Request.URL.Path, c.Request.Method) |
|||
if err != nil || !ok { |
|||
common.FailWithBusiErrorWithHttpStatus(c, http.StatusForbidden, common.ErrForbidden) |
|||
c.Abort() |
|||
return |
|||
} |
|||
} |
|||
c.Next() |
|||
} |
|||
} |
|||
|
|||
// AdminMiddleware 仅管理员可访问
|
|||
func AdminMiddleware() gin.HandlerFunc { |
|||
return func(c *gin.Context) { |
|||
if !c.GetBool("is_admin") { |
|||
common.FailWithBusiErrorWithHttpStatus(c, http.StatusForbidden, common.ErrForbidden) |
|||
return |
|||
} |
|||
c.Next() |
|||
} |
|||
} |
|||
@ -0,0 +1,30 @@ |
|||
package middleware |
|||
|
|||
import ( |
|||
"net/http" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
) |
|||
|
|||
// CORSMiddleware 跨域中间件
|
|||
func CORSMiddleware() gin.HandlerFunc { |
|||
return func(c *gin.Context) { |
|||
origin := c.Request.Header.Get("Origin") |
|||
if origin == "" { |
|||
c.Writer.Header().Set("Access-Control-Allow-Origin", "*") |
|||
} else { |
|||
c.Writer.Header().Set("Access-Control-Allow-Origin", origin) |
|||
} |
|||
c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE") |
|||
c.Header("Access-Control-Allow-Headers", "Authorization, Content-Length, Content-Type") |
|||
c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers") |
|||
c.Header("Access-Control-Max-Age", "172800") |
|||
c.Header("Access-Control-Allow-Credentials", "true") |
|||
|
|||
if c.Request.Method == http.MethodOptions { |
|||
c.AbortWithStatus(http.StatusOK) |
|||
return |
|||
} |
|||
c.Next() |
|||
} |
|||
} |
|||
@ -0,0 +1,25 @@ |
|||
package middleware |
|||
|
|||
import ( |
|||
"time" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/logger" |
|||
) |
|||
|
|||
// DefaultLogMiddleware 访问日志：方法、路径、状态码、耗时、来源 IP
|
|||
func DefaultLogMiddleware() gin.HandlerFunc { |
|||
return func(c *gin.Context) { |
|||
start := time.Now() |
|||
c.Next() |
|||
latency := time.Since(start) |
|||
logger.INFO( |
|||
c.Request.Method, |
|||
c.Request.URL.Path, |
|||
"status", c.Writer.Status(), |
|||
"ip", c.ClientIP(), |
|||
"latency", latency.String(), |
|||
) |
|||
} |
|||
} |
|||
@ -0,0 +1,155 @@ |
|||
package middleware |
|||
|
|||
import ( |
|||
"net/http" |
|||
"strings" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/model" |
|||
"laic-backend/service" |
|||
) |
|||
|
|||
// OperationLogMiddleware 记录设备控制写操作（POST/PUT/DELETE）到 operation_log
|
|||
func OperationLogMiddleware() gin.HandlerFunc { |
|||
return func(c *gin.Context) { |
|||
if !isMutating(c.Request.Method) { |
|||
c.Next() |
|||
return |
|||
} |
|||
path := c.Request.URL.Path |
|||
if !isDeviceOperation(path) { |
|||
c.Next() |
|||
return |
|||
} |
|||
c.Next() |
|||
|
|||
userID := c.GetInt64("user_id") |
|||
if userID == 0 { |
|||
return |
|||
} |
|||
result := "success" |
|||
if c.Writer.Status() >= http.StatusBadRequest { |
|||
result = "failed" |
|||
} |
|||
userName := c.GetString("username") |
|||
if name := resolveUserName(userID); name != "" { |
|||
userName = name |
|||
} |
|||
service.DefaultOperationLogService.Record( |
|||
userID, userName, |
|||
resolveModule(path), resolveAction(c.Request.Method, path), |
|||
path, result, c.ClientIP(), |
|||
) |
|||
} |
|||
} |
|||
|
|||
func isMutating(method string) bool { |
|||
return method == http.MethodPost || method == http.MethodPut || method == http.MethodDelete |
|||
} |
|||
|
|||
func isDeviceOperation(path string) bool { |
|||
parts := strings.Split(strings.Trim(path, "/"), "/") |
|||
if len(parts) < 3 || parts[0] != "v1" { |
|||
return false |
|||
} |
|||
switch parts[1] { |
|||
case "docks": |
|||
return len(parts) >= 4 && (parts[3] == "command" || parts[3] == "firmware") |
|||
case "tasks": |
|||
return len(parts) >= 4 && parts[3] == "execute" |
|||
case "alarms": |
|||
return len(parts) >= 4 && (parts[3] == "acknowledge" || parts[3] == "resolve") |
|||
case "live": |
|||
return len(parts) >= 4 && (parts[3] == "start" || parts[3] == "stop") |
|||
default: |
|||
return false |
|||
} |
|||
} |
|||
|
|||
// resolveUserName 从 DB 读取用户显示名（快照），失败回退为空（由调用方回退 phone）
|
|||
func resolveUserName(userID int64) string { |
|||
var u model.User |
|||
if err := common.DB.Select("name").First(&u, userID).Error; err == nil { |
|||
return u.Name |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
// resolveModule 将路径首个资源段映射为中文模块名
|
|||
func resolveModule(path string) string { |
|||
parts := strings.Split(strings.Trim(path, "/"), "/") |
|||
if len(parts) < 2 { |
|||
return "系统" |
|||
} |
|||
switch parts[1] { |
|||
case "users": |
|||
return "用户管理" |
|||
case "roles": |
|||
return "角色管理" |
|||
case "docks": |
|||
return "机巢管理" |
|||
case "drones": |
|||
return "无人机管理" |
|||
case "alarms": |
|||
return "告警管理" |
|||
case "commands": |
|||
return "指令管理" |
|||
case "routes": |
|||
return "航线管理" |
|||
case "tasks": |
|||
return "任务管理" |
|||
case "executions": |
|||
return "执行记录" |
|||
case "firmwares": |
|||
return "固件管理" |
|||
case "live": |
|||
return "直播管理" |
|||
case "videos": |
|||
return "视频管理" |
|||
case "account": |
|||
return "账户中心" |
|||
case "logs": |
|||
return "系统管理" |
|||
case "auth": |
|||
return "认证" |
|||
default: |
|||
return "系统" |
|||
} |
|||
} |
|||
|
|||
// resolveAction 将 HTTP 方法 + 路径末尾段映射为中文动作
|
|||
func resolveAction(method, path string) string { |
|||
seg := path |
|||
if i := strings.LastIndex(seg, "/"); i >= 0 { |
|||
seg = seg[i+1:] |
|||
} |
|||
special := map[string]string{ |
|||
"execute": "执行任务", |
|||
"command": "下发指令", |
|||
"upgrade": "下发升级", |
|||
"pay": "支付订单", |
|||
"download": "下载视频", |
|||
"start": "开始直播", |
|||
"stop": "停止直播", |
|||
"complete": "确认上传", |
|||
"recharge": "充值", |
|||
"resolve": "处理告警", |
|||
"acknowledge": "确认告警", |
|||
"logout": "退出登录", |
|||
} |
|||
if a, ok := special[seg]; ok { |
|||
return a |
|||
} |
|||
switch method { |
|||
case http.MethodPost: |
|||
return "新增" |
|||
case http.MethodPut: |
|||
return "修改" |
|||
case http.MethodDelete: |
|||
return "删除" |
|||
default: |
|||
return method |
|||
} |
|||
} |
|||
@ -0,0 +1,47 @@ |
|||
package middleware |
|||
|
|||
import ( |
|||
"fmt" |
|||
"net/http" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
) |
|||
|
|||
// rateLimitLua 固定窗口计数：首次计数设置过期时间，返回当前窗口计数
|
|||
const rateLimitLua = ` |
|||
local key = KEYS[1] |
|||
local ttl = tonumber(ARGV[1]) |
|||
local n = redis.call('INCR', key) |
|||
if n == 1 then |
|||
redis.call('EXPIRE', key, ttl) |
|||
end |
|||
return n |
|||
` |
|||
|
|||
// RateLimitMiddleware 基于 Redis 固定窗口的按 IP 限流（默认 300 次/分钟）
|
|||
func RateLimitMiddleware(limit int64, windowSeconds int64) gin.HandlerFunc { |
|||
if limit <= 0 { |
|||
limit = 300 |
|||
} |
|||
if windowSeconds <= 0 { |
|||
windowSeconds = 60 |
|||
} |
|||
return func(c *gin.Context) { |
|||
key := fmt.Sprintf("laic:ratelimit:%s", c.ClientIP()) |
|||
n, err := common.GetLuaInt64(rateLimitLua, []string{key}, windowSeconds) |
|||
if err != nil { |
|||
// 限流依赖 Redis，异常时放行（避免 Redis 故障拖垮全站）
|
|||
logger.ERROR("限流计数失败", err) |
|||
c.Next() |
|||
return |
|||
} |
|||
if n > limit { |
|||
common.FailWithBusiErrorWithHttpStatus(c, http.StatusTooManyRequests, common.NewBusiError(http.StatusTooManyRequests, "请求过于频繁，请稍后再试")) |
|||
return |
|||
} |
|||
c.Next() |
|||
} |
|||
} |
|||
@ -0,0 +1,26 @@ |
|||
package middleware |
|||
|
|||
import ( |
|||
"strings" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
"github.com/google/uuid" |
|||
|
|||
"laic-backend/logger" |
|||
) |
|||
|
|||
// TraceMiddleware 为每个请求生成/透传 traceID，便于链路日志追踪
|
|||
func TraceMiddleware() gin.HandlerFunc { |
|||
return func(c *gin.Context) { |
|||
traceID := c.GetHeader("X-Trace-Id") |
|||
if traceID == "" { |
|||
traceID = strings.ReplaceAll(uuid.New().String(), "-", "")[:16] |
|||
} |
|||
logger.SetTraceID(traceID) |
|||
defer logger.CleanTraceID() |
|||
|
|||
c.Header("X-Trace-Id", traceID) |
|||
c.Set("trace_id", traceID) |
|||
c.Next() |
|||
} |
|||
} |
|||
@ -0,0 +1,638 @@ |
|||
package main |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"fmt" |
|||
"log" |
|||
"strings" |
|||
"sync" |
|||
"time" |
|||
|
|||
paho "github.com/eclipse/paho.mqtt.golang" |
|||
) |
|||
|
|||
// downlinkEnvelope 后台下发指令的通用外层（协议文档 §4）
|
|||
type downlinkEnvelope struct { |
|||
RequestID string `json:"requestId"` |
|||
DockID string `json:"dockId"` |
|||
DroneSN string `json:"droneSn"` |
|||
Payload json.RawMessage `json:"payload"` |
|||
} |
|||
|
|||
// commandMsg 后台下发的指令载荷（协议文档 §5.1，位于 payload 内）
|
|||
type commandMsg struct { |
|||
CommandID string `json:"commandId"` |
|||
Type string `json:"type"` |
|||
TTLMs int `json:"ttlMs"` |
|||
Params map[string]any `json:"params"` |
|||
RequestID string // 来自通用外层
|
|||
DroneSN string // 来自通用外层
|
|||
} |
|||
|
|||
// droneTelemetry 是机巢直接生成的无人机模拟状态
|
|||
type droneTelemetry struct { |
|||
armed bool |
|||
flightMode string |
|||
latitude float64 |
|||
longitude float64 |
|||
altitude float64 |
|||
groundSpeed float64 |
|||
heading float64 |
|||
roll float64 |
|||
pitch float64 |
|||
yaw float64 |
|||
batteryPct int |
|||
batteryV float64 |
|||
satellites int |
|||
gpsQuality string |
|||
batteryPctDrain int |
|||
} |
|||
|
|||
// MockDock 模拟机巢：通过 MQTT 上云并直接上报机巢、无人机状态
|
|||
type MockDock struct { |
|||
spec dockSpec |
|||
|
|||
client paho.Client |
|||
|
|||
mu sync.Mutex |
|||
doorState string |
|||
droneOnline bool |
|||
tele droneTelemetry |
|||
inMission bool |
|||
missionPaused bool |
|||
missionCancel chan struct{} |
|||
missionCmdID string |
|||
uploadedRoute []missionWaypoint |
|||
liveSessionID string |
|||
liveStreaming bool |
|||
videoVersion int64 |
|||
} |
|||
|
|||
func newMockDock(spec dockSpec) *MockDock { |
|||
d := &MockDock{ |
|||
spec: spec, |
|||
doorState: "closed", |
|||
tele: droneTelemetry{ |
|||
flightMode: "STANDBY", |
|||
latitude: spec.DroneLat, |
|||
longitude: spec.DroneLon, |
|||
batteryPct: spec.Battery, |
|||
batteryV: 25.2, |
|||
satellites: 22, |
|||
gpsQuality: "RTK_FIX", |
|||
}, |
|||
droneOnline: true, |
|||
} |
|||
if spec.Flying { |
|||
d.doorState = "open" |
|||
d.tele.armed = true |
|||
d.tele.flightMode = "AUTO" |
|||
d.tele.altitude = 58 |
|||
d.tele.groundSpeed = 4.2 |
|||
d.tele.heading = 120 |
|||
} |
|||
return d |
|||
} |
|||
|
|||
func (d *MockDock) start(broker, username, password string) error { |
|||
// 1. MQTT 连接后台 EMQX
|
|||
opts := paho.NewClientOptions(). |
|||
AddBroker(broker). |
|||
SetClientID("mock-dock-" + d.spec.DockID). |
|||
SetUsername(username). |
|||
SetPassword(password). |
|||
SetAutoReconnect(true). |
|||
SetCleanSession(true) |
|||
opts.SetOnConnectHandler(func(c paho.Client) { d.onMqttConnect(c) }) |
|||
d.client = paho.NewClient(opts) |
|||
if tok := d.client.Connect(); tok.Wait() && tok.Error() != nil { |
|||
return fmt.Errorf("MQTT 连接失败: %w", tok.Error()) |
|||
} |
|||
|
|||
go d.publishLoop() |
|||
|
|||
log.Printf("[%s] 机巢启动（MQTT 直接模拟无人机数据）", d.spec.DockID) |
|||
return nil |
|||
} |
|||
|
|||
func (d *MockDock) topic(sub string) string { |
|||
return fmt.Sprintf("dock-edge/v1/dock/%s/%s", d.spec.DockID, sub) |
|||
} |
|||
|
|||
// publish 上发一条不含 droneSn 的消息（外层统一包 envelope）
|
|||
func (d *MockDock) publish(topic string, qos byte, retained bool, payload any) { |
|||
d.publishEnvelope(topic, qos, retained, "", "", payload) |
|||
} |
|||
|
|||
// publishDrone 上发一条携带 droneSn 的消息
|
|||
func (d *MockDock) publishDrone(topic string, qos byte, retained bool, droneSN string, payload any) { |
|||
d.publishEnvelope(topic, qos, retained, "", droneSN, payload) |
|||
} |
|||
|
|||
func (d *MockDock) publishEnvelope(topic string, qos byte, retained bool, requestID, droneSN string, payload any) { |
|||
if d.client == nil || !d.client.IsConnected() { |
|||
return |
|||
} |
|||
b, err := json.Marshal(d.wrap(requestID, droneSN, payload)) |
|||
if err != nil { |
|||
return |
|||
} |
|||
d.client.Publish(topic, qos, retained, b) |
|||
} |
|||
|
|||
func (d *MockDock) publishVideoState(payload any, version int64) { |
|||
if d.client == nil || !d.client.IsConnected() { |
|||
return |
|||
} |
|||
b, err := json.Marshal(map[string]any{ |
|||
"requestId": nil, |
|||
"eventId": fmt.Sprintf("video-%s-%d", d.spec.DockID, version), |
|||
"version": version, |
|||
"dockId": d.spec.DockID, |
|||
"droneSn": nil, |
|||
"timestamp": time.Now().UnixMilli(), |
|||
"payload": payload, |
|||
}) |
|||
if err != nil { |
|||
return |
|||
} |
|||
d.client.Publish(d.topic("state/video"), 1, true, b) |
|||
} |
|||
|
|||
// wrap 构造通用外层；requestId / droneSn 为空时序列化为 null
|
|||
func (d *MockDock) wrap(requestID, droneSN string, payload any) map[string]any { |
|||
return map[string]any{ |
|||
"requestId": nilIfEmpty(requestID), |
|||
"dockId": d.spec.DockID, |
|||
"droneSn": nilIfEmpty(droneSN), |
|||
"timestamp": time.Now().UnixMilli(), |
|||
"payload": payload, |
|||
} |
|||
} |
|||
|
|||
func (d *MockDock) onMqttConnect(c paho.Client) { |
|||
log.Printf("[%s] MQTT 已连接", d.spec.DockID) |
|||
if tok := c.Subscribe(d.topic("command"), 1, d.onCommand); tok.Wait() && tok.Error() != nil { |
|||
log.Printf("[%s] 订阅指令失败: %v", d.spec.DockID, tok.Error()) |
|||
} |
|||
d.publishStatusOnline() |
|||
d.publishStateDock() |
|||
d.publishStateDrone() |
|||
d.publishStateVideo() |
|||
} |
|||
|
|||
func (d *MockDock) publishLoop() { |
|||
tick1 := time.NewTicker(time.Second) |
|||
tick5 := time.NewTicker(5 * time.Second) |
|||
defer tick1.Stop() |
|||
defer tick5.Stop() |
|||
for { |
|||
select { |
|||
case <-tick1.C: |
|||
d.publishTelemetry() |
|||
case <-tick5.C: |
|||
d.publishStateDrone() |
|||
d.publishStateDock() |
|||
} |
|||
} |
|||
} |
|||
|
|||
// ---- 指令下发 ----
|
|||
|
|||
func (d *MockDock) onCommand(_ paho.Client, msg paho.Message) { |
|||
var env downlinkEnvelope |
|||
if err := json.Unmarshal(msg.Payload(), &env); err != nil { |
|||
log.Printf("[%s] 解析指令外层失败: %v", d.spec.DockID, err) |
|||
return |
|||
} |
|||
var cmd commandMsg |
|||
if err := json.Unmarshal(env.Payload, &cmd); err != nil { |
|||
log.Printf("[%s] 解析指令载荷失败: %v", d.spec.DockID, err) |
|||
return |
|||
} |
|||
cmd.RequestID = env.RequestID |
|||
cmd.DroneSN = env.DroneSN |
|||
log.Printf("[%s] 收到指令 %s (commandId=%s)", d.spec.DockID, cmd.Type, cmd.CommandID) |
|||
|
|||
switch { |
|||
case strings.HasPrefix(cmd.Type, "dock."): |
|||
d.handleDockCommand(cmd) |
|||
case strings.HasPrefix(cmd.Type, "drone."): |
|||
d.handleDroneCommand(cmd) |
|||
case strings.HasPrefix(cmd.Type, "workflow."): |
|||
d.handleWorkflowCommand(cmd) |
|||
case strings.HasPrefix(cmd.Type, "video."): |
|||
d.handleVideoCommand(cmd) |
|||
default: |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, false, "UNSUPPORTED_COMMAND") |
|||
} |
|||
} |
|||
|
|||
func (d *MockDock) handleDockCommand(cmd commandMsg) { |
|||
switch cmd.Type { |
|||
case "dock.open": |
|||
d.setDoor("open") |
|||
case "dock.close": |
|||
d.setDoor("closed") |
|||
case "dock.reset": |
|||
d.setDoor("closed") |
|||
case "dock.prepare_takeoff", "dock.complete_takeoff", "dock.prepare_landing", "dock.complete_landing", |
|||
"dock.charge_start", "dock.drone_power_on", "dock.drone_power_off", |
|||
"dock.centering_loose", "dock.centering_tight", "dock.clear_alarm", "dock.emergency_stop": |
|||
// 无状态变化，仅确认
|
|||
default: |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, false, "UNSUPPORTED_COMMAND") |
|||
return |
|||
} |
|||
d.publishStateDock() |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, true, "OK") |
|||
} |
|||
|
|||
func (d *MockDock) handleDroneCommand(cmd commandMsg) { |
|||
switch cmd.Type { |
|||
case "drone.takeoff": |
|||
d.mu.Lock() |
|||
d.tele.armed = true |
|||
d.tele.flightMode = "AUTO" |
|||
d.tele.altitude = 50 |
|||
d.tele.groundSpeed = 4.2 |
|||
d.doorState = "open" |
|||
d.mu.Unlock() |
|||
case "drone.land", "drone.return": |
|||
if d.cancelMission() { |
|||
// 任务协程收到取消后会自行返航落地
|
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, true, "OK") |
|||
return |
|||
} |
|||
d.snapHomeLanded() |
|||
case "drone.mission_upload": |
|||
wps := parseWaypoints(cmd.Params) |
|||
d.mu.Lock() |
|||
d.uploadedRoute = wps |
|||
d.mu.Unlock() |
|||
log.Printf("[%s] 已缓存航线，航点数=%d", d.spec.DockID, len(wps)) |
|||
case "drone.mission_start": |
|||
d.mu.Lock() |
|||
wps := append([]missionWaypoint(nil), d.uploadedRoute...) |
|||
d.mu.Unlock() |
|||
if len(wps) == 0 { |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, false, "NO_ROUTE") |
|||
return |
|||
} |
|||
if !d.beginMission(cmd.CommandID) { |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, false, "MISSION_IN_PROGRESS") |
|||
return |
|||
} |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, true, "OK") |
|||
go d.flyUploadedMission(wps) |
|||
return |
|||
case "drone.mission_pause": |
|||
d.setMissionPaused(true) |
|||
case "drone.mission_resume": |
|||
d.setMissionPaused(false) |
|||
case "drone.mission_cancel": |
|||
d.cancelMission() |
|||
default: |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, false, "UNSUPPORTED_COMMAND") |
|||
return |
|||
} |
|||
d.publishStateDrone() |
|||
d.publishStateDock() |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, true, "OK") |
|||
} |
|||
|
|||
func (d *MockDock) handleVideoCommand(cmd commandMsg) { |
|||
streamSessionID, _ := cmd.Params["streamSessionId"].(string) |
|||
switch cmd.Type { |
|||
case "video.start_stream": |
|||
if streamSessionID == "" { |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, false, "SESSION_ID_REQUIRED") |
|||
return |
|||
} |
|||
d.mu.Lock() |
|||
if d.liveStreaming && d.liveSessionID != streamSessionID { |
|||
d.mu.Unlock() |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, false, "STREAM_IN_PROGRESS") |
|||
return |
|||
} |
|||
d.liveSessionID = streamSessionID |
|||
d.liveStreaming = true |
|||
d.mu.Unlock() |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, true, "OK") |
|||
d.publishStateVideo() |
|||
case "video.stop_stream": |
|||
d.mu.Lock() |
|||
matches := streamSessionID != "" && streamSessionID == d.liveSessionID |
|||
if matches { |
|||
d.liveStreaming = false |
|||
} |
|||
d.mu.Unlock() |
|||
if !matches { |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, false, "SESSION_NOT_FOUND") |
|||
return |
|||
} |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, true, "OK") |
|||
d.publishStateVideo() |
|||
default: |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, false, "UNSUPPORTED_COMMAND") |
|||
} |
|||
} |
|||
|
|||
func (d *MockDock) handleWorkflowCommand(cmd commandMsg) { |
|||
switch cmd.Type { |
|||
case "workflow.cancel", "workflow.stop_task": |
|||
d.cancelMission() |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, true, "OK") |
|||
return |
|||
case "workflow.one_key_return", "workflow.one_key_landing": |
|||
if !d.cancelMission() { |
|||
go d.rtlAndLand() |
|||
} |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, true, "OK") |
|||
return |
|||
} |
|||
|
|||
if d.isRepeatCommand(cmd.CommandID) { |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, true, "OK") |
|||
return |
|||
} |
|||
if !d.beginMission(cmd.CommandID) { |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, false, "MISSION_IN_PROGRESS") |
|||
return |
|||
} |
|||
d.ack(cmd.RequestID, cmd.DroneSN, cmd.CommandID, true, "OK") |
|||
|
|||
switch cmd.Type { |
|||
case "workflow.start_task": |
|||
go d.runStartTask(cmd) |
|||
case "workflow.one_key_takeoff": |
|||
go d.runOneKeyTakeoff(cmd) |
|||
default: |
|||
go d.runGenericWorkflow(cmd) |
|||
} |
|||
} |
|||
|
|||
func (d *MockDock) setDoor(state string) { |
|||
d.mu.Lock() |
|||
d.doorState = state |
|||
d.mu.Unlock() |
|||
} |
|||
|
|||
// ---- 上发各类消息 ----
|
|||
|
|||
func (d *MockDock) ack(requestID, droneSN, commandID string, accepted bool, resultCode string) { |
|||
d.publishEnvelope(d.topic("command/ack"), 1, false, requestID, droneSN, map[string]any{ |
|||
"commandId": commandID, |
|||
"accepted": accepted, |
|||
"resultCode": resultCode, |
|||
}) |
|||
} |
|||
|
|||
func (d *MockDock) publishStatusOnline() { |
|||
d.publish(d.topic("status/online"), 1, true, map[string]any{ |
|||
"status": "online", |
|||
"bootId": "boot-" + d.spec.DockID, |
|||
"dockIdSource": "dmi_product_serial", |
|||
"softwareVersion": "1.3.0", |
|||
"protocolVersion": "1.0", |
|||
"uptimeSec": 86400, |
|||
"timeSynced": true, |
|||
"mqttConnected": true, |
|||
"modbusConnected": true, |
|||
"mavlinkConnected": false, |
|||
"updating": false, |
|||
"name": d.spec.Name, |
|||
"location": d.spec.Location, |
|||
"latitude": d.spec.Latitude, |
|||
"longitude": d.spec.Longitude, |
|||
}) |
|||
} |
|||
|
|||
func (d *MockDock) publishStateDock() { |
|||
d.mu.Lock() |
|||
door := d.doorState |
|||
flying := d.tele.armed || d.tele.altitude > 0 |
|||
d.mu.Unlock() |
|||
|
|||
chargingState := "charging" |
|||
chargingVoltage := 25.2 |
|||
chargingCurrent := 8.4 |
|||
dronePresent := true |
|||
if flying { |
|||
chargingState = "idle" |
|||
chargingVoltage = 0 |
|||
chargingCurrent = 0 |
|||
dronePresent = false |
|||
door = "open" |
|||
} |
|||
|
|||
alarms := []string{} |
|||
emergency := false |
|||
if d.spec.Alarm { |
|||
alarms = []string{"DOCK_EMERGENCY_STOP"} |
|||
emergency = true |
|||
} |
|||
|
|||
d.publish(d.topic("state/dock"), 1, true, map[string]any{ |
|||
"plcConnected": true, |
|||
"controlMode": "auto", |
|||
"door": map[string]any{"left": door, "right": door}, |
|||
"centering": map[string]any{"leftRight": "loose", "frontBack": "loose"}, |
|||
"chargingState": chargingState, |
|||
"chargingVoltage": chargingVoltage, |
|||
"chargingCurrent": chargingCurrent, |
|||
"emergencyStop": emergency, |
|||
"alarmCodes": alarms, |
|||
"chargeStartComplete": chargingState == "charging", |
|||
"dronePowerOnComplete": dronePresent, |
|||
"dronePowerOffComplete": false, |
|||
"dronePresent": dronePresent, |
|||
"takeoffPreparationComplete": false, |
|||
"hangarActionComplete": false, |
|||
"landingPreparationComplete": false, |
|||
"landingExecutionComplete": false, |
|||
"doorCloseComplete": door == "closed", |
|||
"doorOpenComplete": door == "open", |
|||
"centeringLooseComplete": false, |
|||
"centeringTightComplete": false, |
|||
"resetComplete": true, |
|||
"environment": dockEnvironment(d.spec.DockID), |
|||
}) |
|||
} |
|||
|
|||
func dockEnvironment(dockID string) map[string]any { |
|||
switch dockID { |
|||
case "dock-2": |
|||
return map[string]any{ |
|||
"rain": false, "windSpeed": 3.6, "outsideTemperature": 22.4, |
|||
"outsideHumidity": 61.0, "insideTemperature": 26.1, "insideHumidity": 48.0, |
|||
} |
|||
case "dock-3": |
|||
return map[string]any{ |
|||
"rain": true, "windSpeed": 5.1, "outsideTemperature": 19.8, |
|||
"outsideHumidity": 78.0, "insideTemperature": 24.6, "insideHumidity": 62.0, |
|||
} |
|||
default: |
|||
return map[string]any{ |
|||
"rain": false, "windSpeed": 1.8, "outsideTemperature": 25.1, |
|||
"outsideHumidity": 55.0, "insideTemperature": 27.3, "insideHumidity": 50.0, |
|||
} |
|||
} |
|||
} |
|||
|
|||
func (d *MockDock) publishStateDrone() { |
|||
d.mu.Lock() |
|||
t := d.tele |
|||
online := d.droneOnline |
|||
d.mu.Unlock() |
|||
|
|||
d.publishDrone(d.topic("state/drone"), 1, true, d.spec.DroneSN, map[string]any{ |
|||
"droneSn": d.spec.DroneSN, |
|||
"name": d.spec.DroneName, |
|||
"currentSysId": 1, |
|||
"online": online, |
|||
"armed": t.armed, |
|||
"flightMode": t.flightMode, |
|||
"flightModeCode": 11, |
|||
"latitude": t.latitude, |
|||
"longitude": t.longitude, |
|||
"altitude": t.altitude, |
|||
"groundSpeed": t.groundSpeed, |
|||
"roll": t.roll, |
|||
"pitch": t.pitch, |
|||
"yaw": t.yaw, |
|||
"batteryPercent": float64(t.batteryPct), |
|||
"batteryVoltage": t.batteryV, |
|||
"batteryCurrent": 0.0, |
|||
"satellites": t.satellites, |
|||
"gpsQuality": t.gpsQuality, |
|||
"linkQuality": 98.0, |
|||
"homeSet": true, |
|||
"alarmCodes": []string{}, |
|||
}) |
|||
} |
|||
|
|||
func (d *MockDock) publishTelemetry() { |
|||
d.mu.Lock() |
|||
t := d.tele |
|||
d.mu.Unlock() |
|||
|
|||
d.publishDrone(d.topic("telemetry"), 0, false, d.spec.DroneSN, map[string]any{ |
|||
"ts": time.Now().UnixMilli(), |
|||
"longitude": t.longitude, |
|||
"latitude": t.latitude, |
|||
"altitude": t.altitude, |
|||
"groundSpeed": t.groundSpeed, |
|||
"roll": t.roll, |
|||
"pitch": t.pitch, |
|||
"yaw": t.yaw, |
|||
"batteryPct": t.batteryPct, |
|||
"batteryV": t.batteryV, |
|||
"satellites": t.satellites, |
|||
"gpsQuality": t.gpsQuality, |
|||
"linkQuality": 96, |
|||
"flightMode": t.flightMode, |
|||
"armed": boolToInt(t.armed), |
|||
}) |
|||
} |
|||
|
|||
func (d *MockDock) publishStateVideo() { |
|||
d.mu.Lock() |
|||
streamSessionID := d.liveSessionID |
|||
streaming := d.liveStreaming |
|||
d.videoVersion++ |
|||
version := d.videoVersion |
|||
d.mu.Unlock() |
|||
|
|||
phase := "idle" |
|||
if streaming { |
|||
phase = "streaming" |
|||
} |
|||
provider := "fake" |
|||
d.publishVideoState(map[string]any{ |
|||
"provider": provider, |
|||
"phase": phase, |
|||
"inputOnline": streaming, |
|||
"streaming": streaming, |
|||
"inputCodec": "h264", |
|||
"uplinkProtocol": "srt", |
|||
"streamSessionId": nilIfEmpty(streamSessionID), |
|||
"width": nil, |
|||
"height": nil, |
|||
"frameRate": nil, |
|||
"bitrateBps": liveBitrate(streaming), |
|||
"retryCount": 0, |
|||
"stopReason": nil, |
|||
"errorCode": nil, |
|||
"updatedAt": time.Now().UnixMilli(), |
|||
}, version) |
|||
} |
|||
|
|||
func liveBitrate(streaming bool) int { |
|||
if streaming { |
|||
return 1500000 |
|||
} |
|||
return 0 |
|||
} |
|||
|
|||
func (d *MockDock) publishWorkflow(cmd commandMsg, taskID, missionID, state, step string) { |
|||
resultCode := any(nil) |
|||
switch state { |
|||
case "succeeded": |
|||
resultCode = "OK" |
|||
case "cancelled": |
|||
resultCode = "CANCELLED" |
|||
case "failed": |
|||
resultCode = "TIMEOUT" |
|||
} |
|||
recordOriginalVideo := any(nil) |
|||
if cmd.Type == "workflow.start_task" { |
|||
recordOriginalVideo = asBool(cmd.Params["recordOriginalVideo"]) |
|||
} |
|||
d.publishEnvelope(d.topic("state/workflow"), 1, true, cmd.RequestID, "", map[string]any{ |
|||
"commandId": cmd.CommandID, |
|||
"type": cmd.Type, |
|||
"taskId": nilIfEmpty(taskID), |
|||
"missionId": nilIfEmpty(missionID), |
|||
"recordOriginalVideo": recordOriginalVideo, |
|||
"state": state, |
|||
"step": step, |
|||
"resultCode": resultCode, |
|||
"updatedAt": time.Now().UnixMilli(), |
|||
}) |
|||
} |
|||
|
|||
// ---- 工具函数 ----
|
|||
|
|||
func boolToInt(b bool) int8 { |
|||
if b { |
|||
return 1 |
|||
} |
|||
return 0 |
|||
} |
|||
|
|||
func nilIfEmpty(s string) any { |
|||
if s == "" { |
|||
return nil |
|||
} |
|||
return s |
|||
} |
|||
|
|||
func flightModeFromBaseMode(base string) string { |
|||
return base |
|||
} |
|||
|
|||
func gpsQualityFromFixType(fix int) string { |
|||
switch fix { |
|||
case 6: |
|||
return "RTK_FIX" |
|||
case 5: |
|||
return "RTK_FLOAT" |
|||
case 3: |
|||
return "3D_FIX" |
|||
case 2: |
|||
return "2D_FIX" |
|||
default: |
|||
return "NO_FIX" |
|||
} |
|||
} |
|||
@ -0,0 +1,73 @@ |
|||
package main |
|||
|
|||
import ( |
|||
"log" |
|||
"os" |
|||
"os/signal" |
|||
"syscall" |
|||
"time" |
|||
) |
|||
|
|||
// dockSpec 描述机巢及其直接模拟上报的绑定无人机。
|
|||
type dockSpec struct { |
|||
DockID string |
|||
Name string |
|||
Location string |
|||
DroneSN string |
|||
DroneName string |
|||
Latitude float64 |
|||
Longitude float64 |
|||
DroneLat float64 |
|||
DroneLon float64 |
|||
Flying bool |
|||
Battery int |
|||
Alarm bool |
|||
} |
|||
|
|||
func main() { |
|||
broker := envOr("MOCK_MQTT_BROKER", "tcp://127.0.0.1:1883") |
|||
username := envOr("MOCK_MQTT_USERNAME", "laic") |
|||
password := envOr("MOCK_MQTT_PASSWORD", "") |
|||
|
|||
specs := []dockSpec{ |
|||
{ |
|||
DockID: "dock-1", Name: "青山湖 01 号机巢", Location: "青山湖科技城 A 区", |
|||
DroneSN: "JG-UAV-001", DroneName: "巡检无人机 001", |
|||
Latitude: 30.262438, Longitude: 119.802631, DroneLat: 30.262438, DroneLon: 119.802631, |
|||
Battery: 86, |
|||
}, |
|||
{ |
|||
DockID: "dock-2", Name: "青山湖 02 号机巢", Location: "青山湖科技城 B 区", |
|||
DroneSN: "JG-UAV-002", DroneName: "巡检无人机 002", |
|||
Latitude: 30.255917, Longitude: 119.818264, DroneLat: 30.256500, DroneLon: 119.818900, |
|||
Flying: true, Battery: 72, |
|||
}, |
|||
{ |
|||
DockID: "dock-3", Name: "青山湖 03 号机巢", Location: "青山湖科技城 C 区", |
|||
DroneSN: "JG-UAV-003", DroneName: "巡检无人机 003", |
|||
Latitude: 30.248610, Longitude: 119.791450, DroneLat: 30.248610, DroneLon: 119.791450, |
|||
Battery: 41, Alarm: true, |
|||
}, |
|||
} |
|||
|
|||
for _, spec := range specs { |
|||
dock := newMockDock(spec) |
|||
if err := dock.start(broker, username, password); err != nil { |
|||
log.Fatalf("启动机巢 %s 失败: %v", spec.DockID, err) |
|||
} |
|||
} |
|||
|
|||
log.Println("模拟设备已启动（机巢 MQTT 直接上报机巢与无人机数据），按 Ctrl+C 退出") |
|||
quit := make(chan os.Signal, 1) |
|||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM) |
|||
<-quit |
|||
log.Println("正在退出...") |
|||
time.Sleep(200 * time.Millisecond) |
|||
} |
|||
|
|||
func envOr(key, def string) string { |
|||
if v := os.Getenv(key); v != "" { |
|||
return v |
|||
} |
|||
return def |
|||
} |
|||
@ -0,0 +1,705 @@ |
|||
package main |
|||
|
|||
import ( |
|||
"bytes" |
|||
"io" |
|||
"log" |
|||
"math" |
|||
"net/http" |
|||
"sort" |
|||
"strconv" |
|||
"time" |
|||
) |
|||
|
|||
const ( |
|||
missionTick = 200 * time.Millisecond |
|||
missionTimeScale = 10.0 // mock 加速倍数，便于在短时间内跑完航线
|
|||
defaultCruiseSpeed = 8.0 // m/s
|
|||
minLegDuration = 2 * time.Second |
|||
maxLegDuration = 12 * time.Second |
|||
maxHoldSec = 8 |
|||
defaultTakeoffAlt = 50.0 |
|||
) |
|||
|
|||
// missionWaypoint 与后台 workflow.start_task params.waypoints 字段对齐
|
|||
type missionWaypoint struct { |
|||
Seq int |
|||
Longitude float64 |
|||
Latitude float64 |
|||
Altitude float64 |
|||
Speed float64 |
|||
Yaw float64 |
|||
HoldSec int |
|||
} |
|||
|
|||
func (d *MockDock) beginMission(commandID string) bool { |
|||
d.mu.Lock() |
|||
defer d.mu.Unlock() |
|||
if d.inMission { |
|||
return false |
|||
} |
|||
d.inMission = true |
|||
d.missionPaused = false |
|||
d.missionCmdID = commandID |
|||
d.missionCancel = make(chan struct{}) |
|||
return true |
|||
} |
|||
|
|||
func (d *MockDock) endMission() { |
|||
d.mu.Lock() |
|||
defer d.mu.Unlock() |
|||
d.inMission = false |
|||
d.missionPaused = false |
|||
d.missionCmdID = "" |
|||
d.missionCancel = nil |
|||
} |
|||
|
|||
func (d *MockDock) isRepeatCommand(commandID string) bool { |
|||
d.mu.Lock() |
|||
defer d.mu.Unlock() |
|||
return commandID != "" && d.inMission && d.missionCmdID == commandID |
|||
} |
|||
|
|||
// cancelMission 通知当前任务中止，返回是否有正在执行的任务
|
|||
func (d *MockDock) cancelMission() bool { |
|||
d.mu.Lock() |
|||
defer d.mu.Unlock() |
|||
if !d.inMission || d.missionCancel == nil { |
|||
return false |
|||
} |
|||
select { |
|||
case <-d.missionCancel: |
|||
default: |
|||
close(d.missionCancel) |
|||
} |
|||
d.missionPaused = false |
|||
return true |
|||
} |
|||
|
|||
func (d *MockDock) setMissionPaused(paused bool) { |
|||
d.mu.Lock() |
|||
defer d.mu.Unlock() |
|||
if d.inMission { |
|||
d.missionPaused = paused |
|||
} |
|||
} |
|||
|
|||
func (d *MockDock) currentCancel() <-chan struct{} { |
|||
d.mu.Lock() |
|||
defer d.mu.Unlock() |
|||
return d.missionCancel |
|||
} |
|||
|
|||
func (d *MockDock) currentLLA() (lat, lon, alt float64) { |
|||
d.mu.Lock() |
|||
defer d.mu.Unlock() |
|||
return d.tele.latitude, d.tele.longitude, d.tele.altitude |
|||
} |
|||
|
|||
func (d *MockDock) setPose(lat, lon, alt, speed, heading float64) { |
|||
d.mu.Lock() |
|||
d.tele.latitude = lat |
|||
d.tele.longitude = lon |
|||
d.tele.altitude = alt |
|||
d.tele.groundSpeed = speed |
|||
d.tele.heading = heading |
|||
d.tele.yaw = heading |
|||
if d.tele.batteryPct > 20 && speed > 0.5 { |
|||
d.tele.batteryPctDrain++ |
|||
if d.tele.batteryPctDrain >= 20 { |
|||
d.tele.batteryPctDrain = 0 |
|||
d.tele.batteryPct-- |
|||
d.tele.batteryV = 22.0 + float64(d.tele.batteryPct)*0.037 |
|||
} |
|||
} |
|||
d.mu.Unlock() |
|||
} |
|||
|
|||
func (d *MockDock) setArmedFlying(alt float64) { |
|||
d.mu.Lock() |
|||
d.tele.armed = true |
|||
d.tele.flightMode = "AUTO" |
|||
d.tele.altitude = alt |
|||
d.doorState = "open" |
|||
d.mu.Unlock() |
|||
} |
|||
|
|||
func (d *MockDock) snapHomeLanded() { |
|||
d.mu.Lock() |
|||
d.tele.armed = false |
|||
d.tele.flightMode = "STANDBY" |
|||
d.tele.groundSpeed = 0 |
|||
d.tele.altitude = 0 |
|||
d.tele.latitude = d.spec.DroneLat |
|||
d.tele.longitude = d.spec.DroneLon |
|||
d.tele.heading = 0 |
|||
d.tele.yaw = 0 |
|||
d.doorState = "closed" |
|||
d.mu.Unlock() |
|||
} |
|||
|
|||
// runStartTask 执行 workflow.start_task：按航点飞行并持续改写遥测坐标
|
|||
func (d *MockDock) runStartTask(cmd commandMsg) { |
|||
defer d.endMission() |
|||
|
|||
taskID, missionID := extractTaskIDs(cmd) |
|||
waypoints := parseWaypoints(cmd.Params) |
|||
recordOriginalVideo := asBool(cmd.Params["recordOriginalVideo"]) |
|||
originalVideo := parseOriginalVideo(cmd.Params["originalVideo"]) |
|||
d.mu.Lock() |
|||
d.uploadedRoute = waypoints |
|||
d.mu.Unlock() |
|||
|
|||
log.Printf("[%s] 开始执行任务 taskId=%s 航点数=%d", d.spec.DockID, taskID, len(waypoints)) |
|||
for _, wp := range waypoints { |
|||
log.Printf("[%s] 航点 seq=%d lon=%.6f lat=%.6f alt=%.1f speed=%.1f hold=%ds", |
|||
d.spec.DockID, wp.Seq, wp.Longitude, wp.Latitude, wp.Altitude, wp.Speed, wp.HoldSec) |
|||
} |
|||
|
|||
d.setDoor("open") |
|||
d.publishStateDock() |
|||
|
|||
prep := []string{ |
|||
"preparing_dock_takeoff", |
|||
"waiting_drone_online", |
|||
"checking_drone", |
|||
"route_uploading", |
|||
"route_ready", |
|||
} |
|||
for _, step := range prep { |
|||
d.publishWorkflow(cmd, taskID, missionID, "running", step) |
|||
if !d.sleepOrAbort(400 * time.Millisecond) { |
|||
d.finishWorkflow(cmd, taskID, missionID, "cancelled") |
|||
return |
|||
} |
|||
} |
|||
|
|||
takeoffAlt := defaultTakeoffAlt |
|||
if len(waypoints) > 0 && waypoints[0].Altitude > 0 { |
|||
takeoffAlt = waypoints[0].Altitude |
|||
} |
|||
|
|||
d.publishWorkflow(cmd, taskID, missionID, "running", "taking_off") |
|||
d.setArmedFlying(0) |
|||
d.publishStateDrone() |
|||
if !d.climbTo(takeoffAlt) { |
|||
d.returnHomeAndLand(true) |
|||
d.finishWorkflow(cmd, taskID, missionID, "cancelled") |
|||
return |
|||
} |
|||
d.publishStateDrone() |
|||
|
|||
d.publishWorkflow(cmd, taskID, missionID, "running", "finishing_takeoff") |
|||
if !d.sleepOrAbort(400 * time.Millisecond) { |
|||
d.returnHomeAndLand(true) |
|||
d.finishWorkflow(cmd, taskID, missionID, "cancelled") |
|||
return |
|||
} |
|||
|
|||
d.publishWorkflow(cmd, taskID, missionID, "running", "starting_mission") |
|||
if !d.sleepOrAbort(300 * time.Millisecond) { |
|||
d.returnHomeAndLand(true) |
|||
d.finishWorkflow(cmd, taskID, missionID, "cancelled") |
|||
return |
|||
} |
|||
|
|||
d.publishWorkflow(cmd, taskID, missionID, "running", "mission_running") |
|||
if !d.flyWaypoints(waypoints) { |
|||
d.publishWorkflow(cmd, taskID, missionID, "running", "returning_to_home") |
|||
d.returnHomeAndLand(true) |
|||
d.finishWorkflow(cmd, taskID, missionID, "cancelled") |
|||
return |
|||
} |
|||
|
|||
d.publishWorkflow(cmd, taskID, missionID, "running", "returning_to_home") |
|||
if !d.returnHomeAndLand(false) { |
|||
d.returnHomeAndLand(true) |
|||
d.finishWorkflow(cmd, taskID, missionID, "cancelled") |
|||
return |
|||
} |
|||
|
|||
d.finishWorkflow(cmd, taskID, missionID, "succeeded") |
|||
if recordOriginalVideo { |
|||
d.uploadOriginalVideo(cmd, taskID, originalVideo) |
|||
} |
|||
log.Printf("[%s] 任务执行完成 taskId=%s", d.spec.DockID, taskID) |
|||
} |
|||
|
|||
func (d *MockDock) runOneKeyTakeoff(cmd commandMsg) { |
|||
defer d.endMission() |
|||
taskID, missionID := extractTaskIDs(cmd) |
|||
d.setDoor("open") |
|||
d.publishStateDock() |
|||
d.publishWorkflow(cmd, taskID, missionID, "running", "taking_off") |
|||
d.setArmedFlying(0) |
|||
if !d.climbTo(defaultTakeoffAlt) { |
|||
d.finishWorkflow(cmd, taskID, missionID, "cancelled") |
|||
return |
|||
} |
|||
d.publishStateDrone() |
|||
d.publishWorkflow(cmd, taskID, missionID, "succeeded", "idle") |
|||
} |
|||
|
|||
func (d *MockDock) runGenericWorkflow(cmd commandMsg) { |
|||
defer d.endMission() |
|||
taskID, missionID := extractTaskIDs(cmd) |
|||
|
|||
d.setDoor("open") |
|||
d.publishStateDock() |
|||
|
|||
steps := []string{ |
|||
"preparing_dock_takeoff", |
|||
"waiting_drone_online", |
|||
"checking_drone", |
|||
"taking_off", |
|||
"finishing_takeoff", |
|||
"returning_to_home", |
|||
"preparing_dock_landing", |
|||
"returning", |
|||
"finishing_landing", |
|||
} |
|||
for _, step := range steps { |
|||
d.publishWorkflow(cmd, taskID, missionID, "running", step) |
|||
if step == "taking_off" { |
|||
d.setArmedFlying(0) |
|||
if !d.climbTo(defaultTakeoffAlt) { |
|||
d.returnHomeAndLand(true) |
|||
d.finishWorkflow(cmd, taskID, missionID, "cancelled") |
|||
return |
|||
} |
|||
} |
|||
if !d.sleepOrAbort(time.Second) { |
|||
d.returnHomeAndLand(true) |
|||
d.finishWorkflow(cmd, taskID, missionID, "cancelled") |
|||
return |
|||
} |
|||
} |
|||
|
|||
d.returnHomeAndLand(true) |
|||
d.finishWorkflow(cmd, taskID, missionID, "succeeded") |
|||
} |
|||
|
|||
func (d *MockDock) flyUploadedMission(waypoints []missionWaypoint) { |
|||
defer d.endMission() |
|||
log.Printf("[%s] 执行已上传航线，航点数=%d", d.spec.DockID, len(waypoints)) |
|||
d.setArmedFlying(math.Max(d.currentAlt(), defaultTakeoffAlt)) |
|||
if !d.flyWaypoints(waypoints) { |
|||
d.returnHomeAndLand(true) |
|||
return |
|||
} |
|||
d.returnHomeAndLand(true) |
|||
} |
|||
|
|||
func (d *MockDock) flyWaypoints(waypoints []missionWaypoint) bool { |
|||
for i, wp := range waypoints { |
|||
alt := wp.Altitude |
|||
if alt <= 0 { |
|||
_, _, alt = d.currentLLA() |
|||
if alt <= 0 { |
|||
alt = defaultTakeoffAlt |
|||
} |
|||
} |
|||
speed := wp.Speed |
|||
if speed <= 0 { |
|||
speed = defaultCruiseSpeed |
|||
} |
|||
log.Printf("[%s] 飞向航点 %d/%d (seq=%d)", d.spec.DockID, i+1, len(waypoints), wp.Seq) |
|||
if !d.flyTo(wp.Latitude, wp.Longitude, alt, speed, false) { |
|||
return false |
|||
} |
|||
if wp.HoldSec > 0 { |
|||
if !d.holdAt(wp.HoldSec, wp.Yaw) { |
|||
return false |
|||
} |
|||
} else if wp.Yaw != 0 { |
|||
d.mu.Lock() |
|||
d.tele.heading = wp.Yaw |
|||
d.tele.yaw = wp.Yaw |
|||
d.mu.Unlock() |
|||
} |
|||
d.publishStateDrone() |
|||
} |
|||
return true |
|||
} |
|||
|
|||
func (d *MockDock) climbTo(alt float64) bool { |
|||
lat, lon, cur := d.currentLLA() |
|||
if cur < 1 { |
|||
d.setPose(lat, lon, 1, 0, d.currentHeading()) |
|||
} |
|||
return d.flyTo(lat, lon, alt, 5, false) |
|||
} |
|||
|
|||
func (d *MockDock) currentAlt() float64 { |
|||
_, _, alt := d.currentLLA() |
|||
return alt |
|||
} |
|||
|
|||
func (d *MockDock) currentHeading() float64 { |
|||
d.mu.Lock() |
|||
defer d.mu.Unlock() |
|||
return d.tele.heading |
|||
} |
|||
|
|||
func (d *MockDock) returnHomeAndLand(forced bool) bool { |
|||
homeAlt := math.Max(d.currentAlt(), 20) |
|||
if !d.flyTo(d.spec.DroneLat, d.spec.DroneLon, homeAlt, defaultCruiseSpeed, forced) && !forced { |
|||
return false |
|||
} |
|||
if !d.flyTo(d.spec.DroneLat, d.spec.DroneLon, 0, 3, forced) && !forced { |
|||
return false |
|||
} |
|||
d.snapHomeLanded() |
|||
d.publishStateDrone() |
|||
d.publishStateDock() |
|||
return true |
|||
} |
|||
|
|||
func (d *MockDock) rtlAndLand() { |
|||
if !d.beginMission("rtl") { |
|||
return |
|||
} |
|||
defer d.endMission() |
|||
d.returnHomeAndLand(true) |
|||
} |
|||
|
|||
func (d *MockDock) finishWorkflow(cmd commandMsg, taskID, missionID, state string) { |
|||
if state == "cancelled" { |
|||
d.setDoor("closed") |
|||
d.publishStateDock() |
|||
} |
|||
d.publishWorkflow(cmd, taskID, missionID, state, "idle") |
|||
} |
|||
|
|||
// flyTo 按球面插值飞向目标点；forced 时忽略暂停/取消（返航落地用）
|
|||
func (d *MockDock) uploadOriginalVideo(cmd commandMsg, taskID string, video originalVideoUpload) { |
|||
if video.VideoID == 0 || video.ExecutionID == 0 || video.UploadURL == "" { |
|||
d.publishOriginalVideo(cmd, taskID, video, "failed", "UPLOAD_CONFIG_INVALID", 0) |
|||
return |
|||
} |
|||
if video.UploadExpireAt > 0 && time.Now().UnixMilli() >= video.UploadExpireAt { |
|||
d.publishOriginalVideo(cmd, taskID, video, "failed", "UPLOAD_URL_EXPIRED", 0) |
|||
return |
|||
} |
|||
|
|||
body := []byte("mock original video fixture\n") |
|||
req, err := http.NewRequest(http.MethodPut, video.UploadURL, bytes.NewReader(body)) |
|||
if err != nil { |
|||
d.publishOriginalVideo(cmd, taskID, video, "failed", "OSS_PUT_FAILED", 0) |
|||
return |
|||
} |
|||
req.Header.Set("Content-Type", "video/mp4") |
|||
resp, err := (&http.Client{Timeout: 20 * time.Second}).Do(req) |
|||
if err != nil { |
|||
d.publishOriginalVideo(cmd, taskID, video, "failed", "OSS_PUT_FAILED", 0) |
|||
return |
|||
} |
|||
_, _ = io.Copy(io.Discard, resp.Body) |
|||
resp.Body.Close() |
|||
if resp.StatusCode < http.StatusOK || resp.StatusCode >= http.StatusMultipleChoices { |
|||
d.publishOriginalVideo(cmd, taskID, video, "failed", "OSS_PUT_FAILED", 0) |
|||
return |
|||
} |
|||
d.publishOriginalVideo(cmd, taskID, video, "completed", "", int64(len(body))) |
|||
} |
|||
|
|||
func (d *MockDock) publishOriginalVideo(cmd commandMsg, taskID string, video originalVideoUpload, eventType, errorCode string, fileSize int64) { |
|||
d.publishDrone(d.topic("internal/original-video"), 1, false, d.spec.DroneSN, map[string]any{ |
|||
"extension": "laic.mock.original-video.v1", |
|||
"eventId": "mock-original-video-" + strconv.FormatInt(video.VideoID, 10), |
|||
"eventType": eventType, |
|||
"videoId": video.VideoID, |
|||
"executionId": video.ExecutionID, |
|||
"taskId": taskID, |
|||
"commandId": cmd.CommandID, |
|||
"fileSize": fileSize, |
|||
"duration": 1, |
|||
"errorCode": nilIfEmpty(errorCode), |
|||
"uploadedAt": time.Now().UnixMilli(), |
|||
}) |
|||
} |
|||
|
|||
type originalVideoUpload struct { |
|||
VideoID int64 |
|||
ExecutionID int64 |
|||
UploadURL string |
|||
UploadExpireAt int64 |
|||
} |
|||
|
|||
func parseOriginalVideo(value any) originalVideoUpload { |
|||
data, ok := value.(map[string]any) |
|||
if !ok { |
|||
return originalVideoUpload{} |
|||
} |
|||
return originalVideoUpload{ |
|||
VideoID: asInt64(data["videoId"]), |
|||
ExecutionID: asInt64(data["executionId"]), |
|||
UploadURL: asString(data["uploadUrl"]), |
|||
UploadExpireAt: asInt64(data["uploadExpireAt"]), |
|||
} |
|||
} |
|||
|
|||
func asString(value any) string { |
|||
text, _ := value.(string) |
|||
return text |
|||
} |
|||
|
|||
func asBool(value any) bool { |
|||
valueBool, _ := value.(bool) |
|||
return valueBool |
|||
} |
|||
|
|||
func asInt64(value any) int64 { |
|||
switch v := value.(type) { |
|||
case float64: |
|||
return int64(v) |
|||
case string: |
|||
parsed, _ := strconv.ParseInt(v, 10, 64) |
|||
return parsed |
|||
default: |
|||
return 0 |
|||
} |
|||
} |
|||
|
|||
func (d *MockDock) flyTo(lat, lon, alt, speed float64, forced bool) bool { |
|||
if speed <= 0 { |
|||
speed = defaultCruiseSpeed |
|||
} |
|||
fromLat, fromLon, fromAlt := d.currentLLA() |
|||
dist := haversineM(fromLat, fromLon, lat, lon) |
|||
dAlt := math.Abs(alt - fromAlt) |
|||
heading := bearingDeg(fromLat, fromLon, lat, lon) |
|||
if dist < 0.5 { |
|||
heading = d.currentHeading() |
|||
} |
|||
|
|||
horizSec := dist / speed / missionTimeScale |
|||
vertSec := dAlt / 5.0 / missionTimeScale |
|||
sec := math.Max(horizSec, vertSec) |
|||
dur := time.Duration(sec * float64(time.Second)) |
|||
if (dist > 1 || dAlt > 1) && dur < minLegDuration { |
|||
dur = minLegDuration |
|||
} |
|||
if dur > maxLegDuration { |
|||
dur = maxLegDuration |
|||
} |
|||
if dur < missionTick { |
|||
dur = missionTick |
|||
} |
|||
|
|||
steps := int(dur / missionTick) |
|||
if steps < 1 { |
|||
steps = 1 |
|||
} |
|||
for i := 1; i <= steps; i++ { |
|||
if !forced && !d.waitMissionTick() { |
|||
return false |
|||
} |
|||
if forced { |
|||
time.Sleep(missionTick) |
|||
} |
|||
t := float64(i) / float64(steps) |
|||
d.setPose( |
|||
lerp(fromLat, lat, t), |
|||
lerp(fromLon, lon, t), |
|||
lerp(fromAlt, alt, t), |
|||
speed, |
|||
heading, |
|||
) |
|||
} |
|||
d.setPose(lat, lon, alt, 0, heading) |
|||
return true |
|||
} |
|||
|
|||
func (d *MockDock) holdAt(seconds int, yaw float64) bool { |
|||
if seconds > maxHoldSec { |
|||
seconds = maxHoldSec |
|||
} |
|||
if yaw != 0 { |
|||
d.mu.Lock() |
|||
d.tele.heading = yaw |
|||
d.tele.yaw = yaw |
|||
d.tele.groundSpeed = 0 |
|||
d.mu.Unlock() |
|||
} else { |
|||
d.setPoseKeepLLA(0) |
|||
} |
|||
ticks := seconds * int(time.Second/missionTick) |
|||
for i := 0; i < ticks; i++ { |
|||
if !d.waitMissionTick() { |
|||
return false |
|||
} |
|||
} |
|||
return true |
|||
} |
|||
|
|||
func (d *MockDock) setPoseKeepLLA(speed float64) { |
|||
d.mu.Lock() |
|||
d.tele.groundSpeed = speed |
|||
d.mu.Unlock() |
|||
} |
|||
|
|||
func (d *MockDock) waitMissionTick() bool { |
|||
cancel := d.currentCancel() |
|||
for { |
|||
if isClosed(cancel) { |
|||
return false |
|||
} |
|||
d.mu.Lock() |
|||
paused := d.missionPaused |
|||
d.mu.Unlock() |
|||
if !paused { |
|||
if cancel == nil { |
|||
time.Sleep(missionTick) |
|||
return true |
|||
} |
|||
select { |
|||
case <-cancel: |
|||
return false |
|||
case <-time.After(missionTick): |
|||
return true |
|||
} |
|||
} |
|||
if cancel == nil { |
|||
time.Sleep(100 * time.Millisecond) |
|||
continue |
|||
} |
|||
select { |
|||
case <-cancel: |
|||
return false |
|||
case <-time.After(100 * time.Millisecond): |
|||
} |
|||
} |
|||
} |
|||
|
|||
func (d *MockDock) sleepOrAbort(wait time.Duration) bool { |
|||
cancel := d.currentCancel() |
|||
if cancel == nil { |
|||
time.Sleep(wait) |
|||
return true |
|||
} |
|||
t := time.NewTimer(wait) |
|||
defer t.Stop() |
|||
select { |
|||
case <-cancel: |
|||
return false |
|||
case <-t.C: |
|||
return true |
|||
} |
|||
} |
|||
|
|||
func extractTaskIDs(cmd commandMsg) (taskID, missionID string) { |
|||
if cmd.Params == nil { |
|||
return "", "" |
|||
} |
|||
if v, ok := cmd.Params["taskId"].(string); ok { |
|||
taskID = v |
|||
} |
|||
if v, ok := cmd.Params["missionId"].(string); ok { |
|||
missionID = v |
|||
} |
|||
return taskID, missionID |
|||
} |
|||
|
|||
func parseWaypoints(params map[string]any) []missionWaypoint { |
|||
if params == nil { |
|||
return nil |
|||
} |
|||
raw, ok := params["waypoints"] |
|||
if !ok || raw == nil { |
|||
return nil |
|||
} |
|||
arr, ok := raw.([]any) |
|||
if !ok { |
|||
return nil |
|||
} |
|||
out := make([]missionWaypoint, 0, len(arr)) |
|||
for i, item := range arr { |
|||
m, ok := item.(map[string]any) |
|||
if !ok { |
|||
continue |
|||
} |
|||
wp := missionWaypoint{ |
|||
Seq: asInt(m["seq"]), |
|||
Longitude: asFloat(m["longitude"]), |
|||
Latitude: asFloat(m["latitude"]), |
|||
Altitude: asFloat(m["altitude"]), |
|||
Speed: asFloat(m["speed"]), |
|||
Yaw: asFloat(m["yaw"]), |
|||
HoldSec: asInt(m["holdSec"]), |
|||
} |
|||
if wp.Seq == 0 { |
|||
wp.Seq = i |
|||
} |
|||
if wp.Longitude == 0 && wp.Latitude == 0 { |
|||
continue |
|||
} |
|||
out = append(out, wp) |
|||
} |
|||
sort.SliceStable(out, func(i, j int) bool { return out[i].Seq < out[j].Seq }) |
|||
return out |
|||
} |
|||
|
|||
func asFloat(v any) float64 { |
|||
switch n := v.(type) { |
|||
case float64: |
|||
return n |
|||
case float32: |
|||
return float64(n) |
|||
case int: |
|||
return float64(n) |
|||
case int32: |
|||
return float64(n) |
|||
case int64: |
|||
return float64(n) |
|||
case uint64: |
|||
return float64(n) |
|||
default: |
|||
return 0 |
|||
} |
|||
} |
|||
|
|||
func asInt(v any) int { |
|||
return int(asFloat(v)) |
|||
} |
|||
|
|||
func isClosed(ch <-chan struct{}) bool { |
|||
if ch == nil { |
|||
return false |
|||
} |
|||
select { |
|||
case <-ch: |
|||
return true |
|||
default: |
|||
return false |
|||
} |
|||
} |
|||
|
|||
func lerp(a, b, t float64) float64 { |
|||
return a + (b-a)*t |
|||
} |
|||
|
|||
func haversineM(lat1, lon1, lat2, lon2 float64) float64 { |
|||
const earthRadiusM = 6371000.0 |
|||
phi1 := lat1 * math.Pi / 180 |
|||
phi2 := lat2 * math.Pi / 180 |
|||
dPhi := (lat2 - lat1) * math.Pi / 180 |
|||
dLambda := (lon2 - lon1) * math.Pi / 180 |
|||
a := math.Sin(dPhi/2)*math.Sin(dPhi/2) + |
|||
math.Cos(phi1)*math.Cos(phi2)*math.Sin(dLambda/2)*math.Sin(dLambda/2) |
|||
c := 2 * math.Atan2(math.Sqrt(a), math.Sqrt(1-a)) |
|||
return earthRadiusM * c |
|||
} |
|||
|
|||
func bearingDeg(lat1, lon1, lat2, lon2 float64) float64 { |
|||
phi1 := lat1 * math.Pi / 180 |
|||
phi2 := lat2 * math.Pi / 180 |
|||
dLambda := (lon2 - lon1) * math.Pi / 180 |
|||
y := math.Sin(dLambda) * math.Cos(phi2) |
|||
x := math.Cos(phi1)*math.Sin(phi2) - math.Sin(phi1)*math.Cos(phi2)*math.Cos(dLambda) |
|||
deg := math.Atan2(y, x) * 180 / math.Pi |
|||
if deg < 0 { |
|||
deg += 360 |
|||
} |
|||
return deg |
|||
} |
|||
@ -0,0 +1,109 @@ |
|||
2026/08/14 15:49:00 [JG-UAV-001] 无人机启动（MAVLink TCP server 127.0.0.1:5760） |
|||
2026/08/14 15:49:00 [JG-UAV-002] 无人机启动（MAVLink TCP server 127.0.0.1:5761） |
|||
2026/08/14 15:49:01 [dock-1] MQTT 已连接 |
|||
2026/08/14 15:49:01 [dock-1] 机巢启动（MQTT 上云 + MAVLink→127.0.0.1:5760） |
|||
2026/08/14 15:49:01 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:49:01 [dock-2] MQTT 已连接 |
|||
2026/08/14 15:49:01 [dock-2] 机巢启动（MQTT 上云 + MAVLink→127.0.0.1:5761） |
|||
2026/08/14 15:49:01 模拟设备已启动（机巢=MQTT，无人机=MAVLink），按 Ctrl+C 退出 |
|||
2026/08/14 15:49:01 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:50:01 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:50:01 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:50:03 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:50:03 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:51:03 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:51:03 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:51:05 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:51:05 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:52:05 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:52:05 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:52:07 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:52:07 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:52:15 [dock-1] 收到指令 dock.open (commandId=10179956445184) |
|||
2026/08/14 15:52:56 [dock-1] 收到指令 drone.takeoff (commandId=10185279803392) |
|||
2026/08/14 15:52:56 [dock-1] 已向无人机下发 MAVLink 指令 drone.takeoff |
|||
2026/08/14 15:52:56 [dock-1] 收到无人机 MAVLink ACK: drone.takeoff accepted=true |
|||
2026/08/14 15:53:07 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:53:09 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:53:56 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:53:58 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:54:09 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:54:11 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:54:30 [dock-1] 收到指令 workflow.start_task (commandId=10197644873728) |
|||
2026/08/14 15:55:11 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:55:13 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:55:35 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:55:37 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:56:13 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:56:15 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:56:37 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:56:39 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:57:15 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:57:17 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:57:39 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:57:41 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:58:17 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:58:19 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:58:41 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:58:43 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:58:47 [dock-1] 收到指令 workflow.start_task (commandId=10231313338368) |
|||
2026/08/14 15:59:19 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:59:21 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 15:59:52 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 15:59:54 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:00:21 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:00:23 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:00:54 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:00:56 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:01:23 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:01:25 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:01:56 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:01:58 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:02:25 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:02:27 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:02:58 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:03:00 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:03:27 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:03:29 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:04:00 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:04:02 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:04:29 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:04:31 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:05:02 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:05:04 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:05:31 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:05:33 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:06:04 [dock-1] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:06:06 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:06:33 [dock-2] 无人机 MAVLink 通道断开: EOF |
|||
2026/08/14 16:06:35 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:10:58 [JG-UAV-001] 无人机启动（MAVLink TCP server 127.0.0.1:5760） |
|||
2026/08/14 16:10:58 [JG-UAV-002] 无人机启动（MAVLink TCP server 127.0.0.1:5761） |
|||
2026/08/14 16:10:58 [dock-1] MQTT 已连接 |
|||
2026/08/14 16:10:58 [dock-1] 机巢启动（MQTT 上云 + MAVLink→127.0.0.1:5760） |
|||
2026/08/14 16:10:58 [dock-1] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 16:10:58 [dock-2] MQTT 已连接 |
|||
2026/08/14 16:10:58 [dock-2] 机巢启动（MQTT 上云 + MAVLink→127.0.0.1:5761） |
|||
2026/08/14 16:10:58 模拟设备已启动（机巢=MQTT，无人机=MAVLink），按 Ctrl+C 退出 |
|||
2026/08/14 16:10:58 [dock-2] 无人机 MAVLink 通道已建立 |
|||
2026/08/14 17:08:55 [dock-1] 收到指令 workflow.start_task (commandId=10782857760768) |
|||
2026/08/14 17:09:09 [dock-1] 收到指令 dock.open (commandId=10784631296000) |
|||
2026/08/14 17:09:11 [dock-1] 收到指令 workflow.start_task (commandId=10784979947520) |
|||
2026/08/14 17:09:13 [dock-1] 收到指令 dock.close (commandId=10785230032896) |
|||
2026/08/14 17:09:15 [dock-1] 收到指令 dock.self_check (commandId=10785475792896) |
|||
2026/08/14 17:09:16 [dock-1] 收到指令 dock.reset (commandId=10785655754752) |
|||
2026/08/14 17:09:22 [dock-1] 收到指令 drone.takeoff (commandId=10786424360960) |
|||
2026/08/14 17:09:22 [dock-1] 已向无人机下发 MAVLink 指令 drone.takeoff |
|||
2026/08/14 17:09:22 [dock-1] 收到无人机 MAVLink ACK: drone.takeoff accepted=true |
|||
2026/08/14 17:09:24 [dock-1] 收到指令 drone.land (commandId=10786613104640) |
|||
2026/08/14 17:09:24 [dock-1] 已向无人机下发 MAVLink 指令 drone.land |
|||
2026/08/14 17:09:24 [dock-1] 收到无人机 MAVLink ACK: drone.land accepted=true |
|||
2026/08/14 17:09:26 [dock-1] 收到指令 drone.return (commandId=10786925318144) |
|||
2026/08/14 17:09:26 [dock-1] 已向无人机下发 MAVLink 指令 drone.return |
|||
2026/08/14 17:09:26 [dock-1] 收到无人机 MAVLink ACK: drone.return accepted=true |
|||
2026/08/14 17:09:28 [dock-1] 收到指令 drone.hover (commandId=10787145781248) |
|||
2026/08/14 17:09:30 [dock-1] 收到指令 drone.emergency_stop (commandId=10787445149696) |
|||
2026/08/14 17:10:30 [dock-2] 收到指令 dock.open (commandId=10795332276224) |
|||
2026/08/14 17:10:32 [dock-2] 收到指令 dock.close (commandId=10795596517376) |
|||
2026/08/14 17:10:33 [dock-2] 收到指令 dock.reset (commandId=10795735977984) |
|||
2026/08/14 17:10:35 [dock-2] 收到指令 dock.self_check (commandId=10795919740928) |
|||
@ -0,0 +1,37 @@ |
|||
package model |
|||
|
|||
import "time" |
|||
|
|||
// AlarmCode 告警码字典
|
|||
type AlarmCode struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
Code string `gorm:"column:code;type:VARCHAR(64);not null;uniqueIndex:idx_code" json:"code"` |
|||
Category string `gorm:"column:category;type:VARCHAR(32);not null" json:"category"` // dock / drone
|
|||
MessageCn string `gorm:"column:message_cn;type:VARCHAR(256);not null" json:"messageCn"` |
|||
Level string `gorm:"column:level;type:VARCHAR(16);default:critical" json:"level"` |
|||
Source string `gorm:"column:source;type:VARCHAR(128);default:''" json:"source"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (AlarmCode) TableName() string { |
|||
return "alarm_code" |
|||
} |
|||
|
|||
// Alarm 告警记录
|
|||
type Alarm struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
DockID string `gorm:"column:dock_id;type:VARCHAR(64);not null" json:"dockId"` |
|||
Code string `gorm:"column:code;type:VARCHAR(64);not null" json:"code"` |
|||
DeviceType string `gorm:"column:device_type;type:VARCHAR(8);not null" json:"deviceType"` // dock / drone
|
|||
MessageCn string `gorm:"column:message_cn;type:VARCHAR(256);not null" json:"messageCn"` |
|||
Level string `gorm:"column:level;type:VARCHAR(16);not null" json:"level"` |
|||
Status string `gorm:"column:status;type:VARCHAR(16);default:active" json:"status"` // active / acknowledged / resolved
|
|||
TriggeredAt *time.Time `gorm:"column:triggered_at" json:"triggeredAt"` |
|||
ResolvedAt *time.Time `gorm:"column:resolved_at" json:"resolvedAt"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
} |
|||
|
|||
func (Alarm) TableName() string { |
|||
return "alarm" |
|||
} |
|||
@ -0,0 +1,91 @@ |
|||
package model |
|||
|
|||
import "time" |
|||
|
|||
// TrafficOrder 云媒体流量订单
|
|||
type TrafficOrder struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null" json:"userId"` |
|||
AmountGb int `gorm:"column:amount_gb;type:INT;not null" json:"amountGb"` |
|||
UnitPrice float64 `gorm:"column:unit_price;type:DECIMAL(6,2)" json:"unitPrice"` |
|||
TotalPrice float64 `gorm:"column:total_price;type:DECIMAL(10,2)" json:"totalPrice"` |
|||
PayStatus string `gorm:"column:pay_status;type:VARCHAR(16);default:unpaid" json:"payStatus"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
PaidAt *time.Time `gorm:"column:paid_at" json:"paidAt"` |
|||
} |
|||
|
|||
func (TrafficOrder) TableName() string { |
|||
return "traffic_order" |
|||
} |
|||
|
|||
// TrafficUsageLog 云媒体消费流水（账本，不可变）
|
|||
type TrafficUsageLog struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null" json:"userId"` |
|||
SourceType string `gorm:"column:source_type;type:VARCHAR(16)" json:"sourceType"` // live / replay / download
|
|||
SourceID int64 `gorm:"column:source_id;type:BIGINT" json:"sourceId"` |
|||
BytesUsed int64 `gorm:"column:bytes_used;type:BIGINT" json:"bytesUsed"` |
|||
BalanceBefore int64 `gorm:"column:balance_before;type:BIGINT" json:"balanceBefore"` |
|||
BalanceAfter int64 `gorm:"column:balance_after;type:BIGINT" json:"balanceAfter"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
} |
|||
|
|||
func (TrafficUsageLog) TableName() string { |
|||
return "traffic_usage_log" |
|||
} |
|||
|
|||
// SimCard SIM 卡
|
|||
type SimCard struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null" json:"userId"` |
|||
DockID string `gorm:"column:dock_id;type:VARCHAR(64)" json:"dockId"` |
|||
Carrier string `gorm:"column:carrier;type:VARCHAR(16)" json:"carrier"` // 移动/联通/电信
|
|||
Phone string `gorm:"column:phone;type:VARCHAR(16)" json:"phone"` |
|||
Iccid string `gorm:"column:iccid;type:VARCHAR(32)" json:"iccid"` // 权威字段
|
|||
PlanGb int `gorm:"column:plan_gb;type:INT" json:"planGb"` |
|||
UsedGb float64 `gorm:"column:used_gb;type:DECIMAL(10,4);default:0" json:"usedGb"` |
|||
ExpiredAt *time.Time `gorm:"column:expired_at;type:DATE" json:"expiredAt"` |
|||
Status string `gorm:"column:status;type:VARCHAR(16);default:active" json:"status"` // active / expired
|
|||
CarrierStatus string `gorm:"column:carrier_status;type:VARCHAR(16);default:normal" json:"carrierStatus"` // normal / suspended / arrears / cancelled
|
|||
LastSyncAt *time.Time `gorm:"column:last_sync_at" json:"lastSyncAt"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (SimCard) TableName() string { |
|||
return "sim_card" |
|||
} |
|||
|
|||
// SimRechargeLog SIM 卡充值记录
|
|||
type SimRechargeLog struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
SimCardID int64 `gorm:"column:sim_card_id;type:BIGINT;not null" json:"simCardId"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null" json:"userId"` |
|||
AmountGb int `gorm:"column:amount_gb;type:INT;not null" json:"amountGb"` |
|||
UnitPrice float64 `gorm:"column:unit_price;type:DECIMAL(6,2)" json:"unitPrice"` |
|||
TotalPrice float64 `gorm:"column:total_price;type:DECIMAL(10,2)" json:"totalPrice"` |
|||
PayStatus string `gorm:"column:pay_status;type:VARCHAR(16);default:paid" json:"payStatus"` |
|||
RechargeStatus string `gorm:"column:recharge_status;type:VARCHAR(16);default:pending" json:"rechargeStatus"` // pending / success / failed
|
|||
RechargeMsg string `gorm:"column:recharge_msg;type:VARCHAR(256);default:''" json:"rechargeMsg"` |
|||
IdempotentKey string `gorm:"column:idempotent_key;type:VARCHAR(64);not null;uniqueIndex:uk_recharge_idem" json:"idempotentKey"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
} |
|||
|
|||
func (SimRechargeLog) TableName() string { |
|||
return "sim_recharge_log" |
|||
} |
|||
|
|||
// SimUsageRecord SIM 卡用量快照
|
|||
type SimUsageRecord struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
SimCardID int64 `gorm:"column:sim_card_id;type:BIGINT;not null;index:idx_record_sim" json:"simCardId"` |
|||
PlanGb int `gorm:"column:plan_gb;type:INT" json:"planGb"` |
|||
UsedGb float64 `gorm:"column:used_gb;type:DECIMAL(10,4)" json:"usedGb"` |
|||
RemainGb float64 `gorm:"column:remain_gb;type:DECIMAL(10,4)" json:"remainGb"` |
|||
CarrierStatus string `gorm:"column:carrier_status;type:VARCHAR(16)" json:"carrierStatus"` |
|||
SyncedAt time.Time `gorm:"column:synced_at;index:idx_record_sim" json:"syncedAt"` |
|||
} |
|||
|
|||
func (SimUsageRecord) TableName() string { |
|||
return "sim_usage_record" |
|||
} |
|||
@ -0,0 +1,25 @@ |
|||
package model |
|||
|
|||
import "time" |
|||
|
|||
// DeviceCommandLog 设备指令日志
|
|||
type DeviceCommandLog struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` // = commandId (Snowflake)
|
|||
DockID string `gorm:"column:dock_id;type:VARCHAR(64);not null" json:"dockId"` |
|||
CommandType string `gorm:"column:command_type;type:VARCHAR(64);not null" json:"commandType"` // dock.open / drone.takeoff / workflow.start_task
|
|||
Params string `gorm:"column:params;type:JSON" json:"params"` |
|||
DroneSN string `gorm:"column:drone_sn;type:VARCHAR(32)" json:"droneSn"` |
|||
RequestID string `gorm:"column:request_id;type:VARCHAR(64)" json:"requestId"` |
|||
TTLMs int `gorm:"column:ttl_ms;type:INT;default:30000" json:"ttlMs"` |
|||
AckAccepted int8 `gorm:"column:ack_accepted;type:TINYINT" json:"ackAccepted"` |
|||
AckResultCode string `gorm:"column:ack_result_code;type:VARCHAR(64)" json:"ackResultCode"` |
|||
Status string `gorm:"column:status;type:VARCHAR(16);default:sent" json:"status"` // sent / acked / timeout / terminal
|
|||
RetryCount int `gorm:"column:retry_count;type:INT;default:0" json:"retryCount"` |
|||
SentAt *time.Time `gorm:"column:sent_at" json:"sentAt"` |
|||
AckedAt *time.Time `gorm:"column:acked_at" json:"ackedAt"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
} |
|||
|
|||
func (DeviceCommandLog) TableName() string { |
|||
return "device_command_log" |
|||
} |
|||
@ -0,0 +1,49 @@ |
|||
package model |
|||
|
|||
import "time" |
|||
|
|||
// Dock 机巢
|
|||
type Dock struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null" json:"userId"` |
|||
DockID string `gorm:"column:dock_id;type:VARCHAR(64);not null;uniqueIndex:idx_dock_id" json:"dockId"` |
|||
Name string `gorm:"column:name;type:VARCHAR(64);default:''" json:"name"` |
|||
Code string `gorm:"column:code;type:VARCHAR(32);default:''" json:"code"` |
|||
SN string `gorm:"column:sn;type:VARCHAR(64);default:''" json:"sn"` |
|||
Iccid string `gorm:"column:iccid;type:VARCHAR(32);default:''" json:"iccid"` // 冗余，权威见 sim_card.iccid
|
|||
Longitude float64 `gorm:"column:longitude;type:DOUBLE;default:0" json:"longitude"` |
|||
Latitude float64 `gorm:"column:latitude;type:DOUBLE;default:0" json:"latitude"` |
|||
Altitude float64 `gorm:"column:altitude;type:DOUBLE;default:0" json:"altitude"` |
|||
Location string `gorm:"column:location;type:VARCHAR(128);default:''" json:"location"` |
|||
Status string `gorm:"column:status;type:VARCHAR(32);default:offline" json:"status"` |
|||
RegisterStatus string `gorm:"column:register_status;type:VARCHAR(16);default:pending" json:"registerStatus"` |
|||
DockIDSource string `gorm:"column:dock_id_source;type:VARCHAR(32);default:''" json:"dockIdSource"` |
|||
SoftwareVer string `gorm:"column:software_ver;type:VARCHAR(16);default:''" json:"softwareVer"` |
|||
ProtocolVer string `gorm:"column:protocol_ver;type:VARCHAR(8);default:'1.0'" json:"protocolVer"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (Dock) TableName() string { |
|||
return "dock" |
|||
} |
|||
|
|||
// Drone 无人机（与机巢默认一对一绑定）
|
|||
type Drone struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null" json:"userId"` |
|||
DroneSN string `gorm:"column:drone_sn;type:VARCHAR(32);not null;uniqueIndex:idx_drone_sn" json:"droneSn"` |
|||
DockID string `gorm:"column:dock_id;type:VARCHAR(64);not null;uniqueIndex:idx_drone_dock" json:"dockId"` |
|||
Name string `gorm:"column:name;type:VARCHAR(64);default:''" json:"name"` |
|||
Code string `gorm:"column:code;type:VARCHAR(32);default:''" json:"code"` |
|||
Model string `gorm:"column:model;type:VARCHAR(32);default:''" json:"model"` |
|||
Status string `gorm:"column:status;type:VARCHAR(32);default:offline" json:"status"` |
|||
Battery int `gorm:"column:battery;type:INT;default:0" json:"battery"` |
|||
FirmwareVer string `gorm:"column:firmware_ver;type:VARCHAR(16);default:''" json:"firmwareVer"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (Drone) TableName() string { |
|||
return "drone" |
|||
} |
|||
@ -0,0 +1,24 @@ |
|||
package model |
|||
|
|||
import "time" |
|||
|
|||
// Firmware 固件版本
|
|||
type Firmware struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
Component string `gorm:"column:component;type:VARCHAR(32);not null" json:"component"` // 固定 dock-edge-agent
|
|||
Version string `gorm:"column:version;type:VARCHAR(32);not null" json:"version"` // 1.3.0
|
|||
Description string `gorm:"column:description;type:VARCHAR(512);default:''" json:"description"` |
|||
FileURL string `gorm:"column:file_url;type:VARCHAR(512);not null" json:"fileUrl"` |
|||
Sha256 string `gorm:"column:sha256;type:VARCHAR(128);not null" json:"sha256"` |
|||
Signature string `gorm:"column:signature;type:TEXT;not null" json:"signature"` |
|||
FileSize int64 `gorm:"column:file_size;type:BIGINT;default:0" json:"fileSize"` |
|||
Mandatory int8 `gorm:"column:mandatory;type:TINYINT;default:0" json:"mandatory"` |
|||
Status string `gorm:"column:status;type:VARCHAR(16);default:draft" json:"status"` // draft / released / deprecated
|
|||
CreatedBy int64 `gorm:"column:created_by;type:BIGINT;not null" json:"createdBy"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (Firmware) TableName() string { |
|||
return "firmware" |
|||
} |
|||
@ -0,0 +1,80 @@ |
|||
package model |
|||
|
|||
import "time" |
|||
|
|||
// LiveSession 直播会话
|
|||
type LiveSession struct { |
|||
ID string `gorm:"primaryKey;column:id;type:VARCHAR(128);not null" json:"id"` |
|||
DockID string `gorm:"column:dock_id;type:VARCHAR(64);not null" json:"dockId"` |
|||
Provider string `gorm:"column:provider;type:VARCHAR(16);default:aliyun" json:"provider"` |
|||
StreamName string `gorm:"column:stream_name;type:VARCHAR(128);not null" json:"streamName"` |
|||
PushURLHash string `gorm:"column:push_url_hash;type:VARCHAR(128)" json:"pushUrlHash"` |
|||
ExpiresAt int64 `gorm:"column:expires_at;type:BIGINT;not null" json:"expiresAt"` |
|||
MaxBitrateBps int64 `gorm:"column:max_bitrate_bps;type:BIGINT;default:1500000" json:"maxBitrateBps"` |
|||
Phase string `gorm:"column:phase;type:VARCHAR(16);default:idle" json:"phase"` |
|||
StopReason string `gorm:"column:stop_reason;type:VARCHAR(32)" json:"stopReason"` |
|||
ErrorCode string `gorm:"column:error_code;type:VARCHAR(64)" json:"errorCode"` |
|||
RequestedBy int64 `gorm:"column:requested_by;type:BIGINT" json:"requestedBy"` |
|||
StopDeadline *time.Time `gorm:"column:stop_deadline" json:"stopDeadline"` |
|||
DeviceStateVersion int64 `gorm:"column:device_state_version;type:BIGINT" json:"deviceStateVersion"` |
|||
DeviceEventID string `gorm:"column:device_event_id;type:VARCHAR(128)" json:"deviceEventId"` |
|||
DeviceUpdatedAt int64 `gorm:"column:device_updated_at;type:BIGINT" json:"deviceUpdatedAt"` |
|||
CloudConfirmedAt *time.Time `gorm:"column:cloud_confirmed_at" json:"cloudConfirmedAt"` |
|||
StartedAt *time.Time `gorm:"column:started_at" json:"startedAt"` |
|||
StoppedAt *time.Time `gorm:"column:stopped_at" json:"stoppedAt"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (LiveSession) TableName() string { |
|||
return "live_session" |
|||
} |
|||
|
|||
// LiveViewerLease 表示用户对直播会话的短期观看租约。
|
|||
type LiveViewerLease struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
StreamSessionID string `gorm:"column:stream_session_id;type:VARCHAR(128);not null" json:"streamSessionId"` |
|||
ViewerID int64 `gorm:"column:viewer_id;type:BIGINT;not null" json:"viewerId"` |
|||
ExpiresAt time.Time `gorm:"column:expires_at;not null" json:"expiresAt"` |
|||
ReleasedAt *time.Time `gorm:"column:released_at" json:"releasedAt"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (LiveViewerLease) TableName() string { |
|||
return "live_viewer_lease" |
|||
} |
|||
|
|||
// Video 视频文件
|
|||
type Video struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null" json:"userId"` |
|||
DroneSN string `gorm:"column:drone_sn;type:VARCHAR(32)" json:"droneSn"` |
|||
ExecutionID int64 `gorm:"column:execution_id;type:BIGINT" json:"executionId"` |
|||
FileName string `gorm:"column:file_name;type:VARCHAR(256)" json:"fileName"` |
|||
FileSize int64 `gorm:"column:file_size;type:BIGINT" json:"fileSize"` |
|||
Duration int `gorm:"column:duration;type:INT" json:"duration"` |
|||
OssKey string `gorm:"column:oss_key;type:VARCHAR(256)" json:"ossKey"` |
|||
OssBucket string `gorm:"column:oss_bucket;type:VARCHAR(128)" json:"ossBucket"` |
|||
ThumbnailKey string `gorm:"column:thumbnail_key;type:VARCHAR(256)" json:"thumbnailKey"` |
|||
Status string `gorm:"column:status;type:VARCHAR(16);default:uploading" json:"status"` |
|||
UploadExpireAt *time.Time `gorm:"column:upload_expire_at" json:"uploadExpireAt"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
} |
|||
|
|||
func (Video) TableName() string { |
|||
return "video" |
|||
} |
|||
|
|||
// DownloadLog 下载记录
|
|||
type DownloadLog struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
VideoID int64 `gorm:"column:video_id;type:BIGINT;not null" json:"videoId"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null" json:"userId"` |
|||
Bytes int64 `gorm:"column:bytes;type:BIGINT;not null" json:"bytes"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
} |
|||
|
|||
func (DownloadLog) TableName() string { |
|||
return "download_log" |
|||
} |
|||
@ -0,0 +1,20 @@ |
|||
package model |
|||
|
|||
import "time" |
|||
|
|||
// OperationLog 操作日志（审计）
|
|||
type OperationLog struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null;index:idx_op_user" json:"userId"` |
|||
UserName string `gorm:"column:user_name;type:VARCHAR(32)" json:"userName"` |
|||
Module string `gorm:"column:module;type:VARCHAR(32)" json:"module"` |
|||
Action string `gorm:"column:action;type:VARCHAR(64)" json:"action"` |
|||
Detail string `gorm:"column:detail;type:VARCHAR(256)" json:"detail"` |
|||
Result string `gorm:"column:result;type:VARCHAR(16)" json:"result"` |
|||
IP string `gorm:"column:ip;type:VARCHAR(64)" json:"ip"` |
|||
CreatedAt time.Time `gorm:"column:created_at;index:idx_op_user" json:"createdAt"` |
|||
} |
|||
|
|||
func (OperationLog) TableName() string { |
|||
return "operation_log" |
|||
} |
|||
@ -0,0 +1,35 @@ |
|||
package model |
|||
|
|||
import "time" |
|||
|
|||
// Route 航线
|
|||
type Route struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null" json:"userId"` |
|||
Name string `gorm:"column:name;type:VARCHAR(128);not null" json:"name"` |
|||
Description string `gorm:"column:description;type:VARCHAR(256)" json:"description"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (Route) TableName() string { |
|||
return "route" |
|||
} |
|||
|
|||
// RouteWaypoint 航线航点
|
|||
type RouteWaypoint struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
RouteID int64 `gorm:"column:route_id;type:BIGINT;not null;uniqueIndex:uk_route_seq" json:"routeId"` |
|||
Seq int `gorm:"column:seq;type:INT;not null;uniqueIndex:uk_route_seq" json:"seq"` |
|||
Longitude float64 `gorm:"column:longitude;type:DOUBLE" json:"longitude"` |
|||
Latitude float64 `gorm:"column:latitude;type:DOUBLE" json:"latitude"` |
|||
Altitude float64 `gorm:"column:altitude;type:DOUBLE" json:"altitude"` |
|||
Speed float64 `gorm:"column:speed;type:DOUBLE" json:"speed"` |
|||
Yaw float64 `gorm:"column:yaw;type:DOUBLE" json:"yaw"` |
|||
HoldSec int `gorm:"column:hold_sec;type:INT;default:0" json:"holdSec"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
} |
|||
|
|||
func (RouteWaypoint) TableName() string { |
|||
return "route_waypoint" |
|||
} |
|||
@ -0,0 +1,62 @@ |
|||
package model |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"time" |
|||
) |
|||
|
|||
// TaskPlan 任务计划
|
|||
type TaskPlan struct { |
|||
ID string `gorm:"primaryKey;column:id;type:VARCHAR(128);not null" json:"id"` |
|||
UserID int64 `gorm:"column:user_id;type:BIGINT;not null" json:"userId"` |
|||
Name string `gorm:"column:name;type:VARCHAR(128);not null" json:"name"` |
|||
DockID string `gorm:"column:dock_id;type:VARCHAR(64);not null" json:"dockId"` |
|||
RouteID int64 `gorm:"column:route_id;type:BIGINT" json:"routeId"` |
|||
ScheduleType string `gorm:"column:schedule_type;type:VARCHAR(16);default:once" json:"scheduleType"` // once / cron
|
|||
ScheduleCron string `gorm:"column:schedule_cron;type:VARCHAR(32)" json:"scheduleCron"` |
|||
VideoPolicy string `gorm:"column:video_policy;type:VARCHAR(16);default:raw" json:"videoPolicy"` |
|||
Status string `gorm:"column:status;type:VARCHAR(16);default:draft" json:"status"` |
|||
CreatedBy int64 `gorm:"column:created_by;type:BIGINT" json:"createdBy"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (TaskPlan) TableName() string { |
|||
return "task_plan" |
|||
} |
|||
|
|||
// TaskExecution 任务执行记录
|
|||
type TaskExecution struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
TaskID string `gorm:"column:task_id;type:VARCHAR(128)" json:"taskId"` |
|||
CommandID string `gorm:"column:command_id;type:VARCHAR(64)" json:"commandId"` |
|||
DockID string `gorm:"column:dock_id;type:VARCHAR(64);not null" json:"dockId"` |
|||
DroneSN string `gorm:"column:drone_sn;type:VARCHAR(32)" json:"droneSn"` |
|||
StartTime *time.Time `gorm:"column:start_time" json:"startTime"` |
|||
EndTime *time.Time `gorm:"column:end_time" json:"endTime"` |
|||
Status string `gorm:"column:status;type:VARCHAR(16);default:pending" json:"status"` |
|||
TrajectoryJSON json.RawMessage `gorm:"column:trajectory_json;type:JSON" json:"trajectoryJson"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
} |
|||
|
|||
func (TaskExecution) TableName() string { |
|||
return "task_execution" |
|||
} |
|||
|
|||
// WorkflowState 工作流状态（任务/一键流程执行步骤跟踪）
|
|||
type WorkflowState struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
DockID string `gorm:"column:dock_id;type:VARCHAR(64);not null" json:"dockId"` |
|||
CommandID string `gorm:"column:command_id;type:VARCHAR(64)" json:"commandId"` |
|||
Type string `gorm:"column:type;type:VARCHAR(64)" json:"type"` |
|||
TaskID string `gorm:"column:task_id;type:VARCHAR(128)" json:"taskId"` |
|||
MissionID string `gorm:"column:mission_id;type:VARCHAR(128)" json:"missionId"` |
|||
State string `gorm:"column:state;type:VARCHAR(16);not null" json:"state"` // idle / running / succeeded / failed / cancelled
|
|||
Step string `gorm:"column:step;type:VARCHAR(64);not null" json:"step"` |
|||
ResultCode string `gorm:"column:result_code;type:VARCHAR(64)" json:"resultCode"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (WorkflowState) TableName() string { |
|||
return "workflow_state" |
|||
} |
|||
@ -0,0 +1,22 @@ |
|||
package model |
|||
|
|||
import "time" |
|||
|
|||
// User 用户表
|
|||
type User struct { |
|||
ID int64 `gorm:"primaryKey;column:id;type:BIGINT;not null" json:"id"` |
|||
Name string `gorm:"column:name;type:VARCHAR(32);not null" json:"name"` |
|||
Phone string `gorm:"column:phone;type:VARCHAR(32);not null;uniqueIndex:idx_phone" json:"phone"` |
|||
Email string `gorm:"column:email;type:VARCHAR(64)" json:"email"` |
|||
Password string `gorm:"column:password;type:VARCHAR(64);not null" json:"-"` |
|||
Role string `gorm:"column:role;type:VARCHAR(16);default:user" json:"role"` |
|||
TrafficBalance int64 `gorm:"column:traffic_balance;type:BIGINT;default:0" json:"trafficBalance"` |
|||
Status int8 `gorm:"column:status;type:TINYINT;default:1" json:"status"` |
|||
LastLogin *time.Time `gorm:"column:last_login" json:"lastLogin"` |
|||
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` |
|||
UpdatedAt time.Time `gorm:"column:updated_at" json:"updatedAt"` |
|||
} |
|||
|
|||
func (User) TableName() string { |
|||
return "user" |
|||
} |
|||
@ -0,0 +1,54 @@ |
|||
package mqtt |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"time" |
|||
) |
|||
|
|||
// Envelope MQTT 消息通用外层（协议文档 §4）
|
|||
//
|
|||
// {
|
|||
// "requestId": "req-uuid" | null,
|
|||
// "dockId": "dock-...",
|
|||
// "droneSn": "..." | null,
|
|||
// "timestamp": 1784700000000,
|
|||
// "payload": { ... }
|
|||
// }
|
|||
type Envelope struct { |
|||
RequestID string `json:"requestId"` |
|||
EventID string `json:"eventId"` |
|||
Version int64 `json:"version"` |
|||
DockID string `json:"dockId"` |
|||
DroneSN string `json:"droneSn"` |
|||
Timestamp int64 `json:"timestamp"` |
|||
Payload json.RawMessage `json:"payload"` |
|||
} |
|||
|
|||
// ParseEnvelope 解析通用外层；payload 缺失（无外层的历史扁平消息）时把整个消息体当作 payload。
|
|||
func ParseEnvelope(data []byte) *Envelope { |
|||
var env Envelope |
|||
if err := json.Unmarshal(data, &env); err != nil { |
|||
return &Envelope{Payload: data} |
|||
} |
|||
if len(env.Payload) == 0 { |
|||
env.Payload = data |
|||
} |
|||
return &env |
|||
} |
|||
|
|||
// NewEnvelope 构造下行外层；payload 为业务字段（内部自动 marshal），timestamp 取当前毫秒。
|
|||
func NewEnvelope(requestID, dockID, droneSN string, payload any) *Envelope { |
|||
var raw json.RawMessage |
|||
if payload != nil { |
|||
if b, err := json.Marshal(payload); err == nil { |
|||
raw = b |
|||
} |
|||
} |
|||
return &Envelope{ |
|||
RequestID: requestID, |
|||
DockID: dockID, |
|||
DroneSN: droneSN, |
|||
Timestamp: time.Now().UnixMilli(), |
|||
Payload: raw, |
|||
} |
|||
} |
|||
@ -0,0 +1,72 @@ |
|||
package mqtt |
|||
|
|||
import ( |
|||
"errors" |
|||
|
|||
mqtt "github.com/eclipse/paho.mqtt.golang" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
) |
|||
|
|||
var ( |
|||
client mqtt.Client |
|||
subTopics = map[string]byte{} |
|||
subHandler mqtt.MessageHandler |
|||
) |
|||
|
|||
// InitMQTT 初始化 MQTT 客户端（不依赖 service，订阅由上层注册）
|
|||
func InitMQTT(config *common.MQTT) { |
|||
if config == nil || config.Broker == "" { |
|||
logger.WARN("MQTT broker 为空，跳过初始化") |
|||
return |
|||
} |
|||
|
|||
opts := mqtt.NewClientOptions(). |
|||
AddBroker(config.Broker). |
|||
SetClientID(config.ClientId). |
|||
SetUsername(config.Username). |
|||
SetPassword(config.Password). |
|||
SetAutoReconnect(true). |
|||
SetCleanSession(true). |
|||
SetOnConnectHandler(func(c mqtt.Client) { |
|||
logger.INFO("MQTT 已连接，重新订阅") |
|||
resubscribe(c) |
|||
}) |
|||
|
|||
client = mqtt.NewClient(opts) |
|||
if token := client.Connect(); token.Wait() && token.Error() != nil { |
|||
logger.ERROR("MQTT 连接失败", token.Error()) |
|||
return |
|||
} |
|||
logger.INFO("MQTT ready") |
|||
} |
|||
|
|||
// Subscribe 记录订阅关系并立即订阅；连接重建时自动重订阅
|
|||
func Subscribe(topics map[string]byte, handler mqtt.MessageHandler) { |
|||
subHandler = handler |
|||
for topic, qos := range topics { |
|||
subTopics[topic] = qos |
|||
} |
|||
if client != nil && client.IsConnected() { |
|||
resubscribe(client) |
|||
} |
|||
} |
|||
|
|||
func resubscribe(c mqtt.Client) { |
|||
for topic, qos := range subTopics { |
|||
if token := c.Subscribe(topic, qos, subHandler); token.Wait() && token.Error() != nil { |
|||
logger.ERROR("订阅失败 topic="+topic, token.Error()) |
|||
} |
|||
} |
|||
} |
|||
|
|||
// Publish 发布消息到指定 topic
|
|||
func Publish(topic string, qos byte, retained bool, payload []byte) error { |
|||
if client == nil || !client.IsConnected() { |
|||
return errors.New("mqtt client not connected") |
|||
} |
|||
token := client.Publish(topic, qos, retained, payload) |
|||
token.Wait() |
|||
return token.Error() |
|||
} |
|||
@ -0,0 +1,171 @@ |
|||
package route |
|||
|
|||
import ( |
|||
"github.com/gin-gonic/gin" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/handler" |
|||
"laic-backend/logger" |
|||
"laic-backend/middleware" |
|||
"laic-backend/websocket" |
|||
) |
|||
|
|||
// InitRouter 初始化 gin 引擎与路由注册
|
|||
func InitRouter(port int32) { |
|||
common.Server(true) |
|||
common.SetWriter(logger.GetLogWriter()) |
|||
|
|||
engine := common.GetEngine() |
|||
engine.Use(middleware.TraceMiddleware(), middleware.CORSMiddleware(), middleware.DefaultLogMiddleware(), middleware.RateLimitMiddleware(300, 60)) |
|||
|
|||
// 健康检查
|
|||
engine.GET("/health", func(c *gin.Context) { |
|||
common.OKWithData(c, gin.H{"status": "up"}) |
|||
}) |
|||
|
|||
// v1 业务路由
|
|||
v1 := common.RouteGroup("/v1") |
|||
v1.Use(middleware.OperationLogMiddleware()) |
|||
|
|||
// 实时监控 WS（设备状态 + 告警推送）
|
|||
v1.GET("/ws/monitor", middleware.AuthMiddleware(), websocket.HandleMonitor) |
|||
|
|||
// 认证（公开）
|
|||
auth := v1.Group("/auth") |
|||
{ |
|||
auth.POST("/register", handler.Register) |
|||
auth.POST("/login", handler.Login) |
|||
auth.POST("/refresh", handler.Refresh) |
|||
auth.POST("/sms-code", handler.SmsCode) |
|||
auth.POST("/reset-password", handler.ResetPassword) |
|||
auth.POST("/logout", middleware.AuthMiddleware(), handler.Logout) |
|||
} |
|||
|
|||
// 账户中心（资料 / 流量 / SIM 卡）
|
|||
account := v1.Group("/account", middleware.AuthMiddleware()) |
|||
{ |
|||
account.GET("/profile", handler.GetProfile) |
|||
account.PUT("/profile", handler.UpdateProfile) |
|||
account.GET("/traffic/balance", handler.GetTrafficBalance) |
|||
account.GET("/traffic/usage", handler.GetUsagePage) |
|||
account.GET("/traffic/orders", handler.GetOrderPage) |
|||
account.POST("/traffic/orders", handler.CreateTrafficOrder) |
|||
account.POST("/traffic/orders/:id/pay", middleware.AdminMiddleware(), handler.PayTrafficOrder) |
|||
account.GET("/sim-cards", handler.ListSimCards) |
|||
account.GET("/sim-recharge-logs", handler.GetSimRechargeLogPage) |
|||
account.POST("/sim-cards/:id/recharge", handler.RechargeSimCard) |
|||
} |
|||
|
|||
// 机巢 & 无人机
|
|||
docks := v1.Group("/docks", middleware.AuthMiddleware()) |
|||
{ |
|||
docks.GET("", handler.GetDockPage) |
|||
docks.POST("", handler.CreateDock) |
|||
docks.GET("/:id", handler.GetDock) |
|||
docks.PUT("/:id", handler.UpdateDock) |
|||
docks.DELETE("/:id", handler.DeleteDock) |
|||
docks.GET("/:id/status", handler.GetDockStatus) |
|||
docks.POST("/:id/command", handler.DispatchCommand) |
|||
docks.POST("/:id/firmware/upgrade", handler.UpgradeFirmware) |
|||
} |
|||
|
|||
drones := v1.Group("/drones", middleware.AuthMiddleware()) |
|||
{ |
|||
drones.GET("", handler.GetDronePage) |
|||
drones.GET("/:id", handler.GetDrone) |
|||
drones.PUT("/:id", handler.UpdateDrone) |
|||
drones.DELETE("/:id", handler.DeleteDrone) |
|||
drones.GET("/:id/telemetry", handler.GetDroneTelemetry) |
|||
} |
|||
|
|||
// 告警
|
|||
alarms := v1.Group("/alarms", middleware.AuthMiddleware()) |
|||
{ |
|||
alarms.GET("", handler.GetAlarmPage) |
|||
alarms.PUT("/:id/acknowledge", handler.AcknowledgeAlarm) |
|||
alarms.PUT("/:id/resolve", handler.ResolveAlarm) |
|||
} |
|||
|
|||
// 指令历史
|
|||
commands := v1.Group("/commands", middleware.AuthMiddleware()) |
|||
{ |
|||
commands.GET("", handler.GetCommandPage) |
|||
commands.GET("/:id", handler.GetCommand) |
|||
} |
|||
|
|||
// 航线
|
|||
routes := v1.Group("/routes", middleware.AuthMiddleware()) |
|||
{ |
|||
routes.GET("", handler.GetRoutePage) |
|||
routes.POST("", handler.CreateRoute) |
|||
routes.GET("/:id", handler.GetRoute) |
|||
routes.PUT("/:id", handler.UpdateRoute) |
|||
routes.DELETE("/:id", handler.DeleteRoute) |
|||
} |
|||
|
|||
// 任务
|
|||
tasks := v1.Group("/tasks", middleware.AuthMiddleware()) |
|||
{ |
|||
tasks.GET("", handler.GetTaskPage) |
|||
tasks.POST("", handler.CreateTask) |
|||
tasks.GET("/:id", handler.GetTask) |
|||
tasks.PUT("/:id", handler.UpdateTask) |
|||
tasks.DELETE("/:id", handler.DeleteTask) |
|||
tasks.POST("/:id/execute", handler.ExecuteTask) |
|||
} |
|||
|
|||
// 执行记录
|
|||
executions := v1.Group("/executions", middleware.AuthMiddleware()) |
|||
{ |
|||
executions.GET("", handler.GetExecutionPage) |
|||
executions.GET("/:id", handler.GetExecution) |
|||
executions.GET("/:id/trajectory", handler.GetTrajectory) |
|||
} |
|||
|
|||
// 已发布固件列表（普通用户可选升级，无需 admin）
|
|||
v1.GET("/firmwares/released", middleware.AuthMiddleware(), handler.GetReleasedFirmwares) |
|||
|
|||
// 固件管理（admin）
|
|||
firmwares := v1.Group("/firmwares", middleware.AuthMiddleware(), middleware.AdminMiddleware()) |
|||
{ |
|||
firmwares.GET("", handler.GetFirmwarePage) |
|||
firmwares.GET("/:id", handler.GetFirmware) |
|||
firmwares.POST("", handler.CreateFirmware) |
|||
firmwares.PUT("/:id", handler.UpdateFirmware) |
|||
firmwares.DELETE("/:id", handler.DeleteFirmware) |
|||
} |
|||
|
|||
// 直播
|
|||
live := v1.Group("/live", middleware.AuthMiddleware()) |
|||
{ |
|||
live.GET("", handler.GetLivePage) |
|||
live.POST("/:dockId/start", handler.StartLive) |
|||
live.POST("/:dockId/stop", handler.StopLive) |
|||
live.GET("/:dockId/play-url", handler.GetLivePlayURL) |
|||
live.POST("/:dockId/sessions", handler.CreateLiveSession) |
|||
live.GET("/:dockId/sessions/:streamSessionId", handler.GetLiveSession) |
|||
live.POST("/:dockId/sessions/:streamSessionId/heartbeat", handler.HeartbeatLiveSession) |
|||
live.DELETE("/:dockId/sessions/:streamSessionId/viewers/me", handler.LeaveLiveSession) |
|||
} |
|||
|
|||
// 视频（OSS 上传/下载）
|
|||
videos := v1.Group("/videos", middleware.AuthMiddleware()) |
|||
{ |
|||
videos.POST("", handler.CreateVideoUpload) |
|||
videos.POST("/:id/complete", handler.CompleteVideoUpload) |
|||
videos.GET("", handler.GetVideoPage) |
|||
videos.GET("/:id", handler.GetVideo) |
|||
videos.POST("/:id/download", handler.DownloadVideo) |
|||
} |
|||
|
|||
// 系统管理（admin）
|
|||
users := v1.Group("/users", middleware.AuthMiddleware(), middleware.AdminMiddleware()) |
|||
{ |
|||
users.GET("", handler.GetUserPage) |
|||
users.POST("", handler.CreateUser) |
|||
users.PUT("/:id", handler.UpdateUser) |
|||
users.DELETE("/:id", handler.DeleteUser) |
|||
} |
|||
v1.GET("/roles", middleware.AuthMiddleware(), middleware.AdminMiddleware(), handler.GetRoles) |
|||
v1.GET("/logs/operation", middleware.AuthMiddleware(), handler.GetOperationLogPage) |
|||
} |
|||
@ -0,0 +1,94 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"errors" |
|||
"time" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type AccountService struct{} |
|||
|
|||
var DefaultAccountService = &AccountService{} |
|||
|
|||
// GetProfile 查询个人资料
|
|||
func (s *AccountService) GetProfile(userID int64) (*model.User, *common.BusiError) { |
|||
var user model.User |
|||
if err := common.DB.First(&user, userID).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrUserNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &user, nil |
|||
} |
|||
|
|||
// UpdateProfile 更新个人资料（姓名/邮箱）
|
|||
func (s *AccountService) UpdateProfile(userID int64, req *vo.ProfileUpdateReq) (*model.User, *common.BusiError) { |
|||
var user model.User |
|||
if err := common.DB.First(&user, userID).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrUserNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
updates := map[string]any{"updated_at": time.Now()} |
|||
if req.Name != "" { |
|||
updates["name"] = req.Name |
|||
} |
|||
if req.Email != "" { |
|||
updates["email"] = req.Email |
|||
} |
|||
if err := common.DB.Model(&user).Updates(updates).Error; err != nil { |
|||
logger.ERROR("更新个人资料失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
common.DB.First(&user, userID) |
|||
return &user, nil |
|||
} |
|||
|
|||
// GetTrafficBalance 查询云媒体流量余额
|
|||
func (s *AccountService) GetTrafficBalance(userID int64) (*vo.TrafficBalanceVO, *common.BusiError) { |
|||
return DefaultBillingService.GetBalance(userID) |
|||
} |
|||
|
|||
// GetUsagePage 流量消费流水分页
|
|||
func (s *AccountService) GetUsagePage(userID int64, req *vo.UsagePageReq) (*common.PageResponse[model.TrafficUsageLog], *common.BusiError) { |
|||
return DefaultBillingService.GetUsagePage(userID, req) |
|||
} |
|||
|
|||
// GetOrderPage 流量订单分页
|
|||
func (s *AccountService) GetOrderPage(userID int64, req *vo.OrderPageReq) (*common.PageResponse[model.TrafficOrder], *common.BusiError) { |
|||
return DefaultBillingService.GetOrderPage(userID, req) |
|||
} |
|||
|
|||
// CreateTrafficOrder 创建流量充值订单
|
|||
func (s *AccountService) CreateTrafficOrder(userID int64, req *vo.TrafficOrderCreateReq) (*model.TrafficOrder, *common.BusiError) { |
|||
return DefaultBillingService.CreateOrder(userID, req.AmountGb) |
|||
} |
|||
|
|||
// PayTrafficOrder 订单支付入账（admin）
|
|||
func (s *AccountService) PayTrafficOrder(orderID int64) (*model.TrafficOrder, *common.BusiError) { |
|||
return DefaultBillingService.PayOrder(orderID) |
|||
} |
|||
|
|||
// ListSimCards SIM 卡列表
|
|||
func (s *AccountService) ListSimCards(userID int64, req *vo.SimCardPageReq) (*common.PageResponse[model.SimCard], *common.BusiError) { |
|||
return DefaultBillingService.ListSimCards(userID, req) |
|||
} |
|||
|
|||
// GetSimRechargeLogPage SIM 卡充值记录分页
|
|||
func (s *AccountService) GetSimRechargeLogPage(userID int64, req *vo.SimRechargeLogPageReq) (*common.PageResponse[model.SimRechargeLog], *common.BusiError) { |
|||
return DefaultBillingService.GetSimRechargeLogPage(userID, req) |
|||
} |
|||
|
|||
// RechargeSimCard SIM 卡充值
|
|||
func (s *AccountService) RechargeSimCard(userID, simCardID int64, req *vo.SimRechargeReq) (*model.SimRechargeLog, *common.BusiError) { |
|||
return DefaultBillingService.RechargeSimCard(userID, simCardID, req.AmountGb) |
|||
} |
|||
@ -0,0 +1,200 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"errors" |
|||
"time" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type AlarmService struct{} |
|||
|
|||
var DefaultAlarmService = &AlarmService{} |
|||
|
|||
// GetPage 告警分页列表（按所属机巢过滤）
|
|||
func (s *AlarmService) GetPage(userID int64, isAdmin bool, req *vo.AlarmPageReq) (*common.PageResponse[model.Alarm], *common.BusiError) { |
|||
db := common.DB.Model(&model.Alarm{}).Scopes(withDockFilter(userID, isAdmin)) |
|||
if req.Status != "" { |
|||
db = db.Where("status = ?", req.Status) |
|||
} |
|||
if req.Level != "" { |
|||
db = db.Where("level = ?", req.Level) |
|||
} |
|||
if req.DeviceType != "" { |
|||
db = db.Where("device_type = ?", req.DeviceType) |
|||
} |
|||
if req.DockID != "" { |
|||
db = db.Where("dock_id = ?", req.DockID) |
|||
} |
|||
if req.Keyword != "" { |
|||
kw := "%" + req.Keyword + "%" |
|||
db = db.Where("code LIKE ? OR message_cn LIKE ?", kw, kw) |
|||
} |
|||
|
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
logger.ERROR("统计告警失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.Alarm |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
logger.ERROR("查询告警列表失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// Acknowledge 确认告警（active → acknowledged）
|
|||
func (s *AlarmService) Acknowledge(userID int64, isAdmin bool, id int64) *common.BusiError { |
|||
var alarm model.Alarm |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).First(&alarm, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return common.ErrAlarmNotFound |
|||
} |
|||
return common.ErrInternal |
|||
} |
|||
if alarm.Status == "active" { |
|||
if err := common.DB.Model(&model.Alarm{}).Where("id = ?", id).Update("status", "acknowledged").Error; err != nil { |
|||
logger.ERROR("确认告警失败", err) |
|||
return common.ErrInternal |
|||
} |
|||
broadcast("alarm.acknowledged", map[string]any{ |
|||
"id": id, |
|||
"dockId": alarm.DockID, |
|||
"code": alarm.Code, |
|||
}) |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
// Resolve 关闭告警（active/acknowledged → resolved）
|
|||
func (s *AlarmService) Resolve(userID int64, isAdmin bool, id int64) *common.BusiError { |
|||
var alarm model.Alarm |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).First(&alarm, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return common.ErrAlarmNotFound |
|||
} |
|||
return common.ErrInternal |
|||
} |
|||
if alarm.Status != "resolved" { |
|||
now := time.Now() |
|||
if err := common.DB.Model(&model.Alarm{}).Where("id = ?", id). |
|||
Updates(map[string]any{"status": "resolved", "resolved_at": now}).Error; err != nil { |
|||
logger.ERROR("关闭告警失败", err) |
|||
return common.ErrInternal |
|||
} |
|||
broadcast("alarm.resolved", map[string]any{ |
|||
"id": id, |
|||
"dockId": alarm.DockID, |
|||
"code": alarm.Code, |
|||
"resolvedAt": now, |
|||
}) |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
// SyncByState 根据设备最新 alarmCodes 做 diff，生成/关闭告警
|
|||
// deviceType: dock / drone
|
|||
func (s *AlarmService) SyncByState(dockID, deviceType string, codes []string) { |
|||
key := s.alarmKey(dockID, deviceType) |
|||
|
|||
last, err := common.SetMembers(key) |
|||
if err != nil { |
|||
last = []string{} |
|||
} |
|||
lastSet := toSet(last) |
|||
newSet := toSet(codes) |
|||
|
|||
// 新增告警 → active
|
|||
for code := range newSet { |
|||
if !lastSet[code] { |
|||
s.createAlarm(dockID, deviceType, code) |
|||
} |
|||
} |
|||
// 移除告警 → resolved
|
|||
for code := range lastSet { |
|||
if !newSet[code] { |
|||
s.resolveAlarm(dockID, code) |
|||
} |
|||
} |
|||
|
|||
// 更新当前告警码集合
|
|||
_ = common.Delete(key) |
|||
for _, code := range codes { |
|||
_ = common.SetAdd(key, code) |
|||
} |
|||
} |
|||
|
|||
func (s *AlarmService) createAlarm(dockID, deviceType, code string) { |
|||
messageCn, level := code, "critical" |
|||
var ac model.AlarmCode |
|||
if err := common.DB.Where("code = ?", code).First(&ac).Error; err == nil { |
|||
messageCn, level = ac.MessageCn, ac.Level |
|||
} |
|||
|
|||
id, err := tool.NextID() |
|||
if err != nil { |
|||
logger.ERROR("生成告警 ID 失败", err) |
|||
return |
|||
} |
|||
now := time.Now() |
|||
alarm := model.Alarm{ |
|||
ID: id, |
|||
DockID: dockID, |
|||
Code: code, |
|||
DeviceType: deviceType, |
|||
MessageCn: messageCn, |
|||
Level: level, |
|||
Status: "active", |
|||
TriggeredAt: &now, |
|||
CreatedAt: now, |
|||
} |
|||
if err := common.DB.Create(&alarm).Error; err != nil { |
|||
logger.ERROR("写入告警失败", err) |
|||
return |
|||
} |
|||
|
|||
broadcast("alarm.created", map[string]any{ |
|||
"dockId": dockID, |
|||
"alarm": alarm, |
|||
}) |
|||
} |
|||
|
|||
func (s *AlarmService) resolveAlarm(dockID, code string) { |
|||
now := time.Now() |
|||
res := common.DB.Model(&model.Alarm{}).Where("dock_id = ? AND code = ? AND status = 'active'", dockID, code). |
|||
Updates(map[string]any{"status": "resolved", "resolved_at": now}) |
|||
if res.Error != nil { |
|||
logger.ERROR("关闭告警失败", res.Error) |
|||
return |
|||
} |
|||
if res.RowsAffected > 0 { |
|||
broadcast("alarm.resolved", map[string]any{ |
|||
"dockId": dockID, |
|||
"code": code, |
|||
"resolvedAt": now, |
|||
}) |
|||
} |
|||
} |
|||
|
|||
func (s *AlarmService) alarmKey(dockID, deviceType string) string { |
|||
if deviceType == "drone" { |
|||
return cache.DroneAlarmKeyOf(dockID) |
|||
} |
|||
return cache.DockAlarmKeyOf(dockID) |
|||
} |
|||
|
|||
func toSet(items []string) map[string]bool { |
|||
set := make(map[string]bool, len(items)) |
|||
for _, it := range items { |
|||
set[it] = true |
|||
} |
|||
return set |
|||
} |
|||
@ -0,0 +1,478 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"errors" |
|||
"fmt" |
|||
"strconv" |
|||
"strings" |
|||
"time" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
const ( |
|||
gbBytes = 1024 * 1024 * 1024 // 1 GB 字节数
|
|||
defaultTrafficUnitPrice = 10.0 // 云媒体流量单价（元/GB）
|
|||
) |
|||
|
|||
// trafficDeductLua 原子扣减：key 不存在时用 MySQL 快照兜底初始化，再判断并扣减。
|
|||
// 返回 {status, before, after}：status=1 成功，0 余额不足。
|
|||
const trafficDeductLua = ` |
|||
local key = KEYS[1] |
|||
local delta = tonumber(ARGV[1]) |
|||
local fallback = tonumber(ARGV[2]) |
|||
local before = 0 |
|||
if redis.call('EXISTS', key) == 0 then |
|||
redis.call('SET', key, fallback) |
|||
before = fallback |
|||
else |
|||
before = tonumber(redis.call('GET', key)) |
|||
end |
|||
if before < delta then |
|||
return {0, before, before} |
|||
end |
|||
local after = redis.call('DECRBY', key, delta) |
|||
return {1, before, after} |
|||
` |
|||
|
|||
type BillingService struct{} |
|||
|
|||
var DefaultBillingService = &BillingService{} |
|||
|
|||
// CarrierAPI 运营商接口抽象（查询用量 / 充值）
|
|||
type CarrierAPI interface { |
|||
QueryUsage(iccid string) (usedGb float64, carrierStatus string, err error) |
|||
Recharge(iccid string, amountGb int) (orderNo string, err error) |
|||
} |
|||
|
|||
// MockCarrierAPI 运营商 Mock（未接入真实通道前使用）
|
|||
type MockCarrierAPI struct{} |
|||
|
|||
func (m *MockCarrierAPI) QueryUsage(iccid string) (float64, string, error) { |
|||
return 0.0, "normal", nil |
|||
} |
|||
|
|||
func (m *MockCarrierAPI) Recharge(iccid string, amountGb int) (string, error) { |
|||
return fmt.Sprintf("MOCK-%d", time.Now().UnixNano()), nil |
|||
} |
|||
|
|||
// carrier 当前使用的运营商通道（后续可替换为真实实现）
|
|||
var carrier CarrierAPI = &MockCarrierAPI{} |
|||
|
|||
// mysqlBalance 读取用户 MySQL 流量快照
|
|||
func (b *BillingService) mysqlBalance(userID int64) (int64, error) { |
|||
var u model.User |
|||
if err := common.DB.Select("traffic_balance").First(&u, userID).Error; err != nil { |
|||
return 0, err |
|||
} |
|||
return u.TrafficBalance, nil |
|||
} |
|||
|
|||
// GetBalance 查询流量余额（Redis 权威，缺 key 时用 MySQL 快照兜底）
|
|||
func (b *BillingService) GetBalance(userID int64) (*vo.TrafficBalanceVO, *common.BusiError) { |
|||
balance, err := b.currentBalance(userID) |
|||
if err != nil { |
|||
logger.ERROR("查询流量余额失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &vo.TrafficBalanceVO{ |
|||
BalanceBytes: balance, |
|||
BalanceGb: float64(balance) / float64(gbBytes), |
|||
}, nil |
|||
} |
|||
|
|||
// currentBalance 返回 Redis 权威余额；缺 key 时先用 MySQL 快照 SetNX 初始化
|
|||
func (b *BillingService) currentBalance(userID int64) (int64, error) { |
|||
key := cache.TrafficKeyOf(userID) |
|||
bal, err := common.GetInt64(key) |
|||
if err == nil { |
|||
return bal, nil |
|||
} |
|||
fallback, err := b.mysqlBalance(userID) |
|||
if err != nil { |
|||
return 0, err |
|||
} |
|||
if _, err := common.SetNX(key, fallback); err != nil { |
|||
return 0, err |
|||
} |
|||
// SetNX 失败说明并发下已有值，重读一次
|
|||
bal, err = common.GetInt64(key) |
|||
if err != nil { |
|||
return fallback, nil |
|||
} |
|||
return bal, nil |
|||
} |
|||
|
|||
// Deduct 扣减流量（直播/回放/下载计费），原子 Lua + 账本落库
|
|||
func (b *BillingService) Deduct(userID int64, bytes int64, sourceType string, sourceID int64) (bool, *common.BusiError) { |
|||
if bytes <= 0 { |
|||
return true, nil |
|||
} |
|||
fallback, err := b.mysqlBalance(userID) |
|||
if err != nil { |
|||
logger.ERROR("读取流量快照失败", err) |
|||
return false, common.ErrInternal |
|||
} |
|||
key := cache.TrafficKeyOf(userID) |
|||
res, err := common.GetLuaInt64s(trafficDeductLua, []string{key}, bytes, fallback) |
|||
if err != nil { |
|||
logger.ERROR("流量扣减 Lua 执行失败", err) |
|||
return false, common.ErrInternal |
|||
} |
|||
if len(res) < 3 { |
|||
return false, common.ErrInternal |
|||
} |
|||
status, before, after := res[0], res[1], res[2] |
|||
if status == 0 { |
|||
return false, common.ErrTrafficNotEnough |
|||
} |
|||
|
|||
log := &model.TrafficUsageLog{ |
|||
ID: mustID(), |
|||
UserID: userID, |
|||
SourceType: sourceType, |
|||
SourceID: sourceID, |
|||
BytesUsed: bytes, |
|||
BalanceBefore: before, |
|||
BalanceAfter: after, |
|||
CreatedAt: time.Now(), |
|||
} |
|||
if err := common.DB.Create(log).Error; err != nil { |
|||
logger.ERROR("流量账本落库失败", err) |
|||
} |
|||
return true, nil |
|||
} |
|||
|
|||
// credit 充值入账：Redis 权威自增 + MySQL 快照同步
|
|||
func (b *BillingService) credit(userID int64, bytes int64) error { |
|||
key := cache.TrafficKeyOf(userID) |
|||
if _, err := b.currentBalance(userID); err != nil { |
|||
return err |
|||
} |
|||
if _, err := common.IncrBy(key, bytes); err != nil { |
|||
return err |
|||
} |
|||
return common.DB.Model(&model.User{}).Where("id = ?", userID). |
|||
UpdateColumn("traffic_balance", gorm.Expr("traffic_balance + ?", bytes)).Error |
|||
} |
|||
|
|||
// GetUsagePage 流量消费流水分页
|
|||
func (b *BillingService) GetUsagePage(userID int64, req *vo.UsagePageReq) (*common.PageResponse[model.TrafficUsageLog], *common.BusiError) { |
|||
db := common.DB.Model(&model.TrafficUsageLog{}).Where("user_id = ?", userID) |
|||
if req.SourceType != "" { |
|||
db = db.Where("source_type = ?", req.SourceType) |
|||
} |
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.TrafficUsageLog |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// GetOrderPage 流量订单分页
|
|||
func (b *BillingService) GetOrderPage(userID int64, req *vo.OrderPageReq) (*common.PageResponse[model.TrafficOrder], *common.BusiError) { |
|||
db := common.DB.Model(&model.TrafficOrder{}).Where("user_id = ?", userID) |
|||
if req.PayStatus != "" { |
|||
db = db.Where("pay_status = ?", req.PayStatus) |
|||
} |
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.TrafficOrder |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// CreateOrder 创建流量充值订单（待支付）
|
|||
func (b *BillingService) CreateOrder(userID int64, amountGb int) (*model.TrafficOrder, *common.BusiError) { |
|||
order := &model.TrafficOrder{ |
|||
ID: mustID(), |
|||
UserID: userID, |
|||
AmountGb: amountGb, |
|||
UnitPrice: defaultTrafficUnitPrice, |
|||
TotalPrice: float64(amountGb) * defaultTrafficUnitPrice, |
|||
PayStatus: "unpaid", |
|||
CreatedAt: time.Now(), |
|||
} |
|||
if err := common.DB.Create(order).Error; err != nil { |
|||
logger.ERROR("创建流量订单失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return order, nil |
|||
} |
|||
|
|||
// PayOrder 订单支付入账（admin 确认支付）
|
|||
func (b *BillingService) PayOrder(orderID int64) (*model.TrafficOrder, *common.BusiError) { |
|||
var order model.TrafficOrder |
|||
if err := common.DB.First(&order, orderID).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrOrderNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
if order.PayStatus == "paid" { |
|||
return nil, common.ErrOrderPaid |
|||
} |
|||
|
|||
bytes := int64(order.AmountGb) * gbBytes |
|||
now := time.Now() |
|||
if err := common.DB.Transaction(func(tx *gorm.DB) error { |
|||
if err := tx.Model(&order).Updates(map[string]any{ |
|||
"pay_status": "paid", |
|||
"paid_at": now, |
|||
}).Error; err != nil { |
|||
return err |
|||
} |
|||
return tx.Model(&model.User{}).Where("id = ?", order.UserID). |
|||
UpdateColumn("traffic_balance", gorm.Expr("traffic_balance + ?", bytes)).Error |
|||
}); err != nil { |
|||
logger.ERROR("订单支付失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
// Redis 权威余额入账（幂等：多次支付由 pay_status 校验拦截）
|
|||
if err := b.credit(order.UserID, bytes); err != nil { |
|||
logger.ERROR("订单支付 Redis 入账失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
order.PayStatus = "paid" |
|||
order.PaidAt = &now |
|||
return &order, nil |
|||
} |
|||
|
|||
// ListSimCards SIM 卡列表
|
|||
func (b *BillingService) ListSimCards(userID int64, req *vo.SimCardPageReq) (*common.PageResponse[model.SimCard], *common.BusiError) { |
|||
db := common.DB.Model(&model.SimCard{}).Where("user_id = ?", userID) |
|||
if req.Status != "" { |
|||
db = db.Where("status = ?", req.Status) |
|||
} |
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.SimCard |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// GetSimRechargeLogPage SIM 卡充值记录分页
|
|||
func (b *BillingService) GetSimRechargeLogPage(userID int64, req *vo.SimRechargeLogPageReq) (*common.PageResponse[model.SimRechargeLog], *common.BusiError) { |
|||
db := common.DB.Model(&model.SimRechargeLog{}).Where("user_id = ?", userID) |
|||
if req.SimCardID > 0 { |
|||
db = db.Where("sim_card_id = ?", req.SimCardID) |
|||
} |
|||
if req.RechargeStatus != "" { |
|||
db = db.Where("recharge_status = ?", req.RechargeStatus) |
|||
} |
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.SimRechargeLog |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// RechargeSimCard SIM 卡充值（运营商下单 + 回写状态）
|
|||
func (b *BillingService) RechargeSimCard(userID, simCardID int64, amountGb int) (*model.SimRechargeLog, *common.BusiError) { |
|||
var card model.SimCard |
|||
if err := common.DB.Where("id = ? AND user_id = ?", simCardID, userID).First(&card).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrSimCardNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
log := &model.SimRechargeLog{ |
|||
ID: mustID(), |
|||
SimCardID: simCardID, |
|||
UserID: userID, |
|||
AmountGb: amountGb, |
|||
UnitPrice: defaultTrafficUnitPrice, |
|||
TotalPrice: float64(amountGb) * defaultTrafficUnitPrice, |
|||
PayStatus: "paid", |
|||
RechargeStatus: "pending", |
|||
IdempotentKey: fmt.Sprintf("%d-%d", simCardID, time.Now().UnixNano()), |
|||
CreatedAt: time.Now(), |
|||
} |
|||
if err := common.DB.Create(log).Error; err != nil { |
|||
logger.ERROR("创建 SIM 充值记录失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
orderNo, err := carrier.Recharge(card.Iccid, amountGb) |
|||
if err != nil { |
|||
log.RechargeStatus = "failed" |
|||
log.RechargeMsg = err.Error() |
|||
common.DB.Model(log).Updates(map[string]any{"recharge_status": "failed", "recharge_msg": err.Error()}) |
|||
return log, nil |
|||
} |
|||
|
|||
updates := map[string]any{ |
|||
"recharge_status": "success", |
|||
"recharge_msg": orderNo, |
|||
} |
|||
common.DB.Model(log).Updates(updates) |
|||
common.DB.Model(&card).Updates(map[string]any{ |
|||
"plan_gb": gorm.Expr("plan_gb + ?", amountGb), |
|||
"status": "active", |
|||
}) |
|||
log.RechargeStatus = "success" |
|||
log.RechargeMsg = orderNo |
|||
return log, nil |
|||
} |
|||
|
|||
// mustID 生成雪花 ID（失败返回 0，由 DB 约束兜底）
|
|||
func mustID() int64 { |
|||
id, err := tool.NextID() |
|||
if err != nil { |
|||
logger.ERROR("生成雪花 ID 失败", err) |
|||
return 0 |
|||
} |
|||
return id |
|||
} |
|||
|
|||
// chargeLiveSessions 直播计费（定时任务，每 60s）：扫描进行中的会话按码率估算扣减
|
|||
func (b *BillingService) chargeLiveSessions() { |
|||
var sessions []model.LiveSession |
|||
if err := common.DB.Where("phase IN ('starting','streaming')").Find(&sessions).Error; err != nil { |
|||
logger.ERROR("扫描直播会话失败", err) |
|||
return |
|||
} |
|||
now := time.Now() |
|||
for i := range sessions { |
|||
b.chargeLiveSession(&sessions[i], now) |
|||
} |
|||
} |
|||
|
|||
// chargeLiveSession 对单个直播会话按时长增量扣减；余额耗尽则停止推流
|
|||
func (b *BillingService) chargeLiveSession(session *model.LiveSession, now time.Time) { |
|||
var dock model.Dock |
|||
if err := common.DB.Select("user_id").Where("dock_id = ?", session.DockID).First(&dock).Error; err != nil { |
|||
return |
|||
} |
|||
|
|||
lastKey := cache.LiveBilledKeyOf(session.ID) |
|||
lastTs := int64(0) |
|||
if v, err := common.GetInt64(lastKey); err == nil { |
|||
lastTs = v |
|||
} else if session.StartedAt != nil { |
|||
lastTs = session.StartedAt.Unix() |
|||
} else { |
|||
lastTs = session.CreatedAt.Unix() |
|||
} |
|||
|
|||
nowTs := now.Unix() |
|||
if nowTs <= lastTs { |
|||
return |
|||
} |
|||
bytes := (nowTs - lastTs) * session.MaxBitrateBps / 8 |
|||
sourceID, _ := strconv.ParseInt(session.ID, 10, 64) |
|||
|
|||
if bytes > 0 { |
|||
ok, _ := b.Deduct(dock.UserID, bytes, "live", sourceID) |
|||
if !ok { |
|||
if _, busiErr := DefaultLiveService.stopForBilling(session.DockID, session.ID); busiErr != nil { |
|||
logger.ERROR("直播余额不足停止推流失败", busiErr) |
|||
} |
|||
logger.WARN("直播余额耗尽，停止推流:", session.DockID) |
|||
} |
|||
} |
|||
_ = common.SetValue(lastKey, nowTs) |
|||
} |
|||
|
|||
// flushBalanceSnapshot 余额快照刷新（定时任务，每 30min）：Redis 权威余额回写 MySQL
|
|||
func (b *BillingService) flushBalanceSnapshot() { |
|||
keys, err := common.GetKeysWithPrefix(cache.TrafficKeyPrefix) |
|||
if err != nil { |
|||
logger.ERROR("扫描流量余额 key 失败", err) |
|||
return |
|||
} |
|||
for _, key := range keys { |
|||
idStr := strings.TrimPrefix(key, cache.TrafficKeyPrefix) |
|||
userID, err := strconv.ParseInt(idStr, 10, 64) |
|||
if err != nil { |
|||
continue |
|||
} |
|||
bal, err := common.GetInt64(key) |
|||
if err != nil { |
|||
continue |
|||
} |
|||
if err := common.DB.Model(&model.User{}).Where("id = ?", userID). |
|||
UpdateColumn("traffic_balance", bal).Error; err != nil { |
|||
logger.ERROR("回写余额快照失败", err) |
|||
} |
|||
} |
|||
} |
|||
|
|||
// syncSimUsage SIM 卡用量同步（定时任务，每 1h）：运营商查询回填 + 用量快照 + 阈值告警
|
|||
func (b *BillingService) syncSimUsage() { |
|||
var cards []model.SimCard |
|||
if err := common.DB.Where("status = ?", "active").Find(&cards).Error; err != nil { |
|||
logger.ERROR("扫描 SIM 卡失败", err) |
|||
return |
|||
} |
|||
now := time.Now() |
|||
for i := range cards { |
|||
b.syncSimCard(&cards[i], now) |
|||
} |
|||
} |
|||
|
|||
func (b *BillingService) syncSimCard(card *model.SimCard, now time.Time) { |
|||
usedGb, carrierStatus, err := carrier.QueryUsage(card.Iccid) |
|||
if err != nil { |
|||
logger.WARN("查询 SIM 用量失败:", card.Iccid, err) |
|||
return |
|||
} |
|||
|
|||
updates := map[string]any{ |
|||
"used_gb": usedGb, |
|||
"carrier_status": carrierStatus, |
|||
"last_sync_at": now, |
|||
} |
|||
if carrierStatus == "cancelled" { |
|||
updates["status"] = "expired" |
|||
} |
|||
if err := common.DB.Model(&model.SimCard{}).Where("id = ?", card.ID).Updates(updates).Error; err != nil { |
|||
logger.ERROR("回填 SIM 用量失败", err) |
|||
} |
|||
|
|||
remainGb := float64(card.PlanGb) - usedGb |
|||
record := &model.SimUsageRecord{ |
|||
ID: mustID(), |
|||
SimCardID: card.ID, |
|||
PlanGb: card.PlanGb, |
|||
UsedGb: usedGb, |
|||
RemainGb: remainGb, |
|||
CarrierStatus: carrierStatus, |
|||
SyncedAt: now, |
|||
} |
|||
if err := common.DB.Create(record).Error; err != nil { |
|||
logger.ERROR("写入 SIM 用量快照失败", err) |
|||
} |
|||
|
|||
if remainGb < float64(card.PlanGb)*0.1 { |
|||
logger.WARN("SIM 卡流量即将耗尽:", card.Iccid, "剩余", remainGb, "GB") |
|||
} |
|||
if carrierStatus == "suspended" || carrierStatus == "arrears" { |
|||
logger.WARN("SIM 卡运营商状态异常:", card.Iccid, carrierStatus) |
|||
} |
|||
} |
|||
@ -0,0 +1,90 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
) |
|||
|
|||
// BootstrapCasbin 引导默认 RBAC 策略（幂等）：p 策略（角色→路径→方法）+ 用户角色映射 g 策略
|
|||
func BootstrapCasbin() { |
|||
e := common.CasbinEnforcer |
|||
if e == nil { |
|||
return |
|||
} |
|||
|
|||
// 清理旧的 admin 全量策略，防止历史 "admin /* *" 残留导致越权（幂等）
|
|||
if _, err := e.RemoveFilteredPolicy(0, "admin"); err != nil { |
|||
logger.ERROR("清理 admin 旧策略失败", err) |
|||
} |
|||
|
|||
// p 策略：
|
|||
// admin —— 平台管理职能：查看用户/机巢/无人机，管理固件，管理主账号云媒体流量
|
|||
// user —— 业务操作：机巢/无人机/任务/航线等（自注册普通用户）
|
|||
// keyMatch2 下 "/prefix/*" 匹配所有子路径（含多级），再补 "/prefix" 精确匹配列表/新增基路径。
|
|||
adminPolicies := [][]string{ |
|||
// 查看所有用户信息（暂只读）
|
|||
{"admin", "/v1/users", "GET"}, |
|||
{"admin", "/v1/users/*", "GET"}, |
|||
// 查看机巢 / 无人机（含状态与遥测）
|
|||
{"admin", "/v1/docks", "GET"}, |
|||
{"admin", "/v1/docks/*", "GET"}, |
|||
{"admin", "/v1/drones", "GET"}, |
|||
{"admin", "/v1/drones/*", "GET"}, |
|||
// 固件包管理
|
|||
{"admin", "/v1/firmwares", "*"}, |
|||
{"admin", "/v1/firmwares/*", "*"}, |
|||
// 主账号云媒体流量管理（流量余额/订单/SIM 卡）
|
|||
{"admin", "/v1/account/traffic", "*"}, |
|||
{"admin", "/v1/account/traffic/*", "*"}, |
|||
{"admin", "/v1/account/sim-cards", "*"}, |
|||
{"admin", "/v1/account/sim-cards/*", "*"}, |
|||
// 平台监控：角色字典
|
|||
{"admin", "/v1/roles", "GET"}, |
|||
} |
|||
|
|||
userPrefixes := []string{ |
|||
"/v1/docks", |
|||
"/v1/drones", |
|||
"/v1/alarms", |
|||
"/v1/commands", |
|||
"/v1/tasks", |
|||
"/v1/routes", |
|||
"/v1/executions", |
|||
"/v1/account", |
|||
"/v1/live", |
|||
"/v1/videos", |
|||
"/v1/ws", |
|||
} |
|||
|
|||
policies := append([][]string{}, adminPolicies...) |
|||
for _, prefix := range userPrefixes { |
|||
policies = append(policies, []string{"user", prefix, "*"}, []string{"user", prefix + "/*", "*"}) |
|||
} |
|||
// 普通用户可读取已发布固件列表（选择升级）
|
|||
policies = append(policies, []string{"user", "/v1/firmwares/released", "GET"}) |
|||
// 普通用户查看自己的操作记录
|
|||
policies = append(policies, []string{"user", "/v1/logs/operation", "GET"}) |
|||
|
|||
for _, p := range policies { |
|||
args := make([]any, len(p)) |
|||
for i, v := range p { |
|||
args[i] = v |
|||
} |
|||
if _, err := e.AddPolicy(args...); err != nil { |
|||
logger.ERROR("添加 Casbin 策略失败", err) |
|||
} |
|||
} |
|||
|
|||
// g 策略：同步现有用户角色映射
|
|||
var users []model.User |
|||
if err := common.DB.Find(&users).Error; err != nil { |
|||
logger.ERROR("查询用户同步 Casbin 角色失败", err) |
|||
return |
|||
} |
|||
for _, u := range users { |
|||
if err := common.SetUserRole(u.ID, u.Role); err != nil { |
|||
logger.ERROR("同步用户角色失败", err) |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,304 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"errors" |
|||
"fmt" |
|||
"strconv" |
|||
"strings" |
|||
"time" |
|||
|
|||
"github.com/google/uuid" |
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/mqtt" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type CommandService struct{} |
|||
|
|||
var DefaultCommandService = &CommandService{} |
|||
|
|||
const maxRetryCount = 2 |
|||
|
|||
// Dispatch 下发指令：校验在线 → 落库 → MQTT 发布
|
|||
func (s *CommandService) Dispatch(userID int64, isAdmin bool, id int64, req *vo.CommandReq) (*model.DeviceCommandLog, *common.BusiError) { |
|||
var dock model.Dock |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&dock, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrDockNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
dockID := dock.DockID |
|||
|
|||
online, _ := common.SetMemberExists(cache.OnlineDockSetKey, dockID) |
|||
if !online { |
|||
return nil, common.ErrDockOffline |
|||
} |
|||
|
|||
commandID, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
requestID := strings.ReplaceAll(uuid.New().String(), "-", "") |
|||
ttlMs := req.TTLMs |
|||
if ttlMs <= 0 { |
|||
ttlMs = 30000 |
|||
} |
|||
|
|||
paramsJSON := "{}" |
|||
if req.Params != nil { |
|||
if b, e := json.Marshal(req.Params); e == nil { |
|||
paramsJSON = string(b) |
|||
} |
|||
} |
|||
|
|||
now := time.Now() |
|||
cmdLog := &model.DeviceCommandLog{ |
|||
ID: commandID, |
|||
DockID: dockID, |
|||
CommandType: req.Type, |
|||
Params: paramsJSON, |
|||
DroneSN: req.DroneSN, |
|||
RequestID: requestID, |
|||
TTLMs: ttlMs, |
|||
Status: "sent", |
|||
SentAt: &now, |
|||
CreatedAt: now, |
|||
} |
|||
if err := common.DB.Create(cmdLog).Error; err != nil { |
|||
logger.ERROR("写入指令日志失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
if err := s.publish(cmdLog, req.Params); err != nil { |
|||
logger.ERROR("下发指令失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return cmdLog, nil |
|||
} |
|||
|
|||
// HandleAck 处理指令应答，按外层 requestId 更新一次发送尝试。
|
|||
func (s *CommandService) HandleAck(dockID, requestID, commandIDStr string, accepted bool, resultCode string) { |
|||
if requestID == "" { |
|||
logger.WARN("指令应答缺少 requestId", dockID) |
|||
return |
|||
} |
|||
|
|||
var cmd model.DeviceCommandLog |
|||
if err := common.DB.Where("dock_id = ? AND request_id = ? AND status = ?", dockID, requestID, "sent").First(&cmd).Error; err != nil { |
|||
if !errors.Is(err, gorm.ErrRecordNotFound) { |
|||
logger.ERROR("查询指令应答失败", err) |
|||
} |
|||
return |
|||
} |
|||
if commandIDStr != "" && commandIDStr != strconv.FormatInt(cmd.ID, 10) { |
|||
logger.WARN("指令应答 commandId 与 requestId 不匹配", dockID, requestID) |
|||
return |
|||
} |
|||
|
|||
now := time.Now() |
|||
updates := map[string]any{ |
|||
"ack_accepted": accepted, |
|||
"ack_result_code": resultCode, |
|||
"acked_at": now, |
|||
"status": "terminal", |
|||
} |
|||
if accepted { |
|||
updates["status"] = "acked" |
|||
} |
|||
result := common.DB.Model(&model.DeviceCommandLog{}). |
|||
Where("id = ? AND dock_id = ? AND request_id = ? AND status = ?", cmd.ID, dockID, requestID, "sent"). |
|||
Updates(updates) |
|||
if result.Error != nil { |
|||
logger.ERROR("更新指令应答失败", result.Error) |
|||
return |
|||
} |
|||
if result.RowsAffected == 0 { |
|||
return |
|||
} |
|||
if !accepted { |
|||
failTaskExecution(cmd, resultCode) |
|||
} |
|||
DefaultLiveService.OnCommandAck(&cmd, accepted, resultCode) |
|||
} |
|||
|
|||
func failTaskExecution(cmd model.DeviceCommandLog, reason string) { |
|||
if reason == "" { |
|||
reason = "COMMAND_REJECTED" |
|||
} |
|||
_ = common.DB.Model(&model.TaskExecution{}). |
|||
Where("dock_id = ? AND command_id = ? AND status IN ?", cmd.DockID, strconv.FormatInt(cmd.ID, 10), []string{"pending", "running"}). |
|||
Updates(map[string]any{"status": "failed", "end_time": time.Now()}) |
|||
} |
|||
|
|||
// DispatchToDock 持久化并下发已完成权限与在线校验的设备指令。
|
|||
func (s *CommandService) DispatchToDock(dockID, commandType string, params map[string]any) (*model.DeviceCommandLog, error) { |
|||
commandID, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
requestID := strings.ReplaceAll(uuid.New().String(), "-", "") |
|||
paramsJSON, err := json.Marshal(params) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
now := time.Now() |
|||
cmd := &model.DeviceCommandLog{ |
|||
ID: commandID, |
|||
DockID: dockID, |
|||
CommandType: commandType, |
|||
Params: string(paramsJSON), |
|||
RequestID: requestID, |
|||
TTLMs: 30000, |
|||
Status: "sent", |
|||
SentAt: &now, |
|||
CreatedAt: now, |
|||
} |
|||
if err := common.DB.Create(cmd).Error; err != nil { |
|||
return nil, err |
|||
} |
|||
if err := s.publish(cmd, params); err != nil { |
|||
return nil, err |
|||
} |
|||
return cmd, nil |
|||
} |
|||
|
|||
// StartRetryScanner 启动指令超时重试扫描器
|
|||
func (s *CommandService) StartRetryScanner() { |
|||
go func() { |
|||
ticker := time.NewTicker(10 * time.Second) |
|||
defer ticker.Stop() |
|||
for range ticker.C { |
|||
s.retryTimeout() |
|||
} |
|||
}() |
|||
} |
|||
|
|||
func (s *CommandService) retryTimeout() { |
|||
var pending []model.DeviceCommandLog |
|||
if err := common.DB.Where("status = 'sent'").Find(&pending).Error; err != nil { |
|||
return |
|||
} |
|||
now := time.Now() |
|||
for _, cmd := range pending { |
|||
if cmd.SentAt == nil { |
|||
continue |
|||
} |
|||
ttl := int64(cmd.TTLMs) |
|||
if ttl <= 0 { |
|||
ttl = 30000 |
|||
} |
|||
if now.Sub(*cmd.SentAt).Milliseconds() <= ttl { |
|||
continue |
|||
} |
|||
|
|||
if cmd.RetryCount >= maxRetryCount { |
|||
result := common.DB.Model(&model.DeviceCommandLog{}).Where("id = ? AND status = ?", cmd.ID, "sent").Update("status", "timeout") |
|||
if result.Error == nil && result.RowsAffected > 0 { |
|||
failTaskExecution(cmd, "COMMAND_ACK_TIMEOUT") |
|||
} |
|||
continue |
|||
} |
|||
|
|||
// 重试：复用 commandId，换新 requestId（工控机按 commandId 幂等）
|
|||
requestID := strings.ReplaceAll(uuid.New().String(), "-", "") |
|||
_ = common.DB.Model(&model.DeviceCommandLog{}).Where("id = ?", cmd.ID).Updates(map[string]any{ |
|||
"request_id": requestID, |
|||
"retry_count": cmd.RetryCount + 1, |
|||
"sent_at": now, |
|||
}) |
|||
|
|||
var params map[string]any |
|||
if cmd.Params != "" { |
|||
_ = json.Unmarshal([]byte(cmd.Params), ¶ms) |
|||
} |
|||
b, _ := json.Marshal(s.commandEnvelope(&cmd, requestID, params)) |
|||
topic := fmt.Sprintf("dock-edge/v1/dock/%s/command", cmd.DockID) |
|||
_ = mqtt.Publish(topic, 1, false, b) |
|||
} |
|||
|
|||
var acked []model.DeviceCommandLog |
|||
if err := common.DB.Where("status = 'acked'").Find(&acked).Error; err != nil { |
|||
return |
|||
} |
|||
for _, cmd := range acked { |
|||
if cmd.AckedAt == nil { |
|||
continue |
|||
} |
|||
ttl := int64(cmd.TTLMs) |
|||
if ttl <= 0 { |
|||
ttl = 30000 |
|||
} |
|||
if now.Sub(*cmd.AckedAt).Milliseconds() > ttl { |
|||
failTaskExecution(cmd, "WORKFLOW_TIMEOUT") |
|||
} |
|||
} |
|||
} |
|||
|
|||
// commandEnvelope 构造下行指令通用外层（文档 §5.1）
|
|||
func (s *CommandService) commandEnvelope(cmd *model.DeviceCommandLog, requestID string, params map[string]any) *mqtt.Envelope { |
|||
return mqtt.NewEnvelope(requestID, cmd.DockID, cmd.DroneSN, map[string]any{ |
|||
"commandId": strconv.FormatInt(cmd.ID, 10), |
|||
"type": cmd.CommandType, |
|||
"ttlMs": cmd.TTLMs, |
|||
"params": params, |
|||
}) |
|||
} |
|||
|
|||
func (s *CommandService) publish(cmd *model.DeviceCommandLog, params map[string]any) error { |
|||
b, err := json.Marshal(s.commandEnvelope(cmd, cmd.RequestID, params)) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
topic := fmt.Sprintf("dock-edge/v1/dock/%s/command", cmd.DockID) |
|||
return mqtt.Publish(topic, 1, false, b) |
|||
} |
|||
|
|||
// GetPage 指令历史分页列表（按所属机巢过滤）
|
|||
func (s *CommandService) GetPage(userID int64, isAdmin bool, req *vo.CommandPageReq) (*common.PageResponse[model.DeviceCommandLog], *common.BusiError) { |
|||
db := common.DB.Model(&model.DeviceCommandLog{}).Scopes(withDockFilter(userID, isAdmin)) |
|||
if req.Status != "" { |
|||
db = db.Where("status = ?", req.Status) |
|||
} |
|||
if req.CommandType != "" { |
|||
db = db.Where("command_type = ?", req.CommandType) |
|||
} |
|||
if req.DockID != "" { |
|||
db = db.Where("dock_id = ?", req.DockID) |
|||
} |
|||
if req.Keyword != "" { |
|||
kw := "%" + req.Keyword + "%" |
|||
db = db.Where("dock_id LIKE ? OR command_type LIKE ? OR drone_sn LIKE ?", kw, kw, kw) |
|||
} |
|||
|
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
logger.ERROR("统计指令日志失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.DeviceCommandLog |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
logger.ERROR("查询指令日志失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// GetDetail 指令日志详情
|
|||
func (s *CommandService) GetDetail(userID int64, isAdmin bool, id int64) (*model.DeviceCommandLog, *common.BusiError) { |
|||
var cmd model.DeviceCommandLog |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).First(&cmd, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrCommandNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &cmd, nil |
|||
} |
|||
@ -0,0 +1,355 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"errors" |
|||
"time" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type DockService struct{} |
|||
|
|||
var DefaultDockService = &DockService{} |
|||
|
|||
// withUserFilter admin 不过滤，普通用户按 user_id 过滤（GORM scope）
|
|||
func withUserFilter(userID int64, isAdmin bool) func(*gorm.DB) *gorm.DB { |
|||
if isAdmin { |
|||
return func(db *gorm.DB) *gorm.DB { return db } |
|||
} |
|||
return func(db *gorm.DB) *gorm.DB { return db.Where("user_id = ?", userID) } |
|||
} |
|||
|
|||
// withDockFilter admin 不过滤，普通用户按所属机巢过滤（用于无 user_id 的 dock 关联表，如 alarm / command_log）
|
|||
func withDockFilter(userID int64, isAdmin bool) func(*gorm.DB) *gorm.DB { |
|||
if isAdmin { |
|||
return func(db *gorm.DB) *gorm.DB { return db } |
|||
} |
|||
return func(db *gorm.DB) *gorm.DB { |
|||
return db.Where("dock_id IN (SELECT dock_id FROM dock WHERE user_id = ?)", userID) |
|||
} |
|||
} |
|||
|
|||
// GetPage 机巢分页列表
|
|||
func (s *DockService) GetPage(userID int64, isAdmin bool, req *vo.DockPageReq) (*common.PageResponse[model.Dock], *common.BusiError) { |
|||
db := common.DB.Model(&model.Dock{}).Scopes(withUserFilter(userID, isAdmin)) |
|||
if req.Status != "" { |
|||
db = db.Where("status = ?", req.Status) |
|||
} |
|||
if req.RegisterStatus != "" { |
|||
db = db.Where("register_status = ?", req.RegisterStatus) |
|||
} |
|||
if req.Keyword != "" { |
|||
kw := "%" + req.Keyword + "%" |
|||
db = db.Where("name LIKE ? OR dock_id LIKE ? OR code LIKE ?", kw, kw, kw) |
|||
} |
|||
|
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
logger.ERROR("统计机巢失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.Dock |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
logger.ERROR("查询机巢列表失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// GetDetail 机巢详情（含实时状态）
|
|||
func (s *DockService) GetDetail(userID int64, isAdmin bool, id int64) (*vo.DockVO, *common.BusiError) { |
|||
var dock model.Dock |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&dock, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrDockNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
online, realtime := s.realtime(dock.DockID) |
|||
return &vo.DockVO{Dock: &dock, Online: online, Realtime: realtime}, nil |
|||
} |
|||
|
|||
// GetStatus 机巢纯实时状态
|
|||
func (s *DockService) GetStatus(userID int64, isAdmin bool, id int64) (*vo.DockStatusVO, *common.BusiError) { |
|||
var dock model.Dock |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).Select("dock_id").First(&dock, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrDockNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
online, realtime := s.realtime(dock.DockID) |
|||
return &vo.DockStatusVO{DockID: dock.DockID, Online: online, Realtime: realtime}, nil |
|||
} |
|||
|
|||
// Create 手动登记机巢（归属当前用户；admin 可指定 userID）
|
|||
func (s *DockService) Create(userID int64, isAdmin bool, req *vo.DockCreateReq) (*model.Dock, *common.BusiError) { |
|||
owner := userID |
|||
if isAdmin && req.UserID != 0 { |
|||
owner = req.UserID |
|||
} |
|||
|
|||
id, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
now := time.Now() |
|||
dock := &model.Dock{ |
|||
ID: id, |
|||
UserID: owner, |
|||
DockID: req.DockID, |
|||
Name: req.Name, |
|||
Code: req.Code, |
|||
SN: req.SN, |
|||
Iccid: req.Iccid, |
|||
Location: req.Location, |
|||
Longitude: req.Longitude, |
|||
Latitude: req.Latitude, |
|||
Altitude: req.Altitude, |
|||
Status: "offline", |
|||
RegisterStatus: "registered", |
|||
DockIDSource: "manual", |
|||
ProtocolVer: "1.0", |
|||
CreatedAt: now, |
|||
UpdatedAt: now, |
|||
} |
|||
if err := common.DB.Create(dock).Error; err != nil { |
|||
if code, _ := common.ParseError(err); code == 1062 { |
|||
return nil, common.ErrDockExists |
|||
} |
|||
logger.ERROR("登记机巢失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return dock, nil |
|||
} |
|||
|
|||
// Update 更新/认领机巢
|
|||
func (s *DockService) Update(userID int64, isAdmin bool, id int64, req *vo.DockUpdateReq) (*model.Dock, *common.BusiError) { |
|||
var dock model.Dock |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&dock, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrDockNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
updates := map[string]any{"updated_at": time.Now()} |
|||
if req.Name != "" { |
|||
updates["name"] = req.Name |
|||
} |
|||
if req.Code != "" { |
|||
updates["code"] = req.Code |
|||
} |
|||
if req.Location != nil { |
|||
updates["location"] = *req.Location |
|||
} |
|||
if req.Longitude != nil { |
|||
updates["longitude"] = *req.Longitude |
|||
} |
|||
if req.Latitude != nil { |
|||
updates["latitude"] = *req.Latitude |
|||
} |
|||
if req.Altitude != nil { |
|||
updates["altitude"] = *req.Altitude |
|||
} |
|||
|
|||
if isAdmin && req.UserID != 0 { |
|||
updates["user_id"] = req.UserID |
|||
} |
|||
if req.RegisterAs == "registered" { |
|||
updates["register_status"] = "registered" |
|||
if isAdmin && req.UserID != 0 { |
|||
updates["user_id"] = req.UserID |
|||
} else { |
|||
updates["user_id"] = userID // 普通用户认领 → 归属自己
|
|||
} |
|||
} |
|||
|
|||
if err := common.DB.Transaction(func(tx *gorm.DB) error { |
|||
if err := tx.Model(&dock).Updates(updates).Error; err != nil { |
|||
return err |
|||
} |
|||
if owner, ok := updates["user_id"]; ok { |
|||
if err := tx.Model(&model.Drone{}).Where("dock_id = ?", dock.DockID).Update("user_id", owner).Error; err != nil { |
|||
return err |
|||
} |
|||
} |
|||
return nil |
|||
}); err != nil { |
|||
logger.ERROR("更新机巢失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
common.DB.First(&dock, id) |
|||
return &dock, nil |
|||
} |
|||
|
|||
// Delete 硬删除机巢及其关联设备数据
|
|||
func (s *DockService) Delete(userID int64, isAdmin bool, id int64) *common.BusiError { |
|||
var dock model.Dock |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&dock, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return common.ErrDockNotFound |
|||
} |
|||
logger.ERROR("查询待删除机巢失败", err) |
|||
return common.ErrInternal |
|||
} |
|||
|
|||
var taskIDs []string |
|||
var droneSNs []string |
|||
var simCardIDs []int64 |
|||
if err := common.DB.Transaction(func(tx *gorm.DB) error { |
|||
if err := tx.Model(&model.TaskPlan{}).Where("dock_id = ?", dock.DockID).Pluck("id", &taskIDs).Error; err != nil { |
|||
return err |
|||
} |
|||
if err := tx.Model(&model.Drone{}).Where("dock_id = ?", dock.DockID).Pluck("drone_sn", &droneSNs).Error; err != nil { |
|||
return err |
|||
} |
|||
if err := tx.Model(&model.SimCard{}).Where("dock_id = ?", dock.DockID).Pluck("id", &simCardIDs).Error; err != nil { |
|||
return err |
|||
} |
|||
|
|||
if len(droneSNs) > 0 { |
|||
var videoIDs []int64 |
|||
if err := tx.Model(&model.Video{}).Where("drone_sn IN ?", droneSNs).Pluck("id", &videoIDs).Error; err != nil { |
|||
return err |
|||
} |
|||
if len(videoIDs) > 0 { |
|||
if err := tx.Where("video_id IN ?", videoIDs).Delete(&model.DownloadLog{}).Error; err != nil { |
|||
return err |
|||
} |
|||
if err := tx.Where("id IN ?", videoIDs).Delete(&model.Video{}).Error; err != nil { |
|||
return err |
|||
} |
|||
} |
|||
} |
|||
if len(simCardIDs) > 0 { |
|||
if err := tx.Where("sim_card_id IN ?", simCardIDs).Delete(&model.SimUsageRecord{}).Error; err != nil { |
|||
return err |
|||
} |
|||
if err := tx.Where("sim_card_id IN ?", simCardIDs).Delete(&model.SimRechargeLog{}).Error; err != nil { |
|||
return err |
|||
} |
|||
if err := tx.Where("id IN ?", simCardIDs).Delete(&model.SimCard{}).Error; err != nil { |
|||
return err |
|||
} |
|||
} |
|||
for _, item := range []any{ |
|||
&model.Alarm{}, &model.DeviceCommandLog{}, &model.WorkflowState{}, |
|||
&model.TaskExecution{}, &model.TaskPlan{}, &model.LiveSession{}, &model.Drone{}, |
|||
} { |
|||
if err := tx.Where("dock_id = ?", dock.DockID).Delete(item).Error; err != nil { |
|||
return err |
|||
} |
|||
} |
|||
return tx.Delete(&dock).Error |
|||
}); err != nil { |
|||
logger.ERROR("硬删除机巢失败", err) |
|||
return common.ErrInternal |
|||
} |
|||
|
|||
for _, taskID := range taskIDs { |
|||
DefaultScheduler.RemoveTask(taskID) |
|||
} |
|||
_ = common.SetRemove(cache.OnlineDockSetKey, dock.DockID) |
|||
_ = common.Delete(cache.DockHeartbeatKeyOf(dock.DockID)) |
|||
_ = common.Delete(cache.DockStatusKeyOf(dock.DockID)) |
|||
return nil |
|||
} |
|||
|
|||
// realtime 读取机巢实时状态（Redis Hash + 在线集合）
|
|||
func (s *DockService) realtime(dockID string) (bool, map[string]string) { |
|||
realtime, err := common.HashGetAll(cache.DockStatusKeyOf(dockID)) |
|||
if err != nil { |
|||
realtime = map[string]string{} |
|||
} |
|||
online, _ := common.SetMemberExists(cache.OnlineDockSetKey, dockID) |
|||
return online, realtime |
|||
} |
|||
|
|||
// EnsureFromOnline 设备自发现：不存在则登记为 pending，存在则更新在线状态与软件版本
|
|||
func (s *DockService) EnsureFromOnline(dockID string, st statusOnline) { |
|||
online := st.Status == "online" |
|||
status := "offline" |
|||
if online { |
|||
status = "online" |
|||
} |
|||
|
|||
var dock model.Dock |
|||
err := common.DB.Where("dock_id = ?", dockID).First(&dock).Error |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
id, e := tool.NextID() |
|||
if e != nil { |
|||
logger.ERROR("生成机巢 ID 失败", e) |
|||
return |
|||
} |
|||
now := time.Now() |
|||
dock = model.Dock{ |
|||
ID: id, |
|||
UserID: 0, // 待认领
|
|||
DockID: dockID, |
|||
Name: st.Name, |
|||
Code: dockID, |
|||
SN: dockID, |
|||
Location: st.Location, |
|||
Latitude: st.Latitude, |
|||
Longitude: st.Longitude, |
|||
RegisterStatus: "pending", |
|||
DockIDSource: "auto", |
|||
SoftwareVer: st.SoftwareVersion, |
|||
ProtocolVer: st.ProtocolVersion, |
|||
Status: status, |
|||
CreatedAt: now, |
|||
UpdatedAt: now, |
|||
} |
|||
if err := common.DB.Create(&dock).Error; err != nil { |
|||
logger.ERROR("登记机巢失败", err) |
|||
return |
|||
} |
|||
} else if err == nil { |
|||
updates := map[string]any{"status": status, "updated_at": time.Now()} |
|||
if st.SoftwareVersion != "" { |
|||
updates["software_ver"] = st.SoftwareVersion |
|||
} |
|||
if st.ProtocolVersion != "" { |
|||
updates["protocol_ver"] = st.ProtocolVersion |
|||
} |
|||
if dock.Name == "" && st.Name != "" { |
|||
updates["name"] = st.Name |
|||
} |
|||
if dock.Location == "" && st.Location != "" { |
|||
updates["location"] = st.Location |
|||
} |
|||
if dock.Latitude == 0 && st.Latitude != 0 { |
|||
updates["latitude"] = st.Latitude |
|||
} |
|||
if dock.Longitude == 0 && st.Longitude != 0 { |
|||
updates["longitude"] = st.Longitude |
|||
} |
|||
if err := common.DB.Model(&dock).Updates(updates).Error; err != nil { |
|||
logger.ERROR("更新机巢在线状态失败", err) |
|||
} |
|||
} |
|||
|
|||
// 在线集合
|
|||
if online { |
|||
_ = common.SetAdd(cache.OnlineDockSetKey, dockID) |
|||
refreshDeviceHeartbeat(cache.DockHeartbeatKeyOf(dockID)) |
|||
} else { |
|||
_ = common.SetRemove(cache.OnlineDockSetKey, dockID) |
|||
_ = common.Delete(cache.DockHeartbeatKeyOf(dockID)) |
|||
} |
|||
|
|||
broadcast("dock.status", map[string]any{ |
|||
"dockId": dockID, |
|||
"online": online, |
|||
"softwareVer": st.SoftwareVersion, |
|||
"protocolVer": st.ProtocolVersion, |
|||
}) |
|||
} |
|||
@ -0,0 +1,206 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"errors" |
|||
"time" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type DroneService struct{} |
|||
|
|||
var DefaultDroneService = &DroneService{} |
|||
|
|||
// GetPage 无人机分页列表
|
|||
func (s *DroneService) GetPage(userID int64, isAdmin bool, req *vo.DronePageReq) (*common.PageResponse[model.Drone], *common.BusiError) { |
|||
db := common.DB.Model(&model.Drone{}).Scopes(withUserFilter(userID, isAdmin)) |
|||
if req.Status != "" { |
|||
db = db.Where("status = ?", req.Status) |
|||
} |
|||
if req.Keyword != "" { |
|||
kw := "%" + req.Keyword + "%" |
|||
db = db.Where("drone_sn LIKE ? OR name LIKE ? OR code LIKE ? OR model LIKE ?", kw, kw, kw, kw) |
|||
} |
|||
|
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
logger.ERROR("统计无人机失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.Drone |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
logger.ERROR("查询无人机列表失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// GetDetail 无人机详情（含实时遥测）
|
|||
func (s *DroneService) GetDetail(userID int64, isAdmin bool, id int64) (*vo.DroneVO, *common.BusiError) { |
|||
var drone model.Drone |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&drone, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrDroneNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
online, realtime := s.realtime(drone.DockID, drone.DroneSN) |
|||
return &vo.DroneVO{Drone: &drone, Online: online, Realtime: realtime}, nil |
|||
} |
|||
|
|||
// GetTelemetry 无人机实时遥测（当前值，历史曲线走 TDengine，见后续阶段）
|
|||
func (s *DroneService) GetTelemetry(userID int64, isAdmin bool, id int64) (map[string]any, *common.BusiError) { |
|||
var drone model.Drone |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&drone, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrDroneNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
online, realtime := s.realtime(drone.DockID, drone.DroneSN) |
|||
return map[string]any{ |
|||
"droneSn": drone.DroneSN, |
|||
"online": online, |
|||
"realtime": realtime, |
|||
}, nil |
|||
} |
|||
|
|||
// Update 更新无人机基础信息
|
|||
func (s *DroneService) Update(userID int64, isAdmin bool, id int64, req *vo.DroneUpdateReq) (*model.Drone, *common.BusiError) { |
|||
var drone model.Drone |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&drone, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrDroneNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
updates := map[string]any{"updated_at": time.Now()} |
|||
if req.Name != "" { |
|||
updates["name"] = req.Name |
|||
} |
|||
if req.Code != "" { |
|||
updates["code"] = req.Code |
|||
} |
|||
if req.Model != "" { |
|||
updates["model"] = req.Model |
|||
} |
|||
|
|||
if err := common.DB.Model(&drone).Updates(updates).Error; err != nil { |
|||
logger.ERROR("更新无人机失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
common.DB.First(&drone, id) |
|||
return &drone, nil |
|||
} |
|||
|
|||
// Delete 删除无人机
|
|||
func (s *DroneService) Delete(userID int64, isAdmin bool, id int64) *common.BusiError { |
|||
res := common.DB.Scopes(withUserFilter(userID, isAdmin)).Delete(&model.Drone{}, id) |
|||
if res.Error != nil { |
|||
logger.ERROR("删除无人机失败", res.Error) |
|||
return common.ErrInternal |
|||
} |
|||
if res.RowsAffected == 0 { |
|||
return common.ErrDroneNotFound |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
// realtime 读取无人机实时遥测（Redis Hash，按 dock_id 存储）+ 在线集合
|
|||
func (s *DroneService) realtime(dockID, droneSN string) (bool, map[string]string) { |
|||
realtime, err := common.HashGetAll(cache.DroneTelemetryKeyOf(dockID)) |
|||
if err != nil { |
|||
realtime = map[string]string{} |
|||
} |
|||
online, _ := common.SetMemberExists(cache.OnlineDroneSetKey, droneSN) |
|||
return online, realtime |
|||
} |
|||
|
|||
// EnsureFromState 无人机自动关联：按 droneSn 或 dockId 创建/更新（1:1 绑定）
|
|||
func (s *DroneService) EnsureFromState(dockID, droneSN, name string, online bool, battery int, firmwareVer string) { |
|||
status := "offline" |
|||
if online { |
|||
status = "online" |
|||
} |
|||
if droneSN == "" { |
|||
return |
|||
} |
|||
|
|||
var dock model.Dock |
|||
if err := common.DB.Select("user_id").Where("dock_id = ?", dockID).First(&dock).Error; err != nil { |
|||
if !errors.Is(err, gorm.ErrRecordNotFound) { |
|||
logger.ERROR("查询无人机关联机巢失败", err) |
|||
} |
|||
return |
|||
} |
|||
|
|||
var drone model.Drone |
|||
err := common.DB.Where("drone_sn = ? OR dock_id = ?", droneSN, dockID).First(&drone).Error |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
id, e := tool.NextID() |
|||
if e != nil { |
|||
logger.ERROR("生成无人机 ID 失败", e) |
|||
return |
|||
} |
|||
now := time.Now() |
|||
drone = model.Drone{ |
|||
ID: id, |
|||
UserID: dock.UserID, |
|||
DroneSN: droneSN, |
|||
DockID: dockID, |
|||
Name: name, |
|||
Status: status, |
|||
Battery: battery, |
|||
FirmwareVer: firmwareVer, |
|||
CreatedAt: now, |
|||
UpdatedAt: now, |
|||
} |
|||
if err := common.DB.Create(&drone).Error; err != nil { |
|||
logger.ERROR("创建无人机失败", err) |
|||
return |
|||
} |
|||
} else if err == nil { |
|||
updates := map[string]any{ |
|||
"user_id": dock.UserID, |
|||
"drone_sn": droneSN, |
|||
"dock_id": dockID, |
|||
"status": status, |
|||
"battery": battery, |
|||
"updated_at": time.Now(), |
|||
} |
|||
if firmwareVer != "" { |
|||
updates["firmware_ver"] = firmwareVer |
|||
} |
|||
if name != "" { |
|||
updates["name"] = name |
|||
} |
|||
if err := common.DB.Model(&drone).Updates(updates).Error; err != nil { |
|||
logger.ERROR("更新无人机失败", err) |
|||
} |
|||
} |
|||
|
|||
// 在线集合
|
|||
if online { |
|||
_ = common.SetAdd(cache.OnlineDroneSetKey, droneSN) |
|||
refreshDeviceHeartbeat(cache.DroneHeartbeatKeyOf(droneSN)) |
|||
} else { |
|||
_ = common.SetRemove(cache.OnlineDroneSetKey, droneSN) |
|||
_ = common.Delete(cache.DroneHeartbeatKeyOf(droneSN)) |
|||
} |
|||
|
|||
broadcast("drone.status", map[string]any{ |
|||
"dockId": dockID, |
|||
"droneSn": droneSN, |
|||
"online": online, |
|||
"battery": battery, |
|||
"firmwareVer": firmwareVer, |
|||
}) |
|||
} |
|||
@ -0,0 +1,75 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"errors" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type ExecutionService struct{} |
|||
|
|||
var DefaultExecutionService = &ExecutionService{} |
|||
|
|||
// GetPage 执行记录分页列表（按所属机巢过滤）
|
|||
func (s *ExecutionService) GetPage(userID int64, isAdmin bool, req *vo.ExecutionPageReq) (*common.PageResponse[model.TaskExecution], *common.BusiError) { |
|||
db := common.DB.Model(&model.TaskExecution{}).Scopes(withDockFilter(userID, isAdmin)) |
|||
if req.Status != "" { |
|||
db = db.Where("status = ?", req.Status) |
|||
} |
|||
if req.TaskID != "" { |
|||
db = db.Where("task_id = ?", req.TaskID) |
|||
} |
|||
if req.DockID != "" { |
|||
db = db.Where("dock_id = ?", req.DockID) |
|||
} |
|||
|
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
logger.ERROR("统计执行记录失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.TaskExecution |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
logger.ERROR("查询执行记录失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// GetDetail 执行记录详情
|
|||
func (s *ExecutionService) GetDetail(userID int64, isAdmin bool, id int64) (*model.TaskExecution, *common.BusiError) { |
|||
var exec model.TaskExecution |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).First(&exec, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrExecutionNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &exec, nil |
|||
} |
|||
|
|||
// GetTrajectory 执行轨迹（解析 trajectory_json）
|
|||
func (s *ExecutionService) GetTrajectory(userID int64, isAdmin bool, id int64) (*vo.TrajectoryVO, *common.BusiError) { |
|||
var exec model.TaskExecution |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).First(&exec, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrExecutionNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
points := []vo.TrajectoryPoint{} |
|||
if len(exec.TrajectoryJSON) > 0 { |
|||
if err := json.Unmarshal([]byte(exec.TrajectoryJSON), &points); err != nil { |
|||
logger.ERROR("解析轨迹失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
} |
|||
return &vo.TrajectoryVO{ExecutionID: exec.ID, DockID: exec.DockID, Points: points}, nil |
|||
} |
|||
@ -0,0 +1,220 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"errors" |
|||
"fmt" |
|||
"strconv" |
|||
"time" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/mqtt" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type FirmwareService struct{} |
|||
|
|||
var DefaultFirmwareService = &FirmwareService{} |
|||
|
|||
const defaultFirmwareComponent = "dock-edge-agent" |
|||
|
|||
// GetPage 固件分页列表（admin）
|
|||
func (s *FirmwareService) GetPage(req *vo.FirmwarePageReq) (*common.PageResponse[model.Firmware], *common.BusiError) { |
|||
db := common.DB.Model(&model.Firmware{}) |
|||
if req.Status != "" { |
|||
db = db.Where("status = ?", req.Status) |
|||
} |
|||
if req.Component != "" { |
|||
db = db.Where("component = ?", req.Component) |
|||
} |
|||
if req.Keyword != "" { |
|||
kw := "%" + req.Keyword + "%" |
|||
db = db.Where("version LIKE ? OR description LIKE ?", kw, kw) |
|||
} |
|||
|
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
logger.ERROR("统计固件失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.Firmware |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
logger.ERROR("查询固件列表失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// GetReleasedList 已发布固件列表（供普通用户选择升级）
|
|||
func (s *FirmwareService) GetReleasedList() ([]model.Firmware, *common.BusiError) { |
|||
var list []model.Firmware |
|||
if err := common.DB.Where("status = ?", "released").Order("id DESC").Find(&list).Error; err != nil { |
|||
logger.ERROR("查询已发布固件失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return list, nil |
|||
} |
|||
|
|||
// GetDetail 固件详情（admin）
|
|||
func (s *FirmwareService) GetDetail(id int64) (*model.Firmware, *common.BusiError) { |
|||
var fw model.Firmware |
|||
if err := common.DB.First(&fw, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrFirmwareNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &fw, nil |
|||
} |
|||
|
|||
// Create 新增固件版本（admin）
|
|||
func (s *FirmwareService) Create(adminID int64, req *vo.FirmwareCreateReq) (*model.Firmware, *common.BusiError) { |
|||
id, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
component := req.Component |
|||
if component == "" { |
|||
component = defaultFirmwareComponent |
|||
} |
|||
now := time.Now() |
|||
fw := &model.Firmware{ |
|||
ID: id, |
|||
Component: component, |
|||
Version: req.Version, |
|||
Description: req.Description, |
|||
FileURL: req.FileURL, |
|||
Sha256: req.Sha256, |
|||
Signature: req.Signature, |
|||
FileSize: req.FileSize, |
|||
Mandatory: req.Mandatory, |
|||
Status: "draft", |
|||
CreatedBy: adminID, |
|||
CreatedAt: now, |
|||
UpdatedAt: now, |
|||
} |
|||
if err := common.DB.Create(fw).Error; err != nil { |
|||
if code, _ := common.ParseError(err); code == 1062 { |
|||
return nil, common.ErrFirmwareVersionExist |
|||
} |
|||
logger.ERROR("新增固件失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return fw, nil |
|||
} |
|||
|
|||
// Update 编辑固件（admin）
|
|||
func (s *FirmwareService) Update(id int64, req *vo.FirmwareUpdateReq) *common.BusiError { |
|||
var fw model.Firmware |
|||
if err := common.DB.First(&fw, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return common.ErrFirmwareNotFound |
|||
} |
|||
return common.ErrInternal |
|||
} |
|||
|
|||
updates := map[string]any{"updated_at": time.Now()} |
|||
if req.Description != "" { |
|||
updates["description"] = req.Description |
|||
} |
|||
if req.FileURL != "" { |
|||
updates["file_url"] = req.FileURL |
|||
} |
|||
if req.Sha256 != "" { |
|||
updates["sha256"] = req.Sha256 |
|||
} |
|||
if req.Signature != "" { |
|||
updates["signature"] = req.Signature |
|||
} |
|||
if req.FileSize != 0 { |
|||
updates["file_size"] = req.FileSize |
|||
} |
|||
if req.Mandatory != nil { |
|||
updates["mandatory"] = *req.Mandatory |
|||
} |
|||
if req.Status != "" { |
|||
updates["status"] = req.Status |
|||
} |
|||
|
|||
if err := common.DB.Model(&fw).Updates(updates).Error; err != nil { |
|||
logger.ERROR("编辑固件失败", err) |
|||
return common.ErrInternal |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
// Delete 删除固件（admin）
|
|||
func (s *FirmwareService) Delete(id int64) *common.BusiError { |
|||
res := common.DB.Delete(&model.Firmware{}, id) |
|||
if res.Error != nil { |
|||
logger.ERROR("删除固件失败", res.Error) |
|||
return common.ErrInternal |
|||
} |
|||
if res.RowsAffected == 0 { |
|||
return common.ErrFirmwareNotFound |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
// Upgrade 下发 OTA 升级：校验机巢归属+在线 → 校验固件已发布 → MQTT ota/desired
|
|||
func (s *FirmwareService) Upgrade(userID int64, isAdmin bool, dockID int64, firmwareID int64) (*vo.FirmwareUpgradeVO, *common.BusiError) { |
|||
var dock model.Dock |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&dock, dockID).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrDockNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
online, _ := common.SetMemberExists(cache.OnlineDockSetKey, dock.DockID) |
|||
if !online { |
|||
return nil, common.ErrDockOffline |
|||
} |
|||
|
|||
var fw model.Firmware |
|||
if err := common.DB.First(&fw, firmwareID).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrFirmwareNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
if fw.Status != "released" { |
|||
return nil, common.ErrFirmwareNotReleased |
|||
} |
|||
|
|||
updateID, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
payload := map[string]any{ |
|||
"updateId": strconv.FormatInt(updateID, 10), |
|||
"component": fw.Component, |
|||
"version": fw.Version, |
|||
"url": fw.FileURL, |
|||
"sha256": fw.Sha256, |
|||
"signature": fw.Signature, |
|||
"mandatory": fw.Mandatory == 1, |
|||
"issuedAt": time.Now().UnixMilli(), |
|||
} |
|||
b, err := json.Marshal(payload) |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
topic := fmt.Sprintf("dock-edge/v1/dock/%s/ota/desired", dock.DockID) |
|||
if err := mqtt.Publish(topic, 1, true, b); err != nil { |
|||
logger.ERROR("下发 OTA 升级失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
return &vo.FirmwareUpgradeVO{ |
|||
UpdateID: strconv.FormatInt(updateID, 10), |
|||
DockID: dock.DockID, |
|||
Firmware: &fw, |
|||
}, nil |
|||
} |
|||
@ -0,0 +1,90 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"time" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
) |
|||
|
|||
func heartbeatTimeout() uint { |
|||
if common.AppConf != nil && common.AppConf.Heartbeat.TimeoutSeconds > 0 { |
|||
return uint(common.AppConf.Heartbeat.TimeoutSeconds) |
|||
} |
|||
return 30 |
|||
} |
|||
|
|||
func refreshDeviceHeartbeat(key string) { |
|||
if err := common.SetValueWithExpired(key, "1", heartbeatTimeout()); err != nil { |
|||
logger.WARN("刷新设备心跳失败:", err) |
|||
} |
|||
} |
|||
|
|||
func expireDeviceHeartbeats() { |
|||
dockMembers, err := common.SetMembers(cache.OnlineDockSetKey) |
|||
if err != nil { |
|||
logger.WARN("扫描机巢在线状态失败:", err) |
|||
} else { |
|||
for _, dockID := range dockMembers { |
|||
removed, e := common.SetRemoveIfKeyMissing(cache.OnlineDockSetKey, cache.DockHeartbeatKeyOf(dockID), dockID) |
|||
if e != nil { |
|||
logger.WARN("检查机巢心跳失败:", e) |
|||
continue |
|||
} |
|||
if removed { |
|||
markDockOffline(dockID) |
|||
} |
|||
} |
|||
} |
|||
|
|||
droneMembers, err := common.SetMembers(cache.OnlineDroneSetKey) |
|||
if err != nil { |
|||
logger.WARN("扫描无人机在线状态失败:", err) |
|||
return |
|||
} |
|||
for _, droneSN := range droneMembers { |
|||
removed, e := common.SetRemoveIfKeyMissing(cache.OnlineDroneSetKey, cache.DroneHeartbeatKeyOf(droneSN), droneSN) |
|||
if e != nil { |
|||
logger.WARN("检查无人机心跳失败:", e) |
|||
continue |
|||
} |
|||
if removed { |
|||
markDroneOffline(droneSN) |
|||
} |
|||
} |
|||
} |
|||
|
|||
func markDockOffline(dockID string) { |
|||
result := common.DB.Model(&model.Dock{}). |
|||
Where("dock_id = ? AND status = ?", dockID, "online"). |
|||
Update("status", "offline") |
|||
if result.Error != nil { |
|||
logger.WARN("更新机巢离线状态失败:", result.Error) |
|||
return |
|||
} |
|||
if result.RowsAffected > 0 { |
|||
broadcast("dock.status", map[string]any{"dockId": dockID, "online": false}) |
|||
} |
|||
} |
|||
|
|||
func markDroneOffline(droneSN string) { |
|||
result := common.DB.Model(&model.Drone{}). |
|||
Where("drone_sn = ? AND status = ?", droneSN, "online"). |
|||
Update("status", "offline") |
|||
if result.Error != nil { |
|||
logger.WARN("更新无人机离线状态失败:", result.Error) |
|||
return |
|||
} |
|||
if result.RowsAffected > 0 { |
|||
broadcast("drone.status", map[string]any{"droneSn": droneSN, "online": false}) |
|||
} |
|||
} |
|||
|
|||
func heartbeatScanInterval() time.Duration { |
|||
if common.AppConf != nil && common.AppConf.Heartbeat.ScanSeconds > 0 { |
|||
return time.Duration(common.AppConf.Heartbeat.ScanSeconds) * time.Second |
|||
} |
|||
return 5 * time.Second |
|||
} |
|||
@ -0,0 +1,393 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"errors" |
|||
"strconv" |
|||
"time" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
liveprovider "laic-backend/live" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type LiveService struct { |
|||
provider liveprovider.Adapter |
|||
} |
|||
|
|||
var DefaultLiveService = &LiveService{} |
|||
|
|||
func InitLiveProvider(conf common.Live) error { |
|||
provider, err := liveprovider.NewAdapter(conf) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
DefaultLiveService.provider = provider |
|||
return nil |
|||
} |
|||
|
|||
func (s *LiveService) adapter() (liveprovider.Adapter, *common.BusiError) { |
|||
if s.provider == nil || s.provider.Provider() == "disabled" { |
|||
return nil, common.NewBusiError(common.LiveProviderDisabled, "直播服务未配置") |
|||
} |
|||
return s.provider, nil |
|||
} |
|||
|
|||
func (s *LiveService) Join(userID int64, isAdmin bool, dockID string, req *vo.LiveSessionCreateReq) (*vo.LiveSessionVO, *common.BusiError) { |
|||
provider, busiErr := s.adapter() |
|||
if busiErr != nil { |
|||
return nil, busiErr |
|||
} |
|||
var dock model.Dock |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).Where("dock_id = ?", dockID).First(&dock).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrDockNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
online, _ := common.SetMemberExists(cache.OnlineDockSetKey, dockID) |
|||
if !online { |
|||
return nil, common.ErrDockOffline |
|||
} |
|||
|
|||
var session model.LiveSession |
|||
var lease model.LiveViewerLease |
|||
var startParams map[string]any |
|||
err := common.DB.Transaction(func(tx *gorm.DB) error { |
|||
if err := tx.Set("gorm:query_option", "FOR UPDATE").Where("dock_id = ?", dockID).First(&model.Dock{}).Error; err != nil { |
|||
return err |
|||
} |
|||
if err := tx.Where("dock_id = ? AND phase IN ('starting','streaming','reconnecting','stopping')", dockID).Order("created_at DESC").First(&session).Error; err != nil && !errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return err |
|||
} |
|||
if session.ID == "" { |
|||
maxBitrate := req.MaxBitrateBps |
|||
if maxBitrate <= 0 { |
|||
maxBitrate = 1500000 |
|||
} |
|||
streamID, err := tool.NextID() |
|||
if err != nil { |
|||
return err |
|||
} |
|||
expireSec := common.AppConf.Live.AuthExpireSeconds |
|||
if expireSec <= 0 { |
|||
expireSec = 7200 |
|||
} |
|||
expiresAt := time.Now().Add(time.Duration(expireSec) * time.Second) |
|||
streamName := strconv.FormatInt(streamID, 10) |
|||
credentials, err := provider.CreateStream(liveprovider.StreamRequest{StreamName: streamName, ExpiresAt: expiresAt, MaxBitrateBps: maxBitrate}) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
now := time.Now() |
|||
session = model.LiveSession{ID: streamName, DockID: dockID, Provider: credentials.Provider, StreamName: streamName, ExpiresAt: expiresAt.Unix(), MaxBitrateBps: maxBitrate, Phase: "starting", RequestedBy: userID, CreatedAt: now, UpdatedAt: now} |
|||
if err := tx.Create(&session).Error; err != nil { |
|||
return err |
|||
} |
|||
startParams = map[string]any{"provider": credentials.Provider, "streamSessionId": session.ID, "pushUrl": credentials.PushURL, "expiresAt": credentials.ExpiresAt.UnixMilli(), "maxBitrateBps": maxBitrate} |
|||
} |
|||
now := time.Now() |
|||
leaseID, err := tool.NextID() |
|||
if err != nil { |
|||
return err |
|||
} |
|||
lease = model.LiveViewerLease{ID: leaseID, StreamSessionID: session.ID, ViewerID: userID, ExpiresAt: now.Add(time.Duration(leaseSeconds()) * time.Second), CreatedAt: now, UpdatedAt: now} |
|||
return tx.Where("stream_session_id = ? AND viewer_id = ?", session.ID, userID).Assign(map[string]any{"expires_at": lease.ExpiresAt, "released_at": nil, "updated_at": now}).FirstOrCreate(&lease).Error |
|||
}) |
|||
if err != nil { |
|||
logger.ERROR("创建直播观看租约失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
if startParams != nil { |
|||
if _, err := DefaultCommandService.DispatchToDock(dockID, "video.start_stream", startParams); err != nil { |
|||
logger.ERROR("下发推流指令失败", err) |
|||
_ = common.DB.Model(&model.LiveSession{}).Where("id = ?", session.ID).Updates(map[string]any{"phase": "failed", "error_code": "MQTT_PUBLISH_FAILED"}).Error |
|||
return nil, common.ErrInternal |
|||
} |
|||
} |
|||
return &vo.LiveSessionVO{Session: &session, LeaseExpiresAt: lease.ExpiresAt.Unix()}, nil |
|||
} |
|||
|
|||
func leaseSeconds() int { |
|||
seconds := common.AppConf.Live.ViewerLeaseSeconds |
|||
if seconds <= 0 { |
|||
return 30 |
|||
} |
|||
return seconds |
|||
} |
|||
|
|||
func (s *LiveService) GetSession(userID int64, isAdmin bool, dockID, streamSessionID string) (*model.LiveSession, *common.BusiError) { |
|||
var session model.LiveSession |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).Where("id = ? AND dock_id = ?", streamSessionID, dockID).First(&session).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrLiveNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &session, nil |
|||
} |
|||
func (s *LiveService) Heartbeat(userID int64, isAdmin bool, dockID, streamSessionID string) (*vo.LiveSessionVO, *common.BusiError) { |
|||
var session model.LiveSession |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).Where("id = ? AND dock_id = ?", streamSessionID, dockID).First(&session).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrLiveNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
expiresAt := time.Now().Add(time.Duration(leaseSeconds()) * time.Second) |
|||
result := common.DB.Model(&model.LiveViewerLease{}).Where("stream_session_id = ? AND viewer_id = ? AND released_at IS NULL", streamSessionID, userID).Updates(map[string]any{"expires_at": expiresAt, "updated_at": time.Now()}) |
|||
if result.Error != nil || result.RowsAffected == 0 { |
|||
return nil, common.ErrLiveNotFound |
|||
} |
|||
return &vo.LiveSessionVO{Session: &session, LeaseExpiresAt: expiresAt.Unix()}, nil |
|||
} |
|||
|
|||
func (s *LiveService) Leave(userID int64, isAdmin bool, dockID, streamSessionID string) *common.BusiError { |
|||
var session model.LiveSession |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).Where("id = ? AND dock_id = ?", streamSessionID, dockID).First(&session).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return common.ErrLiveNotFound |
|||
} |
|||
return common.ErrInternal |
|||
} |
|||
now := time.Now() |
|||
if err := common.DB.Model(&model.LiveViewerLease{}).Where("stream_session_id = ? AND viewer_id = ? AND released_at IS NULL", streamSessionID, userID).Updates(map[string]any{"released_at": now, "updated_at": now}).Error; err != nil { |
|||
return common.ErrInternal |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func stopGraceSeconds() time.Duration { |
|||
seconds := common.AppConf.Live.StopGraceSeconds |
|||
if seconds <= 0 { |
|||
seconds = 15 |
|||
} |
|||
return time.Duration(seconds) * time.Second |
|||
} |
|||
|
|||
// ReconcileLeases expires stale viewer leases and stops unviewed sessions after the grace period.
|
|||
func (s *LiveService) ReconcileLeases() { |
|||
now := time.Now() |
|||
if err := common.DB.Model(&model.LiveViewerLease{}).Where("released_at IS NULL AND expires_at <= ?", now).Updates(map[string]any{"released_at": now, "updated_at": now}).Error; err != nil { |
|||
logger.ERROR("清理过期直播租约失败", err) |
|||
return |
|||
} |
|||
|
|||
var active []model.LiveSession |
|||
if err := common.DB.Where("phase IN ('starting','streaming','reconnecting')").Find(&active).Error; err != nil { |
|||
logger.ERROR("扫描活动直播会话失败", err) |
|||
return |
|||
} |
|||
for _, session := range active { |
|||
if session.ExpiresAt > 0 && now.Unix() >= session.ExpiresAt { |
|||
if _, busiErr := s.stopForBilling(session.DockID, session.ID); busiErr != nil { |
|||
logger.ERROR("直播会话到期停止失败: "+session.ID, busiErr) |
|||
} |
|||
continue |
|||
} |
|||
var viewers int64 |
|||
if err := common.DB.Model(&model.LiveViewerLease{}).Where("stream_session_id = ? AND released_at IS NULL AND expires_at > ?", session.ID, now).Count(&viewers).Error; err != nil || viewers > 0 { |
|||
if session.StopDeadline != nil { |
|||
_ = common.DB.Model(&model.LiveSession{}).Where("id = ?", session.ID).Update("stop_deadline", nil).Error |
|||
} |
|||
continue |
|||
} |
|||
if session.StopDeadline == nil { |
|||
_ = common.DB.Model(&model.LiveSession{}).Where("id = ? AND stop_deadline IS NULL", session.ID).Update("stop_deadline", now.Add(stopGraceSeconds())).Error |
|||
continue |
|||
} |
|||
if session.StopDeadline.After(now) { |
|||
continue |
|||
} |
|||
if _, busiErr := s.stopForBilling(session.DockID, session.ID); busiErr != nil { |
|||
logger.ERROR("无观看者停止直播失败: "+session.ID, busiErr) |
|||
} |
|||
} |
|||
} |
|||
|
|||
// Start 兼容旧启动接口，统一复用创建或加入会话的租约流程。
|
|||
func (s *LiveService) Start(userID int64, isAdmin bool, dockID string, req *vo.LiveStartReq) (*model.LiveSession, *common.BusiError) { |
|||
result, busiErr := s.Join(userID, isAdmin, dockID, &vo.LiveSessionCreateReq{MaxBitrateBps: req.MaxBitrateBps}) |
|||
if busiErr != nil { |
|||
return nil, busiErr |
|||
} |
|||
return result.Session, nil |
|||
} |
|||
|
|||
// Stop 停止指定机巢最新活动会话；设备必须根据 streamSessionId 防止误停新会话。
|
|||
func (s *LiveService) Stop(userID int64, isAdmin bool, dockID string) *common.BusiError { |
|||
var dock model.Dock |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).Where("dock_id = ?", dockID).First(&dock).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return common.ErrDockNotFound |
|||
} |
|||
return common.ErrInternal |
|||
} |
|||
var active model.LiveSession |
|||
if err := common.DB.Where("dock_id = ? AND phase IN ('starting','streaming','reconnecting','stopping')", dockID).Order("id DESC").First(&active).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return common.ErrLiveNotFound |
|||
} |
|||
return common.ErrInternal |
|||
} |
|||
params := map[string]any{ |
|||
"streamSessionId": active.ID, |
|||
"reason": "user_request", |
|||
} |
|||
if _, err := DefaultCommandService.DispatchToDock(dockID, "video.stop_stream", params); err != nil { |
|||
logger.ERROR("下发停止推流指令失败", err) |
|||
return common.ErrInternal |
|||
} |
|||
if err := common.DB.Model(&model.LiveSession{}).Where("id = ? AND phase IN ('starting','streaming','reconnecting')", active.ID).Updates(map[string]any{"phase": "stopping", "stop_reason": "manual"}).Error; err != nil { |
|||
return common.ErrInternal |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (s *LiveService) stopForBilling(dockID, streamSessionID string) (*model.DeviceCommandLog, *common.BusiError) { |
|||
params := map[string]any{ |
|||
"streamSessionId": streamSessionID, |
|||
"reason": "no_balance", |
|||
} |
|||
cmd, err := DefaultCommandService.DispatchToDock(dockID, "video.stop_stream", params) |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
if err := common.DB.Model(&model.LiveSession{}).Where("id = ? AND dock_id = ? AND phase IN ('starting','streaming','reconnecting')", streamSessionID, dockID).Updates(map[string]any{"phase": "stopping", "stop_reason": "no_balance"}).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
return cmd, nil |
|||
} |
|||
func (s *LiveService) GetPage(userID int64, isAdmin bool, req *vo.LivePageReq) (*common.PageResponse[model.LiveSession], *common.BusiError) { |
|||
db := common.DB.Model(&model.LiveSession{}).Scopes(withDockFilter(userID, isAdmin)) |
|||
if req.Phase != "" { |
|||
db = db.Where("phase = ?", req.Phase) |
|||
} |
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.LiveSession |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// GetPlayURL only returns a URL after both edge and provider confirmation are represented by streaming.
|
|||
func (s *LiveService) GetPlayURL(userID int64, isAdmin bool, dockID string) (*vo.LivePlayURLVO, *common.BusiError) { |
|||
provider, busiErr := s.adapter() |
|||
if busiErr != nil { |
|||
return nil, busiErr |
|||
} |
|||
var session model.LiveSession |
|||
if err := common.DB.Scopes(withDockFilter(userID, isAdmin)).Where("dock_id = ? AND phase = 'streaming'", dockID).Order("id DESC").First(&session).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrLiveNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
var lease model.LiveViewerLease |
|||
if err := common.DB.Where("stream_session_id = ? AND viewer_id = ? AND released_at IS NULL AND expires_at > ?", session.ID, userID, time.Now()).First(&lease).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrLiveLeaseNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
ttl := common.AppConf.Live.PlayURLTTLSeconds |
|||
if ttl <= 0 || ttl > 300 { |
|||
ttl = 300 |
|||
} |
|||
urls, err := provider.CreatePlayURLs(liveprovider.PlayRequest{StreamName: session.StreamName, ExpiresAt: time.Now().Add(time.Duration(ttl) * time.Second)}) |
|||
if err != nil { |
|||
return nil, common.NewBusiError(common.LiveProviderInvalid, "生成播放地址失败") |
|||
} |
|||
return &vo.LivePlayURLVO{StreamName: session.StreamName, PlayURL: urls.HLS, ExpiresAt: urls.ExpiresAt.Unix()}, nil |
|||
} |
|||
|
|||
func (s *LiveService) OnCommandAck(cmd *model.DeviceCommandLog, accepted bool, resultCode string) { |
|||
if cmd.CommandType != "video.start_stream" && cmd.CommandType != "video.stop_stream" { |
|||
return |
|||
} |
|||
var params struct { |
|||
StreamSessionID string `json:"streamSessionId"` |
|||
} |
|||
if err := json.Unmarshal([]byte(cmd.Params), ¶ms); err != nil || params.StreamSessionID == "" { |
|||
logger.ERROR("解析直播指令参数失败", err) |
|||
return |
|||
} |
|||
if accepted { |
|||
return |
|||
} |
|||
|
|||
updates := map[string]any{"error_code": resultCode} |
|||
if cmd.CommandType == "video.start_stream" { |
|||
updates["phase"] = "failed" |
|||
} else { |
|||
updates["phase"] = "streaming" |
|||
} |
|||
if err := common.DB.Model(&model.LiveSession{}). |
|||
Where("id = ? AND dock_id = ?", params.StreamSessionID, cmd.DockID). |
|||
Updates(updates).Error; err != nil { |
|||
logger.ERROR("更新直播指令失败状态失败", err) |
|||
} |
|||
} |
|||
|
|||
// OnVideoState accepts only ordered state from the matching non-terminal stream session.
|
|||
func (s *LiveService) OnVideoState(dockID string, streaming bool, streamSessionID, eventID string, version, updatedAt int64) bool { |
|||
if streamSessionID == "" || eventID == "" || version <= 0 || updatedAt <= 0 { |
|||
return false |
|||
} |
|||
now := time.Now() |
|||
if abs(now.UnixMilli()-updatedAt) > int64((5*time.Minute)/time.Millisecond) { |
|||
logger.WARN("忽略过期视频状态", dockID, streamSessionID) |
|||
return false |
|||
} |
|||
|
|||
var session model.LiveSession |
|||
if err := common.DB.Where("id = ? AND dock_id = ? AND phase IN ('starting','streaming','reconnecting','stopping')", streamSessionID, dockID).First(&session).Error; err != nil { |
|||
return false |
|||
} |
|||
if version <= session.DeviceStateVersion || (session.DeviceEventID != "" && session.DeviceEventID == eventID) || updatedAt < session.DeviceUpdatedAt { |
|||
return false |
|||
} |
|||
updates := map[string]any{ |
|||
"device_state_version": version, |
|||
"device_event_id": eventID, |
|||
"device_updated_at": updatedAt, |
|||
"updated_at": now, |
|||
} |
|||
if streaming { |
|||
provider, busiErr := s.adapter() |
|||
if busiErr != nil { |
|||
return false |
|||
} |
|||
online, queryErr := provider.QueryOnline(streamSessionID) |
|||
if queryErr != nil || !online.Online { |
|||
updates["error_code"] = "PROVIDER_ONLINE_UNVERIFIED" |
|||
return common.DB.Model(&model.LiveSession{}).Where("id = ? AND dock_id = ? AND device_state_version < ?", streamSessionID, dockID, version).Updates(updates).Error == nil |
|||
} |
|||
updates["phase"] = "streaming" |
|||
updates["started_at"] = now |
|||
updates["cloud_confirmed_at"] = now |
|||
} else { |
|||
updates["phase"] = "stopped" |
|||
updates["stop_reason"] = "device" |
|||
updates["stopped_at"] = now |
|||
} |
|||
result := common.DB.Model(&model.LiveSession{}).Where("id = ? AND dock_id = ? AND phase IN ('starting','streaming','reconnecting','stopping') AND device_state_version < ?", streamSessionID, dockID, version).Updates(updates) |
|||
return result.Error == nil && result.RowsAffected == 1 |
|||
} |
|||
|
|||
func abs(value int64) int64 { |
|||
if value < 0 { |
|||
return -value |
|||
} |
|||
return value |
|||
} |
|||
@ -0,0 +1,408 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"fmt" |
|||
"strconv" |
|||
"strings" |
|||
|
|||
paho "github.com/eclipse/paho.mqtt.golang" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/mqtt" |
|||
"laic-backend/websocket" |
|||
) |
|||
|
|||
// 上行消息业务载荷结构（协议文档 §7，均位于通用外层 payload 内）
|
|||
|
|||
// statusOnline §7.1 工控机在线状态
|
|||
type statusOnline struct { |
|||
Status string `json:"status"` // online / offline / degraded / updating
|
|||
BootID string `json:"bootId"` |
|||
DockIDSource string `json:"dockIdSource"` |
|||
SoftwareVersion string `json:"softwareVersion"` |
|||
ProtocolVersion string `json:"protocolVersion"` |
|||
UptimeSec int64 `json:"uptimeSec"` |
|||
TimeSynced bool `json:"timeSynced"` |
|||
MqttConnected bool `json:"mqttConnected"` |
|||
ModbusConnected bool `json:"modbusConnected"` |
|||
MavlinkConnected bool `json:"mavlinkConnected"` |
|||
Updating bool `json:"updating"` |
|||
Name string `json:"name"` |
|||
Location string `json:"location"` |
|||
Latitude float64 `json:"latitude"` |
|||
Longitude float64 `json:"longitude"` |
|||
} |
|||
|
|||
// dockState §7.2 机巢状态（仅解析业务关心字段，其余经 flatten 存 Redis）
|
|||
type dockState struct { |
|||
PlcConnected bool `json:"plcConnected"` |
|||
ControlMode string `json:"controlMode"` |
|||
ChargingState string `json:"chargingState"` |
|||
EmergencyStop bool `json:"emergencyStop"` |
|||
AlarmCodes []string `json:"alarmCodes"` |
|||
DronePresent *bool `json:"dronePresent"` |
|||
} |
|||
|
|||
// droneState §7.3 无人机状态
|
|||
type droneState struct { |
|||
DroneSN string `json:"droneSn"` |
|||
Name string `json:"name"` |
|||
CurrentSysID int `json:"currentSysId"` |
|||
Online bool `json:"online"` |
|||
Armed bool `json:"armed"` |
|||
FlightMode string `json:"flightMode"` |
|||
FlightModeCode int `json:"flightModeCode"` |
|||
Latitude float64 `json:"latitude"` |
|||
Longitude float64 `json:"longitude"` |
|||
Altitude float64 `json:"altitude"` |
|||
GroundSpeed float64 `json:"groundSpeed"` |
|||
Roll float64 `json:"roll"` |
|||
Pitch float64 `json:"pitch"` |
|||
Yaw float64 `json:"yaw"` |
|||
BatteryPercent float64 `json:"batteryPercent"` |
|||
BatteryVoltage float64 `json:"batteryVoltage"` |
|||
BatteryCurrent float64 `json:"batteryCurrent"` |
|||
Satellites int `json:"satellites"` |
|||
GpsQuality string `json:"gpsQuality"` |
|||
LinkQuality float64 `json:"linkQuality"` |
|||
HomeSet bool `json:"homeSet"` |
|||
AlarmCodes []string `json:"alarmCodes"` |
|||
} |
|||
|
|||
// commandAck §5.3 指令应答
|
|||
type commandAck struct { |
|||
CommandID string `json:"commandId"` |
|||
Accepted bool `json:"accepted"` |
|||
ResultCode string `json:"resultCode"` |
|||
} |
|||
|
|||
// otaReported §8 OTA 升级进度上报
|
|||
type otaReported struct { |
|||
UpdateID string `json:"updateId"` |
|||
Component string `json:"component"` |
|||
TargetVersion string `json:"targetVersion"` |
|||
CurrentVersion string `json:"currentVersion"` |
|||
Status string `json:"status"` |
|||
Progress int `json:"progress"` |
|||
Message string `json:"message"` |
|||
ErrorCode *string `json:"errorCode"` |
|||
ReportedAt int64 `json:"reportedAt"` |
|||
} |
|||
|
|||
// videoState §7.8 视频推流状态
|
|||
type videoState struct { |
|||
Version int64 `json:"version"` |
|||
EventID string `json:"eventId"` |
|||
Provider string `json:"provider"` |
|||
Phase string `json:"phase"` |
|||
InputOnline bool `json:"inputOnline"` |
|||
Streaming bool `json:"streaming"` |
|||
InputCodec string `json:"inputCodec"` |
|||
UplinkProtocol string `json:"uplinkProtocol"` |
|||
StreamSessionID string `json:"streamSessionId"` |
|||
Width *int `json:"width"` |
|||
Height *int `json:"height"` |
|||
FrameRate *float64 `json:"frameRate"` |
|||
BitrateBps uint64 `json:"bitrateBps"` |
|||
RetryCount int `json:"retryCount"` |
|||
StopReason *string `json:"stopReason"` |
|||
ErrorCode *string `json:"errorCode"` |
|||
UpdatedAt int64 `json:"updatedAt"` |
|||
} |
|||
|
|||
// InitMQTTSubscriber 注册 MQTT 订阅处理器
|
|||
func InitMQTTSubscriber() { |
|||
mqtt.Subscribe(map[string]byte{ |
|||
"dock-edge/v1/dock/+/status/online": 1, |
|||
"dock-edge/v1/dock/+/state/dock": 1, |
|||
"dock-edge/v1/dock/+/state/drone": 1, |
|||
"dock-edge/v1/dock/+/state/workflow": 1, |
|||
"dock-edge/v1/dock/+/state/video": 1, |
|||
"dock-edge/v1/dock/+/internal/original-video": 1, |
|||
"dock-edge/v1/dock/+/telemetry": 0, |
|||
"dock-edge/v1/dock/+/command/ack": 1, |
|||
"dock-edge/v1/dock/+/ota/reported": 1, |
|||
}, routeMessage) |
|||
} |
|||
|
|||
// routeMessage 按 topic 分发到对应处理器
|
|||
// topic 结构：dock-edge/v1/dock/{dockId}/{category}/{sub}
|
|||
// 消息体为通用外层 Envelope，业务字段在 payload 内。
|
|||
func routeMessage(_ paho.Client, msg paho.Message) { |
|||
topic := msg.Topic() |
|||
parts := strings.Split(topic, "/") |
|||
if len(parts) < 5 { |
|||
logger.WARN("无法解析 MQTT topic:", topic) |
|||
return |
|||
} |
|||
topicDockID := parts[3] |
|||
dockID := topicDockID |
|||
category := parts[4] |
|||
sub := "" |
|||
if len(parts) > 5 { |
|||
sub = parts[5] |
|||
} |
|||
|
|||
env := mqtt.ParseEnvelope(msg.Payload()) |
|||
if env.DockID != "" && env.DockID != topicDockID { |
|||
logger.WARN("MQTT envelope dockId 与 topic 不一致", topicDockID, env.DockID) |
|||
return |
|||
} |
|||
if env.DockID != "" { |
|||
dockID = env.DockID |
|||
} |
|||
|
|||
switch category { |
|||
case "status": |
|||
if sub == "online" { |
|||
handleStatus(dockID, env) |
|||
} |
|||
case "state": |
|||
switch sub { |
|||
case "dock": |
|||
handleStateDock(dockID, env) |
|||
case "drone": |
|||
handleStateDrone(dockID, env) |
|||
case "workflow": |
|||
handleWorkflow(dockID, env) |
|||
case "video": |
|||
handleVideo(dockID, env) |
|||
} |
|||
case "telemetry": |
|||
handleTelemetry(dockID, env) |
|||
case "command": |
|||
if sub == "ack" { |
|||
handleCommandAck(dockID, env) |
|||
} |
|||
case "ota": |
|||
if sub == "reported" { |
|||
handleOtaReported(dockID, env) |
|||
} |
|||
case "internal": |
|||
if sub == "original-video" { |
|||
handleOriginalVideo(dockID, env) |
|||
} |
|||
default: |
|||
logger.DEBUG("未知 topic:", topic) |
|||
} |
|||
} |
|||
|
|||
// handleStatus 在线/离线 → 设备自发现 + 在线集合 + 软件版本
|
|||
func handleStatus(dockID string, env *mqtt.Envelope) { |
|||
var st statusOnline |
|||
if err := json.Unmarshal(env.Payload, &st); err != nil { |
|||
logger.ERROR("解析 status/online 失败", err) |
|||
return |
|||
} |
|||
DefaultDockService.EnsureFromOnline(dockID, st) |
|||
if st.Status == "online" { |
|||
refreshDeviceHeartbeat(cache.DockHeartbeatKeyOf(dockID)) |
|||
} |
|||
} |
|||
|
|||
// handleStateDock 机巢状态 → Redis Hash + 告警 diff
|
|||
func handleStateDock(dockID string, env *mqtt.Envelope) { |
|||
var raw map[string]any |
|||
_ = json.Unmarshal(env.Payload, &raw) |
|||
_ = common.HashSetValues(cache.DockStatusKeyOf(dockID), flatten(raw)) |
|||
refreshDeviceHeartbeat(cache.DockHeartbeatKeyOf(dockID)) |
|||
|
|||
var st dockState |
|||
if err := json.Unmarshal(env.Payload, &st); err != nil { |
|||
logger.ERROR("解析 state/dock 失败", err) |
|||
return |
|||
} |
|||
DefaultAlarmService.SyncByState(dockID, "dock", st.AlarmCodes) |
|||
} |
|||
|
|||
// handleStateDrone 无人机状态 → 自动关联 + Redis 遥测 + 告警 diff
|
|||
func handleStateDrone(dockID string, env *mqtt.Envelope) { |
|||
var raw map[string]any |
|||
_ = json.Unmarshal(env.Payload, &raw) |
|||
_ = common.HashSetValues(cache.DroneTelemetryKeyOf(dockID), flatten(raw)) |
|||
|
|||
var st droneState |
|||
if err := json.Unmarshal(env.Payload, &st); err != nil { |
|||
logger.ERROR("解析 state/drone 失败", err) |
|||
return |
|||
} |
|||
droneSN := st.DroneSN |
|||
if droneSN == "" { |
|||
droneSN = env.DroneSN |
|||
} |
|||
battery := int(st.BatteryPercent) |
|||
DefaultDroneService.EnsureFromState(dockID, droneSN, st.Name, st.Online, battery, "") |
|||
if droneSN != "" && st.Online { |
|||
refreshDeviceHeartbeat(cache.DroneHeartbeatKeyOf(droneSN)) |
|||
} |
|||
DefaultAlarmService.SyncByState(dockID, "drone", st.AlarmCodes) |
|||
DefaultTelemetryStore.SetDroneSN(dockID, droneSN) |
|||
} |
|||
|
|||
// handleTelemetry 高频遥测 → Redis 最新值 + TDengine 批量写入 + 轨迹采集
|
|||
func handleTelemetry(dockID string, env *mqtt.Envelope) { |
|||
var raw map[string]any |
|||
if err := json.Unmarshal(env.Payload, &raw); err != nil { |
|||
logger.ERROR("解析 telemetry 失败", err) |
|||
return |
|||
} |
|||
_ = common.HashSetValues(cache.DroneTelemetryKeyOf(dockID), flatten(raw)) |
|||
|
|||
var p TelemetryPoint |
|||
if err := json.Unmarshal(env.Payload, &p); err == nil { |
|||
DefaultTelemetryStore.Append(dockID, p) |
|||
DefaultTrajectoryStore.Append(dockID, p) |
|||
} |
|||
if env.DroneSN != "" { |
|||
DefaultTelemetryStore.SetDroneSN(dockID, env.DroneSN) |
|||
refreshDeviceHeartbeat(cache.DroneHeartbeatKeyOf(env.DroneSN)) |
|||
} |
|||
} |
|||
|
|||
// handleWorkflow 工作流状态 → Redis Hash + WS 推送 + 持久化 + 终态回写任务
|
|||
func handleWorkflow(dockID string, env *mqtt.Envelope) { |
|||
var raw map[string]any |
|||
_ = json.Unmarshal(env.Payload, &raw) |
|||
_ = common.HashSetValues(cache.WorkflowKeyOf(dockID), flatten(raw)) |
|||
|
|||
var st WorkflowStateIn |
|||
if err := json.Unmarshal(env.Payload, &st); err != nil { |
|||
logger.ERROR("解析 state/workflow 失败", err) |
|||
return |
|||
} |
|||
DefaultWorkflowService.Upsert(dockID, env.RequestID, &st) |
|||
broadcast("workflow.state", map[string]any{ |
|||
"dockId": dockID, |
|||
"commandId": st.CommandID, |
|||
"taskId": st.TaskID, |
|||
"missionId": st.MissionID, |
|||
"state": st.State, |
|||
"step": st.Step, |
|||
"resultCode": st.ResultCode, |
|||
}) |
|||
} |
|||
|
|||
// handleVideo 视频推流状态 → Redis Hash + WS 推送 + 直播会话联动
|
|||
func handleVideo(dockID string, env *mqtt.Envelope) { |
|||
var raw map[string]any |
|||
_ = json.Unmarshal(env.Payload, &raw) |
|||
var st videoState |
|||
if err := json.Unmarshal(env.Payload, &st); err != nil { |
|||
logger.ERROR("解析 state/video 失败", err) |
|||
return |
|||
} |
|||
if st.EventID == "" { |
|||
st.EventID = env.EventID |
|||
} |
|||
if st.Version == 0 { |
|||
st.Version = env.Version |
|||
} |
|||
if st.StreamSessionID == "" || st.EventID == "" || st.Version <= 0 || st.UpdatedAt <= 0 { |
|||
logger.WARN("忽略缺少顺序元数据的视频状态", dockID, st.StreamSessionID) |
|||
return |
|||
} |
|||
if !DefaultLiveService.OnVideoState(dockID, st.Streaming, st.StreamSessionID, st.EventID, st.Version, st.UpdatedAt) { |
|||
return |
|||
} |
|||
_ = common.HashSetValues(cache.VideoKeyOf(dockID), flatten(raw)) |
|||
broadcast("video.state", map[string]any{ |
|||
"dockId": dockID, |
|||
"streaming": st.Streaming, |
|||
"streamSessionId": st.StreamSessionID, |
|||
"bitrate": st.BitrateBps, |
|||
"phase": st.Phase, |
|||
"errorCode": st.ErrorCode, |
|||
}) |
|||
} |
|||
|
|||
// handleOriginalVideo 处理后台与 Mock 的内部原始视频上传事件。
|
|||
func handleOriginalVideo(dockID string, env *mqtt.Envelope) { |
|||
if env.DockID != "" && env.DockID != dockID { |
|||
logger.WARN("原始视频事件机巢不匹配", dockID, env.DockID) |
|||
return |
|||
} |
|||
var event OriginalVideoEvent |
|||
if err := json.Unmarshal(env.Payload, &event); err != nil { |
|||
logger.ERROR("解析原始视频事件失败", err) |
|||
return |
|||
} |
|||
if err := DefaultVideoService.CompleteOriginalVideo(dockID, &event); err != nil { |
|||
logger.ERROR("处理原始视频事件失败", err) |
|||
return |
|||
} |
|||
broadcast("video.original", map[string]any{ |
|||
"dockId": dockID, "videoId": event.VideoID, "executionId": event.ExecutionID, "eventType": event.EventType, |
|||
}) |
|||
} |
|||
|
|||
// handleCommandAck 指令应答 → 更新 device_command_log
|
|||
func handleCommandAck(dockID string, env *mqtt.Envelope) { |
|||
var ack commandAck |
|||
if err := json.Unmarshal(env.Payload, &ack); err != nil { |
|||
logger.ERROR("解析 command/ack 失败", err) |
|||
return |
|||
} |
|||
DefaultCommandService.HandleAck(dockID, env.RequestID, ack.CommandID, ack.Accepted, ack.ResultCode) |
|||
} |
|||
|
|||
// handleOtaReported OTA 升级进度 → Redis 最新值 + WS 推送
|
|||
func handleOtaReported(dockID string, env *mqtt.Envelope) { |
|||
var r otaReported |
|||
if err := json.Unmarshal(env.Payload, &r); err != nil { |
|||
logger.ERROR("解析 ota/reported 失败", err) |
|||
return |
|||
} |
|||
_ = common.HashSetValues(cache.OtaStatusKeyOf(dockID), map[string]any{ |
|||
"updateId": r.UpdateID, |
|||
"component": r.Component, |
|||
"targetVersion": r.TargetVersion, |
|||
"currentVersion": r.CurrentVersion, |
|||
"status": r.Status, |
|||
"progress": strconv.Itoa(r.Progress), |
|||
"message": r.Message, |
|||
}) |
|||
broadcast("ota.progress", map[string]any{ |
|||
"dockId": dockID, |
|||
"updateId": r.UpdateID, |
|||
"targetVersion": r.TargetVersion, |
|||
"status": r.Status, |
|||
"progress": r.Progress, |
|||
"message": r.Message, |
|||
}) |
|||
} |
|||
|
|||
// flatten 将 JSON map 展平为 Redis Hash 可存储的 string 值
|
|||
func flatten(data map[string]any) map[string]any { |
|||
out := make(map[string]any, len(data)) |
|||
for k, v := range data { |
|||
switch val := v.(type) { |
|||
case string: |
|||
out[k] = val |
|||
case float64: |
|||
out[k] = strconv.FormatFloat(val, 'f', -1, 64) |
|||
case bool: |
|||
out[k] = strconv.FormatBool(val) |
|||
case nil: |
|||
out[k] = "" |
|||
default: |
|||
if b, err := json.Marshal(val); err == nil { |
|||
out[k] = string(b) |
|||
} else { |
|||
out[k] = fmt.Sprint(val) |
|||
} |
|||
} |
|||
} |
|||
return out |
|||
} |
|||
|
|||
// broadcast 向有权访问该机巢的 WS 客户端推送脱敏事件。
|
|||
func broadcast(eventType string, data map[string]any) { |
|||
dockID, _ := data["dockId"].(string) |
|||
if dockID == "" { |
|||
return |
|||
} |
|||
websocket.DefaultHub.BroadcastToDockJSON(dockID, eventType, data) |
|||
} |
|||
@ -0,0 +1,80 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"time" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type OperationLogService struct{} |
|||
|
|||
var DefaultOperationLogService = &OperationLogService{} |
|||
|
|||
// BootstrapOperationLog 确保基础运行时表存在（历史库未跑 DDL 时兜底）。
|
|||
func BootstrapOperationLog() { |
|||
if err := common.DB.AutoMigrate(&model.OperationLog{}, &model.LiveViewerLease{}); err != nil { |
|||
logger.ERROR("迁移基础运行时表失败", err) |
|||
} |
|||
} |
|||
|
|||
// Record 写入一条操作日志
|
|||
func (s *OperationLogService) Record(userID int64, userName, module, action, detail, result, ip string) { |
|||
if userID == 0 { |
|||
return |
|||
} |
|||
id, err := tool.NextID() |
|||
if err != nil { |
|||
logger.ERROR("生成操作日志 ID 失败", err) |
|||
return |
|||
} |
|||
log := &model.OperationLog{ |
|||
ID: id, |
|||
UserID: userID, |
|||
UserName: userName, |
|||
Module: module, |
|||
Action: action, |
|||
Detail: detail, |
|||
Result: result, |
|||
IP: ip, |
|||
CreatedAt: time.Now(), |
|||
} |
|||
if err := common.DB.Create(log).Error; err != nil { |
|||
logger.ERROR("写入操作日志失败", err) |
|||
} |
|||
} |
|||
|
|||
// GetPage 操作日志分页（admin 全量，user 仅本人）
|
|||
func (s *OperationLogService) GetPage(req *vo.OperationLogPageReq) (*common.PageResponse[model.OperationLog], *common.BusiError) { |
|||
db := common.DB.Model(&model.OperationLog{}) |
|||
if req.UserID > 0 { |
|||
db = db.Where("user_id = ?", req.UserID) |
|||
} |
|||
if req.Module != "" { |
|||
db = db.Where("module = ?", req.Module) |
|||
} |
|||
if req.Action != "" { |
|||
db = db.Where("action = ?", req.Action) |
|||
} |
|||
if req.Result != "" { |
|||
db = db.Where("result = ?", req.Result) |
|||
} |
|||
if req.Keyword != "" { |
|||
kw := "%" + req.Keyword + "%" |
|||
db = db.Where("module LIKE ? OR action LIKE ? OR detail LIKE ?", kw, kw, kw) |
|||
} |
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
logger.ERROR("统计操作日志失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.OperationLog |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
logger.ERROR("查询操作日志失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
@ -0,0 +1,190 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"errors" |
|||
"time" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type RouteService struct{} |
|||
|
|||
var DefaultRouteService = &RouteService{} |
|||
|
|||
// GetPage 航线分页列表
|
|||
func (s *RouteService) GetPage(userID int64, isAdmin bool, req *vo.RoutePageReq) (*common.PageResponse[model.Route], *common.BusiError) { |
|||
db := common.DB.Model(&model.Route{}).Scopes(withUserFilter(userID, isAdmin)) |
|||
if req.Keyword != "" { |
|||
kw := "%" + req.Keyword + "%" |
|||
db = db.Where("name LIKE ? OR description LIKE ?", kw, kw) |
|||
} |
|||
|
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
logger.ERROR("统计航线失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.Route |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
logger.ERROR("查询航线列表失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// GetDetail 航线详情（含航点）
|
|||
func (s *RouteService) GetDetail(userID int64, isAdmin bool, id int64) (*vo.RouteVO, *common.BusiError) { |
|||
var route model.Route |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&route, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrRouteNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
waypoints, err := s.waypoints(id) |
|||
if err != nil { |
|||
logger.ERROR("查询航点失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &vo.RouteVO{Route: &route, Waypoints: waypoints}, nil |
|||
} |
|||
|
|||
// Create 新增航线（含航点）
|
|||
func (s *RouteService) Create(userID int64, req *vo.RouteCreateReq) (*vo.RouteVO, *common.BusiError) { |
|||
id, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
now := time.Now() |
|||
route := &model.Route{ |
|||
ID: id, |
|||
UserID: userID, |
|||
Name: req.Name, |
|||
Description: req.Description, |
|||
CreatedAt: now, |
|||
UpdatedAt: now, |
|||
} |
|||
waypoints, err := buildWaypoints(id, req.Waypoints) |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
err = common.DB.Transaction(func(tx *gorm.DB) error { |
|||
if err := tx.Create(route).Error; err != nil { |
|||
return err |
|||
} |
|||
if len(waypoints) > 0 { |
|||
return tx.Create(&waypoints).Error |
|||
} |
|||
return nil |
|||
}) |
|||
if err != nil { |
|||
logger.ERROR("新增航线失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &vo.RouteVO{Route: route, Waypoints: waypoints}, nil |
|||
} |
|||
|
|||
// Update 编辑航线（提供 waypoints 则整体替换航点）
|
|||
func (s *RouteService) Update(userID int64, isAdmin bool, id int64, req *vo.RouteUpdateReq) (*vo.RouteVO, *common.BusiError) { |
|||
var route model.Route |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&route, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrRouteNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
err := common.DB.Transaction(func(tx *gorm.DB) error { |
|||
updates := map[string]any{"updated_at": time.Now()} |
|||
if req.Name != "" { |
|||
updates["name"] = req.Name |
|||
} |
|||
if req.Description != "" { |
|||
updates["description"] = req.Description |
|||
} |
|||
if err := tx.Model(&route).Updates(updates).Error; err != nil { |
|||
return err |
|||
} |
|||
if req.Waypoints != nil { |
|||
if err := tx.Where("route_id = ?", id).Delete(&model.RouteWaypoint{}).Error; err != nil { |
|||
return err |
|||
} |
|||
waypoints, err := buildWaypoints(id, *req.Waypoints) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
if len(waypoints) > 0 { |
|||
return tx.Create(&waypoints).Error |
|||
} |
|||
} |
|||
return nil |
|||
}) |
|||
if err != nil { |
|||
logger.ERROR("更新航线失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
common.DB.First(&route, id) |
|||
waypoints, _ := s.waypoints(id) |
|||
return &vo.RouteVO{Route: &route, Waypoints: waypoints}, nil |
|||
} |
|||
|
|||
// Delete 删除航线（含航点）
|
|||
func (s *RouteService) Delete(userID int64, isAdmin bool, id int64) *common.BusiError { |
|||
err := common.DB.Transaction(func(tx *gorm.DB) error { |
|||
res := tx.Scopes(withUserFilter(userID, isAdmin)).Delete(&model.Route{}, id) |
|||
if res.Error != nil { |
|||
return res.Error |
|||
} |
|||
if res.RowsAffected == 0 { |
|||
return gorm.ErrRecordNotFound |
|||
} |
|||
return tx.Where("route_id = ?", id).Delete(&model.RouteWaypoint{}).Error |
|||
}) |
|||
if err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return common.ErrRouteNotFound |
|||
} |
|||
logger.ERROR("删除航线失败", err) |
|||
return common.ErrInternal |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (s *RouteService) waypoints(routeID int64) ([]model.RouteWaypoint, error) { |
|||
var waypoints []model.RouteWaypoint |
|||
err := common.DB.Where("route_id = ?", routeID).Order("seq ASC").Find(&waypoints).Error |
|||
return waypoints, err |
|||
} |
|||
|
|||
// buildWaypoints 将航点请求转换为模型，Seq 从 1 递增
|
|||
func buildWaypoints(routeID int64, reqs []vo.RouteWaypointReq) ([]model.RouteWaypoint, error) { |
|||
out := make([]model.RouteWaypoint, 0, len(reqs)) |
|||
now := time.Now() |
|||
for i, w := range reqs { |
|||
id, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
out = append(out, model.RouteWaypoint{ |
|||
ID: id, |
|||
RouteID: routeID, |
|||
Seq: i + 1, |
|||
Longitude: w.Longitude, |
|||
Latitude: w.Latitude, |
|||
Altitude: w.Altitude, |
|||
Speed: w.Speed, |
|||
Yaw: w.Yaw, |
|||
HoldSec: w.HoldSec, |
|||
CreatedAt: now, |
|||
}) |
|||
} |
|||
return out, nil |
|||
} |
|||
@ -0,0 +1,104 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"sync" |
|||
"time" |
|||
|
|||
"github.com/robfig/cron/v3" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
) |
|||
|
|||
// Scheduler 后台定时任务调度器：固定周期计费任务 + 任务计划 cron 自动执行
|
|||
type Scheduler struct { |
|||
cron *cron.Cron |
|||
mu sync.Mutex |
|||
entries map[string]cron.EntryID // taskID -> cron entry
|
|||
} |
|||
|
|||
var DefaultScheduler = &Scheduler{} |
|||
|
|||
// Start 启动调度器：注册固定周期任务，加载并注册启用的 cron 任务计划
|
|||
func (s *Scheduler) Start() { |
|||
s.cron = cron.New() |
|||
s.entries = make(map[string]cron.EntryID) |
|||
|
|||
// 直播租约与会话对账：清理过期租约并处理超时/无人观看会话
|
|||
interval := common.AppConf.Live.ReconcileIntervalSeconds |
|||
if interval <= 0 { |
|||
interval = 5 |
|||
} |
|||
_, _ = s.cron.AddFunc("@every "+(time.Duration(interval)*time.Second).String(), DefaultLiveService.ReconcileLeases) |
|||
// 直播计费：每 60s 扫描进行中的 live_session 按码率扣减
|
|||
_, _ = s.cron.AddFunc("@every 60s", DefaultBillingService.chargeLiveSessions) |
|||
// 余额快照：每 30min 将 Redis 权威余额回写 MySQL
|
|||
_, _ = s.cron.AddFunc("@every 30m", DefaultBillingService.flushBalanceSnapshot) |
|||
// SIM 卡用量：每 1h 从运营商同步用量
|
|||
_, _ = s.cron.AddFunc("@every 1h", DefaultBillingService.syncSimUsage) |
|||
// 设备心跳：清理未再收到真实 MQTT 消息的在线设备
|
|||
_, _ = s.cron.AddFunc("@every "+heartbeatScanInterval().String(), expireDeviceHeartbeats) |
|||
|
|||
s.reloadTaskPlans() |
|||
s.cron.Start() |
|||
logger.INFO("scheduler started") |
|||
} |
|||
|
|||
// reloadTaskPlans 从 DB 加载启用中的 cron 任务计划并注册
|
|||
func (s *Scheduler) reloadTaskPlans() { |
|||
var tasks []model.TaskPlan |
|||
if err := common.DB.Where("schedule_type = ? AND status = ?", "cron", "enabled").Find(&tasks).Error; err != nil { |
|||
logger.ERROR("加载 cron 任务计划失败", err) |
|||
return |
|||
} |
|||
for i := range tasks { |
|||
s.upsertTask(&tasks[i]) |
|||
} |
|||
} |
|||
|
|||
// upsertTask 注册或更新单个任务计划的 cron 条目
|
|||
func (s *Scheduler) upsertTask(task *model.TaskPlan) { |
|||
s.mu.Lock() |
|||
defer s.mu.Unlock() |
|||
if old, ok := s.entries[task.ID]; ok { |
|||
s.cron.Remove(old) |
|||
delete(s.entries, task.ID) |
|||
} |
|||
if task.ScheduleType != "cron" || task.Status != "enabled" || task.ScheduleCron == "" { |
|||
return |
|||
} |
|||
id, err := s.cron.AddFunc(task.ScheduleCron, func() { s.executeTask(task.ID) }) |
|||
if err != nil { |
|||
logger.ERROR("注册 cron 任务计划失败: "+task.ID, err) |
|||
return |
|||
} |
|||
s.entries[task.ID] = id |
|||
} |
|||
|
|||
// SyncTask 任务创建/更新后同步调度（按 ID 重新加载）
|
|||
func (s *Scheduler) SyncTask(taskID string) { |
|||
var task model.TaskPlan |
|||
if err := common.DB.First(&task, "id = ?", taskID).Error; err != nil { |
|||
s.RemoveTask(taskID) |
|||
return |
|||
} |
|||
s.upsertTask(&task) |
|||
} |
|||
|
|||
// RemoveTask 移除任务计划的 cron 条目
|
|||
func (s *Scheduler) RemoveTask(taskID string) { |
|||
s.mu.Lock() |
|||
defer s.mu.Unlock() |
|||
if old, ok := s.entries[taskID]; ok { |
|||
s.cron.Remove(old) |
|||
delete(s.entries, taskID) |
|||
} |
|||
} |
|||
|
|||
// executeTask 定时触发任务执行（admin 视角，绕过用户过滤）
|
|||
func (s *Scheduler) executeTask(taskID string) { |
|||
if _, busiErr := DefaultTaskService.Execute(0, true, taskID); busiErr != nil { |
|||
logger.WARN("定时执行任务失败:", taskID, busiErr.Msg) |
|||
} |
|||
} |
|||
@ -0,0 +1,152 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"errors" |
|||
"time" |
|||
|
|||
"golang.org/x/crypto/bcrypt" |
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type SystemService struct{} |
|||
|
|||
var DefaultSystemService = &SystemService{} |
|||
|
|||
// GetUserPage 用户分页列表（admin 全量）
|
|||
func (s *SystemService) GetUserPage(req *vo.UserPageReq) (*common.PageResponse[model.User], *common.BusiError) { |
|||
db := common.DB.Model(&model.User{}) |
|||
if req.Keyword != "" { |
|||
kw := "%" + req.Keyword + "%" |
|||
db = db.Where("name LIKE ? OR phone LIKE ? OR email LIKE ?", kw, kw, kw) |
|||
} |
|||
if req.Role != "" { |
|||
db = db.Where("role = ?", req.Role) |
|||
} |
|||
if req.Status != 0 { |
|||
db = db.Where("status = ?", req.Status) |
|||
} |
|||
|
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
logger.ERROR("统计用户失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.User |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
logger.ERROR("查询用户列表失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// CreateUser 管理员创建用户
|
|||
func (s *SystemService) CreateUser(req *vo.UserCreateReq) (*model.User, *common.BusiError) { |
|||
var count int64 |
|||
if err := common.DB.Model(&model.User{}).Where("phone = ?", req.Phone).Count(&count).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
if count > 0 { |
|||
return nil, common.ErrUserPhoneExists |
|||
} |
|||
|
|||
id, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
hash, err := bcrypt.GenerateFromPassword([]byte(req.Password), bcrypt.DefaultCost) |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
role := req.Role |
|||
if role == "" { |
|||
role = common.RoleUser |
|||
} |
|||
now := time.Now() |
|||
user := &model.User{ |
|||
ID: id, |
|||
Name: req.Name, |
|||
Phone: req.Phone, |
|||
Email: req.Email, |
|||
Password: string(hash), |
|||
Role: role, |
|||
Status: 1, |
|||
CreatedAt: now, |
|||
UpdatedAt: now, |
|||
} |
|||
if err := common.DB.Create(user).Error; err != nil { |
|||
if code, _ := common.ParseError(err); code == 1062 { |
|||
return nil, common.ErrUserPhoneExists |
|||
} |
|||
logger.ERROR("创建用户失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
_ = common.SetUserRole(id, role) |
|||
return user, nil |
|||
} |
|||
|
|||
// UpdateUser 管理员编辑用户（角色/状态/密码等）
|
|||
func (s *SystemService) UpdateUser(id int64, req *vo.UserUpdateReq) (*model.User, *common.BusiError) { |
|||
var user model.User |
|||
if err := common.DB.First(&user, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrUserNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
updates := map[string]any{"updated_at": time.Now()} |
|||
if req.Name != "" { |
|||
updates["name"] = req.Name |
|||
} |
|||
if req.Email != "" { |
|||
updates["email"] = req.Email |
|||
} |
|||
if req.Password != "" { |
|||
hash, err := bcrypt.GenerateFromPassword([]byte(req.Password), bcrypt.DefaultCost) |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
updates["password"] = string(hash) |
|||
} |
|||
if req.Role != "" { |
|||
updates["role"] = req.Role |
|||
} |
|||
if req.Status != nil { |
|||
updates["status"] = *req.Status |
|||
} |
|||
|
|||
if err := common.DB.Model(&user).Updates(updates).Error; err != nil { |
|||
logger.ERROR("更新用户失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
common.DB.First(&user, id) |
|||
if req.Role != "" { |
|||
_ = common.SetUserRole(id, req.Role) |
|||
} |
|||
return &user, nil |
|||
} |
|||
|
|||
// DeleteUser 删除用户
|
|||
func (s *SystemService) DeleteUser(id int64) *common.BusiError { |
|||
res := common.DB.Delete(&model.User{}, id) |
|||
if res.Error != nil { |
|||
logger.ERROR("删除用户失败", res.Error) |
|||
return common.ErrInternal |
|||
} |
|||
if res.RowsAffected == 0 { |
|||
return common.ErrUserNotFound |
|||
} |
|||
_ = common.SetUserRole(id, "") |
|||
return nil |
|||
} |
|||
|
|||
// GetRoles 角色列表
|
|||
func (s *SystemService) GetRoles() []string { |
|||
return []string{common.RoleAdmin, common.RoleUser} |
|||
} |
|||
@ -0,0 +1,372 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"errors" |
|||
"fmt" |
|||
"strconv" |
|||
"strings" |
|||
"time" |
|||
|
|||
"github.com/google/uuid" |
|||
"github.com/robfig/cron/v3" |
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type TaskService struct{} |
|||
|
|||
var DefaultTaskService = &TaskService{} |
|||
|
|||
// GetPage 任务分页列表
|
|||
func (s *TaskService) GetPage(userID int64, isAdmin bool, req *vo.TaskPageReq) (*common.PageResponse[vo.TaskPlanVO], *common.BusiError) { |
|||
db := common.DB.Model(&model.TaskPlan{}).Scopes(withUserFilter(userID, isAdmin)) |
|||
if req.Status != "" { |
|||
db = db.Where("status = ?", req.Status) |
|||
} |
|||
if req.ScheduleType != "" { |
|||
db = db.Where("schedule_type = ?", req.ScheduleType) |
|||
} |
|||
if req.Keyword != "" { |
|||
db = db.Where("name LIKE ?", "%"+req.Keyword+"%") |
|||
} |
|||
|
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
logger.ERROR("统计任务失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
var tasks []model.TaskPlan |
|||
if err := db.Scopes(req.Paginate).Order("created_at DESC").Find(&tasks).Error; err != nil { |
|||
logger.ERROR("查询任务列表失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
latestByTaskID := make(map[string]*model.TaskExecution, len(tasks)) |
|||
if len(tasks) > 0 { |
|||
taskIDs := make([]string, 0, len(tasks)) |
|||
for _, task := range tasks { |
|||
taskIDs = append(taskIDs, task.ID) |
|||
} |
|||
var executions []model.TaskExecution |
|||
if err := common.DB.Where("task_id IN ?", taskIDs).Order("created_at DESC, id DESC").Find(&executions).Error; err != nil { |
|||
logger.ERROR("查询任务最近执行记录失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
for i := range executions { |
|||
exec := &executions[i] |
|||
if _, exists := latestByTaskID[exec.TaskID]; !exists { |
|||
latestByTaskID[exec.TaskID] = exec |
|||
} |
|||
} |
|||
} |
|||
|
|||
list := make([]vo.TaskPlanVO, 0, len(tasks)) |
|||
for _, task := range tasks { |
|||
latest := latestByTaskID[task.ID] |
|||
list = append(list, vo.TaskPlanVO{ |
|||
TaskPlan: task, |
|||
PlanStatus: taskPlanStatus(task.Status, latest), |
|||
LatestExecution: latest, |
|||
}) |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
func taskPlanStatus(status string, latest *model.TaskExecution) string { |
|||
if status == "disabled" { |
|||
return "paused" |
|||
} |
|||
if latest != nil && (latest.Status == "pending" || latest.Status == "running") { |
|||
return "running" |
|||
} |
|||
return "pending" |
|||
} |
|||
|
|||
// GetDetail 任务详情
|
|||
func (s *TaskService) GetDetail(userID int64, isAdmin bool, id string) (*model.TaskPlan, *common.BusiError) { |
|||
var task model.TaskPlan |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&task, "id = ?", id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrTaskNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &task, nil |
|||
} |
|||
|
|||
func validateTaskSchedule(scheduleType, scheduleCron string) *common.BusiError { |
|||
if scheduleType == "once" { |
|||
return nil |
|||
} |
|||
if scheduleType != "cron" || scheduleCron == "" { |
|||
return common.NewBusiError(common.ParamError, "定时任务必须配置执行时间") |
|||
} |
|||
if _, err := cron.ParseStandard(scheduleCron); err != nil { |
|||
return common.NewBusiError(common.ParamError, "定时执行表达式无效") |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
// Create 新增任务
|
|||
func (s *TaskService) Create(userID int64, req *vo.TaskCreateReq) (*model.TaskPlan, *common.BusiError) { |
|||
scheduleType := req.ScheduleType |
|||
if scheduleType == "" { |
|||
scheduleType = "once" |
|||
} |
|||
if busiErr := validateTaskSchedule(scheduleType, req.ScheduleCron); busiErr != nil { |
|||
return nil, busiErr |
|||
} |
|||
videoPolicy := req.VideoPolicy |
|||
if videoPolicy == "" { |
|||
videoPolicy = "raw" |
|||
} |
|||
now := time.Now() |
|||
task := &model.TaskPlan{ |
|||
ID: strings.ReplaceAll(uuid.New().String(), "-", ""), |
|||
UserID: userID, |
|||
Name: req.Name, |
|||
DockID: req.DockID, |
|||
RouteID: req.RouteID, |
|||
ScheduleType: scheduleType, |
|||
ScheduleCron: req.ScheduleCron, |
|||
VideoPolicy: videoPolicy, |
|||
Status: "enabled", |
|||
CreatedBy: userID, |
|||
CreatedAt: now, |
|||
UpdatedAt: now, |
|||
} |
|||
if err := common.DB.Create(task).Error; err != nil { |
|||
logger.ERROR("新增任务失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
DefaultScheduler.SyncTask(task.ID) |
|||
return task, nil |
|||
} |
|||
|
|||
// Update 编辑任务
|
|||
func (s *TaskService) Update(userID int64, isAdmin bool, id string, req *vo.TaskUpdateReq) *common.BusiError { |
|||
var task model.TaskPlan |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&task, "id = ?", id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return common.ErrTaskNotFound |
|||
} |
|||
return common.ErrInternal |
|||
} |
|||
|
|||
scheduleType := task.ScheduleType |
|||
if req.ScheduleType != "" { |
|||
scheduleType = req.ScheduleType |
|||
} |
|||
scheduleCron := task.ScheduleCron |
|||
if req.ScheduleCron != nil { |
|||
scheduleCron = *req.ScheduleCron |
|||
} |
|||
if scheduleType == "once" { |
|||
scheduleCron = "" |
|||
} |
|||
if busiErr := validateTaskSchedule(scheduleType, scheduleCron); busiErr != nil { |
|||
return busiErr |
|||
} |
|||
if req.Status != "" && req.Status != "draft" && req.Status != "enabled" && req.Status != "disabled" { |
|||
return common.NewBusiError(common.ParamError, "任务状态无效") |
|||
} |
|||
|
|||
updates := map[string]any{"updated_at": time.Now()} |
|||
if req.Name != "" { |
|||
updates["name"] = req.Name |
|||
} |
|||
if req.DockID != "" { |
|||
updates["dock_id"] = req.DockID |
|||
} |
|||
if req.RouteID != nil { |
|||
updates["route_id"] = *req.RouteID |
|||
} |
|||
if req.ScheduleType != "" { |
|||
updates["schedule_type"] = req.ScheduleType |
|||
} |
|||
if req.ScheduleCron != nil { |
|||
updates["schedule_cron"] = *req.ScheduleCron |
|||
} |
|||
if req.ScheduleType != "" && req.ScheduleType == "once" { |
|||
updates["schedule_cron"] = "" |
|||
} |
|||
if req.VideoPolicy != "" { |
|||
updates["video_policy"] = req.VideoPolicy |
|||
} |
|||
if req.Status != "" { |
|||
updates["status"] = req.Status |
|||
} |
|||
|
|||
if err := common.DB.Model(&task).Updates(updates).Error; err != nil { |
|||
logger.ERROR("更新任务失败", err) |
|||
return common.ErrInternal |
|||
} |
|||
DefaultScheduler.SyncTask(id) |
|||
return nil |
|||
} |
|||
|
|||
// Delete 删除任务
|
|||
func (s *TaskService) Delete(userID int64, isAdmin bool, id string) *common.BusiError { |
|||
res := common.DB.Scopes(withUserFilter(userID, isAdmin)).Delete(&model.TaskPlan{}, "id = ?", id) |
|||
if res.Error != nil { |
|||
logger.ERROR("删除任务失败", res.Error) |
|||
return common.ErrInternal |
|||
} |
|||
if res.RowsAffected == 0 { |
|||
return common.ErrTaskNotFound |
|||
} |
|||
DefaultScheduler.RemoveTask(id) |
|||
return nil |
|||
} |
|||
|
|||
// Execute 立即执行任务：落库执行记录 + 下发 workflow.start_task 指令
|
|||
func (s *TaskService) Execute(userID int64, isAdmin bool, id string) (*vo.TaskExecuteVO, *common.BusiError) { |
|||
var task model.TaskPlan |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&task, "id = ?", id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
logger.WARN("执行任务失败，任务不存在:", id) |
|||
return nil, common.ErrTaskNotFound |
|||
} |
|||
logger.ERROR("查询待执行任务失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
if task.Status == "disabled" { |
|||
logger.WARN("执行任务失败，任务已禁用:", id) |
|||
return nil, common.ErrTaskStateErr |
|||
} |
|||
|
|||
online, _ := common.SetMemberExists(cache.OnlineDockSetKey, task.DockID) |
|||
if !online { |
|||
logger.WARN("执行任务失败，机巢离线:", task.DockID, "任务:", id) |
|||
return nil, common.ErrDockOffline |
|||
} |
|||
|
|||
commandID, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
execID, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
requestID := strings.ReplaceAll(uuid.New().String(), "-", "") |
|||
now := time.Now() |
|||
params := map[string]any{ |
|||
"taskId": task.ID, |
|||
"missionId": strings.ReplaceAll(uuid.New().String(), "-", ""), |
|||
"executionId": strconv.FormatInt(execID, 10), |
|||
"recordOriginalVideo": task.VideoPolicy == "raw", |
|||
} |
|||
if task.RouteID != 0 { |
|||
var waypoints []model.RouteWaypoint |
|||
if err := common.DB.Where("route_id = ?", task.RouteID).Order("seq ASC").Find(&waypoints).Error; err != nil { |
|||
logger.ERROR("查询航线航点失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
payload := make([]map[string]any, 0, len(waypoints)) |
|||
for i, wp := range waypoints { |
|||
payload = append(payload, map[string]any{ |
|||
"seq": i, |
|||
"longitude": wp.Longitude, |
|||
"latitude": wp.Latitude, |
|||
"altitude": wp.Altitude, |
|||
"speed": wp.Speed, |
|||
"yaw": wp.Yaw, |
|||
"holdSec": wp.HoldSec, |
|||
}) |
|||
} |
|||
params["waypoints"] = payload |
|||
} |
|||
|
|||
var originalVideo *model.Video |
|||
if task.VideoPolicy == "raw" { |
|||
videoID, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
expireAt := now.Add(10 * time.Minute) |
|||
originalVideo = &model.Video{ |
|||
ID: videoID, |
|||
UserID: task.UserID, |
|||
ExecutionID: execID, |
|||
FileName: fmt.Sprintf("execution-%d-original.mp4", execID), |
|||
OssKey: fmt.Sprintf("media/%d/execution-%d-original.mp4", videoID, execID), |
|||
OssBucket: common.AppConf.OSS.Bucket, |
|||
Status: "pending", |
|||
UploadExpireAt: &expireAt, |
|||
CreatedAt: now, |
|||
} |
|||
uploadURL, err := common.PresignOSS("PUT", originalVideo.OssKey, 600) |
|||
if err != nil { |
|||
logger.ERROR("生成原始视频上传地址失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
params["originalVideo"] = map[string]any{ |
|||
"videoId": strconv.FormatInt(videoID, 10), |
|||
"executionId": strconv.FormatInt(execID, 10), |
|||
"uploadUrl": uploadURL, |
|||
"uploadExpireAt": expireAt.UnixMilli(), |
|||
"fileName": originalVideo.FileName, |
|||
"contentType": "video/mp4", |
|||
} |
|||
} |
|||
|
|||
paramsJSON := "" |
|||
if b, e := json.Marshal(params); e == nil { |
|||
paramsJSON = string(b) |
|||
} |
|||
cmdLog := &model.DeviceCommandLog{ |
|||
ID: commandID, |
|||
DockID: task.DockID, |
|||
CommandType: "workflow.start_task", |
|||
Params: paramsJSON, |
|||
RequestID: requestID, |
|||
TTLMs: 30000, |
|||
Status: "sent", |
|||
SentAt: &now, |
|||
CreatedAt: now, |
|||
} |
|||
exec := &model.TaskExecution{ |
|||
ID: execID, |
|||
TaskID: task.ID, |
|||
CommandID: strconv.FormatInt(commandID, 10), |
|||
DockID: task.DockID, |
|||
Status: "pending", |
|||
TrajectoryJSON: json.RawMessage("[]"), |
|||
CreatedAt: now, |
|||
} |
|||
|
|||
if err := common.DB.Transaction(func(tx *gorm.DB) error { |
|||
if err := tx.Create(cmdLog).Error; err != nil { |
|||
return err |
|||
} |
|||
if err := tx.Create(exec).Error; err != nil { |
|||
return err |
|||
} |
|||
if originalVideo != nil { |
|||
return tx.Create(originalVideo).Error |
|||
} |
|||
return nil |
|||
}); err != nil { |
|||
logger.ERROR("创建执行记录失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
var drone model.Drone |
|||
_ = common.DB.Where("dock_id = ?", task.DockID).First(&drone).Error |
|||
DefaultTrajectoryStore.Begin(task.DockID, execID, drone.DroneSN) |
|||
|
|||
if err := DefaultCommandService.publish(cmdLog, params); err != nil { |
|||
logger.ERROR("下发任务指令失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
return &vo.TaskExecuteVO{Execution: exec, Command: cmdLog}, nil |
|||
} |
|||
@ -0,0 +1,171 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"fmt" |
|||
"strconv" |
|||
"strings" |
|||
"sync" |
|||
"time" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
) |
|||
|
|||
// TelemetryPoint 无人机遥测点（对应 device_telemetry 超级表）
|
|||
type TelemetryPoint struct { |
|||
Ts int64 `json:"ts"` |
|||
Longitude float64 `json:"longitude"` |
|||
Latitude float64 `json:"latitude"` |
|||
Altitude float64 `json:"altitude"` |
|||
GroundSpeed float64 `json:"groundSpeed"` |
|||
Roll float64 `json:"roll"` |
|||
Pitch float64 `json:"pitch"` |
|||
Yaw float64 `json:"yaw"` |
|||
BatteryPct int `json:"batteryPct"` |
|||
BatteryV float64 `json:"batteryV"` |
|||
Satellites int `json:"satellites"` |
|||
GpsQuality string `json:"gpsQuality"` |
|||
LinkQuality int `json:"linkQuality"` |
|||
FlightMode string `json:"flightMode"` |
|||
Armed int8 `json:"armed"` |
|||
} |
|||
|
|||
const ( |
|||
telemetryBatchSize = 500 |
|||
telemetryFlushInterval = 10 * time.Second |
|||
) |
|||
|
|||
type telemetryRecord struct { |
|||
DockID string |
|||
Point TelemetryPoint |
|||
} |
|||
|
|||
// TelemetryStore 遥测缓冲：累计 500 条或 10 秒后批量写入 TDengine
|
|||
type TelemetryStore struct { |
|||
mu sync.Mutex |
|||
buffer []telemetryRecord |
|||
droneSN sync.Map // dockID -> droneSN（来自 state/drone 上报）
|
|||
} |
|||
|
|||
var DefaultTelemetryStore = &TelemetryStore{} |
|||
|
|||
// SetDroneSN 记录机巢当前无人机序列号（作为 device_telemetry 的 tag）
|
|||
func (s *TelemetryStore) SetDroneSN(dockID, sn string) { |
|||
if sn != "" { |
|||
s.droneSN.Store(dockID, sn) |
|||
} |
|||
} |
|||
|
|||
// Append 追加一个遥测点，达到阈值立即刷盘
|
|||
func (s *TelemetryStore) Append(dockID string, p TelemetryPoint) { |
|||
s.mu.Lock() |
|||
s.buffer = append(s.buffer, telemetryRecord{DockID: dockID, Point: p}) |
|||
if len(s.buffer) >= telemetryBatchSize { |
|||
s.flushLocked() |
|||
} |
|||
s.mu.Unlock() |
|||
} |
|||
|
|||
// Start 启动定时刷盘协程
|
|||
func (s *TelemetryStore) Start() { |
|||
go func() { |
|||
ticker := time.NewTicker(telemetryFlushInterval) |
|||
defer ticker.Stop() |
|||
for range ticker.C { |
|||
s.mu.Lock() |
|||
s.flushLocked() |
|||
s.mu.Unlock() |
|||
} |
|||
}() |
|||
} |
|||
|
|||
func (s *TelemetryStore) flushLocked() { |
|||
if len(s.buffer) == 0 { |
|||
return |
|||
} |
|||
records := s.buffer |
|||
s.buffer = nil |
|||
s.write(records) |
|||
} |
|||
|
|||
func (s *TelemetryStore) write(records []telemetryRecord) { |
|||
if common.TD == nil { |
|||
return |
|||
} |
|||
groups := make(map[string][]TelemetryPoint, len(records)) |
|||
for _, r := range records { |
|||
groups[r.DockID] = append(groups[r.DockID], r.Point) |
|||
} |
|||
for dockID, points := range groups { |
|||
s.insert(dockID, s.droneSNValue(dockID), points) |
|||
} |
|||
} |
|||
|
|||
func (s *TelemetryStore) droneSNValue(dockID string) string { |
|||
if v, ok := s.droneSN.Load(dockID); ok { |
|||
return v.(string) |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
// DroneSN 返回机巢当前无人机序列号（来自 state/drone 上报）
|
|||
func (s *TelemetryStore) DroneSN(dockID string) string { |
|||
return s.droneSNValue(dockID) |
|||
} |
|||
|
|||
func (s *TelemetryStore) insert(dockID, droneSN string, points []TelemetryPoint) { |
|||
defer func() { |
|||
if r := recover(); r != nil { |
|||
logger.ERROR("TDengine 写入 panic（已恢复，避免进程崩溃）", fmt.Errorf("%v", r)) |
|||
} |
|||
}() |
|||
var b strings.Builder |
|||
b.WriteString("INSERT INTO ") |
|||
b.WriteString(sanitizeTable(dockID)) |
|||
b.WriteString(" USING device_telemetry TAGS ('") |
|||
b.WriteString(escapeTD(dockID)) |
|||
b.WriteString("', '") |
|||
b.WriteString(escapeTD(droneSN)) |
|||
b.WriteString("') VALUES ") |
|||
for i, p := range points { |
|||
if i > 0 { |
|||
b.WriteString(", ") |
|||
} |
|||
ts := p.Ts |
|||
if ts <= 0 { |
|||
ts = time.Now().UnixMilli() |
|||
} |
|||
fmt.Fprintf(&b, "(%d, %s, %s, %s, %s, %s, %s, %s, %d, %s, %d, '%s', %d, '%s', %d)", |
|||
ts, |
|||
ftoa(p.Longitude), ftoa(p.Latitude), ftoa(p.Altitude), ftoa(p.GroundSpeed), |
|||
ftoa(p.Roll), ftoa(p.Pitch), ftoa(p.Yaw), |
|||
p.BatteryPct, ftoa(p.BatteryV), p.Satellites, |
|||
escapeTD(p.GpsQuality), p.LinkQuality, escapeTD(p.FlightMode), p.Armed, |
|||
) |
|||
} |
|||
if _, err := common.TD.Exec(b.String()); err != nil { |
|||
logger.ERROR("TDengine 批量写入遥测失败", err) |
|||
} |
|||
} |
|||
|
|||
func ftoa(v float64) string { |
|||
return strconv.FormatFloat(v, 'f', -1, 64) |
|||
} |
|||
|
|||
func escapeTD(s string) string { |
|||
return strings.ReplaceAll(s, "'", "\\'") |
|||
} |
|||
|
|||
// sanitizeTable 将 dockID 转为 TDengine 子表名（仅保留字母数字）
|
|||
func sanitizeTable(dockID string) string { |
|||
var b strings.Builder |
|||
b.WriteString("d_") |
|||
for _, r := range dockID { |
|||
if (r >= 'a' && r <= 'z') || (r >= 'A' && r <= 'Z') || (r >= '0' && r <= '9') { |
|||
b.WriteRune(r) |
|||
} else { |
|||
b.WriteByte('_') |
|||
} |
|||
} |
|||
return b.String() |
|||
} |
|||
@ -0,0 +1,133 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"fmt" |
|||
"strconv" |
|||
"strings" |
|||
"sync" |
|||
"time" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
// trajectoryBuffer 单次执行的活动轨迹缓冲
|
|||
type trajectoryBuffer struct { |
|||
ExecutionID int64 |
|||
DockID string |
|||
DroneSN string |
|||
Points []vo.TrajectoryPoint |
|||
} |
|||
|
|||
// TrajectoryStore 轨迹采集：维护每个机巢的活动执行，遥测点入缓冲，终态回写 MySQL + TDengine
|
|||
type TrajectoryStore struct { |
|||
mu sync.Mutex |
|||
active map[string]*trajectoryBuffer // dockID -> 活动执行
|
|||
} |
|||
|
|||
var DefaultTrajectoryStore = &TrajectoryStore{active: make(map[string]*trajectoryBuffer)} |
|||
|
|||
// Begin 建立活动执行关联（任务下发成功后调用）
|
|||
func (s *TrajectoryStore) Begin(dockID string, executionID int64, droneSN string) { |
|||
s.mu.Lock() |
|||
defer s.mu.Unlock() |
|||
s.active[dockID] = &trajectoryBuffer{ |
|||
ExecutionID: executionID, |
|||
DockID: dockID, |
|||
DroneSN: droneSN, |
|||
Points: make([]vo.TrajectoryPoint, 0, 1024), |
|||
} |
|||
} |
|||
|
|||
// Append 将遥测点追加到活动执行的轨迹缓冲
|
|||
func (s *TrajectoryStore) Append(dockID string, p TelemetryPoint) { |
|||
s.mu.Lock() |
|||
defer s.mu.Unlock() |
|||
buf, ok := s.active[dockID] |
|||
if !ok { |
|||
return |
|||
} |
|||
ts := p.Ts |
|||
if ts <= 0 { |
|||
ts = time.Now().UnixMilli() |
|||
} |
|||
buf.Points = append(buf.Points, vo.TrajectoryPoint{ |
|||
Ts: ts, |
|||
Longitude: p.Longitude, |
|||
Latitude: p.Latitude, |
|||
Altitude: p.Altitude, |
|||
GroundSpeed: p.GroundSpeed, |
|||
Yaw: p.Yaw, |
|||
BatteryPct: p.BatteryPct, |
|||
}) |
|||
} |
|||
|
|||
// Finalize 终态回写：轨迹点写 MySQL trajectory_json + TDengine task_trajectory，并清除活动关联
|
|||
func (s *TrajectoryStore) Finalize(dockID string, droneSN string) { |
|||
s.mu.Lock() |
|||
buf, ok := s.active[dockID] |
|||
if ok { |
|||
delete(s.active, dockID) |
|||
} |
|||
s.mu.Unlock() |
|||
if !ok || buf == nil { |
|||
return |
|||
} |
|||
if droneSN != "" { |
|||
buf.DroneSN = droneSN |
|||
} |
|||
|
|||
pointsJSON, err := json.Marshal(buf.Points) |
|||
if err != nil { |
|||
pointsJSON = []byte("[]") |
|||
} |
|||
updates := map[string]any{"trajectory_json": string(pointsJSON)} |
|||
if buf.DroneSN != "" { |
|||
updates["drone_sn"] = buf.DroneSN |
|||
} |
|||
if err := common.DB.Model(&model.TaskExecution{}). |
|||
Where("id = ?", buf.ExecutionID). |
|||
Updates(updates).Error; err != nil { |
|||
logger.ERROR("回写执行轨迹 JSON 失败", err) |
|||
} |
|||
|
|||
s.writeTDengine(buf) |
|||
} |
|||
|
|||
// writeTDengine 轨迹点批量写入 task_trajectory 超级表（子表按 execution_id 隔离）
|
|||
func (s *TrajectoryStore) writeTDengine(buf *trajectoryBuffer) { |
|||
if common.TD == nil || len(buf.Points) == 0 { |
|||
return |
|||
} |
|||
defer func() { |
|||
if r := recover(); r != nil { |
|||
logger.ERROR("TDengine 轨迹写入 panic（已恢复）", fmt.Errorf("%v", r)) |
|||
} |
|||
}() |
|||
|
|||
var b strings.Builder |
|||
b.WriteString("INSERT INTO ") |
|||
b.WriteString(trajectoryTable(buf.ExecutionID)) |
|||
b.WriteString(" USING task_trajectory TAGS (") |
|||
fmt.Fprintf(&b, "%d, '%s', '%s') VALUES ", |
|||
buf.ExecutionID, escapeTD(buf.DockID), escapeTD(buf.DroneSN)) |
|||
for i, p := range buf.Points { |
|||
if i > 0 { |
|||
b.WriteString(", ") |
|||
} |
|||
fmt.Fprintf(&b, "(%d, %s, %s, %s, %s, %s, %d)", |
|||
p.Ts, ftoa(p.Longitude), ftoa(p.Latitude), ftoa(p.Altitude), |
|||
ftoa(p.GroundSpeed), ftoa(p.Yaw), p.BatteryPct) |
|||
} |
|||
if _, err := common.TD.Exec(b.String()); err != nil { |
|||
logger.ERROR("TDengine 批量写入轨迹失败", err) |
|||
} |
|||
} |
|||
|
|||
// trajectoryTable 按执行 ID 生成子表名
|
|||
func trajectoryTable(execID int64) string { |
|||
return "tr_" + strconv.FormatInt(execID, 10) |
|||
} |
|||
@ -0,0 +1,207 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"errors" |
|||
"time" |
|||
|
|||
"golang.org/x/crypto/bcrypt" |
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/cache" |
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/token" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type UserService struct{} |
|||
|
|||
var DefaultUserService = &UserService{} |
|||
|
|||
// Register 注册用户（默认角色 user），成功后直接返回登录态
|
|||
func (s *UserService) Register(name, phone, email, password string) (*vo.LoginResp, *common.BusiError) { |
|||
// 手机号唯一性预检（兼顾并发，唯一键兜底）
|
|||
var count int64 |
|||
if err := common.DB.Model(&model.User{}).Where("phone = ?", phone).Count(&count).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
if count > 0 { |
|||
return nil, common.ErrUserPhoneExists |
|||
} |
|||
|
|||
id, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
hash, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost) |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
user := &model.User{ |
|||
ID: id, |
|||
Name: name, |
|||
Phone: phone, |
|||
Email: email, |
|||
Password: string(hash), |
|||
Role: common.RoleUser, |
|||
Status: 1, |
|||
CreatedAt: time.Now(), |
|||
UpdatedAt: time.Now(), |
|||
} |
|||
if err := common.DB.Create(user).Error; err != nil { |
|||
if code, _ := common.ParseError(err); code == 1062 { |
|||
return nil, common.ErrUserPhoneExists |
|||
} |
|||
logger.ERROR("注册用户失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
_ = common.SetUserRole(id, common.RoleUser) |
|||
|
|||
return s.buildLoginResp(user) |
|||
} |
|||
|
|||
// Login 手机号 + 密码登录
|
|||
func (s *UserService) Login(phone, password string) (*vo.LoginResp, *common.BusiError) { |
|||
var user model.User |
|||
if err := common.DB.Where("phone = ? AND status = 1", phone).First(&user).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrPasswordError // 统一返回"用户名或密码错误"，避免暴露账号是否存在
|
|||
} |
|||
logger.ERROR("登录查询用户失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
if !checkPassword(password, user.Password) { |
|||
return nil, common.ErrPasswordError |
|||
} |
|||
|
|||
// 更新最后登录时间（失败不阻断登录）
|
|||
common.DB.Model(&user).Update("last_login", time.Now()) |
|||
|
|||
return s.buildLoginResp(&user) |
|||
} |
|||
|
|||
// Refresh 用 refresh token 换取新的 access + refresh token（刷新令牌轮换）
|
|||
func (s *UserService) Refresh(refreshToken string) (*vo.LoginResp, *common.BusiError) { |
|||
userID, err := token.ParseRefreshToken(refreshToken) |
|||
if err != nil { |
|||
return nil, common.ErrTokenInvalid |
|||
} |
|||
|
|||
// 校验 refresh token 是否仍存证（登出会删除）
|
|||
refreshKey := cache.RefreshTokenKeyOf(userID, token.GenerateShortID(refreshToken)) |
|||
stored, err := common.GetString(refreshKey) |
|||
if err != nil || stored != refreshToken { |
|||
return nil, common.ErrTokenInvalid |
|||
} |
|||
|
|||
var user model.User |
|||
if err := common.DB.Where("id = ? AND status = 1", userID).First(&user).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrUserNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
// 轮换：删除旧 refresh 存证
|
|||
_ = common.Delete(refreshKey) |
|||
|
|||
return s.buildLoginResp(&user) |
|||
} |
|||
|
|||
// Logout 登出：删除 access 存证（及可选 refresh 存证），使 token 立即失效
|
|||
func (s *UserService) Logout(userID int64, accessToken, refreshToken string) *common.BusiError { |
|||
accessKey := cache.UserTokenKeyOf(userID, token.GenerateShortID(accessToken)) |
|||
_ = common.Delete(accessKey) |
|||
if refreshToken != "" { |
|||
refreshKey := cache.RefreshTokenKeyOf(userID, token.GenerateShortID(refreshToken)) |
|||
_ = common.Delete(refreshKey) |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
// SendSmsCode 生成并"发送"短信验证码（暂存 Redis，短信通道未接入前记日志）
|
|||
func (s *UserService) SendSmsCode(phone string) (string, *common.BusiError) { |
|||
code := tool.RandDigit(6) |
|||
key := cache.SmsCodeKeyOf(phone) |
|||
if err := common.SetValueWithExpired(key, code, 5*60); err != nil { |
|||
logger.ERROR("短信验证码写 Redis 失败", err) |
|||
return "", common.ErrInternal |
|||
} |
|||
// TODO: 接入短信服务商后，此处调用真实发送接口；当前仅在日志输出验证码便于联调
|
|||
logger.INFO("send sms code to", phone, "code:", code) |
|||
return code, nil |
|||
} |
|||
|
|||
// ResetPassword 通过短信验证码重置密码
|
|||
func (s *UserService) ResetPassword(phone, smsCode, newPassword string) *common.BusiError { |
|||
key := cache.SmsCodeKeyOf(phone) |
|||
stored, err := common.GetString(key) |
|||
if err != nil || stored == "" || stored != smsCode { |
|||
return common.ErrSmsCodeError |
|||
} |
|||
|
|||
var user model.User |
|||
if err := common.DB.Where("phone = ?", phone).First(&user).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return common.ErrUserNotFound |
|||
} |
|||
return common.ErrInternal |
|||
} |
|||
|
|||
hash, err := bcrypt.GenerateFromPassword([]byte(newPassword), bcrypt.DefaultCost) |
|||
if err != nil { |
|||
return common.ErrInternal |
|||
} |
|||
if err := common.DB.Model(&user).Updates(map[string]any{ |
|||
"password": string(hash), |
|||
"updated_at": time.Now(), |
|||
}).Error; err != nil { |
|||
return common.ErrInternal |
|||
} |
|||
_ = common.Delete(key) |
|||
return nil |
|||
} |
|||
|
|||
// GetByID 按 ID 查询用户
|
|||
func (s *UserService) GetByID(id int64) (*model.User, *common.BusiError) { |
|||
var user model.User |
|||
if err := common.DB.First(&user, id).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrUserNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &user, nil |
|||
} |
|||
|
|||
// buildLoginResp 生成 token 并存证 Redis，返回登录响应
|
|||
func (s *UserService) buildLoginResp(user *model.User) (*vo.LoginResp, *common.BusiError) { |
|||
roles := []string{user.Role} |
|||
accessToken, refreshToken, err := token.GenerateToken(user.ID, user.Phone, roles) |
|||
if err != nil { |
|||
logger.ERROR("生成 token 失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
accessKey := cache.UserTokenKeyOf(user.ID, token.GenerateShortID(accessToken)) |
|||
refreshKey := cache.RefreshTokenKeyOf(user.ID, token.GenerateShortID(refreshToken)) |
|||
if err := common.SetValueWithExpired(accessKey, "1", token.AccessExpireSeconds()); err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
if err := common.SetValueWithExpired(refreshKey, refreshToken, token.RefreshExpireSeconds()); err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
return &vo.LoginResp{ |
|||
AccessToken: accessToken, |
|||
RefreshToken: refreshToken, |
|||
User: vo.NewUserVO(user), |
|||
}, nil |
|||
} |
|||
|
|||
func checkPassword(password, hash string) bool { |
|||
return bcrypt.CompareHashAndPassword([]byte(hash), []byte(password)) == nil |
|||
} |
|||
@ -0,0 +1,231 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"errors" |
|||
"fmt" |
|||
"net/http" |
|||
"time" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
"laic-backend/vo" |
|||
) |
|||
|
|||
type VideoService struct{} |
|||
|
|||
var DefaultVideoService = &VideoService{} |
|||
|
|||
const downloadURLExpire = 300 // 预签名下载地址有效期（秒）
|
|||
|
|||
type OriginalVideoEvent struct { |
|||
Extension string `json:"extension"` |
|||
EventID string `json:"eventId"` |
|||
EventType string `json:"eventType"` |
|||
VideoID int64 `json:"videoId"` |
|||
ExecutionID int64 `json:"executionId"` |
|||
TaskID string `json:"taskId"` |
|||
CommandID string `json:"commandId"` |
|||
FileSize int64 `json:"fileSize"` |
|||
Duration int `json:"duration"` |
|||
ErrorCode string `json:"errorCode"` |
|||
} |
|||
|
|||
func (s *VideoService) CompleteOriginalVideo(dockID string, event *OriginalVideoEvent) error { |
|||
if event.Extension != "laic.mock.original-video.v1" || event.EventID == "" || event.VideoID <= 0 || event.ExecutionID <= 0 { |
|||
return errors.New("invalid original video event") |
|||
} |
|||
var video model.Video |
|||
if err := common.DB.First(&video, event.VideoID).Error; err != nil { |
|||
return err |
|||
} |
|||
if video.ExecutionID != event.ExecutionID { |
|||
return errors.New("video execution mismatch") |
|||
} |
|||
var execution model.TaskExecution |
|||
if err := common.DB.First(&execution, event.ExecutionID).Error; err != nil { |
|||
return err |
|||
} |
|||
if execution.DockID != dockID || execution.TaskID != event.TaskID || execution.CommandID != event.CommandID { |
|||
return errors.New("execution association mismatch") |
|||
} |
|||
var task model.TaskPlan |
|||
if err := common.DB.First(&task, "id = ?", execution.TaskID).Error; err != nil { |
|||
return err |
|||
} |
|||
if task.DockID != dockID || task.VideoPolicy != "raw" { |
|||
return errors.New("task video policy mismatch") |
|||
} |
|||
if event.EventType == "failed" { |
|||
return common.DB.Model(&video).Where("id = ? AND status <> ?", video.ID, "ready").Update("status", "failed").Error |
|||
} |
|||
if event.EventType != "completed" || video.Status == "ready" { |
|||
return nil |
|||
} |
|||
if event.FileSize <= 0 || event.FileSize > 100<<20 { |
|||
return errors.New("invalid video size") |
|||
} |
|||
if err := headOSS(video.OssKey); err != nil { |
|||
return err |
|||
} |
|||
updates := map[string]any{"status": "ready", "file_size": event.FileSize} |
|||
if event.Duration > 0 { |
|||
updates["duration"] = event.Duration |
|||
} |
|||
return common.DB.Model(&video).Where("id = ? AND status IN ?", video.ID, []string{"pending", "uploading"}).Updates(updates).Error |
|||
} |
|||
|
|||
func headOSS(objectKey string) error { |
|||
url, err := common.PresignOSS("HEAD", objectKey, 60) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
req, err := http.NewRequest(http.MethodHead, url, nil) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
resp, err := (&http.Client{Timeout: 10 * time.Second}).Do(req) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
defer resp.Body.Close() |
|||
if resp.StatusCode < http.StatusOK || resp.StatusCode >= http.StatusMultipleChoices { |
|||
return fmt.Errorf("oss object unavailable: %s", resp.Status) |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
// CreateUpload 申请视频上传：落库 + 返回 OSS 预签名 PUT URL
|
|||
func (s *VideoService) CreateUpload(userID int64, req *vo.VideoUploadReq) (*vo.VideoUploadVO, *common.BusiError) { |
|||
id, err := tool.NextID() |
|||
if err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
ossKey := fmt.Sprintf("media/%d/%s", id, req.FileName) |
|||
expireAt := time.Now().Add(10 * time.Minute) |
|||
|
|||
video := &model.Video{ |
|||
ID: id, |
|||
UserID: userID, |
|||
DroneSN: req.DroneSN, |
|||
ExecutionID: req.ExecutionID, |
|||
FileName: req.FileName, |
|||
FileSize: req.FileSize, |
|||
OssKey: ossKey, |
|||
OssBucket: common.AppConf.OSS.Bucket, |
|||
Status: "uploading", |
|||
UploadExpireAt: &expireAt, |
|||
CreatedAt: time.Now(), |
|||
} |
|||
if err := common.DB.Create(video).Error; err != nil { |
|||
logger.ERROR("创建视频记录失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
uploadURL, err := common.PresignOSS("PUT", ossKey, 600) |
|||
if err != nil { |
|||
logger.ERROR("生成上传预签名 URL 失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &vo.VideoUploadVO{Video: video, UploadURL: uploadURL, ExpireAt: &expireAt}, nil |
|||
} |
|||
|
|||
// Complete 上传完成确认：更新视频状态为 ready
|
|||
func (s *VideoService) Complete(userID int64, videoID int64, req *vo.VideoCompleteReq) (*model.Video, *common.BusiError) { |
|||
var video model.Video |
|||
if err := common.DB.Where("id = ? AND user_id = ?", videoID, userID).First(&video).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrVideoNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
updates := map[string]any{"status": "ready"} |
|||
if req.FileSize > 0 { |
|||
updates["file_size"] = req.FileSize |
|||
} |
|||
if req.Duration > 0 { |
|||
updates["duration"] = req.Duration |
|||
} |
|||
if err := common.DB.Model(&video).Updates(updates).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
common.DB.First(&video, videoID) |
|||
return &video, nil |
|||
} |
|||
|
|||
// GetPage 视频分页
|
|||
func (s *VideoService) GetPage(userID int64, isAdmin bool, req *vo.VideoPageReq) (*common.PageResponse[model.Video], *common.BusiError) { |
|||
db := common.DB.Model(&model.Video{}).Scopes(withUserFilter(userID, isAdmin)) |
|||
if req.Status != "" { |
|||
db = db.Where("status = ?", req.Status) |
|||
} |
|||
if req.Keyword != "" { |
|||
db = db.Where("file_name LIKE ?", "%"+req.Keyword+"%") |
|||
} |
|||
var total int64 |
|||
if err := db.Count(&total).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
var list []model.Video |
|||
if err := db.Scopes(req.Paginate).Order("id DESC").Find(&list).Error; err != nil { |
|||
return nil, common.ErrInternal |
|||
} |
|||
return common.Page(req.Pagination, total, list), nil |
|||
} |
|||
|
|||
// GetDetail 视频详情
|
|||
func (s *VideoService) GetDetail(userID int64, isAdmin bool, videoID int64) (*model.Video, *common.BusiError) { |
|||
var video model.Video |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&video, videoID).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrVideoNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
return &video, nil |
|||
} |
|||
|
|||
// Download 下载视频：校验状态 → 流量扣减 → OSS 预签名 GET → 下载日志
|
|||
func (s *VideoService) Download(userID int64, isAdmin bool, videoID int64) (*vo.VideoDownloadVO, *common.BusiError) { |
|||
var video model.Video |
|||
if err := common.DB.Scopes(withUserFilter(userID, isAdmin)).First(&video, videoID).Error; err != nil { |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
return nil, common.ErrVideoNotFound |
|||
} |
|||
return nil, common.ErrInternal |
|||
} |
|||
if video.Status != "ready" { |
|||
return nil, common.ErrVideoUploading |
|||
} |
|||
|
|||
// 流量扣减（账本落库）
|
|||
if ok, busiErr := DefaultBillingService.Deduct(userID, video.FileSize, "download", videoID); !ok { |
|||
return nil, busiErr |
|||
} |
|||
|
|||
downloadURL, err := common.PresignOSS("GET", video.OssKey, downloadURLExpire) |
|||
if err != nil { |
|||
logger.ERROR("生成下载预签名 URL 失败", err) |
|||
return nil, common.ErrInternal |
|||
} |
|||
|
|||
log := &model.DownloadLog{ |
|||
ID: mustID(), |
|||
VideoID: videoID, |
|||
UserID: userID, |
|||
Bytes: video.FileSize, |
|||
CreatedAt: time.Now(), |
|||
} |
|||
if err := common.DB.Create(log).Error; err != nil { |
|||
logger.ERROR("下载日志落库失败", err) |
|||
} |
|||
|
|||
return &vo.VideoDownloadVO{ |
|||
DownloadURL: downloadURL, |
|||
ExpireAt: time.Now().Add(downloadURLExpire * time.Second).Unix(), |
|||
Bytes: video.FileSize, |
|||
}, nil |
|||
} |
|||
@ -0,0 +1,129 @@ |
|||
package service |
|||
|
|||
import ( |
|||
"errors" |
|||
"strconv" |
|||
"time" |
|||
|
|||
"gorm.io/gorm" |
|||
|
|||
"laic-backend/common" |
|||
"laic-backend/logger" |
|||
"laic-backend/model" |
|||
"laic-backend/tool" |
|||
) |
|||
|
|||
type WorkflowService struct{} |
|||
|
|||
var DefaultWorkflowService = &WorkflowService{} |
|||
|
|||
// WorkflowStateIn 工作流状态上报载荷
|
|||
type WorkflowStateIn struct { |
|||
CommandID string `json:"commandId"` |
|||
Type string `json:"type"` |
|||
TaskID string `json:"taskId"` |
|||
MissionID string `json:"missionId"` |
|||
State string `json:"state"` |
|||
Step string `json:"step"` |
|||
ResultCode string `json:"resultCode"` |
|||
} |
|||
|
|||
// Upsert 持久化工作流状态（按 command_id 去重更新），并在终态时回写任务执行记录
|
|||
func (s *WorkflowService) Upsert(dockID, requestID string, in *WorkflowStateIn) { |
|||
now := time.Now() |
|||
if requestID != "" { |
|||
var cmd model.DeviceCommandLog |
|||
if err := common.DB.Where("dock_id = ? AND request_id = ? AND status IN ?", dockID, requestID, []string{"sent", "acked"}).First(&cmd).Error; err != nil { |
|||
if !errors.Is(err, gorm.ErrRecordNotFound) { |
|||
logger.ERROR("按 requestId 查询工作流指令失败", err) |
|||
} |
|||
return |
|||
} |
|||
expected := strconv.FormatInt(cmd.ID, 10) |
|||
if in.CommandID != "" && in.CommandID != expected { |
|||
logger.WARN("工作流状态 commandId 与 requestId 不匹配", dockID, requestID) |
|||
return |
|||
} |
|||
in.CommandID = expected |
|||
} |
|||
|
|||
if in.CommandID != "" { |
|||
var existing model.WorkflowState |
|||
err := common.DB.Where("dock_id = ? AND command_id = ?", dockID, in.CommandID).First(&existing).Error |
|||
if errors.Is(err, gorm.ErrRecordNotFound) { |
|||
id, e := tool.NextID() |
|||
if e == nil { |
|||
common.DB.Create(&model.WorkflowState{ |
|||
ID: id, |
|||
DockID: dockID, |
|||
CommandID: in.CommandID, |
|||
Type: in.Type, |
|||
TaskID: in.TaskID, |
|||
MissionID: in.MissionID, |
|||
State: in.State, |
|||
Step: in.Step, |
|||
ResultCode: in.ResultCode, |
|||
UpdatedAt: now, |
|||
}) |
|||
} |
|||
} else if err == nil { |
|||
common.DB.Model(&existing).Updates(map[string]any{ |
|||
"dock_id": dockID, |
|||
"type": in.Type, |
|||
"task_id": in.TaskID, |
|||
"mission_id": in.MissionID, |
|||
"state": in.State, |
|||
"step": in.Step, |
|||
"result_code": in.ResultCode, |
|||
"updated_at": now, |
|||
}) |
|||
} |
|||
} |
|||
|
|||
// 回写 task_execution，优先按 command_id 精确关联，兼容旧设备时只回退到最新未终态记录。
|
|||
if (in.CommandID != "" || in.TaskID != "") && (in.State == "running" || isTerminalWorkflowState(in.State)) { |
|||
updates := map[string]any{"status": in.State} |
|||
if in.State == "running" { |
|||
updates["start_time"] = now |
|||
} else { |
|||
updates["end_time"] = now |
|||
updates["start_time"] = gorm.Expr("COALESCE(start_time, ?)", now) |
|||
} |
|||
|
|||
query := common.DB.Model(&model.TaskExecution{}). |
|||
Where("dock_id = ? AND status NOT IN ('succeeded','failed','cancelled')", dockID) |
|||
if in.CommandID != "" { |
|||
query = query.Where("command_id = ?", in.CommandID) |
|||
} else { |
|||
query = query.Where("task_id = ?", in.TaskID) |
|||
var execution model.TaskExecution |
|||
if err := query.Order("created_at DESC, id DESC").First(&execution).Error; err != nil { |
|||
if !errors.Is(err, gorm.ErrRecordNotFound) { |
|||
logger.ERROR("查找任务执行记录失败", err) |
|||
} |
|||
query = nil |
|||
} else { |
|||
query = common.DB.Model(&execution) |
|||
} |
|||
} |
|||
if query != nil { |
|||
if err := query.Updates(updates).Error; err != nil { |
|||
logger.ERROR("回写任务执行状态失败", err) |
|||
} |
|||
} |
|||
} |
|||
|
|||
// 终态回写轨迹（MySQL trajectory_json + TDengine task_trajectory）
|
|||
if isTerminalWorkflowState(in.State) { |
|||
DefaultTrajectoryStore.Finalize(dockID, DefaultTelemetryStore.DroneSN(dockID)) |
|||
} |
|||
} |
|||
|
|||
func isTerminalWorkflowState(state string) bool { |
|||
switch state { |
|||
case "succeeded", "failed", "cancelled": |
|||
return true |
|||
default: |
|||
return false |
|||
} |
|||
} |
|||
@ -0,0 +1,360 @@ |
|||
-- ============================================================= |
|||
-- 嘉谷低空智控平台 数据库初始化脚本 |
|||
-- 引擎: InnoDB / 字符集: utf8mb4 |
|||
-- 说明: 所有 BIGINT 主键均为 Snowflake 雪花 ID，无 AUTO_INCREMENT |
|||
-- ============================================================= |
|||
|
|||
CREATE DATABASE IF NOT EXISTS laic DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci; |
|||
USE laic; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 5.1 用户 |
|||
-- --------------------------------------------------------------- |
|||
CREATE TABLE IF NOT EXISTS `user` ( |
|||
id BIGINT PRIMARY KEY, |
|||
name VARCHAR(32) NOT NULL, |
|||
phone VARCHAR(32) NOT NULL, |
|||
email VARCHAR(64), |
|||
password VARCHAR(64) NOT NULL, |
|||
role VARCHAR(16) DEFAULT 'user', -- admin / user |
|||
traffic_balance BIGINT DEFAULT 0, -- 云媒体剩余流量(字节)，慢快照，权威在 Redis |
|||
status TINYINT DEFAULT 1, -- 1 正常 / 0 禁用 |
|||
last_login DATETIME, |
|||
created_at DATETIME, |
|||
updated_at DATETIME, |
|||
UNIQUE KEY idx_phone (phone) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 5.2 机巢 & 无人机 |
|||
-- --------------------------------------------------------------- |
|||
CREATE TABLE IF NOT EXISTS dock ( |
|||
id BIGINT PRIMARY KEY, |
|||
user_id BIGINT NOT NULL, |
|||
dock_id VARCHAR(64) NOT NULL, -- MQTT dockId (dock-xxx) |
|||
name VARCHAR(64) DEFAULT '', |
|||
code VARCHAR(32) DEFAULT '', |
|||
sn VARCHAR(64) DEFAULT '', |
|||
iccid VARCHAR(32) DEFAULT '', -- 冗余，权威见 sim_card.iccid |
|||
longitude DOUBLE DEFAULT 0, |
|||
latitude DOUBLE DEFAULT 0, |
|||
altitude DOUBLE DEFAULT 0, |
|||
location VARCHAR(128) DEFAULT '', |
|||
status VARCHAR(32) DEFAULT 'offline', |
|||
register_status VARCHAR(16) DEFAULT 'pending', -- pending/registered |
|||
dock_id_source VARCHAR(32) DEFAULT '', |
|||
software_ver VARCHAR(16) DEFAULT '', |
|||
protocol_ver VARCHAR(8) DEFAULT '1.0', |
|||
created_at DATETIME, |
|||
updated_at DATETIME, |
|||
UNIQUE KEY idx_dock_id (dock_id) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS drone ( |
|||
id BIGINT PRIMARY KEY, |
|||
user_id BIGINT NOT NULL, |
|||
drone_sn VARCHAR(32) NOT NULL, -- 飞控 HW_SN_NUM |
|||
dock_id VARCHAR(64) NOT NULL, -- 绑定的机巢 dock_id（1:1） |
|||
name VARCHAR(64) DEFAULT '', |
|||
code VARCHAR(32) DEFAULT '', |
|||
model VARCHAR(32) DEFAULT '', |
|||
status VARCHAR(32) DEFAULT 'offline', |
|||
battery INT DEFAULT 0, |
|||
firmware_ver VARCHAR(16) DEFAULT '', |
|||
created_at DATETIME, |
|||
updated_at DATETIME, |
|||
UNIQUE KEY idx_drone_sn (drone_sn), |
|||
UNIQUE KEY idx_drone_dock (dock_id) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 5.3 告警 |
|||
-- --------------------------------------------------------------- |
|||
CREATE TABLE IF NOT EXISTS alarm_code ( |
|||
id BIGINT PRIMARY KEY, |
|||
code VARCHAR(64) NOT NULL, |
|||
category VARCHAR(32) NOT NULL, -- dock / drone |
|||
message_cn VARCHAR(256) NOT NULL, |
|||
level VARCHAR(16) DEFAULT 'critical', |
|||
source VARCHAR(128) DEFAULT '', |
|||
created_at DATETIME, |
|||
updated_at DATETIME, |
|||
UNIQUE KEY idx_code (code) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS alarm ( |
|||
id BIGINT PRIMARY KEY, |
|||
dock_id VARCHAR(64) NOT NULL, |
|||
code VARCHAR(64) NOT NULL, |
|||
device_type VARCHAR(8) NOT NULL, -- dock / drone |
|||
message_cn VARCHAR(256) NOT NULL, |
|||
level VARCHAR(16) NOT NULL, |
|||
status VARCHAR(16) DEFAULT 'active', -- active / acknowledged / resolved |
|||
triggered_at DATETIME, |
|||
resolved_at DATETIME, |
|||
created_at DATETIME, |
|||
KEY idx_alarm_dock (dock_id, status) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 5.4 指令 & 工作流 |
|||
-- --------------------------------------------------------------- |
|||
CREATE TABLE IF NOT EXISTS device_command_log ( |
|||
id BIGINT PRIMARY KEY, -- = commandId (Snowflake) |
|||
dock_id VARCHAR(64) NOT NULL, |
|||
command_type VARCHAR(64) NOT NULL, -- dock.open / drone.takeoff / workflow.start_task |
|||
params JSON, |
|||
drone_sn VARCHAR(32), |
|||
request_id VARCHAR(64), |
|||
ttl_ms INT DEFAULT 30000, |
|||
ack_accepted TINYINT, |
|||
ack_result_code VARCHAR(64), |
|||
status VARCHAR(16) DEFAULT 'sent', -- sent / acked / timeout / terminal |
|||
retry_count INT DEFAULT 0, |
|||
sent_at DATETIME, |
|||
acked_at DATETIME, |
|||
created_at DATETIME, |
|||
KEY idx_cmd_log_dock (dock_id, created_at) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS workflow_state ( |
|||
id BIGINT PRIMARY KEY, |
|||
dock_id VARCHAR(64) NOT NULL, |
|||
command_id VARCHAR(64), |
|||
type VARCHAR(64), |
|||
task_id VARCHAR(128), |
|||
mission_id VARCHAR(128), |
|||
state VARCHAR(16) NOT NULL, -- idle / running / succeeded / failed / cancelled |
|||
step VARCHAR(64) NOT NULL, |
|||
result_code VARCHAR(64), |
|||
updated_at DATETIME |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 5.5 固件管理 |
|||
-- --------------------------------------------------------------- |
|||
CREATE TABLE IF NOT EXISTS firmware ( |
|||
id BIGINT PRIMARY KEY, |
|||
component VARCHAR(32) NOT NULL, -- 固定 dock-edge-agent |
|||
version VARCHAR(32) NOT NULL, -- 1.3.0 |
|||
description VARCHAR(512) DEFAULT '', |
|||
file_url VARCHAR(512) NOT NULL, |
|||
sha256 VARCHAR(128) NOT NULL, |
|||
signature TEXT NOT NULL, |
|||
file_size BIGINT DEFAULT 0, |
|||
mandatory TINYINT DEFAULT 0, |
|||
status VARCHAR(16) DEFAULT 'draft', -- draft/released/deprecated |
|||
created_by BIGINT NOT NULL, |
|||
created_at DATETIME, |
|||
updated_at DATETIME, |
|||
UNIQUE KEY idx_component_ver (component, version) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 5.6 任务 & 航线 |
|||
-- --------------------------------------------------------------- |
|||
CREATE TABLE IF NOT EXISTS task_plan ( |
|||
id VARCHAR(128) PRIMARY KEY, |
|||
user_id BIGINT NOT NULL, |
|||
name VARCHAR(128) NOT NULL, |
|||
dock_id VARCHAR(64) NOT NULL, |
|||
route_id BIGINT, |
|||
schedule_type VARCHAR(16) DEFAULT 'once', -- once / cron |
|||
schedule_cron VARCHAR(32), |
|||
video_policy VARCHAR(16) DEFAULT 'raw', |
|||
status VARCHAR(16) DEFAULT 'draft', |
|||
created_by BIGINT, |
|||
created_at DATETIME, |
|||
updated_at DATETIME |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS task_execution ( |
|||
id BIGINT PRIMARY KEY, |
|||
task_id VARCHAR(128), |
|||
command_id VARCHAR(64), |
|||
dock_id VARCHAR(64) NOT NULL, |
|||
drone_sn VARCHAR(32), |
|||
start_time DATETIME, |
|||
end_time DATETIME, |
|||
status VARCHAR(16) DEFAULT 'pending', |
|||
trajectory_json JSON, |
|||
created_at DATETIME |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS route ( |
|||
id BIGINT PRIMARY KEY, |
|||
user_id BIGINT NOT NULL, |
|||
name VARCHAR(128) NOT NULL, |
|||
description VARCHAR(256), |
|||
created_at DATETIME, |
|||
updated_at DATETIME |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS route_waypoint ( |
|||
id BIGINT PRIMARY KEY, |
|||
route_id BIGINT NOT NULL, |
|||
seq INT NOT NULL, |
|||
longitude DOUBLE, |
|||
latitude DOUBLE, |
|||
altitude DOUBLE, |
|||
speed DOUBLE, |
|||
yaw DOUBLE, |
|||
hold_sec INT DEFAULT 0, |
|||
created_at DATETIME, |
|||
UNIQUE KEY uk_route_seq (route_id, seq) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 5.7 媒体 |
|||
-- --------------------------------------------------------------- |
|||
CREATE TABLE IF NOT EXISTS live_session ( |
|||
id VARCHAR(128) PRIMARY KEY, -- streamSessionId |
|||
dock_id VARCHAR(64) NOT NULL, |
|||
provider VARCHAR(16) DEFAULT 'aliyun', |
|||
stream_name VARCHAR(128) NOT NULL, -- 直播流名（每次会话唯一） |
|||
push_url_hash VARCHAR(128), |
|||
expires_at BIGINT NOT NULL, |
|||
max_bitrate_bps BIGINT DEFAULT 1500000, |
|||
phase VARCHAR(16) DEFAULT 'idle', |
|||
stop_reason VARCHAR(32), |
|||
error_code VARCHAR(64), |
|||
requested_by BIGINT, |
|||
stop_deadline DATETIME, |
|||
device_state_version BIGINT DEFAULT 0, |
|||
device_event_id VARCHAR(128), |
|||
device_updated_at BIGINT DEFAULT 0, |
|||
cloud_confirmed_at DATETIME, |
|||
started_at DATETIME, |
|||
stopped_at DATETIME, |
|||
created_at DATETIME, |
|||
updated_at DATETIME, |
|||
active_dock_id VARCHAR(64) GENERATED ALWAYS AS ( |
|||
CASE WHEN phase IN ('starting', 'streaming', 'reconnecting', 'stopping') THEN dock_id ELSE NULL END |
|||
) STORED, |
|||
KEY idx_live_session_dock_phase (dock_id, phase, created_at), |
|||
KEY idx_live_session_phase_deadline (phase, stop_deadline), |
|||
UNIQUE KEY uk_live_session_device_event (device_event_id), |
|||
UNIQUE KEY uk_live_session_active_dock (active_dock_id) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS live_viewer_lease ( |
|||
id BIGINT PRIMARY KEY, |
|||
stream_session_id VARCHAR(128) NOT NULL, |
|||
viewer_id BIGINT NOT NULL, |
|||
expires_at DATETIME NOT NULL, |
|||
released_at DATETIME NULL, |
|||
created_at DATETIME NULL, |
|||
updated_at DATETIME NULL, |
|||
UNIQUE KEY uk_live_viewer (stream_session_id, viewer_id), |
|||
KEY idx_live_lease_expire (stream_session_id, expires_at), |
|||
KEY idx_live_lease_active (stream_session_id, released_at, expires_at), |
|||
CONSTRAINT fk_live_lease_session FOREIGN KEY (stream_session_id) REFERENCES live_session(id) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS video ( |
|||
id BIGINT PRIMARY KEY, |
|||
user_id BIGINT NOT NULL, |
|||
drone_sn VARCHAR(32), |
|||
execution_id BIGINT, |
|||
file_name VARCHAR(256), |
|||
file_size BIGINT, |
|||
duration INT, |
|||
oss_key VARCHAR(256), |
|||
oss_bucket VARCHAR(128), -- 预签名上传的目标 bucket |
|||
thumbnail_key VARCHAR(256), |
|||
status VARCHAR(16) DEFAULT 'uploading', |
|||
upload_expire_at DATETIME, -- 预签名 PUT URL 过期时间 |
|||
created_at DATETIME |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS download_log ( |
|||
id BIGINT PRIMARY KEY, |
|||
video_id BIGINT NOT NULL, |
|||
user_id BIGINT NOT NULL, |
|||
bytes BIGINT NOT NULL, |
|||
created_at DATETIME |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 5.8 计费 |
|||
-- --------------------------------------------------------------- |
|||
CREATE TABLE IF NOT EXISTS traffic_order ( |
|||
id BIGINT PRIMARY KEY, |
|||
user_id BIGINT NOT NULL, |
|||
amount_gb INT NOT NULL, -- 购买流量(GB) |
|||
unit_price DECIMAL(6,2), |
|||
total_price DECIMAL(10,2), |
|||
pay_status VARCHAR(16) DEFAULT 'unpaid', -- unpaid/paid/cancelled |
|||
created_at DATETIME, |
|||
paid_at DATETIME |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS traffic_usage_log ( |
|||
id BIGINT PRIMARY KEY, |
|||
user_id BIGINT NOT NULL, |
|||
source_type VARCHAR(16), -- live / replay / download |
|||
source_id BIGINT, |
|||
bytes_used BIGINT, |
|||
balance_before BIGINT, -- 扣减前余额(字节) |
|||
balance_after BIGINT, -- 扣减后余额(字节) |
|||
created_at DATETIME |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS sim_card ( |
|||
id BIGINT PRIMARY KEY, |
|||
user_id BIGINT NOT NULL, |
|||
dock_id VARCHAR(64), |
|||
carrier VARCHAR(16), -- 移动/联通/电信 |
|||
phone VARCHAR(16), |
|||
iccid VARCHAR(32), -- 权威字段（查询/充值的入参） |
|||
plan_gb INT, |
|||
used_gb DECIMAL(10,4) DEFAULT 0, |
|||
expired_at DATE, |
|||
status VARCHAR(16) DEFAULT 'active', -- 平台业务态: active/expired |
|||
carrier_status VARCHAR(16) DEFAULT 'normal', -- 运营商侧状态: normal/suspended/arrears/cancelled |
|||
last_sync_at DATETIME, |
|||
created_at DATETIME, |
|||
updated_at DATETIME |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS sim_recharge_log ( |
|||
id BIGINT PRIMARY KEY, |
|||
sim_card_id BIGINT NOT NULL, |
|||
user_id BIGINT NOT NULL, |
|||
amount_gb INT NOT NULL, |
|||
unit_price DECIMAL(6,2), |
|||
total_price DECIMAL(10,2), |
|||
pay_status VARCHAR(16) DEFAULT 'paid', |
|||
recharge_status VARCHAR(16) DEFAULT 'pending', -- pending/success/failed |
|||
recharge_msg VARCHAR(256) DEFAULT '', |
|||
idempotent_key VARCHAR(64) NOT NULL, -- 平台充值单号，超时重试去重 |
|||
created_at DATETIME, |
|||
UNIQUE KEY uk_recharge_idem (idempotent_key) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
CREATE TABLE IF NOT EXISTS sim_usage_record ( |
|||
id BIGINT PRIMARY KEY, |
|||
sim_card_id BIGINT NOT NULL, |
|||
plan_gb INT, |
|||
used_gb DECIMAL(10,4), |
|||
remain_gb DECIMAL(10,4), |
|||
carrier_status VARCHAR(16), |
|||
synced_at DATETIME, |
|||
KEY idx_record_sim (sim_card_id, synced_at) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 7.7 操作日志 |
|||
-- --------------------------------------------------------------- |
|||
CREATE TABLE IF NOT EXISTS operation_log ( |
|||
id BIGINT PRIMARY KEY, |
|||
user_id BIGINT NOT NULL, |
|||
user_name VARCHAR(32), |
|||
module VARCHAR(32), -- 模块（用户管理/机巢管理/任务管理...） |
|||
action VARCHAR(64), -- 动作（新增/修改/删除/下发指令...） |
|||
detail VARCHAR(256), -- 详情（请求路径） |
|||
result VARCHAR(16), -- success / failed |
|||
ip VARCHAR(64), |
|||
created_at DATETIME, |
|||
KEY idx_op_user (user_id, created_at) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
@ -0,0 +1,144 @@ |
|||
-- ============================================================= |
|||
-- 告警码字典初始化数据 |
|||
-- 说明: 完整枚举对齐 MQTT 协议文档 7.6 节（共 106 条）。 |
|||
-- id 为固定字典 ID，非雪花 ID。按 code 唯一键 INSERT IGNORE，可重复执行。 |
|||
-- category: dock（PLC/机巢/执行机构/电池/充电/空调/链路） / drone（飞控/MAVLink/定位） |
|||
-- ============================================================= |
|||
|
|||
USE laic; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 7.6.2 PLC / 机巢 / 执行机构故障（dock） |
|||
-- --------------------------------------------------------------- |
|||
INSERT IGNORE INTO alarm_code (id, code, category, message_cn, level, source, created_at, updated_at) VALUES |
|||
(1001, 'DOCK_EMERGENCY_STOP', 'dock', '机巢急停已触发', 'emergency', 'dock', NOW(), NOW()), |
|||
(1002, 'DOCK_PLC_ALARM_UNKNOWN', 'dock', 'PLC 总报警已触发，但未解析到具体故障', 'critical', 'dock', NOW(), NOW()), |
|||
(1003, 'CHARGING_ROD_MOTOR_FAULT', 'dock', '充电杆电机报警', 'critical', 'dock', NOW(), NOW()), |
|||
(1004, 'CENTERING_ROD_MOTOR_FAULT', 'dock', '归中杆电机报警', 'critical', 'dock', NOW(), NOW()), |
|||
(1005, 'DRONE_POWER_OFF_TIMEOUT', 'dock', '无人机关机超时', 'critical', 'dock', NOW(), NOW()), |
|||
(1006, 'DRONE_POWER_ON_TIMEOUT', 'dock', '无人机开机超时', 'critical', 'dock', NOW(), NOW()), |
|||
(1007, 'DRONE_POSITION_DETECT_TIMEOUT', 'dock', '无人机在位检测超时', 'critical', 'dock', NOW(), NOW()), |
|||
(1008, 'LEFT_DOOR_POWER_FAULT', 'dock', '左顶门动力故障', 'critical', 'dock', NOW(), NOW()), |
|||
(1009, 'LEFT_DOOR_CLOSE_TIMEOUT', 'dock', '左顶门关门超时或关门传感器未感应', 'critical', 'dock', NOW(), NOW()), |
|||
(1010, 'LEFT_DOOR_OPEN_TIMEOUT', 'dock', '左顶门开门超时或开门传感器未感应', 'critical', 'dock', NOW(), NOW()), |
|||
(1011, 'LEFT_DOOR_RESET_TIMEOUT', 'dock', '左顶门超过时间复位未完成', 'critical', 'dock', NOW(), NOW()), |
|||
(1012, 'LEFT_DOOR_NOT_RESET_ACTION_BLOCKED', 'dock', '左顶门电机未复位完成，动作已阻止', 'critical', 'dock', NOW(), NOW()), |
|||
(1013, 'RIGHT_DOOR_POWER_FAULT', 'dock', '右顶门动力故障', 'critical', 'dock', NOW(), NOW()), |
|||
(1014, 'RIGHT_DOOR_CLOSE_TIMEOUT', 'dock', '右顶门关门超时或关门传感器未感应', 'critical', 'dock', NOW(), NOW()), |
|||
(1015, 'RIGHT_DOOR_OPEN_TIMEOUT', 'dock', '右顶门开门超时或开门传感器未感应', 'critical', 'dock', NOW(), NOW()), |
|||
(1016, 'RIGHT_DOOR_RESET_TIMEOUT', 'dock', '右顶门超过时间复位未完成', 'critical', 'dock', NOW(), NOW()), |
|||
(1017, 'RIGHT_DOOR_NOT_RESET_ACTION_BLOCKED', 'dock', '右顶门电机未复位完成，动作已阻止', 'critical', 'dock', NOW(), NOW()), |
|||
(1018, 'X_CENTER_POWER_FAULT', 'dock', '左右归中动力故障', 'critical', 'dock', NOW(), NOW()), |
|||
(1019, 'X_CENTER_OPEN_TIMEOUT', 'dock', '左右归中打开超时或传感器未感应', 'critical', 'dock', NOW(), NOW()), |
|||
(1020, 'X_CENTER_CLAMP_TIMEOUT', 'dock', '左右归中夹紧超时或传感器未感应', 'critical', 'dock', NOW(), NOW()), |
|||
(1021, 'X_CENTER_RESET_TIMEOUT', 'dock', '左右归中超过时间复位未完成', 'critical', 'dock', NOW(), NOW()), |
|||
(1022, 'X_CENTER_NOT_RESET_ACTION_BLOCKED', 'dock', '左右归中电机未复位完成，动作已阻止', 'critical', 'dock', NOW(), NOW()), |
|||
(1023, 'Y_CENTER_POWER_FAULT', 'dock', '前后归中动力故障', 'critical', 'dock', NOW(), NOW()), |
|||
(1024, 'Y_CENTER_OPEN_TIMEOUT', 'dock', '前后归中打开超时或传感器未感应', 'critical', 'dock', NOW(), NOW()), |
|||
(1025, 'Y_CENTER_CLAMP_TIMEOUT', 'dock', '前后归中夹紧超时或传感器未感应', 'critical', 'dock', NOW(), NOW()), |
|||
(1026, 'Y_CENTER_RESET_TIMEOUT', 'dock', '前后归中超过时间复位未完成', 'critical', 'dock', NOW(), NOW()), |
|||
(1027, 'Y_CENTER_NOT_RESET_ACTION_BLOCKED', 'dock', '前后归中电机未复位完成，动作已阻止', 'critical', 'dock', NOW(), NOW()), |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 7.6.3 电池保护和电池报警（dock） |
|||
-- --------------------------------------------------------------- |
|||
(1028, 'BATTERY_NOT_PRESENT', 'dock', '电池未在位，请检查系留接口连接状态', 'warning', 'dock', NOW(), NOW()), |
|||
(1029, 'BATTERY_CELL_OVER_VOLTAGE', 'dock', '电池单体过压保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1030, 'BATTERY_CELL_UNDER_VOLTAGE', 'dock', '电池单体欠压保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1031, 'BATTERY_PACK_OVER_VOLTAGE', 'dock', '电池整组过压保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1032, 'BATTERY_PACK_UNDER_VOLTAGE', 'dock', '电池整组欠压保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1033, 'BATTERY_CHARGE_OVER_TEMPERATURE', 'dock', '电池充电过温保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1034, 'BATTERY_CHARGE_UNDER_TEMPERATURE', 'dock', '电池充电低温保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1035, 'BATTERY_DISCHARGE_OVER_TEMPERATURE', 'dock', '电池放电过温保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1036, 'BATTERY_DISCHARGE_UNDER_TEMPERATURE', 'dock', '电池放电低温保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1037, 'BATTERY_CHARGE_OVER_CURRENT', 'dock', '电池充电过流保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1038, 'BATTERY_DISCHARGE_OVER_CURRENT', 'dock', '电池放电过流保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1039, 'BATTERY_SHORT_CIRCUIT', 'dock', '电池短路保护', 'critical', 'dock', NOW(), NOW()), |
|||
(1040, 'BATTERY_FRONT_IC_FAULT', 'dock', '电池前端检测 IC 错误', 'critical', 'dock', NOW(), NOW()), |
|||
(1041, 'BATTERY_MOS_LOCKED', 'dock', '电池保护板软件锁定 MOS', 'critical', 'dock', NOW(), NOW()), |
|||
(1042, 'BATTERY_WIRE_BREAK', 'dock', '电池断线告警', 'critical', 'dock', NOW(), NOW()), |
|||
(1043, 'BATTERY_SECONDARY_OVER_VOLTAGE', 'dock', '电池二次过压告警', 'critical', 'dock', NOW(), NOW()), |
|||
(1044, 'BATTERY_SELF_LOCK_SWITCH_OPEN', 'dock', '电池自锁开关断开', 'warning', 'dock', NOW(), NOW()), |
|||
(1045, 'BATTERY_ALARM_UNKNOWN', 'dock', '电池模块报警，具体原因未知', 'critical', 'dock', NOW(), NOW()), |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 7.6.4 充电电源故障（dock） |
|||
-- --------------------------------------------------------------- |
|||
(1046, 'CHARGER_OFFLINE', 'dock', '充电电源未在线，请检查电源和控制盒连接状态', 'critical', 'dock', NOW(), NOW()), |
|||
(1047, 'POWER_INPUT_UNDER_VOLTAGE', 'dock', '充电电源交流输入欠压', 'critical', 'dock', NOW(), NOW()), |
|||
(1048, 'POWER_INPUT_OVER_VOLTAGE', 'dock', '充电电源交流输入过压', 'critical', 'dock', NOW(), NOW()), |
|||
(1049, 'POWER_OUTPUT_OVER_CURRENT', 'dock', '充电电源直流输出过流', 'critical', 'dock', NOW(), NOW()), |
|||
(1050, 'POWER_VOLTAGE_SAMPLING_FAULT', 'dock', '充电电源电压采样异常', 'critical', 'dock', NOW(), NOW()), |
|||
(1051, 'POWER_OVER_TEMPERATURE_OR_FAN_FAULT', 'dock', '充电电源超高温或风扇故障', 'critical', 'dock', NOW(), NOW()), |
|||
(1052, 'POWER_COMMUNICATION_TIMEOUT', 'dock', '充电电源主控通信超时', 'critical', 'dock', NOW(), NOW()), |
|||
(1053, 'CHARGER_FAULT', 'dock', '充电器故障', 'critical', 'dock', NOW(), NOW()), |
|||
(1054, 'POWER_LOAD_DISCONNECTED_SHUTDOWN', 'dock', '充电电源负载断开并关机', 'critical', 'dock', NOW(), NOW()), |
|||
(1055, 'POWER_OVER_VOLTAGE_SHUTDOWN', 'dock', '充电电源过压关机', 'critical', 'dock', NOW(), NOW()), |
|||
(1056, 'BATTERY_REVERSED_OR_OTHER_POWER_FAULT', 'dock', '电池接反或充电电源其他故障', 'critical', 'dock', NOW(), NOW()), |
|||
(1057, 'POWER_NTC_FAULT', 'dock', '充电电源 NTC 故障并停机', 'critical', 'dock', NOW(), NOW()), |
|||
(1058, 'POWER_SHORT_CIRCUIT', 'dock', '充电电源输出短路', 'critical', 'dock', NOW(), NOW()), |
|||
(1059, 'CHARGING_TIMEOUT', 'dock', '充电超时', 'critical', 'dock', NOW(), NOW()), |
|||
(1060, 'POWER_ALARM_UNKNOWN', 'dock', '充电电源未知报警', 'critical', 'dock', NOW(), NOW()), |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 7.6.5 空调故障（dock） |
|||
-- --------------------------------------------------------------- |
|||
(1061, 'AC_INSIDE_TEMPERATURE_SENSOR_FAULT', 'dock', '空调柜内温度传感器故障', 'warning', 'dock', NOW(), NOW()), |
|||
(1062, 'AC_INSIDE_HIGH_TEMPERATURE', 'dock', '空调柜内高温告警', 'warning', 'dock', NOW(), NOW()), |
|||
(1063, 'AC_INSIDE_LOW_TEMPERATURE', 'dock', '空调柜内低温告警', 'warning', 'dock', NOW(), NOW()), |
|||
(1064, 'AC_OUTSIDE_TEMPERATURE_SENSOR_FAULT', 'dock', '空调柜外温度传感器故障', 'warning', 'dock', NOW(), NOW()), |
|||
(1065, 'AC_OUTSIDE_HIGH_TEMPERATURE', 'dock', '空调柜外高温告警', 'warning', 'dock', NOW(), NOW()), |
|||
(1066, 'AC_OUTSIDE_LOW_TEMPERATURE', 'dock', '空调柜外低温告警', 'warning', 'dock', NOW(), NOW()), |
|||
(1067, 'AC_INSIDE_HUMIDITY_SENSOR_FAULT', 'dock', '空调柜内湿度传感器故障', 'warning', 'dock', NOW(), NOW()), |
|||
(1068, 'AC_INSIDE_HIGH_HUMIDITY', 'dock', '空调柜内高湿告警', 'warning', 'dock', NOW(), NOW()), |
|||
(1069, 'AC_INSIDE_LOW_HUMIDITY', 'dock', '空调柜内低湿告警', 'warning', 'dock', NOW(), NOW()), |
|||
(1070, 'AC_OUTSIDE_HUMIDITY_SENSOR_FAULT', 'dock', '空调柜外湿度传感器故障', 'warning', 'dock', NOW(), NOW()), |
|||
(1071, 'AC_OUTSIDE_HIGH_HUMIDITY', 'dock', '空调柜外高湿告警', 'warning', 'dock', NOW(), NOW()), |
|||
(1072, 'AC_OUTSIDE_LOW_HUMIDITY', 'dock', '空调柜外低湿告警', 'warning', 'dock', NOW(), NOW()), |
|||
(1073, 'AC_COMPRESSOR_HIGH_PRESSURE', 'dock', '空调压缩机高压告警', 'critical', 'dock', NOW(), NOW()), |
|||
(1074, 'AC_COMPRESSOR_LOW_PRESSURE', 'dock', '空调压缩机低压告警', 'critical', 'dock', NOW(), NOW()), |
|||
(1075, 'AC_COOLING_FAILURE', 'dock', '空调制冷失效告警', 'critical', 'dock', NOW(), NOW()), |
|||
(1076, 'AC_HEATING_FAILURE', 'dock', '空调制热失效告警', 'critical', 'dock', NOW(), NOW()), |
|||
(1077, 'AC_INSIDE_COIL_TEMPERATURE_SENSOR_FAULT','dock', '空调内盘管温度传感器故障', 'warning', 'dock', NOW(), NOW()), |
|||
(1078, 'AC_INSIDE_COIL_LOW_TEMPERATURE', 'dock', '空调内盘管低温告警', 'warning', 'dock', NOW(), NOW()), |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 7.6.6 通信和边缘服务故障（dock 部分） |
|||
-- --------------------------------------------------------------- |
|||
(1079, 'DOCK_MODBUS_DISCONNECTED', 'dock', '机巢 PLC Modbus 通信中断', 'critical', 'dock', NOW(), NOW()), |
|||
(1080, 'PLC_HEARTBEAT_TIMEOUT', 'dock', '机巢 PLC 心跳超时', 'critical', 'dock', NOW(), NOW()), |
|||
(1081, 'ONE_KEY_TAKEOFF_TIMEOUT', 'dock', '一键起飞超时', 'critical', 'dock', NOW(), NOW()), |
|||
(1082, 'ONE_KEY_LANDING_TIMEOUT', 'dock', '一键降落超时', 'critical', 'dock', NOW(), NOW()); |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 7.6.6 通信和边缘服务故障（drone 部分） |
|||
-- --------------------------------------------------------------- |
|||
INSERT IGNORE INTO alarm_code (id, code, category, message_cn, level, source, created_at, updated_at) VALUES |
|||
(2001, 'DRONE_MAVLINK_DISCONNECTED', 'drone', '无人机 MAVLink 通信中断', 'critical', 'drone', NOW(), NOW()), |
|||
(2002, 'DRONE_SN_UNAVAILABLE', 'drone', '无法读取无人机永久序列号', 'critical', 'drone', NOW(), NOW()), |
|||
(2003, 'MISSION_STORE_WRITE_FAILED', 'drone', '当前航线无法持久化', 'warning', 'drone', NOW(), NOW()), |
|||
(2004, 'DRONE_BATTERY_CRITICAL', 'drone', '无人机电池电量严重不足', 'critical', 'drone', NOW(), NOW()), |
|||
(2005, 'DRONE_GPS_INVALID', 'drone', '无人机定位不可用或质量不足', 'warning', 'drone', NOW(), NOW()), |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 7.6.7 无人机飞控告警（drone） |
|||
-- --------------------------------------------------------------- |
|||
(2006, 'DRONE_GCS_LINK_LOST', 'drone', '飞控与地面站通信链路中断', 'critical', 'drone', NOW(), NOW()), |
|||
(2007, 'DRONE_BATTERY_VOLTAGE_LOW', 'drone', '无人机电池电压低', 'critical', 'drone', NOW(), NOW()), |
|||
(2008, 'DRONE_SERVO_OUTPUT_IMBALANCE', 'drone', '电机或舵机输出不平衡', 'critical', 'drone', NOW(), NOW()), |
|||
(2009, 'DRONE_SERVO_OUTPUT_FAULT', 'drone', '电机或舵机动力输出故障', 'critical', 'drone', NOW(), NOW()), |
|||
(2010, 'DRONE_AUTOPILOT_OVER_TEMPERATURE', 'drone', '飞控系统温度过高', 'critical', 'drone', NOW(), NOW()), |
|||
(2011, 'DRONE_INS_POSITIONING_INVALID', 'drone', '飞控 INS 定位解算未就绪或不可用', 'critical', 'drone', NOW(), NOW()), |
|||
(2012, 'DRONE_OUTSIDE_GEOFENCE', 'drone', '无人机超出电子围栏范围', 'critical', 'drone', NOW(), NOW()), |
|||
(2013, 'DRONE_IMU_DATA_OUT_OF_RANGE', 'drone', 'IMU 数据超出范围', 'critical', 'drone', NOW(), NOW()), |
|||
(2014, 'DRONE_EXCESSIVE_TILT', 'drone', '无人机倾斜姿态过大', 'critical', 'drone', NOW(), NOW()), |
|||
(2015, 'DRONE_SPEED_OUT_OF_RANGE', 'drone', '无人机速度超出范围', 'critical', 'drone', NOW(), NOW()), |
|||
(2016, 'DRONE_RC_NOT_READY', 'drone', '遥控器数据未就绪', 'warning', 'drone', NOW(), NOW()), |
|||
(2017, 'DRONE_MAG0_INTERFERENCE', 'drone', '1 号磁力计受到磁干扰', 'warning', 'drone', NOW(), NOW()), |
|||
(2018, 'DRONE_MAG1_INTERFERENCE', 'drone', '2 号磁力计受到磁干扰', 'warning', 'drone', NOW(), NOW()), |
|||
(2019, 'DRONE_IMU0_FAULT', 'drone', '1 号 IMU 数据异常', 'critical', 'drone', NOW(), NOW()), |
|||
(2020, 'DRONE_IMU1_FAULT', 'drone', '2 号 IMU 数据异常', 'critical', 'drone', NOW(), NOW()), |
|||
(2021, 'DRONE_BAROMETER_FAULT', 'drone', '气压计数据异常', 'critical', 'drone', NOW(), NOW()), |
|||
(2022, 'DRONE_GPS0_FAULT', 'drone', '1 号普通 GPS 数据异常', 'warning', 'drone', NOW(), NOW()), |
|||
(2023, 'DRONE_GPS1_FAULT', 'drone', '2 号普通 GPS 数据异常', 'warning', 'drone', NOW(), NOW()), |
|||
(2024, 'DRONE_RTK_FAULT', 'drone', 'RTK 板卡或数据异常', 'warning', 'drone', NOW(), NOW()); |
|||
@ -0,0 +1,73 @@ |
|||
-- ============================================================= |
|||
-- 种子数据 |
|||
-- ============================================================= |
|||
|
|||
USE laic; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 默认管理员账号 |
|||
-- 手机号: 13800000000 |
|||
-- 密码: admin123456 (bcrypt 加密) |
|||
-- ⚠ 上线前务必修改默认密码 |
|||
-- --------------------------------------------------------------- |
|||
INSERT INTO `user` (id, name, phone, email, password, role, traffic_balance, status, created_at, updated_at) |
|||
VALUES (1, '管理员', '13800000000', 'admin@laic.local', |
|||
'$2a$10$vpTDi8nUl/Lyb4KBjtHLBevsCQ08FB46bLgJG3JxVACzqYOWD3YSq', |
|||
'admin', 0, 1, NOW(), NOW()) |
|||
ON DUPLICATE KEY UPDATE role = 'admin'; |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 演示业务账号（远程监控 / 设备控制走普通用户） |
|||
-- 手机号: 13900000000 |
|||
-- 密码: admin123456 |
|||
-- --------------------------------------------------------------- |
|||
INSERT INTO `user` (id, name, phone, email, password, role, traffic_balance, status, created_at, updated_at) |
|||
VALUES (2, '演示用户', '13900000000', 'demo@laic.local', |
|||
'$2a$10$vpTDi8nUl/Lyb4KBjtHLBevsCQ08FB46bLgJG3JxVACzqYOWD3YSq', |
|||
'user', 10737418240, 1, NOW(), NOW()) |
|||
ON DUPLICATE KEY UPDATE |
|||
password = VALUES(password), |
|||
role = 'user', |
|||
status = 1; |
|||
|
|||
SET @demo_user_id = (SELECT id FROM `user` WHERE phone = '13900000000'); |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 演示机巢 / 无人机（与 mock 设备 dock-1/2/3 对应） |
|||
-- 若 MQTT 已自发现同 dock_id 的 pending 记录，则认领并补全名称坐标。 |
|||
-- --------------------------------------------------------------- |
|||
INSERT INTO dock (id, user_id, dock_id, name, code, sn, iccid, longitude, latitude, altitude, location, status, register_status, dock_id_source, software_ver, protocol_ver, created_at, updated_at) |
|||
VALUES |
|||
(1001, @demo_user_id, 'dock-1', '青山湖 01 号机巢', 'JG-DOCK-001', 'dock-1', '8986000000000000001', 119.802631, 30.262438, 15, '青山湖科技城 A 区', 'offline', 'registered', 'seed', '1.3.0', '1.0', NOW(), NOW()), |
|||
(1002, @demo_user_id, 'dock-2', '青山湖 02 号机巢', 'JG-DOCK-002', 'dock-2', '8986000000000000002', 119.818264, 30.255917, 18, '青山湖科技城 B 区', 'offline', 'registered', 'seed', '1.3.0', '1.0', NOW(), NOW()), |
|||
(1003, @demo_user_id, 'dock-3', '青山湖 03 号机巢', 'JG-DOCK-003', 'dock-3', '8986000000000000003', 119.791450, 30.248610, 12, '青山湖科技城 C 区', 'offline', 'registered', 'seed', '1.3.0', '1.0', NOW(), NOW()) |
|||
ON DUPLICATE KEY UPDATE |
|||
user_id = @demo_user_id, |
|||
name = VALUES(name), |
|||
code = VALUES(code), |
|||
sn = VALUES(sn), |
|||
iccid = VALUES(iccid), |
|||
longitude = VALUES(longitude), |
|||
latitude = VALUES(latitude), |
|||
location = VALUES(location), |
|||
register_status = 'registered'; |
|||
|
|||
INSERT INTO drone (id, user_id, drone_sn, dock_id, name, code, model, status, battery, firmware_ver, created_at, updated_at) |
|||
VALUES |
|||
(2001, @demo_user_id, 'JG-UAV-001', 'dock-1', '巡检无人机 001', 'JG-UAV-001', 'LAIC-X1', 'offline', 86, '1.0.0', NOW(), NOW()), |
|||
(2002, @demo_user_id, 'JG-UAV-002', 'dock-2', '巡检无人机 002', 'JG-UAV-002', 'LAIC-X1', 'offline', 72, '1.0.0', NOW(), NOW()), |
|||
(2003, @demo_user_id, 'JG-UAV-003', 'dock-3', '巡检无人机 003', 'JG-UAV-003', 'LAIC-X1', 'offline', 41, '1.0.0', NOW(), NOW()) |
|||
ON DUPLICATE KEY UPDATE |
|||
user_id = @demo_user_id, |
|||
dock_id = VALUES(dock_id), |
|||
name = VALUES(name), |
|||
code = VALUES(code), |
|||
model = VALUES(model); |
|||
|
|||
DELETE FROM drone WHERE drone_sn = 'JG-UAV-MOCK01'; |
|||
DELETE FROM dock WHERE dock_id IN ('JG-DK-MOCK01', '333'); |
|||
|
|||
-- --------------------------------------------------------------- |
|||
-- 说明: Casbin RBAC 策略（g: user_id -> role / p: role -> path -> method） |
|||
-- 由应用首次启动时通过 gorm-adapter 自动建表并引导，不在本脚本中静态写入。 |
|||
-- --------------------------------------------------------------- |
|||
@ -0,0 +1,27 @@ |
|||
-- 该迁移用于已有数据库；新数据库请直接使用 001_schema.sql 中的最终结构。 |
|||
-- 执行前应确认 live_session 没有重复活动会话，再创建活动机巢唯一约束。 |
|||
|
|||
ALTER TABLE live_session |
|||
ADD COLUMN requested_by BIGINT NULL, |
|||
ADD COLUMN stop_deadline DATETIME NULL, |
|||
ADD COLUMN updated_at DATETIME NULL, |
|||
ADD COLUMN active_dock_id VARCHAR(64) GENERATED ALWAYS AS ( |
|||
CASE WHEN phase IN ('starting', 'streaming', 'reconnecting', 'stopping') THEN dock_id ELSE NULL END |
|||
) STORED, |
|||
ADD KEY idx_live_session_dock_phase (dock_id, phase, created_at), |
|||
ADD KEY idx_live_session_phase_deadline (phase, stop_deadline), |
|||
ADD UNIQUE KEY uk_live_session_active_dock (active_dock_id); |
|||
|
|||
CREATE TABLE IF NOT EXISTS live_viewer_lease ( |
|||
id BIGINT PRIMARY KEY, |
|||
stream_session_id VARCHAR(128) NOT NULL, |
|||
viewer_id BIGINT NOT NULL, |
|||
expires_at DATETIME NOT NULL, |
|||
released_at DATETIME NULL, |
|||
created_at DATETIME NULL, |
|||
updated_at DATETIME NULL, |
|||
UNIQUE KEY uk_live_viewer (stream_session_id, viewer_id), |
|||
KEY idx_live_lease_expire (stream_session_id, expires_at), |
|||
KEY idx_live_lease_active (stream_session_id, released_at, expires_at), |
|||
CONSTRAINT fk_live_lease_session FOREIGN KEY (stream_session_id) REFERENCES live_session(id) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
@ -0,0 +1,115 @@ |
|||
-- 已部署数据库的可重复执行升级脚本。 |
|||
-- 执行前应先收敛同一 dock 的重复活动会话，否则活动会话唯一索引会拒绝创建。 |
|||
|
|||
DELIMITER $$ |
|||
|
|||
DROP PROCEDURE IF EXISTS upgrade_live_control $$ |
|||
CREATE PROCEDURE upgrade_live_control() |
|||
BEGIN |
|||
DECLARE item_count INT DEFAULT 0; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.columns |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND column_name = 'requested_by'; |
|||
IF item_count = 0 THEN |
|||
ALTER TABLE live_session ADD COLUMN requested_by BIGINT NULL; |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.columns |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND column_name = 'stop_deadline'; |
|||
IF item_count = 0 THEN |
|||
ALTER TABLE live_session ADD COLUMN stop_deadline DATETIME NULL; |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.columns |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND column_name = 'updated_at'; |
|||
IF item_count = 0 THEN |
|||
ALTER TABLE live_session ADD COLUMN updated_at DATETIME NULL; |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.columns |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND column_name = 'device_state_version'; |
|||
IF item_count = 0 THEN |
|||
ALTER TABLE live_session ADD COLUMN device_state_version BIGINT DEFAULT 0; |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.columns |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND column_name = 'device_event_id'; |
|||
IF item_count = 0 THEN |
|||
ALTER TABLE live_session ADD COLUMN device_event_id VARCHAR(128); |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.columns |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND column_name = 'device_updated_at'; |
|||
IF item_count = 0 THEN |
|||
ALTER TABLE live_session ADD COLUMN device_updated_at BIGINT DEFAULT 0; |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.columns |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND column_name = 'cloud_confirmed_at'; |
|||
IF item_count = 0 THEN |
|||
ALTER TABLE live_session ADD COLUMN cloud_confirmed_at DATETIME NULL; |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.statistics |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND index_name = 'uk_live_session_device_event'; |
|||
IF item_count = 0 THEN |
|||
CREATE UNIQUE INDEX uk_live_session_device_event ON live_session (device_event_id); |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.columns |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND column_name = 'active_dock_id'; |
|||
IF item_count = 0 THEN |
|||
ALTER TABLE live_session ADD COLUMN active_dock_id VARCHAR(64) GENERATED ALWAYS AS ( |
|||
CASE WHEN phase IN ('starting', 'streaming', 'reconnecting', 'stopping') THEN dock_id ELSE NULL END |
|||
) STORED; |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.statistics |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND index_name = 'idx_live_session_dock_phase'; |
|||
IF item_count = 0 THEN |
|||
CREATE INDEX idx_live_session_dock_phase ON live_session (dock_id, phase, created_at); |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.statistics |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND index_name = 'idx_live_session_phase_deadline'; |
|||
IF item_count = 0 THEN |
|||
CREATE INDEX idx_live_session_phase_deadline ON live_session (phase, stop_deadline); |
|||
END IF; |
|||
|
|||
SELECT COUNT(*) INTO item_count |
|||
FROM information_schema.statistics |
|||
WHERE table_schema = DATABASE() AND table_name = 'live_session' AND index_name = 'uk_live_session_active_dock'; |
|||
IF item_count = 0 THEN |
|||
CREATE UNIQUE INDEX uk_live_session_active_dock ON live_session (active_dock_id); |
|||
END IF; |
|||
|
|||
CREATE TABLE IF NOT EXISTS live_viewer_lease ( |
|||
id BIGINT PRIMARY KEY, |
|||
stream_session_id VARCHAR(128) NOT NULL, |
|||
viewer_id BIGINT NOT NULL, |
|||
expires_at DATETIME NOT NULL, |
|||
released_at DATETIME NULL, |
|||
created_at DATETIME NULL, |
|||
updated_at DATETIME NULL, |
|||
UNIQUE KEY uk_live_viewer (stream_session_id, viewer_id), |
|||
KEY idx_live_lease_expire (stream_session_id, expires_at), |
|||
KEY idx_live_lease_active (stream_session_id, released_at, expires_at), |
|||
CONSTRAINT fk_live_lease_session FOREIGN KEY (stream_session_id) REFERENCES live_session(id) |
|||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; |
|||
END $$ |
|||
|
|||
CALL upgrade_live_control() $$ |
|||
DROP PROCEDURE upgrade_live_control $$ |
|||
|
|||
DELIMITER ; |
|||
@ -0,0 +1,41 @@ |
|||
-- TDengine 时序库 schema（与 MySQL DDL 分离，需用 taos / REST 手动执行） |
|||
-- 对应 DESIGN.md §5.9 |
|||
|
|||
CREATE DATABASE IF NOT EXISTS laic VGROUPS 2 DURATION 10d KEEP 3650d; |
|||
|
|||
-- 遥测超级表（device_telemetry）：列顺序必须与 service/telemetry_store.go insert() 的 VALUES 元组一致 |
|||
CREATE STABLE IF NOT EXISTS laic.device_telemetry ( |
|||
ts TIMESTAMP, |
|||
longitude DOUBLE, |
|||
latitude DOUBLE, |
|||
altitude DOUBLE, |
|||
ground_speed DOUBLE, |
|||
roll DOUBLE, |
|||
pitch DOUBLE, |
|||
yaw DOUBLE, |
|||
battery_pct INT, |
|||
battery_v DOUBLE, |
|||
satellites INT, |
|||
gps_quality VARCHAR(16), |
|||
link_quality INT, |
|||
flight_mode VARCHAR(16), |
|||
armed TINYINT |
|||
) TAGS ( |
|||
dock_id VARCHAR(64), |
|||
drone_sn VARCHAR(32) |
|||
); |
|||
|
|||
-- 轨迹超级表（任务回放用） |
|||
CREATE STABLE IF NOT EXISTS laic.task_trajectory ( |
|||
ts TIMESTAMP, |
|||
longitude DOUBLE, |
|||
latitude DOUBLE, |
|||
altitude DOUBLE, |
|||
ground_speed DOUBLE, |
|||
yaw DOUBLE, |
|||
battery_pct INT |
|||
) TAGS ( |
|||
execution_id BIGINT, |
|||
dock_id VARCHAR(64), |
|||
drone_sn VARCHAR(32) |
|||
); |
|||
@ -0,0 +1,121 @@ |
|||
package token |
|||
|
|||
import ( |
|||
"crypto/sha256" |
|||
"encoding/hex" |
|||
"errors" |
|||
"time" |
|||
|
|||
"github.com/golang-jwt/jwt/v5" |
|||
) |
|||
|
|||
var ( |
|||
jwtSecret []byte |
|||
accessExpireH = 7 * 24 // 默认 7 天
|
|||
refreshExpireH = 30 * 24 // 默认 30 天
|
|||
) |
|||
|
|||
// Init 初始化 JWT 密钥与有效期（main 启动时从配置加载）
|
|||
func Init(secret string, accessH, refreshH int) { |
|||
jwtSecret = []byte(secret) |
|||
if accessH > 0 { |
|||
accessExpireH = accessH |
|||
} |
|||
if refreshH > 0 { |
|||
refreshExpireH = refreshH |
|||
} |
|||
} |
|||
|
|||
// Claims access token 载荷
|
|||
type Claims struct { |
|||
UserID int64 `json:"user_id"` |
|||
Username string `json:"username"` |
|||
Roles []string `json:"roles"` |
|||
jwt.RegisteredClaims |
|||
} |
|||
|
|||
// RefreshClaims refresh token 载荷
|
|||
type RefreshClaims struct { |
|||
UserID int64 `json:"user_id"` |
|||
jwt.RegisteredClaims |
|||
} |
|||
|
|||
// GenerateToken 生成 access + refresh token
|
|||
func GenerateToken(userID int64, username string, roles []string) (string, string, error) { |
|||
now := time.Now() |
|||
claims := Claims{ |
|||
UserID: userID, |
|||
Username: username, |
|||
Roles: roles, |
|||
RegisteredClaims: jwt.RegisteredClaims{ |
|||
ExpiresAt: jwt.NewNumericDate(now.Add(time.Duration(accessExpireH) * time.Hour)), |
|||
IssuedAt: jwt.NewNumericDate(now), |
|||
NotBefore: jwt.NewNumericDate(now), |
|||
}, |
|||
} |
|||
access, err := jwt.NewWithClaims(jwt.SigningMethodHS256, claims).SignedString(jwtSecret) |
|||
if err != nil { |
|||
return "", "", err |
|||
} |
|||
|
|||
refreshClaims := RefreshClaims{ |
|||
UserID: userID, |
|||
RegisteredClaims: jwt.RegisteredClaims{ |
|||
ExpiresAt: jwt.NewNumericDate(now.Add(time.Duration(refreshExpireH) * time.Hour)), |
|||
IssuedAt: jwt.NewNumericDate(now), |
|||
NotBefore: jwt.NewNumericDate(now), |
|||
}, |
|||
} |
|||
refresh, err := jwt.NewWithClaims(jwt.SigningMethodHS256, refreshClaims).SignedString(jwtSecret) |
|||
if err != nil { |
|||
return "", "", err |
|||
} |
|||
return access, refresh, nil |
|||
} |
|||
|
|||
// ParseToken 解析并校验 access token
|
|||
func ParseToken(tokenString string) (*Claims, error) { |
|||
token, err := jwt.ParseWithClaims(tokenString, &Claims{}, keyFunc) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
if claims, ok := token.Claims.(*Claims); ok && token.Valid { |
|||
return claims, nil |
|||
} |
|||
return nil, errors.New("invalid token") |
|||
} |
|||
|
|||
// ParseRefreshToken 解析并校验 refresh token，返回用户 ID
|
|||
func ParseRefreshToken(tokenString string) (int64, error) { |
|||
token, err := jwt.ParseWithClaims(tokenString, &RefreshClaims{}, keyFunc) |
|||
if err != nil { |
|||
return 0, err |
|||
} |
|||
if claims, ok := token.Claims.(*RefreshClaims); ok && token.Valid { |
|||
return claims.UserID, nil |
|||
} |
|||
return 0, errors.New("invalid refresh token") |
|||
} |
|||
|
|||
func keyFunc(t *jwt.Token) (any, error) { |
|||
if _, ok := t.Method.(*jwt.SigningMethodHMAC); !ok { |
|||
return nil, errors.New("unexpected signing method") |
|||
} |
|||
return jwtSecret, nil |
|||
} |
|||
|
|||
// GenerateShortID 生成 token 的短摘要，作为 Redis 存储 key 的一部分
|
|||
func GenerateShortID(tokenString string) string { |
|||
hash := sha256.Sum256([]byte(tokenString)) |
|||
return hex.EncodeToString(hash[:8]) |
|||
} |
|||
|
|||
// AccessExpireSeconds 返回 access token 有效期（秒），用于 Redis TTL
|
|||
func AccessExpireSeconds() uint { |
|||
return uint(accessExpireH * 3600) |
|||
} |
|||
|
|||
// RefreshExpireSeconds 返回 refresh token 有效期（秒），用于 Redis TTL
|
|||
func RefreshExpireSeconds() uint { |
|||
return uint(refreshExpireH * 3600) |
|||
} |
|||
@ -0,0 +1,18 @@ |
|||
package tool |
|||
|
|||
import "math" |
|||
|
|||
const earthRadiusM = 6371000.0 |
|||
|
|||
// Haversine 计算两个经纬度坐标之间的球面距离（单位：米）
|
|||
func Haversine(lat1, lon1, lat2, lon2 float64) float64 { |
|||
phi1 := lat1 * math.Pi / 180 |
|||
phi2 := lat2 * math.Pi / 180 |
|||
dPhi := (lat2 - lat1) * math.Pi / 180 |
|||
dLambda := (lon2 - lon1) * math.Pi / 180 |
|||
|
|||
a := math.Sin(dPhi/2)*math.Sin(dPhi/2) + |
|||
math.Cos(phi1)*math.Cos(phi2)*math.Sin(dLambda/2)*math.Sin(dLambda/2) |
|||
c := 2 * math.Atan2(math.Sqrt(a), math.Sqrt(1-a)) |
|||
return earthRadiusM * c |
|||
} |
|||
@ -0,0 +1,117 @@ |
|||
package tool |
|||
|
|||
import ( |
|||
"errors" |
|||
"sync" |
|||
"time" |
|||
) |
|||
|
|||
const ( |
|||
// 起始时间戳 (2023-01-01 00:00:00 UTC)
|
|||
epoch int64 = 1672531200000 |
|||
|
|||
timestampBits = 28 // 时间戳位数（约17年）
|
|||
workerIDBits = 5 // 工作机器ID所占位数
|
|||
sequenceBits = 12 // 序列号所占位数
|
|||
|
|||
maxWorkerID = -1 ^ (-1 << workerIDBits) |
|||
maxSequence = -1 ^ (-1 << sequenceBits) |
|||
|
|||
workerIDShift = sequenceBits |
|||
timestampShift = sequenceBits + workerIDBits |
|||
) |
|||
|
|||
type Snowflake struct { |
|||
mu sync.Mutex |
|||
timestamp int64 |
|||
workerID int64 |
|||
sequence int64 |
|||
lastTime int64 |
|||
} |
|||
|
|||
var DefaultSnowflake *Snowflake |
|||
|
|||
func init() { |
|||
DefaultSnowflake = NewSnowflake(1) |
|||
} |
|||
|
|||
func NewSnowflake(workerID int64) *Snowflake { |
|||
if workerID < 0 || workerID > maxWorkerID { |
|||
panic(errors.New("worker ID must be between 0 and 1023")) |
|||
} |
|||
return &Snowflake{ |
|||
timestamp: 0, |
|||
workerID: workerID, |
|||
sequence: 0, |
|||
lastTime: -1, |
|||
} |
|||
} |
|||
|
|||
// NextID 生成下一个ID
|
|||
func (s *Snowflake) NextID() (int64, error) { |
|||
s.mu.Lock() |
|||
defer s.mu.Unlock() |
|||
|
|||
now := time.Now().UnixMilli() |
|||
now = (now - epoch) & (-1 ^ (-1 << timestampBits)) |
|||
if now <= 0 { |
|||
now = 1 |
|||
} |
|||
|
|||
if now < s.lastTime { |
|||
waitTime := s.lastTime - now |
|||
if waitTime > 100 { |
|||
now = s.lastTime + 1 |
|||
if now > (1<<timestampBits)-1 { |
|||
return -1, errors.New("timestamp overflow due to clock moved backwards") |
|||
} |
|||
} else { |
|||
time.Sleep(time.Duration(waitTime) * time.Millisecond) |
|||
now = (time.Now().UnixMilli() - epoch) & (-1 ^ (-1 << timestampBits)) |
|||
if now <= 0 { |
|||
now = 1 |
|||
} |
|||
if now < s.lastTime { |
|||
now = s.lastTime + 1 |
|||
if now > (1<<timestampBits)-1 { |
|||
return -1, errors.New("timestamp overflow due to clock moved backwards") |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
if s.lastTime == now { |
|||
s.sequence = (s.sequence + 1) & maxSequence |
|||
if s.sequence == 0 { |
|||
for now <= s.lastTime { |
|||
now = (time.Now().UnixMilli() - epoch) & (-1 ^ (-1 << timestampBits)) |
|||
if now <= 0 { |
|||
now = s.lastTime + 1 |
|||
break |
|||
} |
|||
} |
|||
} |
|||
} else { |
|||
s.sequence = 0 |
|||
} |
|||
|
|||
s.lastTime = now |
|||
|
|||
id := (now << timestampShift) | |
|||
(s.workerID << workerIDShift) | |
|||
s.sequence |
|||
|
|||
if id == 0 { |
|||
s.sequence = 1 |
|||
id = (now << timestampShift) | |
|||
(s.workerID << workerIDShift) | |
|||
s.sequence |
|||
} |
|||
|
|||
return id, nil |
|||
} |
|||
|
|||
// NextID 全局函数，使用默认实例生成ID
|
|||
func NextID() (int64, error) { |
|||
return DefaultSnowflake.NextID() |
|||
} |
|||
@ -0,0 +1,35 @@ |
|||
package tool |
|||
|
|||
import ( |
|||
"crypto/rand" |
|||
"math/big" |
|||
) |
|||
|
|||
const ( |
|||
letters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789" |
|||
digits = "0123456789" |
|||
) |
|||
|
|||
// RandString 生成指定长度的随机字母数字串
|
|||
func RandString(n int) string { |
|||
return randFrom(letters, n) |
|||
} |
|||
|
|||
// RandDigit 生成指定长度的随机数字串（短信验证码等）
|
|||
func RandDigit(n int) string { |
|||
return randFrom(digits, n) |
|||
} |
|||
|
|||
func randFrom(charset string, n int) string { |
|||
b := make([]byte, n) |
|||
max := big.NewInt(int64(len(charset))) |
|||
for i := range b { |
|||
idx, err := rand.Int(rand.Reader, max) |
|||
if err != nil { |
|||
b[i] = charset[0] |
|||
continue |
|||
} |
|||
b[i] = charset[idx.Int64()] |
|||
} |
|||
return string(b) |
|||
} |
|||
@ -0,0 +1,28 @@ |
|||
package tool |
|||
|
|||
import "time" |
|||
|
|||
const ( |
|||
TimeLayout = "2006-01-02 15:04:05" |
|||
DateLayout = "2006-01-02" |
|||
TimeLayoutMs = "2006-01-02 15:04:05.999" |
|||
) |
|||
|
|||
func Now() time.Time { |
|||
return time.Now() |
|||
} |
|||
|
|||
// FormatTime 格式化时间为标准字符串
|
|||
func FormatTime(t time.Time) string { |
|||
return t.Format(TimeLayout) |
|||
} |
|||
|
|||
// FormatTimeMs 格式化时间为毫秒精度字符串
|
|||
func FormatTimeMs(t time.Time) string { |
|||
return t.Format(TimeLayoutMs) |
|||
} |
|||
|
|||
// ParseTime 解析标准时间字符串
|
|||
func ParseTime(s string) (time.Time, error) { |
|||
return time.ParseInLocation(TimeLayout, s, time.Local) |
|||
} |
|||
@ -0,0 +1,50 @@ |
|||
package vo |
|||
|
|||
import "laic-backend/common" |
|||
|
|||
// ProfileUpdateReq 更新个人资料
|
|||
type ProfileUpdateReq struct { |
|||
Name string `json:"name" validate:"max=32"` |
|||
Email string `json:"email" validate:"omitempty,email"` |
|||
} |
|||
|
|||
// TrafficBalanceVO 云媒体流量余额（Redis 权威，字节为基准）
|
|||
type TrafficBalanceVO struct { |
|||
BalanceBytes int64 `json:"balanceBytes"` |
|||
BalanceGb float64 `json:"balanceGb"` |
|||
} |
|||
|
|||
// UsagePageReq 流量消费流水分页
|
|||
type UsagePageReq struct { |
|||
common.Pagination |
|||
SourceType string `json:"sourceType" form:"sourceType"` // live / replay / download
|
|||
} |
|||
|
|||
// OrderPageReq 流量订单分页
|
|||
type OrderPageReq struct { |
|||
common.Pagination |
|||
PayStatus string `json:"payStatus" form:"payStatus"` // unpaid / paid
|
|||
} |
|||
|
|||
// TrafficOrderCreateReq 创建流量充值订单
|
|||
type TrafficOrderCreateReq struct { |
|||
AmountGb int `json:"amountGb" validate:"required,min=1"` |
|||
} |
|||
|
|||
// SimCardPageReq SIM 卡分页
|
|||
type SimCardPageReq struct { |
|||
common.Pagination |
|||
Status string `json:"status" form:"status"` // active / expired
|
|||
} |
|||
|
|||
// SimRechargeLogPageReq SIM 卡充值记录分页
|
|||
type SimRechargeLogPageReq struct { |
|||
common.Pagination |
|||
SimCardID int64 `json:"simCardId" form:"simCardId"` |
|||
RechargeStatus string `json:"rechargeStatus" form:"rechargeStatus"` |
|||
} |
|||
|
|||
// SimRechargeReq SIM 卡充值
|
|||
type SimRechargeReq struct { |
|||
AmountGb int `json:"amountGb" validate:"required,min=1"` |
|||
} |
|||
@ -0,0 +1,13 @@ |
|||
package vo |
|||
|
|||
import "laic-backend/common" |
|||
|
|||
// AlarmPageReq 告警分页查询
|
|||
type AlarmPageReq struct { |
|||
common.Pagination |
|||
Status string `json:"status" form:"status"` // active / acknowledged / resolved
|
|||
Level string `json:"level" form:"level"` |
|||
DeviceType string `json:"deviceType" form:"deviceType"` // dock / drone
|
|||
DockID string `json:"dockId" form:"dockId"` |
|||
Keyword string `json:"keyword" form:"keyword"` // 匹配 code / message_cn
|
|||
} |
|||
@ -0,0 +1,81 @@ |
|||
package vo |
|||
|
|||
import ( |
|||
"time" |
|||
|
|||
"laic-backend/model" |
|||
) |
|||
|
|||
// UserLoginReq 登录请求
|
|||
type UserLoginReq struct { |
|||
Phone string `json:"phone" validate:"required,phone"` |
|||
Password string `json:"password" validate:"required"` |
|||
} |
|||
|
|||
// RegisterReq 注册请求
|
|||
type RegisterReq struct { |
|||
Name string `json:"name" validate:"required,max=32"` |
|||
Phone string `json:"phone" validate:"required,phone"` |
|||
Email string `json:"email" validate:"omitempty,email"` |
|||
Password string `json:"password" validate:"required,password"` |
|||
} |
|||
|
|||
// SmsCodeReq 发送短信验证码请求
|
|||
type SmsCodeReq struct { |
|||
Phone string `json:"phone" validate:"required,phone"` |
|||
} |
|||
|
|||
// ResetPasswordReq 重置密码请求
|
|||
type ResetPasswordReq struct { |
|||
Phone string `json:"phone" validate:"required,phone"` |
|||
SmsCode string `json:"smsCode" validate:"required"` |
|||
NewPassword string `json:"newPassword" validate:"required,password"` |
|||
} |
|||
|
|||
// RefreshReq 刷新 token 请求
|
|||
type RefreshReq struct { |
|||
RefreshToken string `json:"refreshToken" validate:"required"` |
|||
} |
|||
|
|||
// LogoutReq 登出请求（refreshToken 可选，用于同时注销刷新令牌）
|
|||
type LogoutReq struct { |
|||
RefreshToken string `json:"refreshToken"` |
|||
} |
|||
|
|||
// UserVO 用户信息（脱敏，不含密码）
|
|||
type UserVO struct { |
|||
ID int64 `json:"id"` |
|||
Name string `json:"name"` |
|||
Phone string `json:"phone"` |
|||
Email string `json:"email"` |
|||
Role string `json:"role"` |
|||
TrafficBalance int64 `json:"trafficBalance"` |
|||
Status int8 `json:"status"` |
|||
LastLogin *time.Time `json:"lastLogin"` |
|||
CreatedAt time.Time `json:"createdAt"` |
|||
} |
|||
|
|||
// LoginResp 登录/刷新响应
|
|||
type LoginResp struct { |
|||
AccessToken string `json:"accessToken"` |
|||
RefreshToken string `json:"refreshToken"` |
|||
User *UserVO `json:"user"` |
|||
} |
|||
|
|||
// NewUserVO 由 model.User 构建脱敏 VO
|
|||
func NewUserVO(u *model.User) *UserVO { |
|||
if u == nil { |
|||
return nil |
|||
} |
|||
return &UserVO{ |
|||
ID: u.ID, |
|||
Name: u.Name, |
|||
Phone: u.Phone, |
|||
Email: u.Email, |
|||
Role: u.Role, |
|||
TrafficBalance: u.TrafficBalance, |
|||
Status: u.Status, |
|||
LastLogin: u.LastLogin, |
|||
CreatedAt: u.CreatedAt, |
|||
} |
|||
} |
|||
@ -0,0 +1,20 @@ |
|||
package vo |
|||
|
|||
import "laic-backend/common" |
|||
|
|||
// CommandReq 指令下发请求
|
|||
type CommandReq struct { |
|||
Type string `json:"type" validate:"required,max=64"` // dock.open / drone.takeoff / workflow.start_task
|
|||
Params map[string]any `json:"params"` |
|||
DroneSN string `json:"droneSn" validate:"max=32"` |
|||
TTLMs int `json:"ttlMs"` |
|||
} |
|||
|
|||
// CommandPageReq 指令历史分页查询
|
|||
type CommandPageReq struct { |
|||
common.Pagination |
|||
Status string `json:"status" form:"status"` // sent / acked / timeout / terminal
|
|||
CommandType string `json:"commandType" form:"commandType"` |
|||
DockID string `json:"dockId" form:"dockId"` |
|||
Keyword string `json:"keyword" form:"keyword"` |
|||
} |
|||
@ -0,0 +1,54 @@ |
|||
package vo |
|||
|
|||
import ( |
|||
"laic-backend/common" |
|||
"laic-backend/model" |
|||
) |
|||
|
|||
// DockPageReq 机巢分页查询
|
|||
type DockPageReq struct { |
|||
common.Pagination |
|||
Status string `json:"status" form:"status"` |
|||
RegisterStatus string `json:"registerStatus" form:"registerStatus"` |
|||
Keyword string `json:"keyword" form:"keyword"` // 匹配 name / dockId / code
|
|||
} |
|||
|
|||
// DockCreateReq 手动登记机巢
|
|||
type DockCreateReq struct { |
|||
DockID string `json:"dockId" validate:"required,max=64"` |
|||
Name string `json:"name" validate:"max=64"` |
|||
Code string `json:"code" validate:"max=32"` |
|||
SN string `json:"sn" validate:"max=64"` |
|||
Iccid string `json:"iccid" validate:"max=32"` |
|||
Location string `json:"location" validate:"max=128"` |
|||
Longitude float64 `json:"longitude"` |
|||
Latitude float64 `json:"latitude"` |
|||
Altitude float64 `json:"altitude"` |
|||
UserID int64 `json:"userId"` // 仅 admin 可指定归属用户，普通用户忽略
|
|||
} |
|||
|
|||
// DockUpdateReq 更新/认领机巢
|
|||
type DockUpdateReq struct { |
|||
Name string `json:"name" validate:"max=64"` |
|||
Code string `json:"code" validate:"max=32"` |
|||
Location *string `json:"location" validate:"omitempty,max=128"` |
|||
Longitude *float64 `json:"longitude"` |
|||
Latitude *float64 `json:"latitude"` |
|||
Altitude *float64 `json:"altitude"` |
|||
UserID int64 `json:"userId"` // 仅 admin 可改归属
|
|||
RegisterAs string `json:"registerAs"` // 空则不修改；registered 表示完成登记
|
|||
} |
|||
|
|||
// DockVO 机巢详情（含实时状态）
|
|||
type DockVO struct { |
|||
*model.Dock |
|||
Online bool `json:"online"` |
|||
Realtime map[string]string `json:"realtime"` |
|||
} |
|||
|
|||
// DockStatusVO 机巢实时状态
|
|||
type DockStatusVO struct { |
|||
DockID string `json:"dockId"` |
|||
Online bool `json:"online"` |
|||
Realtime map[string]string `json:"realtime"` |
|||
} |
|||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue