127 lines
2.8 KiB
Go
127 lines
2.8 KiB
Go
/*
|
|
* Copyright 2020 The Dragonfly 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 dferrors
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"google.golang.org/grpc/status"
|
|
|
|
commonv1 "d7y.io/api/v2/pkg/apis/common/v1"
|
|
)
|
|
|
|
// common and framework errors
|
|
var (
|
|
ErrInvalidArgument = errors.New("invalid argument")
|
|
ErrInvalidHeader = errors.New("invalid Header")
|
|
ErrDataNotFound = errors.New("data not found")
|
|
ErrEmptyValue = errors.New("empty value")
|
|
ErrConvertFailed = errors.New("convert failed")
|
|
ErrEndOfStream = errors.New("end of stream")
|
|
ErrNoCandidateNode = errors.New("no candidate server node")
|
|
)
|
|
|
|
// IsEndOfStream returns true if the error is end of stream.
|
|
func IsEndOfStream(err error) bool {
|
|
return err == ErrEndOfStream
|
|
}
|
|
|
|
type DfError struct {
|
|
Code commonv1.Code
|
|
Message string
|
|
}
|
|
|
|
func (s *DfError) Error() string {
|
|
return fmt.Sprintf("[%d]%s", s.Code, s.Message)
|
|
}
|
|
|
|
func New(code commonv1.Code, msg string) *DfError {
|
|
return &DfError{
|
|
Code: code,
|
|
Message: msg,
|
|
}
|
|
}
|
|
|
|
func Newf(code commonv1.Code, format string, a ...any) *DfError {
|
|
return &DfError{
|
|
Code: code,
|
|
Message: fmt.Sprintf(format, a...),
|
|
}
|
|
}
|
|
|
|
func CheckError(err error, code commonv1.Code) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
|
|
e, ok := err.(*DfError)
|
|
|
|
return ok && e.Code == code
|
|
}
|
|
|
|
// ConvertGRPCErrorToDfError converts grpc error to DfError, if it exists.
|
|
func ConvertGRPCErrorToDfError(err error) error {
|
|
for _, d := range status.Convert(err).Details() {
|
|
switch internal := d.(type) {
|
|
case *commonv1.GrpcDfError:
|
|
return &DfError{
|
|
Code: internal.Code,
|
|
Message: internal.Message,
|
|
}
|
|
}
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// IsGRPCDfError checks if the error is a GRPCDfError.
|
|
func IsGRPCDfError(err error) (*DfError, bool) {
|
|
for _, d := range status.Convert(err).Details() {
|
|
switch internal := d.(type) {
|
|
case *commonv1.GrpcDfError:
|
|
return &DfError{
|
|
Code: internal.Code,
|
|
Message: internal.Message,
|
|
}, true
|
|
}
|
|
}
|
|
|
|
var de *DfError
|
|
ok := errors.As(err, &de)
|
|
if ok {
|
|
return de, true
|
|
}
|
|
|
|
return nil, false
|
|
}
|
|
|
|
// ConvertDfErrorToGRPCError converts DfError to grpc error, if it is.
|
|
func ConvertDfErrorToGRPCError(err error) error {
|
|
if v, ok := err.(*DfError); ok {
|
|
s, e := status.Convert(err).WithDetails(
|
|
&commonv1.GrpcDfError{
|
|
Code: v.Code,
|
|
Message: v.Message,
|
|
})
|
|
if e == nil {
|
|
err = s.Err()
|
|
}
|
|
}
|
|
return err
|
|
}
|