/*
 *
 * Copyright 2020 gRPC authors.
 *
 * 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 status implements errors returned by gRPC. These errors are // serialized and transmitted on the wire between server and client, and allow // for additional data to be transmitted via the Details field in the status // proto. gRPC service handlers should return an error created by this // package, and gRPC clients should expect a corresponding error to be // returned from the RPC call. // // This package upholds the invariants that a non-nil error may not // contain an OK code, and an OK code must result in a nil error.
package status import ( spb ) // Status represents an RPC status code, message, and details. It is immutable // and should be created with New, Newf, or FromProto. type Status struct { s *spb.Status } // New returns a Status representing c and msg. func ( codes.Code, string) *Status { return &Status{s: &spb.Status{Code: int32(), Message: }} } // Newf returns New(c, fmt.Sprintf(format, a...)). func ( codes.Code, string, ...interface{}) *Status { return New(, fmt.Sprintf(, ...)) } // FromProto returns a Status representing s. func ( *spb.Status) *Status { return &Status{s: proto.Clone().(*spb.Status)} } // Err returns an error representing c and msg. If c is OK, returns nil. func ( codes.Code, string) error { return New(, ).Err() } // Errorf returns Error(c, fmt.Sprintf(format, a...)). func ( codes.Code, string, ...interface{}) error { return Err(, fmt.Sprintf(, ...)) } // Code returns the status code contained in s. func ( *Status) () codes.Code { if == nil || .s == nil { return codes.OK } return codes.Code(.s.Code) } // Message returns the message contained in s. func ( *Status) () string { if == nil || .s == nil { return "" } return .s.Message } // Proto returns s's status as an spb.Status proto message. func ( *Status) () *spb.Status { if == nil { return nil } return proto.Clone(.s).(*spb.Status) } // Err returns an immutable error representing s; returns nil if s.Code() is OK. func ( *Status) () error { if .Code() == codes.OK { return nil } return &Error{s: } } // WithDetails returns a new status with the provided details messages appended to the status. // If any errors are encountered, it returns nil and the first error encountered. func ( *Status) ( ...proto.Message) (*Status, error) { if .Code() == codes.OK { return nil, errors.New("no error details for status with code OK") } // s.Code() != OK implies that s.Proto() != nil. := .Proto() for , := range { , := ptypes.MarshalAny() if != nil { return nil, } .Details = append(.Details, ) } return &Status{s: }, nil } // Details returns a slice of details messages attached to the status. // If a detail cannot be decoded, the error is returned in place of the detail. func ( *Status) () []interface{} { if == nil || .s == nil { return nil } := make([]interface{}, 0, len(.s.Details)) for , := range .s.Details { := &ptypes.DynamicAny{} if := ptypes.UnmarshalAny(, ); != nil { = append(, ) continue } = append(, .Message) } return } func ( *Status) () string { return fmt.Sprintf("rpc error: code = %s desc = %s", .Code(), .Message()) } // Error wraps a pointer of a status proto. It implements error and Status, // and a nil *Error should never be returned by this package. type Error struct { s *Status } func ( *Error) () string { return .s.String() } // GRPCStatus returns the Status represented by se. func ( *Error) () *Status { return .s } // Is implements future error.Is functionality. // A Error is equivalent if the code and message are identical. func ( *Error) ( error) bool { , := .(*Error) if ! { return false } return proto.Equal(.s.s, .s.s) } // IsRestrictedControlPlaneCode returns whether the status includes a code // restricted for control plane usage as defined by gRFC A54. func ( *Status) bool { switch .Code() { case codes.InvalidArgument, codes.NotFound, codes.AlreadyExists, codes.FailedPrecondition, codes.Aborted, codes.OutOfRange, codes.DataLoss: return true } return false }