Update servidor.go
This commit is contained in:
@@ -1,6 +1,8 @@
|
|||||||
package models
|
package models
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"crypto/rand"
|
||||||
|
"encoding/hex"
|
||||||
"log"
|
"log"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -8,6 +10,12 @@ import (
|
|||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func generateAgentToken() string {
|
||||||
|
b := make([]byte, 16)
|
||||||
|
rand.Read(b)
|
||||||
|
return "agent_" + hex.EncodeToString(b)
|
||||||
|
}
|
||||||
|
|
||||||
type Servidor struct {
|
type Servidor struct {
|
||||||
gorm.Model
|
gorm.Model
|
||||||
Nombre string `json:"nombre" gorm:"column:nombre"`
|
Nombre string `json:"nombre" gorm:"column:nombre"`
|
||||||
@@ -73,6 +81,9 @@ func GetAllServidoresSelect() ([]Servidor, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func CreateServidor(servidor Servidor) error {
|
func CreateServidor(servidor Servidor) error {
|
||||||
|
if servidor.AgentToken == "" {
|
||||||
|
servidor.AgentToken = generateAgentToken()
|
||||||
|
}
|
||||||
if err := app.Http.Database.DB.Create(&servidor).Error; err != nil {
|
if err := app.Http.Database.DB.Create(&servidor).Error; err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user