change file name

2026-02-04 12:40:11 +00:00 · 2022-12-21 03:04:30 +08:00
parent d98ec9f61c
commit 182fa7ade2
2 changed files with 0 additions and 0 deletions
--- a/node/controller/controller.go
+++ b/node/controller/controller.go
@@ -0,0 +1,166 @@
+package controller
+
+import (
+	"errors"
+	"fmt"
+	"github.com/Yuzuki616/V2bX/api/iprecoder"
+	"github.com/Yuzuki616/V2bX/api/panel"
+	"github.com/Yuzuki616/V2bX/conf"
+	"github.com/Yuzuki616/V2bX/core"
+	"github.com/xtls/xray-core/common/task"
+	"log"
+	"time"
+)
+
+type Node struct {
+	server                    *core.Core
+	clientInfo                panel.ClientInfo
+	apiClient                 panel.Panel
+	nodeInfo                  *panel.NodeInfo
+	Tag                       string
+	userList                  []panel.UserInfo
+	ipRecorder                iprecoder.IpRecorder
+	nodeInfoMonitorPeriodic   *task.Periodic
+	userReportPeriodic        *task.Periodic
+	onlineIpReportPeriodic    *task.Periodic
+	DynamicSpeedLimitPeriodic *task.Periodic
+	*conf.ControllerConfig
+}
+
+// New return a Node service with default parameters.
+func New(server *core.Core, api panel.Panel, config *conf.ControllerConfig) *Node {
+	controller := &Node{
+		server:           server,
+		ControllerConfig: config,
+		apiClient:        api,
+	}
+	return controller
+}
+
+// Start implement the Start() function of the service interface
+func (c *Node) Start() error {
+	c.clientInfo = c.apiClient.Describe()
+	// First fetch Node Info
+	var err error
+	c.nodeInfo, err = c.apiClient.GetNodeInfo()
+	if err != nil {
+		return fmt.Errorf("get node info failed: %s", err)
+	}
+	c.Tag = c.buildNodeTag()
+	// Add new tag
+	err = c.addNewTag(c.nodeInfo)
+	if err != nil {
+		return fmt.Errorf("add new tag failed: %s", err)
+	}
+	// Update user
+	c.userList, err = c.apiClient.GetUserList()
+	if err != nil {
+		return fmt.Errorf("get user list failed: %s", err)
+	}
+	if len(c.userList) == 0 {
+		return errors.New("add users failed: not have any user")
+	}
+	err = c.addNewUser(c.userList, c.nodeInfo)
+	if err != nil {
+		return err
+	}
+	if err := c.server.AddInboundLimiter(c.Tag, c.nodeInfo, c.userList); err != nil {
+		return fmt.Errorf("add inbound limiter failed: %s", err)
+	}
+	// Add Rule Manager
+	if !c.DisableGetRule {
+		if err := c.server.UpdateRule(c.Tag, c.nodeInfo.Rules); err != nil {
+			log.Printf("Update rule filed: %s", err)
+		}
+	}
+	// fetch node info task
+	c.nodeInfoMonitorPeriodic = &task.Periodic{
+		Interval: time.Duration(c.nodeInfo.BaseConfig.PullInterval.(int)) * time.Second,
+		Execute:  c.nodeInfoMonitor,
+	}
+	// fetch user list task
+	c.userReportPeriodic = &task.Periodic{
+		Interval: time.Duration(c.nodeInfo.BaseConfig.PushInterval.(int)) * time.Second,
+		Execute:  c.reportUserTraffic,
+	}
+	log.Printf("[%s: %d] Start monitor node status", c.nodeInfo.NodeType, c.nodeInfo.NodeId)
+	// delay to start nodeInfoMonitor
+	go func() {
+		time.Sleep(time.Duration(c.nodeInfo.BaseConfig.PullInterval.(int)) * time.Second)
+		_ = c.nodeInfoMonitorPeriodic.Start()
+	}()
+
+	log.Printf("[%s: %d] Start report node status", c.nodeInfo.NodeType, c.nodeInfo.NodeId)
+	// delay to start userReport
+	go func() {
+		time.Sleep(time.Duration(c.nodeInfo.BaseConfig.PushInterval.(int)) * time.Second)
+		_ = c.userReportPeriodic.Start()
+	}()
+	if c.EnableDynamicSpeedLimit {
+		// Check dynamic speed limit task
+		c.DynamicSpeedLimitPeriodic = &task.Periodic{
+			Interval: time.Duration(c.DynamicSpeedLimitConfig.Periodic) * time.Second,
+			Execute:  c.dynamicSpeedLimit,
+		}
+		go func() {
+			time.Sleep(time.Duration(c.DynamicSpeedLimitConfig.Periodic) * time.Second)
+			_ = c.DynamicSpeedLimitPeriodic.Start()
+		}()
+		log.Printf("[%s: %d] Start dynamic speed limit", c.nodeInfo.NodeType, c.nodeInfo.NodeId)
+	}
+	if c.EnableIpRecorder {
+		switch c.IpRecorderConfig.Type {
+		case "Recorder":
+			c.ipRecorder = iprecoder.NewRecorder(c.IpRecorderConfig.RecorderConfig)
+		case "Redis":
+			c.ipRecorder = iprecoder.NewRedis(c.IpRecorderConfig.RedisConfig)
+		default:
+			log.Printf("recorder type: %s is not vail, disable recorder", c.IpRecorderConfig.Type)
+			return nil
+		}
+		// report and fetch online ip list task
+		c.onlineIpReportPeriodic = &task.Periodic{
+			Interval: time.Duration(c.IpRecorderConfig.Periodic) * time.Second,
+			Execute:  c.reportOnlineIp,
+		}
+		go func() {
+			time.Sleep(time.Duration(c.IpRecorderConfig.Periodic) * time.Second)
+			_ = c.onlineIpReportPeriodic.Start()
+		}()
+		log.Printf("[%s: %d] Start report online ip", c.nodeInfo.NodeType, c.nodeInfo.NodeId)
+	}
+	return nil
+}
+
+// Close implement the Close() function of the service interface
+func (c *Node) Close() error {
+	if c.nodeInfoMonitorPeriodic != nil {
+		err := c.nodeInfoMonitorPeriodic.Close()
+		if err != nil {
+			log.Panicf("node info periodic close failed: %s", err)
+		}
+	}
+	if c.nodeInfoMonitorPeriodic != nil {
+		err := c.userReportPeriodic.Close()
+		if err != nil {
+			log.Panicf("user report periodic close failed: %s", err)
+		}
+	}
+	if c.onlineIpReportPeriodic != nil {
+		err := c.onlineIpReportPeriodic.Close()
+		if err != nil {
+			log.Panicf("online ip report periodic close failed: %s", err)
+		}
+	}
+	if c.DynamicSpeedLimitPeriodic != nil {
+		err := c.DynamicSpeedLimitPeriodic.Close()
+		if err != nil {
+			log.Panicf("dynamic speed limit periodic close failed: %s", err)
+		}
+	}
+	return nil
+}
+
+func (c *Node) buildNodeTag() string {
+	return fmt.Sprintf("%s_%s_%d", c.nodeInfo.NodeType, c.ListenIP, c.nodeInfo.NodeId)
+}