go17.go 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. // +build go1.7
  2. /*
  3. *
  4. * Copyright 2016 gRPC authors.
  5. *
  6. * Licensed under the Apache License, Version 2.0 (the "License");
  7. * you may not use this file except in compliance with the License.
  8. * You may obtain a copy of the License at
  9. *
  10. * http://www.apache.org/licenses/LICENSE-2.0
  11. *
  12. * Unless required by applicable law or agreed to in writing, software
  13. * distributed under the License is distributed on an "AS IS" BASIS,
  14. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15. * See the License for the specific language governing permissions and
  16. * limitations under the License.
  17. *
  18. */
  19. package transport
  20. import (
  21. "context"
  22. "net"
  23. "net/http"
  24. "google.golang.org/grpc/codes"
  25. "google.golang.org/grpc/status"
  26. netctx "golang.org/x/net/context"
  27. )
  28. // dialContext connects to the address on the named network.
  29. func dialContext(ctx context.Context, network, address string) (net.Conn, error) {
  30. return (&net.Dialer{}).DialContext(ctx, network, address)
  31. }
  32. // ContextErr converts the error from context package into a status error.
  33. func ContextErr(err error) error {
  34. switch err {
  35. case context.DeadlineExceeded, netctx.DeadlineExceeded:
  36. return status.Error(codes.DeadlineExceeded, err.Error())
  37. case context.Canceled, netctx.Canceled:
  38. return status.Error(codes.Canceled, err.Error())
  39. }
  40. return status.Errorf(codes.Internal, "Unexpected error from context packet: %v", err)
  41. }
  42. // contextFromRequest returns a context from the HTTP Request.
  43. func contextFromRequest(r *http.Request) context.Context {
  44. return r.Context()
  45. }