mirrors/nginx-ui
1
0
Fork 0
mirror of https://github.com/0xJacky/nginx-ui.git synced 2025-05-11 10:25:52 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

enhance: add settings for skip tls cert check

Browse source
This commit is contained in:
Jacky 2024-07-30 15:10:02 +08:00
parent 013d810678
commit f1c0f8ddca
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D
8 changed files with 400 additions and 394 deletions
Download patch file Download diff file Expand all files Collapse all files

163
internal/analytic/node.go
View file

@ -1,43 +1,44 @@
package analytic
import (
"crypto/tls"
"encoding/json"
"github.com/0xJacky/Nginx-UI/internal/logger"
"github.com/0xJacky/Nginx-UI/internal/upgrader"
"github.com/0xJacky/Nginx-UI/model"
"github.com/shirou/gopsutil/v3/load"
"github.com/shirou/gopsutil/v3/net"
"io"
"net/http"
"net/url"
"sync"
"time"
"crypto/tls"
"encoding/json"
"github.com/0xJacky/Nginx-UI/internal/logger"
"github.com/0xJacky/Nginx-UI/internal/upgrader"
"github.com/0xJacky/Nginx-UI/model"
"github.com/0xJacky/Nginx-UI/settings"
"github.com/shirou/gopsutil/v3/load"
"github.com/shirou/gopsutil/v3/net"
"io"
"net/http"
"net/url"
"sync"
"time"
)
type NodeInfo struct {
NodeRuntimeInfo upgrader.RuntimeInfo `json:"node_runtime_info"`
Version string `json:"version"`
CPUNum int `json:"cpu_num"`
MemoryTotal string `json:"memory_total"`
DiskTotal string `json:"disk_total"`
NodeRuntimeInfo upgrader.RuntimeInfo `json:"node_runtime_info"`
Version string `json:"version"`
CPUNum int `json:"cpu_num"`
MemoryTotal string `json:"memory_total"`
DiskTotal string `json:"disk_total"`
}
type NodeStat struct {
AvgLoad *load.AvgStat `json:"avg_load"`
CPUPercent float64 `json:"cpu_percent"`
MemoryPercent float64 `json:"memory_percent"`
DiskPercent float64 `json:"disk_percent"`
Network net.IOCountersStat `json:"network"`
Status bool `json:"status"`
ResponseAt time.Time `json:"response_at"`
AvgLoad *load.AvgStat `json:"avg_load"`
CPUPercent float64 `json:"cpu_percent"`
MemoryPercent float64 `json:"memory_percent"`
DiskPercent float64 `json:"disk_percent"`
Network net.IOCountersStat `json:"network"`
Status bool `json:"status"`
ResponseAt time.Time `json:"response_at"`
}
type Node struct {
EnvironmentID int `json:"environment_id,omitempty"`
*model.Environment
NodeStat
NodeInfo
EnvironmentID int `json:"environment_id,omitempty"`
*model.Environment
NodeStat
NodeInfo
}
var mutex sync.Mutex
@ -47,74 +48,74 @@ type TNodeMap map[int]*Node
var NodeMap TNodeMap
func init() {
NodeMap = make(TNodeMap)
NodeMap = make(TNodeMap)
}
func GetNode(env *model.Environment) (n *Node) {
if env == nil {
// this should never happen
logger.Error("env is nil")
return
}
if !env.Enabled {
return &Node{
Environment: env,
}
}
n, ok := NodeMap[env.ID]
if !ok {
n = &Node{}
}
n.Environment = env
return n
if env == nil {
// this should never happen
logger.Error("env is nil")
return
}
if !env.Enabled {
return &Node{
Environment: env,
}
}
n, ok := NodeMap[env.ID]
if !ok {
n = &Node{}
}
n.Environment = env
return n
}
func InitNode(env *model.Environment) (n *Node) {
n = &Node{
Environment: env,
}
n = &Node{
Environment: env,
}
u, err := url.JoinPath(env.URL, "/api/node")
u, err := url.JoinPath(env.URL, "/api/node")
if err != nil {
logger.Error(err)
return
}
if err != nil {
logger.Error(err)
return
}
client := http.Client{
Transport: &http.Transport{
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
},
}
client := http.Client{
Transport: &http.Transport{
TLSClientConfig: &tls.Config{InsecureSkipVerify: settings.ServerSettings.InsecureSkipVerify},
},
}
req, err := http.NewRequest("GET", u, nil)
if err != nil {
logger.Error(err)
return
}
req, err := http.NewRequest("GET", u, nil)
if err != nil {
logger.Error(err)
return
}
req.Header.Set("X-Node-Secret", env.Token)
req.Header.Set("X-Node-Secret", env.Token)
resp, err := client.Do(req)
resp, err := client.Do(req)
if err != nil {
logger.Error(err)
return
}
if err != nil {
logger.Error(err)
return
}
defer resp.Body.Close()
bytes, _ := io.ReadAll(resp.Body)
defer resp.Body.Close()
bytes, _ := io.ReadAll(resp.Body)
if resp.StatusCode != http.StatusOK {
logger.Error(string(bytes))
return
}
if resp.StatusCode != http.StatusOK {
logger.Error(string(bytes))
return
}
err = json.Unmarshal(bytes, &n.NodeInfo)
if err != nil {
logger.Error(err)
return
}
err = json.Unmarshal(bytes, &n.NodeInfo)
if err != nil {
logger.Error(err)
return
}
return
return
}