-
Notifications
You must be signed in to change notification settings - Fork 1.9k
/
client.go
35 lines (29 loc) · 891 Bytes
/
client.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
package logmon
import (
"context"
"github.com/hashicorp/nomad/client/logmon/proto"
"github.com/hashicorp/nomad/helper/pluginutils/grpcutils"
)
type logmonClient struct {
client proto.LogMonClient
// doneCtx is closed when the plugin exits
doneCtx context.Context
}
func (c *logmonClient) Start(cfg *LogConfig) error {
req := &proto.StartRequest{
LogDir: cfg.LogDir,
StdoutFileName: cfg.StdoutLogFile,
StderrFileName: cfg.StderrLogFile,
MaxFiles: uint32(cfg.MaxFiles),
MaxFileSizeMb: uint32(cfg.MaxFileSizeMB),
StdoutFifo: cfg.StdoutFifo,
StderrFifo: cfg.StderrFifo,
}
_, err := c.client.Start(context.Background(), req)
return grpcutils.HandleGrpcErr(err, c.doneCtx)
}
func (c *logmonClient) Stop() error {
req := &proto.StopRequest{}
_, err := c.client.Stop(context.Background(), req)
return grpcutils.HandleGrpcErr(err, c.doneCtx)
}