Move process to go-process

This commit is contained in:
Jae Kwon 2015-11-08 17:43:48 -08:00
parent e12f9d10e7
commit 37b6255e42
4 changed files with 8 additions and 84 deletions

View File

@ -1,4 +1,3 @@
package main
import (
@ -11,10 +10,10 @@ import (
"sync"
"time"
. "github.com/tendermint/go-common"
pcm "github.com/tendermint/go-process"
"github.com/tendermint/go-wire"
. "github.com/tendermint/tendermint/cmd/barak/types"
. "github.com/tendermint/go-common"
pcm "github.com/tendermint/tendermint/process"
"github.com/tendermint/tendermint/rpc/server"
)
@ -137,7 +136,7 @@ func (brk *Barak) StopProcess(label string, kill bool) error {
return fmt.Errorf("Process does not exist: %v", label)
}
err := pcm.Stop(proc, kill)
err := proc.StopProcess(kill)
return err
}

View File

@ -16,12 +16,12 @@ import (
"reflect"
"time"
. "github.com/tendermint/tendermint/cmd/barak/types"
. "github.com/tendermint/go-common"
cfg "github.com/tendermint/go-config"
pcm "github.com/tendermint/tendermint/process"
"github.com/tendermint/tendermint/rpc/server"
pcm "github.com/tendermint/go-process"
"github.com/tendermint/go-wire"
. "github.com/tendermint/tendermint/cmd/barak/types"
"github.com/tendermint/tendermint/rpc/server"
)
const BarakVersion = "0.0.1"
@ -162,7 +162,7 @@ func StartProcess(wait bool, label string, execPath string, args []string, input
if err != nil {
return nil, err
}
proc, err := pcm.Create(label, execPath, args, inFile, outFile)
proc, err := pcm.StartProcess(label, execPath, args, inFile, outFile)
if err != nil {
return nil, err
}

View File

@ -1,7 +1,7 @@
package types
import (
pcm "github.com/tendermint/tendermint/process"
pcm "github.com/tendermint/go-process"
)
type ResponseStatus struct {

View File

@ -1,75 +0,0 @@
package process
import (
"fmt"
"io"
"os"
"os/exec"
"time"
)
type Process struct {
Label string
ExecPath string
Args []string
Pid int
StartTime time.Time
EndTime time.Time
Cmd *exec.Cmd `json:"-"`
ExitState *os.ProcessState `json:"-"`
InputFile io.Reader `json:"-"`
OutputFile io.WriteCloser `json:"-"`
WaitCh chan struct{} `json:"-"`
}
// execPath: command name
// args: args to command. (should not include name)
func Create(label string, execPath string, args []string, inFile io.Reader, outFile io.WriteCloser) (*Process, error) {
cmd := exec.Command(execPath, args...)
cmd.Stdout = outFile
cmd.Stderr = outFile
cmd.Stdin = inFile
if err := cmd.Start(); err != nil {
return nil, err
}
proc := &Process{
Label: label,
ExecPath: execPath,
Args: args,
Pid: cmd.Process.Pid,
StartTime: time.Now(),
Cmd: cmd,
ExitState: nil,
InputFile: inFile,
OutputFile: outFile,
WaitCh: make(chan struct{}),
}
go func() {
err := proc.Cmd.Wait()
if err != nil {
fmt.Printf("Process exit: %v\n", err)
if exitError, ok := err.(*exec.ExitError); ok {
proc.ExitState = exitError.ProcessState
}
}
proc.ExitState = proc.Cmd.ProcessState
proc.EndTime = time.Now() // TODO make this goroutine-safe
err = proc.OutputFile.Close()
if err != nil {
fmt.Printf("Error closing output file for %v: %v\n", proc.Label, err)
}
close(proc.WaitCh)
}()
return proc, nil
}
func Stop(proc *Process, kill bool) error {
defer proc.OutputFile.Close()
if kill {
fmt.Printf("Killing process %v\n", proc.Cmd.Process)
return proc.Cmd.Process.Kill()
} else {
fmt.Printf("Stopping process %v\n", proc.Cmd.Process)
return proc.Cmd.Process.Signal(os.Interrupt)
}
}