38 lines
1.2 KiB
Go
38 lines
1.2 KiB
Go
package dao
|
||
|
||
import "gorm.io/gorm"
|
||
|
||
type Bluetooth struct {
|
||
ID uint64 `gorm:"primaryKey"`
|
||
DeviceId string `gorm:"unique;type varchar(64);not null;comment:'设备Id，唯一标识每个蓝牙设备'"`
|
||
DeviceName string `gorm:"type varchar(64);comment:'蓝牙名称'"`
|
||
DeviceType string `gorm:"type varchar(64);comment:'设备类型'"`
|
||
UserId int32 `gorm:"index;comment:'用户ID'"`
|
||
}
|
||
|
||
func AddBluetooth(data *Bluetooth) error {
|
||
if err := s.db.Where("device_id = ? AND user_id = ?", data.DeviceId, data.UserId).FirstOrCreate(data).Error; err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func DeleteBluetooth(deviceid string, userid int32) error {
|
||
var data *Bluetooth
|
||
if err := s.db.Where("device_id = ? AND user_id = ?", deviceid, userid).First(&data).Error; err != nil && err != gorm.ErrRecordNotFound {
|
||
return err
|
||
}
|
||
if err := s.db.Delete(&data).Error; err != nil && err != gorm.ErrRecordNotFound {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func GetAllBluetooth(userid int32) ([]*Bluetooth, error) {
|
||
var list []*Bluetooth
|
||
if err := s.db.Where("user_id = ?", userid).Find(&list).Error; err != nil && err != gorm.ErrRecordNotFound {
|
||
return nil, err
|
||
}
|
||
return list, nil
|
||
}
|