Files
panels/services/gateway-service/internal/api/handlers/error.go

70 lines
1.9 KiB
Go

// Copyright 2023 Declan Teevan
//
// 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 handlers
import (
"errors"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/fiber/v2/log"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
)
func ErrorHandler(c *fiber.Ctx, err error) error {
c.Set(fiber.HeaderContentType, fiber.MIMEApplicationJSONCharsetUTF8)
code := fiber.StatusInternalServerError
msg := err.Error()
// Retrieval of codes from fiber.Errors
var e *fiber.Error
if errors.As(err, &e) {
code = e.Code
} else {
// Retrival of codes from gRPC errors.
status, ok := status.FromError(err)
if ok {
msg = status.Message()
switch status.Code() {
case codes.NotFound:
code = fiber.StatusNotFound
case codes.InvalidArgument:
code = fiber.StatusUnprocessableEntity
case codes.AlreadyExists:
code = fiber.StatusConflict
case codes.PermissionDenied:
code = fiber.StatusForbidden
case codes.Unauthenticated:
code = fiber.StatusUnauthorized
case codes.Internal:
code = fiber.StatusInternalServerError
case codes.Unavailable:
code = fiber.StatusBadGateway
msg = "Service unavaliable for request."
default:
code = fiber.StatusInternalServerError
msg = "Something went wrong."
log.Error(err)
}
} else {
msg = "Something unexpected went wrong."
log.Error(err)
}
}
return c.Status(code).JSON(fiber.Map{"status": "failure", "msg": msg})
}