60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package model
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Controller struct {
|
|
gorm.Model
|
|
CID string `gorm:"uniqueIndex;column:cid" json:"cid"` // Controller ID
|
|
CName string `gorm:"column:cname" json:"cname"` // Device ID
|
|
Key string `gorm:"column:key" json:"key"` // Service ID
|
|
}
|
|
|
|
func (s *dbHandler) AddController(r io.Reader) (*Controller, error) {
|
|
decoder := json.NewDecoder(r)
|
|
var controller = &Controller{}
|
|
|
|
err := decoder.Decode(controller)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
controller.CID = uuid.NewString()
|
|
controller.Key = controller.CID
|
|
|
|
tx := s.db.Create(controller)
|
|
if tx.Error != nil {
|
|
return nil, tx.Error
|
|
}
|
|
|
|
tx.First(&controller, "cid=?", controller.CID)
|
|
fmt.Println(controller)
|
|
return controller, nil
|
|
}
|
|
|
|
func (s *dbHandler) GetControllers() ([]*Controller, error) {
|
|
var list []*Controller
|
|
|
|
result := s.db.Find(&list)
|
|
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
|
|
return list, nil
|
|
}
|
|
|
|
func (s *dbHandler) IsExistController(cid string) bool {
|
|
var controller = Controller{}
|
|
|
|
result := s.db.First(&controller, "cid=?", cid)
|
|
|
|
return result.Error == nil
|
|
}
|