2022-07-28 23:00:05 +08:00
|
|
|
package node
|
2022-06-02 01:35:41 +08:00
|
|
|
|
|
|
|
import (
|
2023-06-09 12:36:49 +08:00
|
|
|
"fmt"
|
2023-07-29 18:47:47 +08:00
|
|
|
|
2023-07-29 19:27:15 +08:00
|
|
|
"github.com/InazumaV/V2bX/api/panel"
|
|
|
|
"github.com/InazumaV/V2bX/conf"
|
|
|
|
vCore "github.com/InazumaV/V2bX/core"
|
2022-06-02 01:35:41 +08:00
|
|
|
)
|
|
|
|
|
2022-07-28 23:00:05 +08:00
|
|
|
type Node struct {
|
2023-05-16 09:15:29 +08:00
|
|
|
controllers []*Controller
|
2022-06-02 01:35:41 +08:00
|
|
|
}
|
|
|
|
|
2022-10-10 11:31:15 +08:00
|
|
|
func New() *Node {
|
|
|
|
return &Node{}
|
2022-06-02 01:35:41 +08:00
|
|
|
}
|
|
|
|
|
2023-08-20 15:13:52 +08:00
|
|
|
func (n *Node) Start(nodes []conf.NodeConfig, core vCore.Core) error {
|
2023-05-16 09:15:29 +08:00
|
|
|
n.controllers = make([]*Controller, len(nodes))
|
2023-08-20 15:13:52 +08:00
|
|
|
for i := range nodes {
|
|
|
|
p, err := panel.New(&nodes[i].ApiConfig)
|
2022-12-18 23:31:06 +08:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2022-10-10 11:31:15 +08:00
|
|
|
// Register controller service
|
2023-08-20 15:13:52 +08:00
|
|
|
n.controllers[i] = NewController(core, p, &nodes[i].Options)
|
2022-12-18 23:31:06 +08:00
|
|
|
err = n.controllers[i].Start()
|
2022-10-10 11:31:15 +08:00
|
|
|
if err != nil {
|
2023-06-09 12:36:49 +08:00
|
|
|
return fmt.Errorf("start node controller [%s-%s-%d] error: %s",
|
2023-08-20 15:13:52 +08:00
|
|
|
nodes[i].ApiConfig.APIHost,
|
|
|
|
nodes[i].ApiConfig.NodeType,
|
|
|
|
nodes[i].ApiConfig.NodeID,
|
2023-06-09 12:36:49 +08:00
|
|
|
err)
|
2022-09-07 23:02:02 +08:00
|
|
|
}
|
|
|
|
}
|
2022-06-02 01:35:41 +08:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2022-10-10 11:31:15 +08:00
|
|
|
func (n *Node) Close() {
|
|
|
|
for _, c := range n.controllers {
|
|
|
|
err := c.Close()
|
2022-10-10 08:59:00 +08:00
|
|
|
if err != nil {
|
2022-10-10 11:31:15 +08:00
|
|
|
panic(err)
|
2022-10-10 08:59:00 +08:00
|
|
|
}
|
|
|
|
}
|
2022-10-10 11:31:15 +08:00
|
|
|
n.controllers = nil
|
2022-06-02 01:35:41 +08:00
|
|
|
}
|