diff --git a/api/panel/panel.go b/api/panel/panel.go index 5d46915..cd202f8 100644 --- a/api/panel/panel.go +++ b/api/panel/panel.go @@ -3,6 +3,7 @@ package panel import ( "errors" "fmt" + "net" "strconv" "strings" "time" @@ -18,6 +19,7 @@ import ( type Client struct { client *resty.Client APIHost string + APISendIP string Token string NodeType string NodeId int @@ -29,7 +31,14 @@ type Client struct { } func New(c *conf.ApiConfig) (*Client, error) { - client := resty.New() + var client *resty.Client + if c.APISendIP != "" { + client = resty.NewWithLocalAddr(&net.TCPAddr{ + IP: net.ParseIP(c.APISendIP), + }) + } else { + client = resty.New() + } client.SetRetryCount(3) if c.Timeout > 0 { client.SetTimeout(time.Duration(c.Timeout) * time.Second) @@ -69,12 +78,13 @@ func New(c *conf.ApiConfig) (*Client, error) { "token": c.Key, }) return &Client{ - client: client, - Token: c.Key, - APIHost: c.APIHost, - NodeType: c.NodeType, - NodeId: c.NodeID, - UserList: &UserListBody{}, - AliveMap: &AliveMap{}, + client: client, + Token: c.Key, + APIHost: c.APIHost, + APISendIP: c.APISendIP, + NodeType: c.NodeType, + NodeId: c.NodeID, + UserList: &UserListBody{}, + AliveMap: &AliveMap{}, }, nil } diff --git a/conf/node.go b/conf/node.go index 1d7bd91..df1d25c 100644 --- a/conf/node.go +++ b/conf/node.go @@ -25,6 +25,7 @@ type rawNodeConfig struct { type ApiConfig struct { APIHost string `json:"ApiHost"` + APISendIP string `json:"ApiSendIP"` NodeID int `json:"NodeID"` Key string `json:"ApiKey"` NodeType string `json:"NodeType"`