mirror of
https://github.com/kemko/nomad.git
synced 2026-01-02 00:15:43 +03:00
a4cc76bd3e
* client: refactor cgroups management in client * client: fingerprint numa topology * client: plumb numa and cgroups changes to drivers * client: cleanup task resource accounting * client: numa client and config plumbing * lib: add a stack implementation * tools: remove ec2info tool * plugins: fixup testing for cgroups / numa changes * build: update makefile and package tests and cl
38 lines
1000 B
Go
38 lines
1000 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package executor
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/go-plugin"
|
|
"github.com/hashicorp/nomad/drivers/shared/executor/proto"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type ExecutorPlugin struct {
|
|
// TODO: support backwards compatibility with pre 0.9 NetRPC plugin
|
|
plugin.NetRPCUnsupportedPlugin
|
|
logger hclog.Logger
|
|
fsIsolation bool
|
|
}
|
|
|
|
func (p *ExecutorPlugin) GRPCServer(broker *plugin.GRPCBroker, s *grpc.Server) error {
|
|
if p.fsIsolation {
|
|
proto.RegisterExecutorServer(s, &grpcExecutorServer{impl: NewExecutorWithIsolation(p.logger)})
|
|
} else {
|
|
proto.RegisterExecutorServer(s, &grpcExecutorServer{impl: NewExecutor(p.logger)})
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *ExecutorPlugin) GRPCClient(ctx context.Context, broker *plugin.GRPCBroker, c *grpc.ClientConn) (interface{}, error) {
|
|
return &grpcExecutorClient{
|
|
client: proto.NewExecutorClient(c),
|
|
doneCtx: ctx,
|
|
logger: p.logger,
|
|
}, nil
|
|
}
|