From 51f088196a3e3d0622f5dcbaed278b1b798bb6cb Mon Sep 17 00:00:00 2001 From: Timur Guseynov Date: Thu, 28 Dec 2023 15:55:02 +0300 Subject: [PATCH] use websocket.CloseInternalServerErr code instead of number --- internal/rest/response.go | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) diff --git a/internal/rest/response.go b/internal/rest/response.go index 637dcc6..2149256 100755 --- a/internal/rest/response.go +++ b/internal/rest/response.go @@ -167,29 +167,25 @@ func Respond(ctx context.Context, w http.ResponseWriter, data interface{}, code } func WebsocketErrorHandler(ctx context.Context, err error) { - // TODO: replace status codes - - serverErrorStatusCode := 1011 - switch errors.Cause(err) { case ErrNotFound: - websocketRespondError(ctx, err, serverErrorStatusCode) + websocketRespondError(ctx, err, websocket.CloseInternalServerErr) return case ErrInvalidID: - websocketRespondError(ctx, err, serverErrorStatusCode) + websocketRespondError(ctx, err, websocket.CloseInternalServerErr) return case ErrValidation: - websocketRespondError(ctx, err, serverErrorStatusCode) + websocketRespondError(ctx, err, websocket.CloseInternalServerErr) return case ErrUnauthorized: - websocketRespondError(ctx, err, serverErrorStatusCode) + websocketRespondError(ctx, err, websocket.CloseInternalServerErr) return case ErrForbidden: - websocketRespondError(ctx, err, serverErrorStatusCode) + websocketRespondError(ctx, err, websocket.CloseInternalServerErr) return } @@ -202,11 +198,11 @@ func WebsocketErrorHandler(ctx context.Context, err error) { websocketRespond(ctx, v) return case ResponseError: - websocketRespondError(ctx, e.Err, serverErrorStatusCode) + websocketRespondError(ctx, e.Err, websocket.CloseInternalServerErr) return } - websocketRespondError(ctx, err, serverErrorStatusCode) + websocketRespondError(ctx, err, websocket.CloseInternalServerErr) } func errorHandler(ctx context.Context, w http.ResponseWriter, err error) {