95 lines
3.3 KiB
Go
95 lines
3.3 KiB
Go
//
|
||
// Copyright (C) 2020 IOTech Ltd
|
||
//
|
||
// SPDX-License-Identifier: Apache-2.0
|
||
|
||
package models
|
||
|
||
import (
|
||
"github.com/winc-link/hummingbird/internal/pkg/constants"
|
||
"github.com/winc-link/hummingbird/internal/pkg/utils"
|
||
"gorm.io/gorm"
|
||
"strings"
|
||
//"gitlab.com/tedge/edgex/internal/pkg/constants"
|
||
)
|
||
|
||
// DeviceService and its properties are defined in the APIv2 specification:
|
||
// https://app.swaggerhub.com/apis-docs/EdgeXFoundry1/core-metadata/2.x#/DeviceService
|
||
// Model fields are same as the DTOs documented by this swagger. Exceptions, if any, are noted below.
|
||
type DeviceService struct {
|
||
Timestamps `gorm:"embedded"`
|
||
Id string `gorm:"id;primaryKey;not null;type:string;size:255;comment:主键"`
|
||
Name string `gorm:"type:string;size:255;comment:名字"`
|
||
BaseAddress string `gorm:"type:string;size:255;comment:地址"`
|
||
DeviceLibraryId string `gorm:"uniqueIndex;type:string;size:255;comment:驱动ID"`
|
||
Config MapStringInterface `gorm:"type:string;size:255;comment:配置"`
|
||
DockerContainerId string `gorm:"type:string;size:255;comment:docker容器ID"`
|
||
ExpertMode bool `gorm:"comment:扩展模式"`
|
||
ExpertModeContent string `gorm:"comment:扩展内容"`
|
||
DockerParamsSwitch bool `gorm:"comment:docker启动参数开关"`
|
||
DockerParams string `gorm:"type:text;comment:docker启动参数"`
|
||
ContainerName string `gorm:"type:string;size:255;comment:容器名字"`
|
||
LogLevel constants.LogLevel `gorm:"default:1;comment:日志等级"`
|
||
DriverType int `gorm:"default:1;not null;comment:驱动类别，1：驱动，2：三方应用"`
|
||
RunStatus int `gorm:"-"`
|
||
ImageExist bool `gorm:"-"`
|
||
Platform constants.IotPlatform
|
||
}
|
||
|
||
func (d *DeviceService) TableName() string {
|
||
return "device_service"
|
||
}
|
||
|
||
func (d *DeviceService) Get() interface{} {
|
||
return *d
|
||
}
|
||
|
||
func (d *DeviceService) IsRunning() bool {
|
||
return d.RunStatus == constants.RunStatusStarted
|
||
}
|
||
|
||
func (d *DeviceService) IsStopped() bool {
|
||
return d.RunStatus == constants.RunStatusStopped
|
||
}
|
||
|
||
func (d *DeviceService) GetBaseAddress() string {
|
||
if d.BaseAddress == "" {
|
||
return constants.DefaultDriverBaseAddress
|
||
}
|
||
return d.BaseAddress
|
||
}
|
||
|
||
func (d *DeviceService) GetPort() string {
|
||
tmpAddr := strings.Split(d.BaseAddress, ":")
|
||
if len(tmpAddr) >= 2 {
|
||
return tmpAddr[1]
|
||
}
|
||
return ""
|
||
}
|
||
|
||
func (d *DeviceService) IsDriver() bool {
|
||
return d.DriverType == constants.DriverLibTypeDefault
|
||
}
|
||
|
||
type DeviceServiceExtendConf struct {
|
||
ConfigFilePath string
|
||
Mount []string
|
||
Port int
|
||
}
|
||
|
||
func (d *DeviceService) BeforeCreate(tx *gorm.DB) (err error) {
|
||
var mqttAuth MqttAuth
|
||
mqttAuth.Id = utils.RandomNum()
|
||
mqttAuth.ResourceType = constants.DriverResource
|
||
mqttAuth.ResourceId = d.Id
|
||
mqttAuth.ClientId = utils.GenUUID()
|
||
mqttAuth.UserName = "edge-driver" + d.Id
|
||
mqttAuth.Password = utils.GenUUID()
|
||
return tx.Model(&MqttAuth{}).Create(&mqttAuth).Error
|
||
}
|
||
|
||
func (d *DeviceService) BeforeDelete(tx *gorm.DB) (err error) {
|
||
var mqttAuth MqttAuth
|
||
return tx.Model(&MqttAuth{}).Where("resource_type = ? and resource_id = ?", constants.DriverResource, d.Id).Delete(&mqttAuth).Error
|
||
}
|