添加snapShot功能,用于刚加入集群收到一大堆sql的处理

chenshijun

2020-01-15 6f220f830ab10690098f9c53144d31fcab2be33d

添加snapShot功能,用于刚加入集群收到一大堆sql的处理

 agent.go

@@ -62,9 +62,9 @@
}

// Create create serf agent with config
func Create(conf *Config) (*Agent, error) {
func Create(conf *Config, snapshotPath string) (*Agent, error) {
    // config cover to serf config
    serfConf, err := conf.convertToSerf()
    serfConf, err := conf.convertToSerf(snapshotPath)
    if err != nil {
        return nil, err
    }

 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"
@@ -44,7 +46,12 @@
    BroadcastInterval  = 5
    MaxQueryRespSize   = 50 * 1024 * 1024
    MaxQuerySize       = 50 * 1024 * 1024
    MaxUserEventSize   = 5 * 1024
    MaxUserEventSize   = 9 * 1024
    ReplayOnJoinDefault = false
    SnapshotPathDefault = "./serfSnapShot"
    MaxEventBufferCount = 2048

    TcpTransportPort = 30194 //tcp传输大数据量接口
)

// DefaultConfig default config
@@ -84,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)
@@ -119,9 +148,18 @@
        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
}