apisix/runner/server/response.py

#
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You 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.
#
from __future__ import annotations
from a6pluginproto.Err import Code as A6ErrCode

RUNNER_SUCCESS_CODE = 200
RUNNER_SUCCESS_MESSAGE = "OK"
RUNNER_ERROR_CODE = 500
RUNNER_ERROR_MESSAGE = "ERR"

errorCodes = [
    A6ErrCode.Code.CONF_TOKEN_NOT_FOUND,
    A6ErrCode.Code.BAD_REQUEST,
    A6ErrCode.Code.SERVICE_UNAVAILABLE,
]


class Response:
    def __init__(self, code: int = 0, message: str = '', data: bytes = b'', ty: int = 0):
        self.__code = code
        self.__message = message
        self.__type = ty
        self.__data = data

    def __eq__(self, other: Response) -> bool:
        return self.code == other.code and \
               self.message == other.message and \
               self.data == other.data and \
               self.type == other.type

    @property
    def code(self) -> int:
        return self.__code

    @property
    def message(self) -> str:
        return self.__message

    @property
    def data(self) -> bytes:
        return self.__data

    @property
    def type(self) -> int:
        return self.__type