pulumi/pkg/resource/provider/host.go

109 lines
3.5 KiB
Go
Raw Permalink Normal View History

2018-05-22 19:43:36 +00:00
// 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 provider
import (
"io"
"os"
"strings"
"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/contract"
"github.com/pulumi/pulumi/sdk/v3/go/common/util/rpcutil"
pulumirpc "github.com/pulumi/pulumi/sdk/v3/proto/go"
"golang.org/x/net/context"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
"google.golang.org/grpc/grpclog"
)
// HostClient is a client interface into the host's engine RPC interface.
type HostClient struct {
conn *grpc.ClientConn
client pulumirpc.EngineClient
}
// NewHostClient dials the target address, connects over gRPC, and returns a client interface.
func NewHostClient(addr string) (*HostClient, error) {
// Provider client is sensitive to GRPC info logging to stdout, so ensure they are dropped.
// See https://github.com/pulumi/pulumi/issues/7156
grpclog.SetLoggerV2(grpclog.NewLoggerV2(io.Discard, io.Discard, os.Stderr))
2020-04-20 22:25:51 +00:00
conn, err := grpc.Dial(
addr,
grpc.WithTransportCredentials(insecure.NewCredentials()),
2020-04-20 22:25:51 +00:00
grpc.WithUnaryInterceptor(rpcutil.OpenTracingClientInterceptor()),
grpc.WithStreamInterceptor(rpcutil.OpenTracingStreamClientInterceptor()),
2020-04-20 22:25:51 +00:00
rpcutil.GrpcChannelOptions(),
)
if err != nil {
return nil, err
}
return &HostClient{
conn: conn,
client: pulumirpc.NewEngineClient(conn),
}, nil
}
// Close closes and renders the connection and client unusable.
func (host *HostClient) Close() error {
return host.conn.Close()
}
// EngineConn provides the engine gRPC client connection.
func (host *HostClient) EngineConn() *grpc.ClientConn {
return host.conn
}
func (host *HostClient) log(
context context.Context, sev diag.Severity, urn resource.URN, msg string, ephemeral bool,
) error {
var rpcsev pulumirpc.LogSeverity
switch sev {
case diag.Debug:
rpcsev = pulumirpc.LogSeverity_DEBUG
turn on the golangci-lint exhaustive linter (#15028) Turn on the golangci-lint exhaustive linter. This is the first step towards catching more missing cases during development rather than in tests, or in production. This might be best reviewed commit-by-commit, as the first commit turns on the linter with the `default-signifies-exhaustive: true` option set, which requires a lot less changes in the current codebase. I think it's probably worth doing the second commit as well, as that will get us the real benefits, even though we end up with a little bit more churn. However it means all the `switch` statements are covered, which isn't the case after the first commit, since we do have a lot of `default` statements that just call `assert.Fail`. Fixes #14601 ## Checklist - [x] I have run `make tidy` to update any new dependencies - [x] I have run `make lint` to verify my code passes the lint check - [x] I have formatted my code using `gofumpt` <!--- Please provide details if the checkbox below is to be left unchecked. --> - [ ] I have added tests that prove my fix is effective or that my feature works <!--- User-facing changes require a CHANGELOG entry. --> - [ ] I have run `make changelog` and committed the `changelog/pending/<file>` documenting my change <!-- If the change(s) in this PR is a modification of an existing call to the Pulumi Cloud, then the service should honor older versions of the CLI where this change would not exist. You must then bump the API version in /pkg/backend/httpstate/client/api.go, as well as add it to the service. --> - [ ] Yes, there are changes in this PR that warrants bumping the Pulumi Cloud API version <!-- @Pulumi employees: If yes, you must submit corresponding changes in the service repo. -->
2024-01-17 16:50:41 +00:00
case diag.Info, diag.Infoerr:
rpcsev = pulumirpc.LogSeverity_INFO
case diag.Warning:
rpcsev = pulumirpc.LogSeverity_WARNING
case diag.Error:
rpcsev = pulumirpc.LogSeverity_ERROR
default:
contract.Failf("Unrecognized log severity type: %v", sev)
}
_, err := host.client.Log(context, &pulumirpc.LogRequest{
Severity: rpcsev,
Message: strings.ToValidUTF8(msg, "<22>"),
Urn: string(urn),
Ephemeral: ephemeral,
})
return err
}
// Log logs a global message, including errors and warnings.
func (host *HostClient) Log(
context context.Context, sev diag.Severity, urn resource.URN, msg string,
) error {
return host.log(context, sev, urn, msg, false)
}
// LogStatus logs a global status message, including errors and warnings. Status messages will
// appear in the `Info` column of the progress display, but not in the final output.
func (host *HostClient) LogStatus(
context context.Context, sev diag.Severity, urn resource.URN, msg string,
) error {
return host.log(context, sev, urn, msg, true)
}