Function scenario_module_functions

tsan_stress_simplejson.py:426–473

Concurrent scanstring + encode_basestring_ascii (module-level).

()

def scenario_module_functions():
    """Concurrent scanstring + encode_basestring_ascii (module-level)."""
    scanstring = _speedups.scanstring
    enc_ascii = _speedups.encode_basestring_ascii

    strings_to_encode = [
        "hello world",
        "\"quoted\" with \\ backslash",
        "\u00e9\u00e8\u00ea \u4e2d\u6587 emoji-\U0001F600",
        "control\n\t\r chars",
        "x" * 256,
    ]
    # For scanstring: the opening quote has been consumed; pass end=1.
    scan_sources = [
        r'"a simple string"',
        r'"with \"escapes\" and \n newlines"',
        r'"unicode \u00e9\u00e8\u4e2d\u6587 stuff"',
        r'"backslashes \\ and slashes \/"',
    ]

    def enc_worker():
        deadline = time.monotonic() + DURATION
        for i in range(ITERATIONS):
            try:
                enc_ascii(strings_to_encode[i % len(strings_to_encode)])
            except Exception:
                pass
            if time.monotonic() > deadline:
                break

    def scan_worker():
        deadline = time.monotonic() + DURATION
        for i in range(ITERATIONS):
            src = scan_sources[i % len(scan_sources)]
            try:
                # scanstring(basestring, end, encoding, strict)
                scanstring(src, 1, "utf-8", 1)
            except Exception:
                pass
            if time.monotonic() > deadline:
                break

    half = max(THREADS // 2, 1)
    run_scenario(
        "module-level scanstring + encode_basestring_ascii",
        [enc_worker, scan_worker],
        [half, max(THREADS - half, 1)],
    )


# --------------------------- main ------------------------------------------- #