122 lines
4 KiB
Go
122 lines
4 KiB
Go
// Copyright 2016-2018, Pulumi Corporation.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package plugin
|
|
|
|
import (
|
|
"fmt"
|
|
"sync/atomic"
|
|
|
|
pbempty "github.com/golang/protobuf/ptypes/empty"
|
|
"github.com/pkg/errors"
|
|
"golang.org/x/net/context"
|
|
"google.golang.org/grpc"
|
|
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/diag"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/resource"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/util/rpcutil"
|
|
lumirpc "github.com/pulumi/pulumi/sdk/v3/proto/go"
|
|
)
|
|
|
|
// hostServer is the server side of the host RPC machinery.
|
|
type hostServer struct {
|
|
host Host // the host for this RPC server.
|
|
ctx *Context // the associated plugin context.
|
|
addr string // the address the host is listening on.
|
|
cancel chan bool // a channel that can cancel the server.
|
|
done chan error // a channel that resolves when the server completes.
|
|
|
|
// hostServer contains little bits of state that can't be saved in the language host.
|
|
rootUrn atomic.Value // a root resource URN that has been saved via SetRootResource
|
|
}
|
|
|
|
// newHostServer creates a new host server wired up to the given host and context.
|
|
func newHostServer(host Host, ctx *Context) (*hostServer, error) {
|
|
// New up an engine RPC server.
|
|
engine := &hostServer{
|
|
host: host,
|
|
ctx: ctx,
|
|
cancel: make(chan bool),
|
|
}
|
|
|
|
// Fire up a gRPC server and start listening for incomings.
|
|
port, done, err := rpcutil.Serve(0, engine.cancel, []func(*grpc.Server) error{
|
|
func(srv *grpc.Server) error {
|
|
lumirpc.RegisterEngineServer(srv, engine)
|
|
return nil
|
|
},
|
|
}, ctx.tracingSpan)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
engine.addr = fmt.Sprintf("127.0.0.1:%d", port)
|
|
engine.done = done
|
|
engine.rootUrn.Store("")
|
|
|
|
return engine, nil
|
|
}
|
|
|
|
// Address returns the address at which the engine's RPC server may be reached.
|
|
func (eng *hostServer) Address() string {
|
|
return eng.addr
|
|
}
|
|
|
|
// Cancel signals that the engine should be terminated, awaits its termination, and returns any errors that result.
|
|
func (eng *hostServer) Cancel() error {
|
|
eng.cancel <- true
|
|
return <-eng.done
|
|
}
|
|
|
|
// Log logs a global message in the engine, including errors and warnings.
|
|
func (eng *hostServer) Log(ctx context.Context, req *lumirpc.LogRequest) (*pbempty.Empty, error) {
|
|
var sev diag.Severity
|
|
switch req.Severity {
|
|
case lumirpc.LogSeverity_DEBUG:
|
|
sev = diag.Debug
|
|
case lumirpc.LogSeverity_INFO:
|
|
sev = diag.Info
|
|
case lumirpc.LogSeverity_WARNING:
|
|
sev = diag.Warning
|
|
case lumirpc.LogSeverity_ERROR:
|
|
sev = diag.Error
|
|
default:
|
|
return nil, errors.Errorf("Unrecognized logging severity: %v", req.Severity)
|
|
}
|
|
|
|
if req.Ephemeral {
|
|
eng.host.LogStatus(sev, resource.URN(req.Urn), req.Message, req.StreamId)
|
|
} else {
|
|
eng.host.Log(sev, resource.URN(req.Urn), req.Message, req.StreamId)
|
|
}
|
|
return &pbempty.Empty{}, nil
|
|
}
|
|
|
|
// GetRootResource returns the current root resource's URN, which will serve as the parent of resources that are
|
|
// otherwise left unparented.
|
|
func (eng *hostServer) GetRootResource(ctx context.Context,
|
|
req *lumirpc.GetRootResourceRequest) (*lumirpc.GetRootResourceResponse, error) {
|
|
var response lumirpc.GetRootResourceResponse
|
|
response.Urn = eng.rootUrn.Load().(string)
|
|
return &response, nil
|
|
}
|
|
|
|
// SetRootResources sets the current root resource's URN. Generally only called on startup when the Stack resource is
|
|
// registered.
|
|
func (eng *hostServer) SetRootResource(ctx context.Context,
|
|
req *lumirpc.SetRootResourceRequest) (*lumirpc.SetRootResourceResponse, error) {
|
|
var response lumirpc.SetRootResourceResponse
|
|
eng.rootUrn.Store(req.GetUrn())
|
|
return &response, nil
|
|
}
|