scriptoriumadmin/featurebase
1
0
Fork 0
mirror of https://github.com/featurebasedb/featurebase.git synced 2026-08-28 10:54:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

factor copying binary to remote hosts into separate function

Browse source
This commit is contained in:
jaffee 2016-12-22 15:03:18 -06:00
parent 01a5d8592f
commit aa2db76247
1 changed files with 43 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

73
cmd/pilosactl/main.go
View file

@ -1432,43 +1432,56 @@ func (cmd *BspawnCommand) Run(ctx context.Context) error {
}
}
func copyBinary(agentConnections []*pilosactl.SSH, pkg, goos, goarch string) error {
binName := path.Base(pkg)
binLoc := path.Join(os.TempDir(), binName)
com := exec.Command("go", "build", "-o", binLoc, pkg)
com.Env = append([]string{"GOOS=" + goos, "GOARCH=" + goarch}, os.Environ()...)
err := com.Run()
if err != nil {
return err
}
agentWriters := make([]io.Writer, len(agentConnections))
for i, conn := range agentConnections {
agentWriters[i], err = conn.OpenFile(binName, "+x")
if err != nil {
return err
}
}
f, err := os.Open(binLoc)
if err != nil {
return err
}
_, err = io.Copy(io.MultiWriter(agentWriters...), f)
if err != nil {
return err
}
for _, w := range agentWriters {
if wc, ok := w.(io.WriteCloser); ok {
err = wc.Close()
if err != nil {
return err
}
}
}
return nil
}
func (cmd *BspawnCommand) spawnRemote(ctx context.Context, runUUID uuid.UUID) error {
agentIndex := 0
agentConnections, err := pilosactl.SSHClients(cmd.AgentHosts, cmd.SSHUser, "", cmd.Stderr)
if err != nil {
return err
}
if cmd.CopyBinary {
binLoc := path.Join(os.TempDir(), "pilosactl")
cmd := exec.Command("go", "build", "-o", binLoc, "github.com/pilosa/pilosa/cmd/pilosactl")
cmd.Env = append([]string{"GOOS=linux", "GOARCH=amd64"}, os.Environ()...)
err := cmd.Run()
if err != nil {
return err
}
agentWriters := make([]io.Writer, len(agentConnections))
for i, conn := range agentConnections {
agentWriters[i], err = conn.OpenFile("pilosactl", "+x")
if err != nil {
return err
}
}
f, err := os.Open(binLoc)
if err != nil {
return err
}
_, err = io.Copy(io.MultiWriter(agentWriters...), f)
if err != nil {
return err
}
for _, w := range agentWriters {
if wc, ok := w.(io.WriteCloser); ok {
err = wc.Close()
if err != nil {
return err
}
}
if cmd.CopyBinary {
err = copyBinary(agentConnections, "github.com/pilosa/pilosa/cmd/pilosactl", "linux", "amd64")
if err != nil {
return err
}
}