554325746@qq.com
2020-05-13 8fec134c09589b1229c84f51fb37add1c4be35cc
sync.go
@@ -1,4 +1,5 @@
package androidSync
//package main
import (
   sdb "basic.com/Android/syncdb.git"
@@ -8,41 +9,44 @@
   "strings"
   "time"
)
type ReceiveSqlInterface2 interface {
   sdb.ReceiveSqlInterface
}
/*
   每次开机后都会调用该接口,该接口会去查询数据库,确实之前是否已经加入过集群,若是已经加入集群,则开机自动加入
 */
func RegisterReceiveSqlInterfaceFromJava(c ReceiveSqlInterface2) {
   sdb.RegisterReceiveSqlInterface(c)
}
func InitAgent(devID string) bool {
   var nodeIps []string
   sqlFindAllCluster := string("select * from " + dBNameCluster)
   clusters, err := sdb.ExecuteQuerySql([]string{sqlFindAllCluster})
   if err == nil && clusters != nil && len(clusters) > 0 {
   if err == nil && clusters != nil && len(clusters) > 0 && clusters[0].Values != nil {
      c := clusters[0]
      sqlFindNodesByClusterId := string("select * from " + dBNameClusterNode + " where cluster_id='" + c.Values[0][1].(string) + "'")
      nodes, err := sdb.ExecuteQuerySql([]string{sqlFindNodesByClusterId})
      if err == nil && nodes != nil && len(nodes) > 0 {
         var nodeIps []string
         for _, n := range nodes {
            if n.Values[0][3].(string) != devID {
            if n.Values != nil && n.Values[0][3].(string) != devID {
               nodeIps = append(nodeIps, n.Values[0][4].(string))
            }
         }
         agent, err = sdb.Init(c.Values[0][0].(string), c.Values[0][2].(string), devID, nodeIps)
         if agent != nil {
            logger.Debug("sync.Agent init success!")
         } else {
            logger.Debug("sync.Agent init fail!")
         }
      }
      //agent, err = sdb.Init(c.Values[0][0].(string), c.Values[0][2].(string), devID, nodeIps)
      //if agent != nil {
      //   logger.Debug("sync.Agent init success!")
      //} else {
      //   logger.Debug("sync.Agent init fail!")
      //}
   }
   return true
}
/*
   之前没有集群。调用此接口进行集群初始化,传入集群其他节点的ip,便于初始化后直接加入集群
   strAddrs = "ip1:port1;ip2:port2;ip3:port3"
*/
func SyncInit(clusterID string, password string, nodeID string, strAddrs string) bool {
func SyncInit(clusterID string, password string, nodeID string, strAddrs string, snapshotPath string) bool {
   var ips []string
   if strAddrs == "" {
      ips = nil
@@ -52,7 +56,7 @@
   pwdFull := syncClusterKeyPrefix + password
   agent, _ = sdb.Init(clusterID, pwdFull, nodeID, ips)
   agent, _ = sdb.Init(clusterID, pwdFull, nodeID, ips, snapshotPath)
   if agent == nil {
      logger.Error("sdb.Init")
      return false
@@ -61,10 +65,6 @@
   return true
}
/*
   初始化时,若没能加入集群,可以通过该接口加入集群。
   strAddrs = "ip1:port1;ip2:port2;ip3:port3"
*/
func JoinByNodeAddrs(strAddrs string) bool {
   if strAddrs == "" {
      logger.Error("strAddrs == \"\"")
@@ -80,15 +80,11 @@
   return true
}
/*
   加入集群,包含初始化节点SyncInit,并根据传入的集群其他节点列表自动加入集群
   strAddrs = "ip1:port1;ip2:port2;ip3:port3"
*/
func JoinCluster(clusterID, password, strAddrs, devID, devIP, devName string) bool {
func JoinCluster(clusterID, password, strAddrs, devID, devIP, devName string, snapshotPath string) bool {
   isOk := SyncInit(clusterID, password, devID, strAddrs)
   isOk := SyncInit(clusterID, password, devID, strAddrs,snapshotPath)
   if isOk { //加入成功
   if isOk {
      logger.Debug("dbSync.Init success")
      if ! syncTableDataFromCluster(clusterID, devID, devIP, devName) {
@@ -109,22 +105,6 @@
   return true
}
/*
   加入集群后,可以通过该接口获取集群的节点信息,不过最好直接查同步库的集群节点表
json vector
[
{Node1},
{Node2},
...
{Noden}
]
type NodeInfo struct {
   ClusterID   string `json:"clusterID"`
   NodeID      string `json:"nodeID"`
   NodeAddress string `json:"nodeAddress"`
   IsAlive     int    `json:"isAlive"` //StatusNone:0, StatusAlive:1, StatusLeaving:2, StatusLeft:3, StatusFailed:4
}
*/
func GetNodes() []byte {
   nodes := agent.GetNodes()
   strNode, err := json.Marshal(nodes)
@@ -135,20 +115,15 @@
   return strNode
}
/*
   加入集群后, 清空本地的同步库数据,并从集群拉取最新的同步库数据
 */
func syncTableDataFromCluster(clusterID, devID, devIP, devName string) bool {
   var err error
   //0.关闭reference
   foreignSql := string("PRAGMA foreign_keys=OFF")
   _, err = sdb.ExecuteWriteSql([]string{foreignSql}, false)
   if err != nil {
      return false
   }
   //1.删除本地的同步库数据
   var sqls []string
   var delSql string
   for _, t := range syncTables {
@@ -162,7 +137,6 @@
      sqls = append(sqls, delSql)
   }
   //2.拉取集群内的同步库数据到本地数据库表中
   var dumpSqls *[]string
   dumpSqls, err = agent.GetTableDataFromCluster(syncTables)
   if dumpSqls != nil {
@@ -173,7 +147,6 @@
   logger.Debug("成功添加当前节点到集群节点中")
   //3.将本节点加入到节点列表中
   timeUnix := time.Now().Unix()
   fmtTimeStr := time.Unix(timeUnix, 0).Format("2006-01-02 15:04:05")
@@ -182,17 +155,14 @@
      (devIP + ":" + strconv.Itoa(sdb.DefaultBindPort)) + "','" + fmtTimeStr + "')"
   sqls = append(sqls, sqlSync)
   //4. 写入数据库
   _, err = sdb.ExecuteWriteSql(sqls, true)
   if err != nil {
      logger.Debug("sdb.ExecuteWriteSql ERROR:", err)
      return false
   }
   //5. 同步该节点到集群
   agent.SyncSql([]string{sqlSync})
   //6.开启reference
   foreignSql = string("PRAGMA foreign_keys=ON")
   _, err = sdb.ExecuteWriteSql([]string{foreignSql}, false)
   if err != nil {
@@ -202,10 +172,6 @@
   return true
}
/*
   操作数据库后,需要调用该接口将对应的sql语句同步到集群
   strSql = "sql1;sql2;sql3;...;sqln"
 */
func SyncSql(strSql string) {
   sqls := strings.Split(strSql, ";")
@@ -213,9 +179,6 @@
   agent.SyncSql(sqls)
}
/*
   更新集群的名字
 */
func UpdateClusterName(clusterName, clusterID string) bool {
   sql := string("update cluster set cluster_name='" + clusterName + "' where cluster_id='" + clusterID + "'")
   _, err := sdb.ExecuteWriteSql([]string{sql}, false)
@@ -226,9 +189,6 @@
   return true
}
/*
   退出集群
 */
func Leave() bool {
   if agent != nil {
@@ -249,4 +209,4 @@
   }
   return true
}
}