gh-ost/vendor/github.com/siddontang/go-mysql/failover/mysql_gtid_handler.go
2017-02-12 13:13:54 +02:00

142 lines
3.0 KiB
Go

package failover
import (
"fmt"
"net"
"github.com/juju/errors"
. "github.com/siddontang/go-mysql/mysql"
)
type MysqlGTIDHandler struct {
Handler
}
func (h *MysqlGTIDHandler) Promote(s *Server) error {
if err := h.WaitRelayLogDone(s); err != nil {
return errors.Trace(err)
}
if err := s.StopSlave(); err != nil {
return errors.Trace(err)
}
return nil
}
func (h *MysqlGTIDHandler) FindBestSlaves(slaves []*Server) ([]*Server, error) {
// MHA use Relay_Master_Log_File and Exec_Master_Log_Pos to determind which is the best slave
bestSlaves := []*Server{}
ps := make([]Position, len(slaves))
lastIndex := -1
for i, slave := range slaves {
pos, err := slave.FetchSlaveExecutePos()
if err != nil {
return nil, errors.Trace(err)
}
ps[i] = pos
if lastIndex == -1 {
lastIndex = i
bestSlaves = []*Server{slave}
} else {
switch ps[lastIndex].Compare(pos) {
case 1:
//do nothing
case -1:
lastIndex = i
bestSlaves = []*Server{slave}
case 0:
// these two slaves have same data,
bestSlaves = append(bestSlaves, slave)
}
}
}
return bestSlaves, nil
}
const changeMasterToWithAuto = `CHANGE MASTER TO
MASTER_HOST = "%s", MASTER_PORT = %s,
MASTER_USER = "%s", MASTER_PASSWORD = "%s",
MASTER_AUTO_POSITION = 1`
func (h *MysqlGTIDHandler) ChangeMasterTo(s *Server, m *Server) error {
if err := h.WaitRelayLogDone(s); err != nil {
return errors.Trace(err)
}
if err := s.StopSlave(); err != nil {
return errors.Trace(err)
}
if err := s.ResetSlave(); err != nil {
return errors.Trace(err)
}
host, port, _ := net.SplitHostPort(m.Addr)
if _, err := s.Execute(fmt.Sprintf(changeMasterToWithAuto,
host, port, m.ReplUser.Name, m.ReplUser.Password)); err != nil {
return errors.Trace(err)
}
if err := s.StartSlave(); err != nil {
return errors.Trace(err)
}
return nil
}
func (h *MysqlGTIDHandler) WaitRelayLogDone(s *Server) error {
if err := s.StopSlaveIOThread(); err != nil {
return errors.Trace(err)
}
r, err := s.SlaveStatus()
if err != nil {
return errors.Trace(err)
}
retrieved, _ := r.GetStringByName(0, "Retrieved_Gtid_Set")
// may only support MySQL version >= 5.6.9
// see http://dev.mysql.com/doc/refman/5.6/en/gtid-functions.html
return h.waitUntilAfterGTIDs(s, retrieved)
}
func (h *MysqlGTIDHandler) WaitCatchMaster(s *Server, m *Server) error {
r, err := m.MasterStatus()
if err != nil {
return errors.Trace(err)
}
masterGTIDSet, _ := r.GetStringByName(0, "Executed_Gtid_Set")
return h.waitUntilAfterGTIDs(s, masterGTIDSet)
}
func (h *MysqlGTIDHandler) CheckGTIDMode(slaves []*Server) error {
for i := 0; i < len(slaves); i++ {
mode, err := slaves[i].MysqlGTIDMode()
if err != nil {
return errors.Trace(err)
} else if mode != GTIDModeOn {
return errors.Errorf("%s use not GTID mode", slaves[i].Addr)
}
}
return nil
}
func (h *MysqlGTIDHandler) waitUntilAfterGTIDs(s *Server, gtids string) error {
_, err := s.Execute(fmt.Sprintf("SELECT WAIT_UNTIL_SQL_THREAD_AFTER_GTIDS('%s')", gtids))
return err
}