gost通讯改为ws

go-gost/x/socket/chain.go

@@ -0,0 +1,111 @@
+package socket
+
+import (
+	"errors"
+	"strings"
+
+	"github.com/go-gost/core/logger"
+	"github.com/go-gost/x/config"
+	parser "github.com/go-gost/x/config/parsing/chain"
+	"github.com/go-gost/x/registry"
+)
+
+func createChain(req createChainRequest) error {
+
+	name := strings.TrimSpace(req.Data.Name)
+	if name == "" {
+		return errors.New("chain name is required")
+	}
+	req.Data.Name = name
+
+	if registry.ChainRegistry().IsRegistered(name) {
+		return errors.New("chain " + name + " already exists")
+	}
+
+	v, err := parser.ParseChain(&req.Data, logger.Default())
+	if err != nil {
+		return errors.New("create chain " + name + " failed: " + err.Error())
+	}
+
+	if err := registry.ChainRegistry().Register(name, v); err != nil {
+		return errors.New("chain " + name + " already exists")
+	}
+
+	config.OnUpdate(func(c *config.Config) error {
+		c.Chains = append(c.Chains, &req.Data)
+		return nil
+	})
+
+	return nil
+}
+
+func updateChain(req updateChainRequest) error {
+
+	name := strings.TrimSpace(req.Chain)
+
+	if !registry.ChainRegistry().IsRegistered(name) {
+		return errors.New("chain " + name + " not found")
+	}
+
+	req.Data.Name = name
+
+	v, err := parser.ParseChain(&req.Data, logger.Default())
+	if err != nil {
+		return errors.New("create chain " + name + " failed: " + err.Error())
+	}
+
+	registry.ChainRegistry().Unregister(name)
+
+	if err := registry.ChainRegistry().Register(name, v); err != nil {
+		return errors.New("chain " + name + " already exists")
+	}
+
+	config.OnUpdate(func(c *config.Config) error {
+		for i := range c.Chains {
+			if c.Chains[i].Name == name {
+				c.Chains[i] = &req.Data
+				break
+			}
+		}
+		return nil
+	})
+
+	return nil
+}
+
+func deleteChain(req deleteChainRequest) error {
+
+	name := strings.TrimSpace(req.Chain)
+
+	if !registry.ChainRegistry().IsRegistered(name) {
+		return errors.New("chain " + name + " not found")
+	}
+	registry.ChainRegistry().Unregister(name)
+
+	config.OnUpdate(func(c *config.Config) error {
+		chains := c.Chains
+		c.Chains = nil
+		for _, s := range chains {
+			if s.Name == name {
+				continue
+			}
+			c.Chains = append(c.Chains, s)
+		}
+		return nil
+	})
+
+	return nil
+}
+
+type createChainRequest struct {
+	Data config.ChainConfig `json:"data"`
+}
+
+type updateChainRequest struct {
+	Chain string             `json:"chain"`
+	Data  config.ChainConfig `json:"data"`
+}
+
+type deleteChainRequest struct {
+	Chain string `json:"chain"`
+}