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