forked from dolthub/vitess
/
vtworkerclient_wrapper.go
42 lines (32 loc) · 1.41 KB
/
vtworkerclient_wrapper.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
// Copyright 2015, Google Inc. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package automation
import (
"bytes"
"fmt"
"golang.org/x/net/context"
log "github.com/golang/glog"
"github.com/youtube/vitess/go/vt/worker/vtworkerclient"
)
// ExecuteVtworker executes the vtworker command in "args" via an RPC to
// "server".
// The output of the RPC, a stream of LoggerEvent messages, is concatenated into
// one output string.
// If a retryable error is encountered (e.g. the vtworker process is already
// executing another command), this function will keep retrying infinitely until
// "ctx" is cancelled.
func ExecuteVtworker(ctx context.Context, server string, args []string) (string, error) {
var output bytes.Buffer
loggerToBufferFunc := createLoggerEventToBufferFunction(&output)
outputLogger := newOutputLogger(loggerToBufferFunc)
startMsg := fmt.Sprintf("Executing remote vtworker command: %v server: %v", args, server)
outputLogger.Infof(startMsg)
log.Info(startMsg)
err := vtworkerclient.RunCommandAndWait(ctx, server, args, loggerToBufferFunc)
endMsg := fmt.Sprintf("Executed remote vtworker command: %v server: %v err: %v", args, server, err)
outputLogger.Infof(endMsg)
// Log full output to log file (but not to the buffer).
log.Infof("%v output (starting on next line):\n%v", endMsg, output.String())
return output.String(), err
}