From d70649cfb61e64fabce40199ad1d53d6a4973f0e Mon Sep 17 00:00:00 2001
From: liuxiaolong <liuxiaolong@aiotlink.com>
Date: 星期一, 31 五月 2021 16:23:28 +0800
Subject: [PATCH] no business

---
 config.go |   69 ++++++++++++++++++++++++++++------
 1 files changed, 56 insertions(+), 13 deletions(-)

diff --git a/config.go b/config.go
index 950e3a1..643226a 100644
--- a/config.go
+++ b/config.go
@@ -19,12 +19,14 @@
 import (
 	"fmt"
 	"net"
+	"os"
 	"strconv"
+	"strings"
 
 	//"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 (
@@ -35,13 +37,19 @@
 	ModeCluster        = "cluster"
 	retryMaxAttempts   = 3
 	groupExpect        = 3
-	DefaultEncryptKey   = "bjbasic@aiotlink"
-	tagKeyClusterID  = "syncer-cluster-name"
+	DefaultEncryptKey  = "bjbasic@aiotlink"
+	tagKeyClusterID    = "syncer-cluster-name"
 	TagKeyClusterPort  = "syncer-cluster-port"
 	TagKeyRPCPort      = "syncer-rpc-port"
-	MaxQueryRespSize = 50 * 1024 *1024
-	MaxQuerySize = 1024 *1024
-	MaxUserEventSize = 1024
+	BroadcastIP        = "255.255.255.255"
+	BroadcastPort      = 30193
+	BroadcastInterval  = 5
+	MaxQueryRespSize   = 50 * 1024 * 1024
+	MaxQuerySize       = 50 * 1024 * 1024
+	MaxUserEventSize   = 9 * 1024
+	ReplayOnJoinDefault = false
+	SnapshotPathDefault = "./serfSnapShot"
+	MaxEventBufferCount = 2048
 )
 
 // DefaultConfig default config
@@ -63,14 +71,14 @@
 type Config struct {
 	// config from serf agent
 	*agent.Config
-	Mode string `json:"mode"`
+	Mode 		string 		`json:"mode"`
 
 	// name to group members into cluster
-	ClusterID 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:"-"`
 }
 
 // readConfigFile reads configuration from config file
@@ -81,8 +89,30 @@
 	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 := SplitHostPort(c.BindAddr, DefaultBindPort)
@@ -115,6 +145,20 @@
 	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
 }
 
@@ -136,4 +180,3 @@
 
 	return addr.IP.String(), addr.Port, nil
 }
-

--
Gitblit v1.8.0