69 lines
1.3 KiB
Go
69 lines
1.3 KiB
Go
|
package raft
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
//------------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Helpers
|
||
|
|
//
|
||
|
|
//------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
//--------------------------------------
|
||
|
|
// Logs
|
||
|
|
//--------------------------------------
|
||
|
|
|
||
|
|
func getLogPath() string {
|
||
|
|
f, _ := ioutil.TempFile("", "raft-log-")
|
||
|
|
f.Close()
|
||
|
|
os.Remove(f.Name())
|
||
|
|
return f.Name()
|
||
|
|
}
|
||
|
|
|
||
|
|
func setupLog(content string) string {
|
||
|
|
f, _ := ioutil.TempFile("", "raft-log-")
|
||
|
|
f.Write([]byte(content))
|
||
|
|
f.Close()
|
||
|
|
return f.Name()
|
||
|
|
}
|
||
|
|
|
||
|
|
//--------------------------------------
|
||
|
|
// Servers
|
||
|
|
//--------------------------------------
|
||
|
|
|
||
|
|
func newTestServer(name string) *Server {
|
||
|
|
path, _ := ioutil.TempDir("", "raft-server-")
|
||
|
|
server, _ := NewServer(name, path)
|
||
|
|
server.AddCommandType(&TestCommand1{})
|
||
|
|
server.AddCommandType(&TestCommand2{})
|
||
|
|
return server
|
||
|
|
}
|
||
|
|
|
||
|
|
//--------------------------------------
|
||
|
|
// Command1
|
||
|
|
//--------------------------------------
|
||
|
|
|
||
|
|
type TestCommand1 struct {
|
||
|
|
Val string `json:"val"`
|
||
|
|
I int `json:"i"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c TestCommand1) Name() string {
|
||
|
|
return "cmd_1"
|
||
|
|
}
|
||
|
|
|
||
|
|
//--------------------------------------
|
||
|
|
// Command2
|
||
|
|
//--------------------------------------
|
||
|
|
|
||
|
|
type TestCommand2 struct {
|
||
|
|
X int `json:"x"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c TestCommand2) Name() string {
|
||
|
|
return "cmd_2"
|
||
|
|
}
|
||
|
|
|