From 0a86db67f14185ab2e08475fcafb64e43f0022dd Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期一, 15 五月 2023 18:15:33 +0800
Subject: [PATCH] add default role tag
---
config.go | 128 ++++++++++++++++++++++++++++++++++++++----
1 files changed, 114 insertions(+), 14 deletions(-)
diff --git a/config.go b/config.go
index 8a25908..ac65ecc 100644
--- a/config.go
+++ b/config.go
@@ -14,16 +14,20 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
-package serf
+package syncdb
import (
+ "context"
"fmt"
+ "net"
+ "os"
"strconv"
+ "strings"
- "github.com/apache/servicecomb-service-center/syncer/pkg/utils"
+ //"github.com/apache/servicecomb-service-center/syncer/pkg/utils"
"github.com/hashicorp/memberlist"
- "github.com/hashicorp/serf/cmd/serf/command/agent"
- "github.com/hashicorp/serf/serf"
+ "basic.com/valib/serf.git/cmd/serf/command/agent"
+ "basic.com/valib/serf.git/serf"
)
const (
@@ -34,20 +38,59 @@
ModeCluster = "cluster"
retryMaxAttempts = 3
groupExpect = 3
- tagKeyClusterName = "syncer-cluster-name"
+ DefaultEncryptKey = "bjbasic@aiotlink"
+ tagKeyClusterID = "syncer-cluster-name"
TagKeyClusterPort = "syncer-cluster-port"
TagKeyRPCPort = "syncer-rpc-port"
+ BroadcastIP = "255.255.255.255"
+ BroadcastPort = 30193
+ BroadcastInterval = 5
+ MaxQueryRespSize = 50 * 1024 * 1024
+ MaxQuerySize = 50 * 1024 * 1024
+ MaxUserEventSize = 9 * 1024 * 10
+ ReplayOnJoinDefault = false
+ SnapshotPathDefault = "./serfSnapShot"
+ MaxEventBufferCount = 2048
)
// DefaultConfig default config
func DefaultConfig() *Config {
agentConf := agent.DefaultConfig()
+ agentConf.QueryResponseSizeLimit = MaxQueryRespSize
+ agentConf.QuerySizeLimit = MaxQuerySize
+ agentConf.UserEventSizeLimit = MaxUserEventSize
agentConf.BindAddr = fmt.Sprintf("0.0.0.0:%d", DefaultBindPort)
agentConf.RPCAddr = fmt.Sprintf("0.0.0.0:%d", DefaultRPCPort)
return &Config{
Mode: ModeSingle,
Config: agentConf,
ClusterPort: DefaultClusterPort,
+ Ctx: context.Background(),
+ }
+}
+
+func (c *Config) MergeConf(s *Config) {
+ if s != nil {
+ if s.Ctx != nil {
+ c.Ctx = s.Ctx
+ } else {
+ c.Ctx = context.Background()
+ }
+ c.BindAddr = s.BindAddr
+ c.RPCAddr = s.RPCAddr
+ c.RPCPort = s.RPCPort
+ //serf蹇収鍦板潃
+ if s.SnapshotPath != "" {
+ c.SnapshotPath = s.SnapshotPath
+ }
+ if s.EncryptKey != "" {
+ //鎶ユ枃鍔犲瘑鐨刱ey
+ c.EncryptKey = s.EncryptKey
+ }
+ if s.RPCAuthKey != "" {
+ //RPC璁よ瘉鐨刱ey
+ c.RPCAuthKey = s.RPCAuthKey
+ }
}
}
@@ -55,14 +98,15 @@
type Config struct {
// config from serf agent
*agent.Config
- Mode string `json:"mode"`
+ Mode string `json:"mode"`
// name to group members into cluster
- ClusterName string `json:"cluster_name"`
+ ClusterID string `json:"cluster_name"`
// port to communicate between cluster members
- ClusterPort int `yaml:"cluster_port"`
- RPCPort int `yaml:"-"`
+ ClusterPort int `yaml:"cluster_port"`
+ RPCPort int `yaml:"-"`
+ Ctx context.Context
}
// readConfigFile reads configuration from config file
@@ -73,11 +117,33 @@
return nil
}
+func isFileRightful(filePath string) bool {
+ if filePath != "" {
+ _, err := os.Stat(filePath)
+ if err != nil && os.IsNotExist(err) {
+ pos := strings.LastIndex(filePath, "/")
+ if pos != -1 {
+ filePath = filePath[0:pos]
+ }
+
+ _, err = os.Stat(filePath)
+ if err == nil || !os.IsNotExist(err) {
+ return true
+ } else {
+ return false
+ }
+ } else {
+ return false
+ }
+ }
+ return false
+}
+
// convertToSerf convert Config to serf.Config
-func (c *Config) convertToSerf() (*serf.Config, error) {
+func (c *Config) convertToSerf(snapshotPath string) (*serf.Config, error) {
serfConf := serf.DefaultConfig()
- bindIP, bindPort, err := utils.SplitHostPort(c.BindAddr, DefaultBindPort)
+ bindIP, bindPort, err := SplitHostPort(c.BindAddr, DefaultBindPort)
if err != nil {
return nil, fmt.Errorf("invalid bind address: %s", err)
}
@@ -96,15 +162,49 @@
serfConf.MemberlistConfig.BindAddr = bindIP
serfConf.MemberlistConfig.BindPort = bindPort
serfConf.NodeName = c.NodeName
- serfConf.Tags = map[string]string{TagKeyRPCPort: strconv.Itoa(c.RPCPort)}
- if c.ClusterName != "" {
- serfConf.Tags[tagKeyClusterName] = c.ClusterName
+ serfConf.Tags = map[string]string{TagKeyRPCPort: strconv.Itoa(c.RPCPort), "role": "slave"}
+
+ if c.ClusterID != "" {
+ serfConf.Tags[tagKeyClusterID] = c.ClusterID
serfConf.Tags[TagKeyClusterPort] = strconv.Itoa(c.ClusterPort)
}
if c.Mode == ModeCluster && c.RetryMaxAttempts <= 0 {
c.RetryMaxAttempts = retryMaxAttempts
}
+
+ c.SnapshotPath = SnapshotPathDefault
+ if isFileRightful(snapshotPath) {
+ c.SnapshotPath = snapshotPath
+ }
+
+ c.ReplayOnJoin = ReplayOnJoinDefault
+
+ serfConf.QueryResponseSizeLimit = c.QueryResponseSizeLimit
+ serfConf.QuerySizeLimit = c.QuerySizeLimit
+ serfConf.UserEventSizeLimit = c.UserEventSizeLimit
+ serfConf.SnapshotPath = c.SnapshotPath
+ serfConf.EventBuffer = MaxEventBufferCount
+
return serfConf, nil
}
+
+// SplitHostPort returns the parts of the address and port. If the port does not exist, use defaultPort.
+func SplitHostPort(address string, defaultPort int) (string, int, error) {
+ _, _, err := net.SplitHostPort(address)
+ if ae, ok := err.(*net.AddrError); ok && ae.Err == "missing port in address" {
+ address = fmt.Sprintf("%s:%d", address, defaultPort)
+ _, _, err = net.SplitHostPort(address)
+ }
+ if err != nil {
+ return "", 0, err
+ }
+
+ addr, err := net.ResolveTCPAddr("tcp", address)
+ if err != nil {
+ return "", 0, err
+ }
+
+ return addr.IP.String(), addr.Port, nil
+}
--
Gitblit v1.8.0