53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
|
// Copyright (C) 2020 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package rpcerrs
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/zeebo/errs"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
|
||
|
|
"storj.io/common/rpc/rpcstatus"
|
||
|
|
)
|
||
|
|
|
||
|
|
// StatusMap is used to apply the correct rpc status code to error classes.
|
||
|
|
type StatusMap map[*errs.Class]rpcstatus.StatusCode
|
||
|
|
|
||
|
|
// Log consolidates logging of original errors with sanitization of internal errors.
|
||
|
|
type Log struct {
|
||
|
|
wrapper *errs.Class
|
||
|
|
log *zap.Logger
|
||
|
|
codeMap StatusMap
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewLog creates a new Log.
|
||
|
|
func NewLog(wrapper *errs.Class, log *zap.Logger, codeMap StatusMap) *Log {
|
||
|
|
return &Log{
|
||
|
|
wrapper: wrapper,
|
||
|
|
log: log,
|
||
|
|
codeMap: codeMap,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Error logs the message and error to the logger and returns the mapped rpcstatus code.
|
||
|
|
func (sanitizer *Log) Error(msg string, err error) error {
|
||
|
|
if sanitizer.wrapper != nil {
|
||
|
|
err = sanitizer.wrapper.Wrap(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if sanitizer.log != nil {
|
||
|
|
sanitizer.log.Error(msg, zap.Error(err))
|
||
|
|
}
|
||
|
|
|
||
|
|
for errClass, code := range sanitizer.codeMap {
|
||
|
|
if errClass.Has(err) {
|
||
|
|
return rpcstatus.Error(code, err.Error())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if sanitizer.wrapper == nil {
|
||
|
|
return rpcstatus.Error(rpcstatus.Internal, msg)
|
||
|
|
}
|
||
|
|
return rpcstatus.Error(rpcstatus.Internal, sanitizer.wrapper.New(msg).Error())
|
||
|
|
}
|