JSON Schema benchmark

benchmark_jsonschema.py

import pyperf
import time

import jsonschema
from jsonschema.exceptions import best_match

SCHEMA =  {
    "type": "object",
    "properties": {
        "displayname": {"oneOf": [{"type": "string"}, {"type": "null"}]},
    },
    "required": ["displayname"],
}
VALIDATOR = jsonschema.Draft7Validator(SCHEMA)


def json_schema(content):
    try:
        VALIDATOR.validate(content)
    except jsonschema.ValidationError:
        pass
    else:
        return content["displayname"]


def manual(content):
    try:
        displayname = content["displayname"]
    except KeyError:
        pass
    else:
        if isinstance(displayname, str):
            return content["displayname"]

runner = pyperf.Runner()

for name, content in (("empty", {}), ("wrong_type", {"displayname": 1}), ("good", {"displayname": "Bob"})):
    runner.bench_func('json_schema_%s' % name, json_schema, content)
    runner.bench_func('manual_%s' % name, manual, {})

See matrix-org/synapse#8445 (comment)