98 lines
1.5 KiB
Go
98 lines
1.5 KiB
Go
|
// Package os runs processes locally
|
||
|
package os
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"os"
|
||
|
"os/exec"
|
||
|
"strconv"
|
||
|
"syscall"
|
||
|
|
||
|
"github.com/micro/go-micro/runtime/process"
|
||
|
)
|
||
|
|
||
|
type Process struct {
|
||
|
}
|
||
|
|
||
|
func (p *Process) Exec(exe *process.Executable) error {
|
||
|
cmd := exec.Command(exe.Binary.Path)
|
||
|
return cmd.Run()
|
||
|
}
|
||
|
|
||
|
func (p *Process) Fork(exe *process.Executable) (*process.PID, error) {
|
||
|
// create command
|
||
|
cmd := exec.Command(exe.Binary.Path, exe.Args...)
|
||
|
// set env vars
|
||
|
cmd.Env = append(cmd.Env, exe.Env...)
|
||
|
|
||
|
in, err := cmd.StdinPipe()
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
out, err := cmd.StdoutPipe()
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
er, err := cmd.StderrPipe()
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
// start the process
|
||
|
if err := cmd.Start(); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return &process.PID{
|
||
|
ID: fmt.Sprintf("%d", cmd.Process.Pid),
|
||
|
Input: in,
|
||
|
Output: out,
|
||
|
Error: er,
|
||
|
}, nil
|
||
|
}
|
||
|
|
||
|
func (p *Process) Kill(pid *process.PID) error {
|
||
|
id, err := strconv.Atoi(pid.ID)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
pr, err := os.FindProcess(id)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
// now kill it
|
||
|
err = pr.Kill()
|
||
|
|
||
|
// return the kill error
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
func (p *Process) Wait(pid *process.PID) error {
|
||
|
id, err := strconv.Atoi(pid.ID)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
pr, err := os.FindProcess(id)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
ps, err := pr.Wait()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if ps.Success() {
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
return fmt.Errorf(ps.String())
|
||
|
}
|
||
|
|
||
|
func NewProcess(opts ...process.Option) process.Process {
|
||
|
return &Process{}
|
||
|
}
|