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

etcd-agent: set up directory for etcd logs

Browse Source
This commit is contained in:
Gyu-Ho Lee 2016-06-20 11:32:14 -07:00
parent fb1f1ce1fd
commit 50f2f984e4
2 changed files with 31 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
tools/functional-tester/etcd-agent/agent.go
View File

@ -18,10 +18,11 @@ import (
"fmt" "fmt"
"os" "os"
"os/exec" "os/exec"
"path" "path/filepath"
"syscall" "syscall"
"time" "time"
"github.com/coreos/etcd/pkg/fileutil"
"github.com/coreos/etcd/pkg/netutil" "github.com/coreos/etcd/pkg/netutil"
"github.com/coreos/etcd/tools/functional-tester/etcd-agent/client" "github.com/coreos/etcd/tools/functional-tester/etcd-agent/client"
) )
@ -36,12 +37,12 @@ const (
type Agent struct { type Agent struct {
state string // the state of etcd process state string // the state of etcd process
cmd *exec.Cmd cmd *exec.Cmd
logfile *os.File logfile *os.File
etcdLogPath string logDir string
} }
func newAgent(etcd, etcdLogPath string) (*Agent, error) { func newAgent(etcd, logDir string) (*Agent, error) {
// check if the file exists // check if the file exists
_, err := os.Stat(etcd) _, err := os.Stat(etcd)
if err != nil { if err != nil {
@ -50,12 +51,18 @@ func newAgent(etcd, etcdLogPath string) (*Agent, error) {
c := exec.Command(etcd) c := exec.Command(etcd)
f, err := os.Create(etcdLogPath) err = fileutil.TouchDirAll(logDir)
if err != nil { if err != nil {
return nil, err return nil, err
} }
return &Agent{state: stateUninitialized, cmd: c, logfile: f, etcdLogPath: etcdLogPath}, nil var f *os.File
f, err = os.Create(filepath.Join(logDir, "etcd.log"))
if err != nil {
return nil, err
}
return &Agent{state: stateUninitialized, cmd: c, logfile: f, logDir: logDir}, nil
} }
// start starts a new etcd process with the given args. // start starts a new etcd process with the given args.
@ -132,14 +139,19 @@ func (a *Agent) cleanup() error {
a.state = stateUninitialized a.state = stateUninitialized
a.logfile.Close() a.logfile.Close()
if err := archiveLogAndDataDir(a.etcdLogPath, a.dataDir()); err != nil { if err := archiveLogAndDataDir(a.logDir, a.dataDir()); err != nil {
return err return err
} }
f, err := os.Create(a.etcdLogPath)
a.logfile = f if err := fileutil.TouchDirAll(a.logDir); err != nil {
return err
}
f, err := os.Create(filepath.Join(a.logDir, "etcd.log"))
if err != nil { if err != nil {
return err return err
} }
a.logfile = f
// https://www.kernel.org/doc/Documentation/sysctl/vm.txt // https://www.kernel.org/doc/Documentation/sysctl/vm.txt
// https://github.com/torvalds/linux/blob/master/fs/drop_caches.c // https://github.com/torvalds/linux/blob/master/fs/drop_caches.c
@ -185,7 +197,7 @@ func (a *Agent) status() client.Status {
} }
func (a *Agent) dataDir() string { func (a *Agent) dataDir() string {
datadir := path.Join(a.cmd.Path, "*.etcd") datadir := filepath.Join(a.cmd.Path, "*.etcd")
args := a.cmd.Args args := a.cmd.Args
// only parse the simple case like "--data-dir /var/lib/etcd" // only parse the simple case like "--data-dir /var/lib/etcd"
for i, arg := range args { for i, arg := range args {
@ -209,20 +221,20 @@ func existDir(fpath string) bool {
return false return false
} }
func archiveLogAndDataDir(log string, datadir string) error { func archiveLogAndDataDir(logDir string, datadir string) error {
dir := path.Join("failure_archive", fmt.Sprint(time.Now().Format(time.RFC3339))) dir := filepath.Join("failure_archive", fmt.Sprint(time.Now().Format(time.RFC3339)))
if existDir(dir) { if existDir(dir) {
dir = path.Join("failure_archive", fmt.Sprint(time.Now().Add(time.Second).Format(time.RFC3339))) dir = filepath.Join("failure_archive", fmt.Sprint(time.Now().Add(time.Second).Format(time.RFC3339)))
} }
if err := os.MkdirAll(dir, 0755); err != nil { if err := fileutil.TouchDirAll(dir); err != nil {
return err return err
} }
if err := os.Rename(log, path.Join(dir, path.Base(log))); err != nil { if err := os.Rename(logDir, filepath.Join(dir, filepath.Base(logDir))); err != nil {
if !os.IsNotExist(err) { if !os.IsNotExist(err) {
return err return err
} }
} }
if err := os.Rename(datadir, path.Join(dir, path.Base(datadir))); err != nil { if err := os.Rename(datadir, filepath.Join(dir, filepath.Base(datadir))); err != nil {
if !os.IsNotExist(err) { if !os.IsNotExist(err) {
return err return err
} }

4
tools/functional-tester/etcd-agent/main.go
View File

@ -26,11 +26,11 @@ var plog = capnslog.NewPackageLogger("github.com/coreos/etcd", "etcd-agent")
func main() { func main() {
etcdPath := flag.String("etcd-path", filepath.Join(os.Getenv("GOPATH"), "bin/etcd"), "the path to etcd binary") etcdPath := flag.String("etcd-path", filepath.Join(os.Getenv("GOPATH"), "bin/etcd"), "the path to etcd binary")
etcdLogPath := flag.String("etcd-log-path", "etcd.log", "the path to etcd log") etcdLogDir := flag.String("etcd-log-dir", "etcd-log", "directory to store etcd logs")
port := flag.String("port", ":9027", "port to serve agent server") port := flag.String("port", ":9027", "port to serve agent server")
flag.Parse() flag.Parse()
a, err := newAgent(*etcdPath, *etcdLogPath) a, err := newAgent(*etcdPath, *etcdLogDir)
if err != nil { if err != nil {
plog.Fatal(err) plog.Fatal(err)
} }