mirror of
https://github.com/wyx2685/V2bX.git
synced 2025-01-22 18:08:14 -05:00
118 lines
3.1 KiB
Go
118 lines
3.1 KiB
Go
package node
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/InazumaV/V2bX/api/iprecoder"
|
|
"github.com/InazumaV/V2bX/api/panel"
|
|
"github.com/InazumaV/V2bX/common/task"
|
|
"github.com/InazumaV/V2bX/conf"
|
|
vCore "github.com/InazumaV/V2bX/core"
|
|
"github.com/InazumaV/V2bX/limiter"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type Controller struct {
|
|
server vCore.Core
|
|
apiClient *panel.Client
|
|
tag string
|
|
limiter *limiter.Limiter
|
|
traffic map[string]int64
|
|
userList []panel.UserInfo
|
|
info *panel.NodeInfo
|
|
ipRecorder iprecoder.IpRecorder
|
|
nodeInfoMonitorPeriodic *task.Task
|
|
userReportPeriodic *task.Task
|
|
renewCertPeriodic *task.Task
|
|
dynamicSpeedLimitPeriodic *task.Task
|
|
onlineIpReportPeriodic *task.Task
|
|
*conf.Options
|
|
}
|
|
|
|
// NewController return a Node controller with default parameters.
|
|
func NewController(server vCore.Core, api *panel.Client, config *conf.Options) *Controller {
|
|
controller := &Controller{
|
|
server: server,
|
|
Options: config,
|
|
apiClient: api,
|
|
}
|
|
return controller
|
|
}
|
|
|
|
// Start implement the Start() function of the service interface
|
|
func (c *Controller) Start() error {
|
|
// First fetch Node Info
|
|
var err error
|
|
node, err := c.apiClient.GetNodeInfo()
|
|
if err != nil {
|
|
return fmt.Errorf("get node info error: %s", err)
|
|
}
|
|
// Update user
|
|
c.userList, err = c.apiClient.GetUserList()
|
|
if err != nil {
|
|
return fmt.Errorf("get user list error: %s", err)
|
|
}
|
|
if len(c.userList) == 0 {
|
|
return errors.New("add users error: not have any user")
|
|
}
|
|
c.tag = c.buildNodeTag(node)
|
|
|
|
// add limiter
|
|
l := limiter.AddLimiter(c.tag, &c.LimitConfig, c.userList)
|
|
// add rule limiter
|
|
if err = l.UpdateRule(&node.Rules); err != nil {
|
|
return fmt.Errorf("update rule error: %s", err)
|
|
}
|
|
c.limiter = l
|
|
if node.Tls || node.Type == "hysteria" {
|
|
err = c.requestCert()
|
|
if err != nil {
|
|
return fmt.Errorf("request cert error: %s", err)
|
|
}
|
|
}
|
|
// Add new tag
|
|
err = c.server.AddNode(c.tag, node, c.Options)
|
|
if err != nil {
|
|
return fmt.Errorf("add new node error: %s", err)
|
|
}
|
|
added, err := c.server.AddUsers(&vCore.AddUsersParams{
|
|
Tag: c.tag,
|
|
Config: c.Options,
|
|
UserInfo: c.userList,
|
|
NodeInfo: node,
|
|
})
|
|
if err != nil {
|
|
return fmt.Errorf("add users error: %s", err)
|
|
}
|
|
log.WithField("tag", c.tag).Infof("Added %d new users", added)
|
|
c.info = node
|
|
c.startTasks(node)
|
|
return nil
|
|
}
|
|
|
|
// Close implement the Close() function of the service interface
|
|
func (c *Controller) Close() error {
|
|
limiter.DeleteLimiter(c.tag)
|
|
if c.nodeInfoMonitorPeriodic != nil {
|
|
c.nodeInfoMonitorPeriodic.Close()
|
|
}
|
|
if c.userReportPeriodic != nil {
|
|
c.userReportPeriodic.Close()
|
|
}
|
|
if c.renewCertPeriodic != nil {
|
|
c.renewCertPeriodic.Close()
|
|
}
|
|
if c.dynamicSpeedLimitPeriodic != nil {
|
|
c.dynamicSpeedLimitPeriodic.Close()
|
|
}
|
|
if c.onlineIpReportPeriodic != nil {
|
|
c.onlineIpReportPeriodic.Close()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Controller) buildNodeTag(node *panel.NodeInfo) string {
|
|
return fmt.Sprintf("%s-%s-%d", c.apiClient.APIHost, node.Type, node.Id)
|
|
}
|