ProtobufException.java

/**
 * 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 org.deephacks.westty.protobuf;

import com.google.common.base.Preconditions;

public class ProtobufException extends RuntimeException {

    private static final long serialVersionUID = -7994691832123397253L;
    private Integer code;
    private String message;

    public ProtobufException(String message, Integer code) {
        this.code = Preconditions.checkNotNull(code);
        this.message = Preconditions.checkNotNull(message);
    }

    public ProtobufException(String message, Integer code, Exception e) {
        super(e);
        this.code = Preconditions.checkNotNull(code);
    }

    public static void throwBadRequest(String message) {
        throw new ProtobufException(message, FailureCode.BAD_REQUEST.code);
    }

    public static void throwUnauthorized(String message) {
        throw new ProtobufException(message, FailureCode.UNAUTHORIZED.code);
    }

    public static void throwForbidden(String message) {
        throw new ProtobufException(message, FailureCode.FORBIDDEN.code);
    }

    public static void throwNotFound(String message) {
        throw new ProtobufException(message, FailureCode.NOT_FOUND.code);
    }

    public static void throwTimeout(String message) {
        throw new ProtobufException(message, FailureCode.TIMEOUT.code);
    }

    public static void throwConflic(String message) {
        throw new ProtobufException(message, FailureCode.CONFLICT.code);
    }

    public static void throwGone(String message) {
        throw new ProtobufException(message, FailureCode.GONE.code);
    }

    public static void throwTooLarge(String message) {
        throw new ProtobufException(message, FailureCode.TOO_LARGE.code);
    }

    public static void throwInternalError(String message) {
        throw new ProtobufException(message, FailureCode.INTERNAL_ERROR.code);
    }

    public static void throwNotImplemented(String message) {
        throw new ProtobufException(message, FailureCode.NOT_IMPLEMENTED.code);
    }

    public static void throwServiceUnavailable(String message) {
        throw new ProtobufException(message, FailureCode.BAD_REQUEST.code);
    }

    public Integer getCode() {
        return code;
    }

    public String getProtobufMessage() {
        return message;
    }

    @Override
    public String getMessage() {
        return codeToString(code) + ": " + message;
    }

    static String codeToString(Integer code) {
        if (code == FailureCode.BAD_REQUEST.code) {
            return FailureCode.BAD_REQUEST.toString();
        } else if (code == FailureCode.UNAUTHORIZED.code) {
            return FailureCode.UNAUTHORIZED.toString();
        } else if (code == FailureCode.FORBIDDEN.code) {
            return FailureCode.FORBIDDEN.toString();
        } else if (code == FailureCode.NOT_FOUND.code) {
            return FailureCode.NOT_FOUND.toString();
        } else if (code == FailureCode.TIMEOUT.code) {
            return FailureCode.TIMEOUT.toString();
        } else if (code == FailureCode.CONFLICT.code) {
            return FailureCode.CONFLICT.toString();
        } else if (code == FailureCode.GONE.code) {
            return FailureCode.GONE.toString();
        } else if (code == FailureCode.TOO_LARGE.code) {
            return FailureCode.TOO_LARGE.toString();
        } else if (code == FailureCode.INTERNAL_ERROR.code) {
            return FailureCode.INTERNAL_ERROR.toString();
        } else if (code == FailureCode.NOT_IMPLEMENTED.code) {
            return FailureCode.NOT_IMPLEMENTED.toString();
        } else if (code == FailureCode.SERVICE_UNAVAILABLE.code) {
            return FailureCode.SERVICE_UNAVAILABLE.toString();
        } else {
            return Integer.toString(code);
        }
    }

    public static enum FailureCode {
        BAD_REQUEST(1), UNAUTHORIZED(2), FORBIDDEN(3), NOT_FOUND(4), TIMEOUT(5), CONFLICT(6), GONE(
                6), TOO_LARGE(7), INTERNAL_ERROR(8), NOT_IMPLEMENTED(9), SERVICE_UNAVAILABLE(10);
        private final Integer code;

        private FailureCode(Integer code) {
            this.code = code;
        }

        public Integer getCode() {
            return code;
        }
    }

}