mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
0abd860f7e
* server/cors.go renamed to http/cors.go * all CORS code removed from Server and PeerServer * Server and PeerServer fulfill http.Handler, now passed to http.Serve * non-HTTP code in PeerServer.Serve moved to PeerServer.Start
101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package tests
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/coreos/raft"
|
|
|
|
"github.com/coreos/etcd/server"
|
|
"github.com/coreos/etcd/store"
|
|
)
|
|
|
|
const (
|
|
testName = "ETCDTEST"
|
|
testClientURL = "localhost:4401"
|
|
testRaftURL = "localhost:7701"
|
|
testSnapshotCount = 10000
|
|
testHeartbeatTimeout = time.Duration(50) * time.Millisecond
|
|
testElectionTimeout = time.Duration(200) * time.Millisecond
|
|
)
|
|
|
|
// Starts a server in a temporary directory.
|
|
func RunServer(f func(*server.Server)) {
|
|
path, _ := ioutil.TempDir("", "etcd-")
|
|
defer os.RemoveAll(path)
|
|
|
|
store := store.New()
|
|
registry := server.NewRegistry(store)
|
|
|
|
serverStats := server.NewRaftServerStats(testName)
|
|
followersStats := server.NewRaftFollowersStats(testName)
|
|
|
|
psConfig := server.PeerServerConfig{
|
|
Name: testName,
|
|
Path: path,
|
|
URL: "http://"+testRaftURL,
|
|
Scheme: "http",
|
|
SnapshotCount: testSnapshotCount,
|
|
MaxClusterSize: 9,
|
|
}
|
|
ps := server.NewPeerServer(psConfig, registry, store, nil, followersStats, serverStats)
|
|
psListener, err := server.NewListener(testRaftURL)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// Create Raft transporter and server
|
|
tls := &server.TLSConfig{Scheme: "http"}
|
|
dialTimeout := (3 * testHeartbeatTimeout) + testElectionTimeout
|
|
responseHeaderTimeout := (3 * testHeartbeatTimeout) + testElectionTimeout
|
|
raftTransporter := server.NewTransporter(tls.Scheme, tls.Client, followersStats, serverStats, registry, testHeartbeatTimeout, dialTimeout, responseHeaderTimeout)
|
|
raftServer, err := raft.NewServer(testName, path, raftTransporter, store, ps, "")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
raftServer.SetElectionTimeout(testElectionTimeout)
|
|
raftServer.SetHeartbeatTimeout(testHeartbeatTimeout)
|
|
ps.SetRaftServer(raftServer)
|
|
|
|
sConfig := server.ServerConfig{
|
|
Name: testName,
|
|
URL: "http://"+testClientURL,
|
|
}
|
|
s := server.New(sConfig, ps, registry, store, nil)
|
|
sListener, err := server.NewListener(testClientURL)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
ps.SetServer(s)
|
|
|
|
// Start up peer server.
|
|
c := make(chan bool)
|
|
go func() {
|
|
c <- true
|
|
ps.Start(false, []string{})
|
|
http.Serve(psListener, ps)
|
|
}()
|
|
<-c
|
|
|
|
// Start up etcd server.
|
|
go func() {
|
|
c <- true
|
|
http.Serve(sListener, s)
|
|
}()
|
|
<-c
|
|
|
|
// Wait to make sure servers have started.
|
|
time.Sleep(50 * time.Millisecond)
|
|
|
|
// Execute the function passed in.
|
|
f(s)
|
|
|
|
// Clean up servers.
|
|
ps.Stop()
|
|
psListener.Close()
|
|
sListener.Close()
|
|
}
|