Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

functional-tester/agent: rename "logger" field

Browse Source 
Signed-off-by: Gyuho Lee <gyuhox@gmail.com>
This commit is contained in:
Gyuho Lee 2018-04-03 09:43:03 -07:00
parent eb0c66f912
commit c2731cde54
2 changed files with 55 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

94
tools/functional-tester/agent/handler.go
View File

@ -36,7 +36,7 @@ func (srv *Server) handleTesterRequest(req *rpcpb.Request) (resp *rpcpb.Response
defer func() {
if err == nil {
srv.last = req.Operation
srv.logger.Info("handler success", zap.String("operation", req.Operation.String()))
srv.lg.Info("handler success", zap.String("operation", req.Operation.String()))
}
}()
@ -78,24 +78,24 @@ func (srv *Server) handleInitialStartEtcd(req *rpcpb.Request) (*rpcpb.Response,
srv.Member = req.Member
srv.Tester = req.Tester
srv.logger.Info("creating base directory", zap.String("path", srv.Member.BaseDir))
srv.lg.Info("creating base directory", zap.String("path", srv.Member.BaseDir))
err := fileutil.TouchDirAll(srv.Member.BaseDir)
if err != nil {
return nil, err
}
srv.logger.Info("created base directory", zap.String("path", srv.Member.BaseDir))
srv.lg.Info("created base directory", zap.String("path", srv.Member.BaseDir))
if err = srv.createEtcdFile(); err != nil {
return nil, err
}
srv.creatEtcdCmd()
srv.logger.Info("starting etcd")
srv.lg.Info("starting etcd")
err = srv.startEtcdCmd()
if err != nil {
return nil, err
}
srv.logger.Info("started etcd", zap.String("command-path", srv.etcdCmd.Path))
srv.lg.Info("started etcd", zap.String("command-path", srv.etcdCmd.Path))
// wait some time for etcd listener start
// before setting up proxy
@ -121,9 +121,9 @@ func (srv *Server) startProxy() error {
return err
}
srv.logger.Info("starting proxy on client traffic", zap.String("url", advertiseClientURL.String()))
srv.lg.Info("starting proxy on client traffic", zap.String("url", advertiseClientURL.String()))
srv.advertiseClientPortToProxy[advertiseClientURLPort] = transport.NewProxy(transport.ProxyConfig{
Logger: srv.logger,
Logger: srv.lg,
From: *advertiseClientURL,
To: *listenClientURL,
})
@ -131,7 +131,7 @@ func (srv *Server) startProxy() error {
case err = <-srv.advertiseClientPortToProxy[advertiseClientURLPort].Error():
return err
case <-time.After(2 * time.Second):
srv.logger.Info("started proxy on client traffic", zap.String("url", advertiseClientURL.String()))
srv.lg.Info("started proxy on client traffic", zap.String("url", advertiseClientURL.String()))
}
}
@ -145,9 +145,9 @@ func (srv *Server) startProxy() error {
return err
}
srv.logger.Info("starting proxy on peer traffic", zap.String("url", advertisePeerURL.String()))
srv.lg.Info("starting proxy on peer traffic", zap.String("url", advertisePeerURL.String()))
srv.advertisePeerPortToProxy[advertisePeerURLPort] = transport.NewProxy(transport.ProxyConfig{
Logger: srv.logger,
Logger: srv.lg,
From: *advertisePeerURL,
To: *listenPeerURL,
})
@ -155,7 +155,7 @@ func (srv *Server) startProxy() error {
case err = <-srv.advertisePeerPortToProxy[advertisePeerURLPort].Error():
return err
case <-time.After(2 * time.Second):
srv.logger.Info("started proxy on peer traffic", zap.String("url", advertisePeerURL.String()))
srv.lg.Info("started proxy on peer traffic", zap.String("url", advertisePeerURL.String()))
}
}
return nil
@ -164,13 +164,13 @@ func (srv *Server) startProxy() error {
func (srv *Server) stopProxy() {
if srv.Member.EtcdClientProxy && len(srv.advertiseClientPortToProxy) > 0 {
for port, px := range srv.advertiseClientPortToProxy {
srv.logger.Info("closing proxy",
srv.lg.Info("closing proxy",
zap.Int("port", port),
zap.String("from", px.From()),
zap.String("to", px.To()),
)
if err := px.Close(); err != nil {
srv.logger.Warn("failed to close proxy", zap.Int("port", port))
srv.lg.Warn("failed to close proxy", zap.Int("port", port))
continue
}
select {
@ -179,7 +179,7 @@ func (srv *Server) stopProxy() {
time.Sleep(time.Second)
case <-time.After(time.Second):
}
srv.logger.Info("closed proxy",
srv.lg.Info("closed proxy",
zap.Int("port", port),
zap.String("from", px.From()),
zap.String("to", px.To()),
@ -189,13 +189,13 @@ func (srv *Server) stopProxy() {
}
if srv.Member.EtcdPeerProxy && len(srv.advertisePeerPortToProxy) > 0 {
for port, px := range srv.advertisePeerPortToProxy {
srv.logger.Info("closing proxy",
srv.lg.Info("closing proxy",
zap.Int("port", port),
zap.String("from", px.From()),
zap.String("to", px.To()),
)
if err := px.Close(); err != nil {
srv.logger.Warn("failed to close proxy", zap.Int("port", port))
srv.lg.Warn("failed to close proxy", zap.Int("port", port))
continue
}
select {
@ -204,7 +204,7 @@ func (srv *Server) stopProxy() {
time.Sleep(time.Second)
case <-time.After(time.Second):
}
srv.logger.Info("closed proxy",
srv.lg.Info("closed proxy",
zap.Int("port", port),
zap.String("from", px.From()),
zap.String("to", px.To()),
@ -215,20 +215,20 @@ func (srv *Server) stopProxy() {
}
func (srv *Server) createEtcdFile() error {
srv.logger.Info("creating etcd log file", zap.String("path", srv.Member.EtcdLogPath))
srv.lg.Info("creating etcd log file", zap.String("path", srv.Member.EtcdLogPath))
var err error
srv.etcdLogFile, err = os.Create(srv.Member.EtcdLogPath)
if err != nil {
return err
}
srv.logger.Info("created etcd log file", zap.String("path", srv.Member.EtcdLogPath))
srv.lg.Info("created etcd log file", zap.String("path", srv.Member.EtcdLogPath))
return nil
}
func (srv *Server) creatEtcdCmd() {
etcdPath, etcdFlags := srv.Member.EtcdExecPath, srv.Member.Etcd.Flags()
u, _ := url.Parse(srv.Member.FailpointHTTPAddr)
srv.logger.Info("creating etcd command",
srv.lg.Info("creating etcd command",
zap.String("etcd-exec-path", etcdPath),
zap.Strings("etcd-flags", etcdFlags),
zap.String("failpoint-http-addr", srv.Member.FailpointHTTPAddr),
@ -248,12 +248,12 @@ func (srv *Server) startEtcdCmd() error {
func (srv *Server) handleRestartEtcd() (*rpcpb.Response, error) {
srv.creatEtcdCmd()
srv.logger.Info("restarting etcd")
srv.lg.Info("restarting etcd")
err := srv.startEtcdCmd()
if err != nil {
return nil, err
}
srv.logger.Info("restarted etcd", zap.String("command-path", srv.etcdCmd.Path))
srv.lg.Info("restarted etcd", zap.String("command-path", srv.etcdCmd.Path))
// wait some time for etcd listener start
// before setting up proxy
@ -273,12 +273,12 @@ func (srv *Server) handleRestartEtcd() (*rpcpb.Response, error) {
func (srv *Server) handleKillEtcd() (*rpcpb.Response, error) {
srv.stopProxy()
srv.logger.Info("killing etcd", zap.String("signal", syscall.SIGTERM.String()))
srv.lg.Info("killing etcd", zap.String("signal", syscall.SIGTERM.String()))
err := stopWithSig(srv.etcdCmd, syscall.SIGTERM)
if err != nil {
return nil, err
}
srv.logger.Info("killed etcd", zap.String("signal", syscall.SIGTERM.String()))
srv.lg.Info("killed etcd", zap.String("signal", syscall.SIGTERM.String()))
return &rpcpb.Response{
Success: true,
@ -290,18 +290,18 @@ func (srv *Server) handleFailArchive() (*rpcpb.Response, error) {
srv.stopProxy()
// exit with stackstrace
srv.logger.Info("killing etcd", zap.String("signal", syscall.SIGQUIT.String()))
srv.lg.Info("killing etcd", zap.String("signal", syscall.SIGQUIT.String()))
err := stopWithSig(srv.etcdCmd, syscall.SIGQUIT)
if err != nil {
return nil, err
}
srv.logger.Info("killed etcd", zap.String("signal", syscall.SIGQUIT.String()))
srv.lg.Info("killed etcd", zap.String("signal", syscall.SIGQUIT.String()))
srv.etcdLogFile.Sync()
srv.etcdLogFile.Close()
// TODO: support separate WAL directory
srv.logger.Info("archiving data", zap.String("base-dir", srv.Member.BaseDir))
srv.lg.Info("archiving data", zap.String("base-dir", srv.Member.BaseDir))
if err = archive(
srv.Member.BaseDir,
srv.Member.EtcdLogPath,
@ -309,17 +309,17 @@ func (srv *Server) handleFailArchive() (*rpcpb.Response, error) {
); err != nil {
return nil, err
}
srv.logger.Info("archived data", zap.String("base-dir", srv.Member.BaseDir))
srv.lg.Info("archived data", zap.String("base-dir", srv.Member.BaseDir))
if err = srv.createEtcdFile(); err != nil {
return nil, err
}
srv.logger.Info("cleaning up page cache")
srv.lg.Info("cleaning up page cache")
if err := cleanPageCache(); err != nil {
srv.logger.Warn("failed to clean up page cache", zap.String("error", err.Error()))
srv.lg.Warn("failed to clean up page cache", zap.String("error", err.Error()))
}
srv.logger.Info("cleaned up page cache")
srv.lg.Info("cleaned up page cache")
return &rpcpb.Response{
Success: true,
@ -329,32 +329,32 @@ func (srv *Server) handleFailArchive() (*rpcpb.Response, error) {
// stop proxy, etcd, delete data directory
func (srv *Server) handleDestroyEtcdAgent() (*rpcpb.Response, error) {
srv.logger.Info("killing etcd", zap.String("signal", syscall.SIGTERM.String()))
srv.lg.Info("killing etcd", zap.String("signal", syscall.SIGTERM.String()))
err := stopWithSig(srv.etcdCmd, syscall.SIGTERM)
if err != nil {
return nil, err
}
srv.logger.Info("killed etcd", zap.String("signal", syscall.SIGTERM.String()))
srv.lg.Info("killed etcd", zap.String("signal", syscall.SIGTERM.String()))
srv.logger.Info("removing base directory", zap.String("dir", srv.Member.BaseDir))
srv.lg.Info("removing base directory", zap.String("dir", srv.Member.BaseDir))
err = os.RemoveAll(srv.Member.BaseDir)
if err != nil {
return nil, err
}
srv.logger.Info("removed base directory", zap.String("dir", srv.Member.BaseDir))
srv.lg.Info("removed base directory", zap.String("dir", srv.Member.BaseDir))
// stop agent server
srv.Stop()
for port, px := range srv.advertiseClientPortToProxy {
srv.logger.Info("closing proxy", zap.Int("client-port", port))
srv.lg.Info("closing proxy", zap.Int("client-port", port))
err := px.Close()
srv.logger.Info("closed proxy", zap.Int("client-port", port), zap.Error(err))
srv.lg.Info("closed proxy", zap.Int("client-port", port), zap.Error(err))
}
for port, px := range srv.advertisePeerPortToProxy {
srv.logger.Info("closing proxy", zap.Int("peer-port", port))
srv.lg.Info("closing proxy", zap.Int("peer-port", port))
err := px.Close()
srv.logger.Info("closed proxy", zap.Int("peer-port", port), zap.Error(err))
srv.lg.Info("closed proxy", zap.Int("peer-port", port), zap.Error(err))
}
return &rpcpb.Response{
@ -365,10 +365,10 @@ func (srv *Server) handleDestroyEtcdAgent() (*rpcpb.Response, error) {
func (srv *Server) handleBlackholePeerPortTxRx() (*rpcpb.Response, error) {
for port, px := range srv.advertisePeerPortToProxy {
srv.logger.Info("blackholing", zap.Int("peer-port", port))
srv.lg.Info("blackholing", zap.Int("peer-port", port))
px.BlackholeTx()
px.BlackholeRx()
srv.logger.Info("blackholed", zap.Int("peer-port", port))
srv.lg.Info("blackholed", zap.Int("peer-port", port))
}
return &rpcpb.Response{
Success: true,
@ -378,10 +378,10 @@ func (srv *Server) handleBlackholePeerPortTxRx() (*rpcpb.Response, error) {
func (srv *Server) handleUnblackholePeerPortTxRx() (*rpcpb.Response, error) {
for port, px := range srv.advertisePeerPortToProxy {
srv.logger.Info("unblackholing", zap.Int("peer-port", port))
srv.lg.Info("unblackholing", zap.Int("peer-port", port))
px.UnblackholeTx()
px.UnblackholeRx()
srv.logger.Info("unblackholed", zap.Int("peer-port", port))
srv.lg.Info("unblackholed", zap.Int("peer-port", port))
}
return &rpcpb.Response{
Success: true,
@ -394,14 +394,14 @@ func (srv *Server) handleDelayPeerPortTxRx() (*rpcpb.Response, error) {
rv := time.Duration(srv.Tester.DelayLatencyMsRv) * time.Millisecond
for port, px := range srv.advertisePeerPortToProxy {
srv.logger.Info("delaying",
srv.lg.Info("delaying",
zap.Int("peer-port", port),
zap.Duration("latency", lat),
zap.Duration("random-variable", rv),
)
px.DelayTx(lat, rv)
px.DelayRx(lat, rv)
srv.logger.Info("delayed",
srv.lg.Info("delayed",
zap.Int("peer-port", port),
zap.Duration("latency", lat),
zap.Duration("random-variable", rv),
@ -416,10 +416,10 @@ func (srv *Server) handleDelayPeerPortTxRx() (*rpcpb.Response, error) {
func (srv *Server) handleUndelayPeerPortTxRx() (*rpcpb.Response, error) {
for port, px := range srv.advertisePeerPortToProxy {
srv.logger.Info("undelaying", zap.Int("peer-port", port))
srv.lg.Info("undelaying", zap.Int("peer-port", port))
px.UndelayTx()
px.UndelayRx()
srv.logger.Info("undelayed", zap.Int("peer-port", port))
srv.lg.Info("undelayed", zap.Int("peer-port", port))
}
return &rpcpb.Response{
Success: true,

17
tools/functional-tester/agent/server.go
View File

@ -34,7 +34,7 @@ import (
// serialized in tester-side
type Server struct {
grpcServer *grpc.Server
logger *zap.Logger
lg *zap.Logger
network string
address string
@ -56,12 +56,12 @@ type Server struct {
// NewServer returns a new agent server.
func NewServer(
logger *zap.Logger,
lg *zap.Logger,
network string,
address string,
) *Server {
return &Server{
logger: logger,
lg: lg,
network: network,
address: address,
last: rpcpb.Operation_NotStarted,
@ -93,34 +93,33 @@ func (srv *Server) StartServe() error {
rpcpb.RegisterTransportServer(srv.grpcServer, srv)
srv.logger.Info(
srv.lg.Info(
"gRPC server started",
zap.String("address", srv.address),
zap.String("listener-address", srv.ln.Addr().String()),
)
err = srv.grpcServer.Serve(srv.ln)
if err != nil && strings.Contains(err.Error(), "use of closed network connection") {
srv.logger.Info(
srv.lg.Info(
"gRPC server is shut down",
zap.String("address", srv.address),
zap.Error(err),
)
} else {
srv.logger.Warn(
srv.lg.Warn(
"gRPC server returned with error",
zap.String("address", srv.address),
zap.Error(err),
)
}
return err
}
// Stop stops serving gRPC server.
func (srv *Server) Stop() {
srv.logger.Info("gRPC server stopping", zap.String("address", srv.address))
srv.lg.Info("gRPC server stopping", zap.String("address", srv.address))
srv.grpcServer.Stop()
srv.logger.Info("gRPC server stopped", zap.String("address", srv.address))
srv.lg.Info("gRPC server stopped", zap.String("address", srv.address))
}
// Transport communicates with etcd tester.