104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package docklog
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"os/exec"
|
|
|
|
hclog "github.com/hashicorp/go-hclog"
|
|
plugin "github.com/hashicorp/go-plugin"
|
|
"github.com/hashicorp/nomad/drivers/docker/docklog/proto"
|
|
"github.com/hashicorp/nomad/plugins/base"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
const PluginName = "docker_logger"
|
|
|
|
// LaunchDockerLogger launches an instance of DockerLogger
|
|
func LaunchDockerLogger(logger hclog.Logger) (DockerLogger, *plugin.Client, error) {
|
|
logger = logger.Named(PluginName)
|
|
bin, err := os.Executable()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
client := plugin.NewClient(&plugin.ClientConfig{
|
|
HandshakeConfig: base.Handshake,
|
|
Plugins: map[string]plugin.Plugin{
|
|
PluginName: &Plugin{impl: NewDockerLogger(logger)},
|
|
},
|
|
Cmd: exec.Command(bin, PluginName),
|
|
AllowedProtocols: []plugin.Protocol{
|
|
plugin.ProtocolGRPC,
|
|
},
|
|
Logger: logger,
|
|
})
|
|
|
|
rpcClient, err := client.Client()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
raw, err := rpcClient.Dispense(PluginName)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
l := raw.(DockerLogger)
|
|
return l, client, nil
|
|
|
|
}
|
|
|
|
func ReattachDockerLogger(reattachCfg *plugin.ReattachConfig) (DockerLogger, *plugin.Client, error) {
|
|
client := plugin.NewClient(&plugin.ClientConfig{
|
|
HandshakeConfig: base.Handshake,
|
|
Plugins: map[string]plugin.Plugin{
|
|
PluginName: &Plugin{impl: NewDockerLogger(hclog.L().Named(PluginName))},
|
|
},
|
|
Reattach: reattachCfg,
|
|
AllowedProtocols: []plugin.Protocol{
|
|
plugin.ProtocolGRPC,
|
|
},
|
|
})
|
|
|
|
rpcClient, err := client.Client()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
raw, err := rpcClient.Dispense(PluginName)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
l := raw.(DockerLogger)
|
|
return l, client, nil
|
|
}
|
|
|
|
// Plugin is the go-plugin implementation
|
|
type Plugin struct {
|
|
plugin.NetRPCUnsupportedPlugin
|
|
impl DockerLogger
|
|
}
|
|
|
|
func NewPlugin(impl DockerLogger) *Plugin {
|
|
return &Plugin{impl: impl}
|
|
}
|
|
|
|
// GRPCServer registered the server side implementation with the grpc server
|
|
func (p *Plugin) GRPCServer(broker *plugin.GRPCBroker, s *grpc.Server) error {
|
|
proto.RegisterDockerLoggerServer(s, &dockerLoggerServer{
|
|
impl: p.impl,
|
|
broker: broker,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
// GRPCClient returns a client side implementation of the plugin
|
|
func (p *Plugin) GRPCClient(ctx context.Context, broker *plugin.GRPCBroker, c *grpc.ClientConn) (interface{}, error) {
|
|
return &dockerLoggerClient{client: proto.NewDockerLoggerClient(c)}, nil
|
|
}
|