mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
90 lines
3.8 KiB
Go
90 lines
3.8 KiB
Go
package model
|
|
|
|
import "time"
|
|
|
|
type Node struct {
|
|
ID uint `json:"id" gorm:"primaryKey"`
|
|
NodeID string `json:"node_id" gorm:"uniqueIndex;size:64;not null"`
|
|
Name string `json:"name" gorm:"size:128;not null"`
|
|
IP string `json:"ip" gorm:"size:64;not null"`
|
|
IPManualOverride bool `json:"ip_manual_override" gorm:"not null;default:false"`
|
|
GeoName string `json:"geo_name" gorm:"size:128"`
|
|
GeoLatitude *float64 `json:"geo_latitude"`
|
|
GeoLongitude *float64 `json:"geo_longitude"`
|
|
GeoManualOverride bool `json:"geo_manual_override" gorm:"not null;default:false"`
|
|
AccessToken string `json:"-" gorm:"column:access_token;size:128;index"`
|
|
AutoUpdateEnabled bool `json:"auto_update_enabled" gorm:"not null;default:false"`
|
|
UpdateRequested bool `json:"update_requested" gorm:"not null;default:false"`
|
|
UpdateChannel string `json:"update_channel" gorm:"size:16;not null;default:'stable'"`
|
|
UpdateTag string `json:"update_tag" gorm:"size:64"`
|
|
RestartOpenrestyRequested bool `json:"restart_openresty_requested" gorm:"not null;default:false"`
|
|
Version string `json:"version" gorm:"size:64;not null"`
|
|
ExtVersion string `json:"ext_version" gorm:"size:64"`
|
|
OpenrestyStatus string `json:"openresty_status" gorm:"size:16;not null;default:'unknown'"`
|
|
OpenrestyMessage string `json:"openresty_message" gorm:"type:text"`
|
|
Status string `json:"status" gorm:"size:16;not null;default:'offline'"`
|
|
CurrentVersion string `json:"current_version" gorm:"size:32"`
|
|
LastSeenAt time.Time `json:"last_seen_at"`
|
|
LastError string `json:"last_error" gorm:"type:text"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
// Node type: edge_node (default) | tunnel_relay | tunnel_client
|
|
NodeType string `json:"node_type" gorm:"size:32;not null;default:'edge_node'"`
|
|
// TunnelRelay specific fields
|
|
RelayBindPort int `json:"relay_bind_port" gorm:"not null;default:0"`
|
|
RelayVhostHTTPPort int `json:"relay_vhost_http_port" gorm:"not null;default:0"`
|
|
RelayAuthToken string `json:"-" gorm:"size:128"`
|
|
RelayAgentAccessAddr string `json:"relay_agent_access_addr" gorm:"size:255"`
|
|
RelayClientAccessAddr string `json:"relay_client_access_addr" gorm:"size:255"`
|
|
RelayClientProxyURL string `json:"relay_client_proxy_url" gorm:"size:512"`
|
|
RelayStatus string `json:"relay_status" gorm:"size:16;not null;default:'unknown'"`
|
|
}
|
|
|
|
func ListNodes() (nodes []*Node, err error) {
|
|
err = DB.Order("id desc").Find(&nodes).Error
|
|
return nodes, err
|
|
}
|
|
|
|
func ListNodesByNodeIDs(nodeIDs []string) (nodes []*Node, err error) {
|
|
if len(nodeIDs) == 0 {
|
|
return []*Node{}, nil
|
|
}
|
|
err = DB.Where("node_id IN ?", nodeIDs).Find(&nodes).Error
|
|
return nodes, err
|
|
}
|
|
|
|
func GetNodeByNodeID(nodeID string) (*Node, error) {
|
|
node := &Node{}
|
|
err := DB.Where("node_id = ?", nodeID).First(node).Error
|
|
return node, err
|
|
}
|
|
|
|
func GetNodeByID(id uint) (*Node, error) {
|
|
node := &Node{}
|
|
err := DB.First(node, id).Error
|
|
return node, err
|
|
}
|
|
|
|
func GetNodeByAccessToken(token string) (*Node, error) {
|
|
node := &Node{}
|
|
err := DB.Where("access_token = ?", token).First(node).Error
|
|
return node, err
|
|
}
|
|
|
|
func (node *Node) Insert() error {
|
|
return DB.Create(node).Error
|
|
}
|
|
|
|
func (node *Node) Update() error {
|
|
return DB.Save(node).Error
|
|
}
|
|
|
|
func (node *Node) Delete() error {
|
|
return DB.Delete(node).Error
|
|
}
|
|
|
|
func ListNodesByType(nodeType string) (nodes []*Node, err error) {
|
|
err = DB.Where("node_type = ?", nodeType).Order("id desc").Find(&nodes).Error
|
|
return nodes, err
|
|
}
|