| | |
| | | return nil, err |
| | | } |
| | | serfConf.MemberlistConfig.Keyring = keyring |
| | | |
| | | serfConf.MemberlistConfig.Delegate = &UserDelegate{} |
| | | |
| | | logger.Info("[INFO] agent: Restored keyring with %d keys from %s", |
| | | len(conf.EncryptKey), conf.EncryptKey) |
| | | |
| | | ltLock.Lock() |
| | | curLTime = QueryLTimeFromDbByGorm() |
| | | ltLock.Unlock() |
| | | |
| | | return &Agent{ |
| | | Agent: serfAgent, |
| | |
| | | } |
| | | |
| | | var SyncDbTablePersonCacheChan = make(chan []byte,0) |
| | | var curLTime uint64 |
| | | var ltLock sync.RWMutex |
| | | |
| | | // HandleEvent Handles serf.EventMemberJoin events, |
| | | // which will wait for members to join until the number of group members is equal to "groupExpect" |
| | |
| | | |
| | | switch ev := event.(type) { |
| | | case serf.UserEvent: |
| | | ltLock.Lock() |
| | | defer ltLock.Unlock() |
| | | if ev.Name == UserEventSyncSql { |
| | | var sqlUe SqlUserEvent |
| | | err := json.Unmarshal(ev.Payload, &sqlUe) |
| | |
| | | return |
| | | } |
| | | if sqlUe.Owner != a.conf.NodeName { |
| | | evTime := uint64(ev.LTime) |
| | | logger.Info("ev.LTime:",evTime,",curLTime:",curLTime,",SqlUserEvent.sql:",sqlUe.Sql) |
| | | if curLTime !=0 && evTime < curLTime{//是处理过的事件 |
| | | logger.Info("already executed event,ev.LTime:",evTime,"SqlUserEvent.sql:",sqlUe.Sql) |
| | | return |
| | | } |
| | | flag, _ := ExecuteSqlByGorm(sqlUe.Sql) |
| | | logger.Info("userEvent exec ",sqlUe.Sql,",Result:",flag) |
| | | if flag { |
| | | curLTime = evTime |
| | | |
| | | ExecuteSqlByGorm([]string{"update sync_serf set lamport_time='"+strconv.FormatUint(curLTime,10)+"'"}) |
| | | } |
| | | flag, e := ExecuteSqlByGorm(sqlUe.Sql) |
| | | |
| | | evTime := uint64(ev.LTime) |
| | | logger.Info("ev.LTime:",evTime,"userEvent exec ",sqlUe.Sql,",Result:",flag,", err:",e) |
| | | } |
| | | } else if ev.Name == UserEventSyncDbTablePersonCache { |
| | | logger.Info("LTime:",ev.LTime,",ev.Payload.len:",len(ev.Payload)) |
| | |
| | | //for _, r := range rows { |
| | | // rowsReturn = append(rowsReturn, *r) |
| | | //} |
| | | var tableNames []string |
| | | err := json.Unmarshal(ev.Payload, &tableNames) |
| | | var fromP QueryTableDataParam |
| | | err := json.Unmarshal(ev.Payload, &fromP) |
| | | if err !=nil { |
| | | logger.Error("Query tableNames unmarshal err") |
| | | return |
| | | } |
| | | logger.Info("Query tableNames:",tableNames) |
| | | datas, err := ExecuteQueryByGorm(tableNames) |
| | | logger.Info("Query tableNames:",fromP.Tables) |
| | | datas, err := ExecuteQueryByGorm(fromP.Tables) |
| | | if err !=nil { |
| | | logger.Error("queryByGorm err") |
| | | return |
| | | } |
| | | bytesReturn, err := json.Marshal(datas) |
| | | logger.Info("results.len: ", len(bytesReturn)) |
| | | if query, ok := event.(*serf.Query); ok { |
| | | if err := query.Respond(bytesReturn); err != nil { |
| | | logger.Error("err: %s\n", err) |
| | | return |
| | | |
| | | var targetNode *memberlist.Node |
| | | nodes := a.Serf().Memberlist().Members() |
| | | if nodes != nil && len(nodes) > 0 { |
| | | for _,n :=range nodes { |
| | | if n.Name == fromP.From { |
| | | targetNode = n |
| | | break |
| | | } |
| | | } |
| | | } |
| | | if targetNode !=nil { |
| | | sendErr := a.Serf().Memberlist().SendToTCP(targetNode, bytesReturn) |
| | | if sendErr != nil { |
| | | logger.Debug("sendToTcp err:",sendErr) |
| | | } |
| | | } else { |
| | | logger.Debug("targetNode is nil") |
| | | } |
| | | |
| | | //var res []*Rows |
| | | //json.Unmarshal(bytesReturn, &res) |
| | | //if query, ok := event.(*serf.Query); ok { |
| | | // if err := query.Respond(bytesReturn); err != nil { |
| | | // logger.Error("err: %s\n", err) |
| | | // return |
| | | // } |
| | | //} |
| | | } |
| | | case serf.MemberEvent: |
| | | if event.EventType() == serf.EventMemberLeave { |
| | |
| | | }() |
| | | } |
| | | |
| | | type QueryTableDataParam struct { |
| | | Tables []string `json:"tables"` |
| | | From string `json:"from"` |
| | | } |
| | | |
| | | var QueryTcpResponseChan = make(chan []byte) |
| | | //GetDbFromCluster get the newest database after join cluster |
| | | //dbPathWrite the path where to write after got a database, |
| | | func (a *Agent) GetTableDataFromCluster(tableNames []string) (*[]string,error) { |
| | |
| | | FilterNodes: strings.Fields(specmembername), |
| | | } |
| | | |
| | | //SQL |
| | | tBytes, _ := json.Marshal(tableNames) |
| | | //get db tables |
| | | var fromP = QueryTableDataParam{ |
| | | Tables: tableNames, |
| | | From: a.conf.NodeName, |
| | | } |
| | | tBytes, _ := json.Marshal(fromP) |
| | | |
| | | resp, err := a.Query(QueryEventUpdateDBData, tBytes, ¶ms) |
| | | if err == nil || !strings.Contains(err.Error(), "cannot contain") { |
| | |
| | | } |
| | | logger.Info("Query.resp.err:",err,"resp:",resp) |
| | | |
| | | var dumpSqls []string |
| | | |
| | | var wg sync.WaitGroup |
| | | wg.Add(1) |
| | | var dumpSqls []string |
| | | go func() { |
| | | defer wg.Done() |
| | | respCh := resp.ResponseCh() |
| | | //respCh := resp.ResponseCh() |
| | | for { |
| | | select { |
| | | case r := <-respCh: |
| | | logger.Info("Query response's len:", len(r.Payload)) |
| | | err := json.Unmarshal(r.Payload, &dumpSqls) |
| | | //case r := <-respCh: |
| | | // logger.Info("Query response's len:", len(r.Payload)) |
| | | // err := json.Unmarshal(r.Payload, &dumpSqls) |
| | | // if err ==nil { |
| | | // logger.Error("dumpSql:",dumpSqls) |
| | | // logger.Error("data dump success") |
| | | // } |
| | | // return |
| | | case msg := <-QueryTcpResponseChan: |
| | | logger.Debug("QueryTcpResponseChan receive msg len:",len(msg)) |
| | | err := json.Unmarshal(msg, &dumpSqls) |
| | | if err ==nil { |
| | | logger.Error("dumpSql:",dumpSqls) |
| | | logger.Error("data dump success") |
| | | logger.Error("dumpSql success:",dumpSqls) |
| | | } else { |
| | | logger.Error("data dump err:",err) |
| | | } |
| | | return |
| | | } |
| | | |
| | | } |
| | | }() |
| | | wg.Wait() |