|
15 | 15 | package agent
|
16 | 16 |
|
17 | 17 | import (
|
| 18 | + "fmt" |
18 | 19 | "net"
|
19 | 20 | "os"
|
20 | 21 | "path/filepath"
|
21 | 22 |
|
22 | 23 | "github.com/coreos/dbtester/dbtesterpb"
|
23 | 24 | "github.com/coreos/dbtester/pkg/ntp"
|
| 25 | + "go.uber.org/zap" |
24 | 26 |
|
25 | 27 | "github.com/coreos/etcd/pkg/netutil"
|
26 |
| - "github.com/coreos/pkg/capnslog" |
27 | 28 | "github.com/gyuho/linux-inspect/df"
|
28 | 29 | "github.com/spf13/cobra"
|
29 | 30 | "google.golang.org/grpc"
|
@@ -58,11 +59,11 @@ var globalFlags flags
|
58 | 59 | func init() {
|
59 | 60 | dn, err := df.GetDevice("/")
|
60 | 61 | if err != nil {
|
61 |
| - plog.Warningf("cannot get disk device mounted at '/' (%v)", err) |
| 62 | + fmt.Printf("cannot get disk device mounted at '/' (%v)\n", err) |
62 | 63 | }
|
63 | 64 | nm, err := netutil.GetDefaultInterfaces()
|
64 | 65 | if err != nil {
|
65 |
| - plog.Warningf("cannot detect default network interface (%v)", err) |
| 66 | + fmt.Printf("cannot detect default network interface (%v)\n", err) |
66 | 67 | }
|
67 | 68 | var nt string
|
68 | 69 | for k := range nm {
|
@@ -102,26 +103,39 @@ var Command = &cobra.Command{
|
102 | 103 |
|
103 | 104 | func commandFunc(cmd *cobra.Command, args []string) error {
|
104 | 105 | no, nerr := ntp.DefaultSync()
|
105 |
| - plog.Infof("npt update output: %q", no) |
106 |
| - plog.Infof("npt update error: %v", nerr) |
| 106 | + fmt.Printf("npt update output: %q\n", no) |
| 107 | + if nerr != nil { |
| 108 | + fmt.Printf("npt update error: %v\n", nerr) |
| 109 | + } |
107 | 110 |
|
108 |
| - f, err := openToAppend(globalFlags.agentLog) |
109 |
| - if err != nil { |
110 |
| - return err |
| 111 | + lcfg := zap.Config{ |
| 112 | + Level: zap.NewAtomicLevelAt(zap.InfoLevel), |
| 113 | + Development: false, |
| 114 | + Sampling: &zap.SamplingConfig{ |
| 115 | + Initial: 100, |
| 116 | + Thereafter: 100, |
| 117 | + }, |
| 118 | + Encoding: "json", |
| 119 | + EncoderConfig: zap.NewProductionEncoderConfig(), |
| 120 | + |
| 121 | + OutputPaths: []string{globalFlags.agentLog}, |
| 122 | + ErrorOutputPaths: []string{globalFlags.agentLog}, |
| 123 | + } |
| 124 | + lg, lerr := lcfg.Build() |
| 125 | + if lerr != nil { |
| 126 | + return lerr |
111 | 127 | }
|
112 |
| - defer f.Close() |
113 |
| - capnslog.SetFormatter(capnslog.NewPrettyFormatter(f, false)) |
114 | 128 |
|
115 | 129 | var (
|
116 | 130 | grpcServer = grpc.NewServer()
|
117 |
| - sender = NewServer() |
| 131 | + sender = NewServer(lg) |
118 | 132 | )
|
119 | 133 | ln, err := net.Listen("tcp", globalFlags.grpcPort)
|
120 | 134 | if err != nil {
|
121 | 135 | return err
|
122 | 136 | }
|
123 | 137 | dbtesterpb.RegisterTransporterServer(grpcServer, sender)
|
124 | 138 |
|
125 |
| - plog.Infof("agent started with gRPC %s (log path %q)", globalFlags.grpcPort, globalFlags.agentLog) |
| 139 | + lg.Info("agent started", zap.String("grpc-server-port", globalFlags.grpcPort), zap.String("agent-log", globalFlags.agentLog)) |
126 | 140 | return grpcServer.Serve(ln)
|
127 | 141 | }
|
0 commit comments