common/prpc, tools/cmd/cproto: prpc client

common/prpc/client.go

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package prpc
 
+import (
+        "bytes"
+        "fmt"
+        "io"
+        "io/ioutil"
+        "net/http"
+        "net/url"
+        "strconv"
+        "strings"
+        "time"
+
+        "github.com/golang/protobuf/proto"
+        "golang.org/x/net/context"
+        "golang.org/x/net/context/ctxhttp"
+        "google.golang.org/grpc"
+        "google.golang.org/grpc/codes"
+        "google.golang.org/grpc/metadata"
+
+        "github.com/luci/luci-go/common/errors"
+        "github.com/luci/luci-go/common/grpcutil"
+        "github.com/luci/luci-go/common/logging"
+        "github.com/luci/luci-go/common/retry"
+)
+
 const (
         // HeaderGRPCCode is a name of the HTTP header that specifies the
         // gRPC code in the response.
         // A pRPC server must always specify it.
         HeaderGRPCCode = "X-Prpc-Grpc-Code"
 )
+
+// DefaultUserAgent is default User-Agent HTTP header for pRPC requests.
+var DefaultUserAgent = "pRPC Client 1.0"
+
+// Client can make pRPC calls.
+type Client struct {
+        C *http.Client // if nil, uses http.DefaultClient
+
+        // ErrBodySize is the number of bytes to read from a HTTP response
+        // with error status and include in the error.
+        // If non-positive, defaults to 256.
+        ErrBodySize int
+
+        Host    string   // host and optionally a port number of the target server.
+        Options *Options // if nil, DefaultOptions() are used.
+}
+
+// renderOptions copies client options and applies opts.
+func (c *Client) renderOptions(opts []grpc.CallOption) *Options {
+        var options *Options
+        if c.Options != nil {
+                cpy := *c.Options
+                options = &cpy
+        } else {
+                options = DefaultOptions()
+        }
+        options.apply(opts)
+        return options
+}
+
+func (c *Client) getHTTPClient() *http.Client {
+        if c.C == nil {
+                return http.DefaultClient
+        }
+        return c.C
+}
+
+// Call makes an RPC.
+// Retries on transient errors according to retry options.
+// Logs HTTP errors.
+//
+// opts must be created by this package.
+// Calling from multiple goroutines concurrently is safe, unless Client is mutated.
+// Called from generated code.
+func (c *Client) Call(ctx context.Context, serviceName, methodName string, in, out proto.Message, opts ...grpc.CallOption) error {
+        options := c.renderOptions(opts)
+
+        reqBody, err := proto.Marshal(in)
+        if err != nil {
+                return err
+        }
+
+        req := prepareRequest(c.Host, serviceName, methodName, len(reqBody), options)
+        ctx = logging.SetFields(ctx, logging.Fields{
+                "host":    c.Host,
+                "service": serviceName,
+                "method":  methodName,
+        })
+
+        // Send the request in a retry loop.
+        var buf bytes.Buffer
+        err = retry.Retry(
+                ctx,
+                retry.TransientOnly(options.Retry),
+                func() error {
+                        logging.Debugf(ctx, "RPC %s/%s.%s", c.Host, serviceName, methodName)

[Vadim Sh., 2016/01/27 01:13:54]

host, serviceName and methodName is already in the fields...

+
+                        // Send the request.
+                        req.Body = ioutil.NopCloser(bytes.NewReader(reqBody))
+                        res, err := ctxhttp.Do(ctx, c.getHTTPClient(), req)
+                        if err != nil {

[Vadim Sh., 2016/01/27 01:13:54]

detect context cancelation here? We don't want to retry it.

+                                return errors.WrapTransient(fmt.Errorf("failed to send request: %s", err))
+                        }
+                        defer res.Body.Close()
+
+                        if options.resHeaderMetadata != nil {
+                                *options.resHeaderMetadata = metadata.MD(res.Header).Copy()
+                        }
+
+                        // Read the response body.
+                        buf.Reset()
+                        var body io.Reader = res.Body
+                        if res.ContentLength > 0 {
+                                buf.Grow(int(res.ContentLength))
+                                body = io.LimitReader(body, res.ContentLength)
+                        }
+                        if _, err = buf.ReadFrom(body); err != nil {
+                                return fmt.Errorf("failed to read response body: %s", err)

[Vadim Sh., 2016/01/27 01:13:54]

this is most probably transient error too (e.g. connectivity error when reading
from the socket)

+                        }
+
+                        if options.resTrailerMetadata != nil {
+                                *options.resTrailerMetadata = metadata.MD(res.Trailer).Copy()
+                        }
+
+                        codeHeader := res.Header.Get(HeaderGRPCCode)
+                        if codeHeader == "" {
+                                // Not a valid pRPC response.
+                                body := buf.String()
+                                bodySize := c.ErrBodySize
+                                if bodySize <= 0 {
+                                        bodySize = 256
+                                }
+                                if len(body) > bodySize {
+                                        body = body[:bodySize] + "..."
+                                }
+                                return fmt.Errorf("HTTP %d: no gRPC code. Body: %s", res.StatusCode, body)
+                        }
+
+                        codeInt, err := strconv.Atoi(codeHeader)
+                        if err != nil {
+                                // Not a valid pRPC response.
+                                return fmt.Errorf("invalid grpc code %q: %s", codeHeader, err)
+                        }
+
+                        code := codes.Code(codeInt)
+                        if code != codes.OK {
+                                desc := strings.TrimSuffix(buf.String(), "\n")
+                                err := grpcutil.Errf(code, "%s", desc)
+                                if isTransientCode(code) {
+                                        err = errors.WrapTransient(err)
+                                }
+                                return err
+                        }
+
+                        return proto.Unmarshal(buf.Bytes(), out) // non-transient error
+                },
+                func(err error, sleepTime time.Duration) {
+                        logging.Fields{
+                                logging.ErrorKey: err,
+                                "sleepTime":      sleepTime,
+                        }.Warningf(ctx, "RPC failed transiently. Will retry in %s", sleepTime)
+                },
+        )
+
+        if err != nil {
+                logging.WithError(err).Warningf(ctx, "RPC failed permanently: %s", err)
+        }
+
+        // We have to unwrap gRPC errors because
+        // grpc.Code and grpc.ErrorDesc functions do not work with error wrappers.
+        // https://github.com/grpc/grpc-go/issues/494
+        return errors.UnwrapAll(err)
+}
+
+// prepareRequest creates an HTTP request for an RPC,
+// except it does not set the request body.
+func prepareRequest(host, serviceName, methodName string, contentLength int, options *Options) *http.Request {
+        if host == "" {
+                panic("Host is not set")
+        }
+        req := &http.Request{
+                Method: "POST",
+                URL: &url.URL{
+                        Scheme: "https",
+                        Host:   host,
+                        Path:   fmt.Sprintf("/prpc/%s/%s", serviceName, methodName),
+                },
+                Header: http.Header{},
+        }
+        if options.Insecure {

[Vadim Sh., 2016/01/27 01:13:54]

can we also do it by default for "localhost"? Very useful when testing stuff
against GAE dev server.

+                req.URL.Scheme = "http"
+        }
+
+        // Set headers.
+        const mediaType = "application/prpc" // binary
+        req.Header.Set("Content-Type", mediaType)
+        req.Header.Set("Accept", mediaType)
+        userAgent := options.UserAgent
+        if userAgent == "" {
+                userAgent = DefaultUserAgent
+        }
+        req.Header.Set("User-Agent", userAgent)
+        req.ContentLength = int64(contentLength)
+        req.Header.Set("Content-Length", strconv.Itoa(contentLength))
+        // TODO(nodir): add "Accept-Encoding: gzip" when pRPC server supports it.
+        return req
+}
+
+func isTransientCode(code codes.Code) bool {
+        switch code {
+        case codes.Internal, codes.Unknown, codes.Unavailable:
+                return true
+
+        default:
+                return false
+        }
+}