36 lines
796 B
Go
36 lines
796 B
Go
|
package consul
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Raft endpoint is used to manipulate the Raft subsystem
|
||
|
|
type Raft struct {
|
||
|
|
server *Server
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Raft) Apply(args []byte, reply *struct{}) error {
|
||
|
|
future := r.server.raft.Apply(args, 0)
|
||
|
|
return future.Error()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Raft) AddPeer(args string, reply *struct{}) error {
|
||
|
|
peer, err := net.ResolveTCPAddr("tcp", args)
|
||
|
|
if err != nil {
|
||
|
|
r.server.logger.Printf("[ERR] Failed to parse peer: %v", err)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
future := r.server.raft.AddPeer(peer)
|
||
|
|
return future.Error()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Raft) RemovePeer(args string, reply *struct{}) error {
|
||
|
|
peer, err := net.ResolveTCPAddr("tcp", args)
|
||
|
|
if err != nil {
|
||
|
|
r.server.logger.Printf("[ERR] Failed to parse peer: %v", err)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
future := r.server.raft.AddPeer(peer)
|
||
|
|
return future.Error()
|
||
|
|
}
|