mirror of
https://github.com/nezhahq/nezha.git
synced 2025-01-22 12:48:14 -05:00
fc98c0919f
* new geoip method * report geoip separately, fix server creation & deletion bugs * fix struct tag * fix write name * remove deleteion list * remove rpc realip header * Revert "remove rpc realip header" This reverts commit 8a5f86cf2d7df87f28cfa2a3b3430f449dd6ed73.
62 lines
2.0 KiB
Go
62 lines
2.0 KiB
Go
package model
|
|
|
|
import (
|
|
"log"
|
|
"sync"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/naiba/nezha/pkg/utils"
|
|
pb "github.com/naiba/nezha/proto"
|
|
)
|
|
|
|
type Server struct {
|
|
Common
|
|
|
|
Name string `json:"name"`
|
|
UUID string `json:"uuid,omitempty" gorm:"unique"`
|
|
Note string `json:"note,omitempty"` // 管理员可见备注
|
|
PublicNote string `json:"public_note,omitempty"` // 公开备注
|
|
DisplayIndex int `json:"display_index"` // 展示排序，越大越靠前
|
|
HideForGuest bool `json:"hide_for_guest,omitempty"` // 对游客隐藏
|
|
EnableDDNS bool `json:"enable_ddns,omitempty"` // 启用DDNS
|
|
DDNSProfilesRaw string `gorm:"default:'[]';column:ddns_profiles_raw" json:"-"`
|
|
|
|
DDNSProfiles []uint64 `gorm:"-" json:"ddns_profiles"` // DDNS配置
|
|
|
|
Host *Host `gorm:"-" json:"host,omitempty"`
|
|
State *HostState `gorm:"-" json:"state,omitempty"`
|
|
GeoIP *GeoIP `gorm:"-" json:"geoip,omitempty"`
|
|
LastActive time.Time `gorm:"-" json:"last_active,omitempty"`
|
|
|
|
TaskClose chan error `gorm:"-" json:"-"`
|
|
TaskCloseLock *sync.Mutex `gorm:"-" json:"-"`
|
|
TaskStream pb.NezhaService_RequestTaskServer `gorm:"-" json:"-"`
|
|
|
|
PrevTransferInSnapshot int64 `gorm:"-" json:"-"` // 上次数据点时的入站使用量
|
|
PrevTransferOutSnapshot int64 `gorm:"-" json:"-"` // 上次数据点时的出站使用量
|
|
}
|
|
|
|
func (s *Server) CopyFromRunningServer(old *Server) {
|
|
s.Host = old.Host
|
|
s.State = old.State
|
|
s.GeoIP = old.GeoIP
|
|
s.LastActive = old.LastActive
|
|
s.TaskClose = old.TaskClose
|
|
s.TaskCloseLock = old.TaskCloseLock
|
|
s.TaskStream = old.TaskStream
|
|
s.PrevTransferInSnapshot = old.PrevTransferInSnapshot
|
|
s.PrevTransferOutSnapshot = old.PrevTransferOutSnapshot
|
|
}
|
|
|
|
func (s *Server) AfterFind(tx *gorm.DB) error {
|
|
if s.DDNSProfilesRaw != "" {
|
|
if err := utils.Json.Unmarshal([]byte(s.DDNSProfilesRaw), &s.DDNSProfiles); err != nil {
|
|
log.Println("NEZHA>> Server.AfterFind:", err)
|
|
return nil
|
|
}
|
|
}
|
|
return nil
|
|
}
|