From fc65716cb14d829c5ad9f79aa909ec0564eea2bd Mon Sep 17 00:00:00 2001
From: chenshijun <csj_sky@126.com>
Date: 星期六, 12 十月 2019 11:36:03 +0800
Subject: [PATCH] 判断SnapshotPath合法性
---
config.go | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++++-------
1 files changed, 52 insertions(+), 7 deletions(-)
diff --git a/config.go b/config.go
index 950e3a1..4b34256 100644
--- a/config.go
+++ b/config.go
@@ -19,7 +19,9 @@
import (
"fmt"
"net"
+ "os"
"strconv"
+ "strings"
//"github.com/apache/servicecomb-service-center/syncer/pkg/utils"
"github.com/hashicorp/memberlist"
@@ -35,13 +37,21 @@
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
+
+ TcpTransportPort = 30194 //tcp浼犺緭澶ф暟鎹噺鎺ュ彛
)
// DefaultConfig default config
@@ -81,8 +91,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 +147,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 +182,3 @@
return addr.IP.String(), addr.Port, nil
}
-
--
Gitblit v1.8.0