from typing import Any

import orjson
from starlette.responses import JSONResponse

from owslib.encoders import JSONEncoder


class ORJSONResponse(JSONResponse):
    encoder = JSONEncoder()

    def render(self, content: Any) -> bytes:
        return orjson.dumps(content, default=self._encode)

    def _encode(self, o: Any) -> Any:
        return self.encoder.default(o)