mirror of
https://github.com/kemko/nomad.git
synced 2026-01-02 00:15:43 +03:00
e190eae395
Before this commit, it was only used for fingerprinting, but not for CPU stats on nodes or tasks. This meant that if the auto-detection failed, setting the cpu_total_compute didn't resolved the issue. This issue was most noticeable on ARM64, as there auto-detection always failed.
39 lines
1.0 KiB
Go
39 lines
1.0 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package executor
|
|
|
|
import (
|
|
"context"
|
|
|
|
hclog "github.com/hashicorp/go-hclog"
|
|
plugin "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
|
|
cpuTotalTicks uint64
|
|
}
|
|
|
|
func (p *ExecutorPlugin) GRPCServer(broker *plugin.GRPCBroker, s *grpc.Server) error {
|
|
if p.fsIsolation {
|
|
proto.RegisterExecutorServer(s, &grpcExecutorServer{impl: NewExecutorWithIsolation(p.logger, p.cpuTotalTicks)})
|
|
} else {
|
|
proto.RegisterExecutorServer(s, &grpcExecutorServer{impl: NewExecutor(p.logger, p.cpuTotalTicks)})
|
|
}
|
|
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
|
|
}
|