|
| 1 | +"""Optional native checks on an explicitly selected isolated local MySQL server.""" |
| 2 | +import re |
| 3 | +import subprocess |
| 4 | +import uuid |
| 5 | + |
| 6 | +FIXTURE_SQL = """CREATE TABLE t( |
| 7 | + id INT PRIMARY KEY, n INT, j JSON, data JSON, body TEXT, d DATETIME, |
| 8 | + KEY idx(n), FULLTEXT KEY(body)); |
| 9 | +CREATE TABLE u(id INT, n INT); |
| 10 | +INSERT INTO t VALUES(1,2,JSON_OBJECT('name','mysql','n',1), |
| 11 | + JSON_OBJECT('name','mysql','n',1),'mysql','2026-01-01'); |
| 12 | +""" |
| 13 | + |
| 14 | + |
| 15 | +def classify_native(returncode, stderr): |
| 16 | + if returncode == 0: |
| 17 | + return dict(state='ACCEPTED', error_code=None) |
| 18 | + match = re.search(r'ERROR (\d+)\b', stderr) |
| 19 | + code = int(match[1]) if match else None |
| 20 | + if code in (1064, 1149): |
| 21 | + state = 'SYNTAX_REJECTED' |
| 22 | + elif code is None or 2000 <= code < 3000: |
| 23 | + state = 'CLIENT_ERROR' |
| 24 | + else: |
| 25 | + state = 'SERVER_ERROR' |
| 26 | + return dict(state=state, error_code=code) |
| 27 | + |
| 28 | + |
| 29 | +def run_native(cases, client, socket, user): |
| 30 | + if not socket or not socket.startswith('/'): |
| 31 | + raise ValueError('Native oracle requires an explicit absolute path to an isolated server socket') |
| 32 | + command = [client, '--no-defaults', '--protocol=SOCKET', '--socket=' + socket, |
| 33 | + '--user=' + user, '--batch', '--raw', '--skip-column-names', |
| 34 | + '--default-character-set=utf8mb4', '--connect-timeout=5'] |
| 35 | + |
| 36 | + def query(sql, schema=None): |
| 37 | + args = command + (['--database', schema] if schema else []) |
| 38 | + return subprocess.run(args, input=sql, text=True, capture_output=True, timeout=30) |
| 39 | + |
| 40 | + metadata = query('SELECT @@version, @@session.sql_mode;') |
| 41 | + if metadata.returncode: |
| 42 | + raise RuntimeError('Cannot query native server metadata: ' + metadata.stderr.strip()) |
| 43 | + version, mode = metadata.stdout.rstrip('\n').split('\t') |
| 44 | + if not re.match(r'^(8\.4\.8|9\.7\.2)(?:-|$)', version): |
| 45 | + raise ValueError('Native server must be MySQL 8.4.8 or 9.7.2; found ' + version) |
| 46 | + schema = 'parsersql_compat_' + uuid.uuid4().hex |
| 47 | + create = query(f'CREATE DATABASE `{schema}`;') |
| 48 | + if create.returncode: |
| 49 | + raise RuntimeError('Cannot create isolated native fixture schema: ' + create.stderr.strip()) |
| 50 | + report = dict(state='RUN', version=version, sql_mode=mode, socket=socket, |
| 51 | + fixture_sql=FIXTURE_SQL, fixture_schema=schema, schema_cleaned_up=False, |
| 52 | + scope='SQL execution/EXPLAIN observations; not a raw-parser or equivalence oracle', |
| 53 | + cases=[]) |
| 54 | + try: |
| 55 | + fixture = query(FIXTURE_SQL, schema) |
| 56 | + if fixture.returncode: |
| 57 | + raise RuntimeError('Native fixture setup failed: ' + fixture.stderr.strip()) |
| 58 | + for case in cases: |
| 59 | + action = case.get('native_action', 'skip') |
| 60 | + if action == 'skip': |
| 61 | + report['cases'].append(dict(id=case['id'], state='NOT VERIFIED', |
| 62 | + reason='No native action configured; DDL/procedures excluded')) |
| 63 | + continue |
| 64 | + if action not in ('execute', 'explain'): |
| 65 | + raise ValueError('Unknown native action: ' + action) |
| 66 | + sql = ('EXPLAIN ' if action == 'explain' else '') + case['sql'] |
| 67 | + # Reapply the captured mode for each connection, keeping every witness comparable. |
| 68 | + escaped_mode = mode.replace("'", "''") |
| 69 | + result = query(f"SET SESSION sql_mode='{escaped_mode}';\n" + sql, schema) |
| 70 | + record = classify_native(result.returncode, result.stderr) |
| 71 | + intended_validity = case.get('native_validity', {}).get(version.split('-')[0], case['validity']) |
| 72 | + record['intended_validity'] = intended_validity |
| 73 | + record['matches_intended_validity'] = ( |
| 74 | + intended_validity == 'valid' if record['state'] == 'ACCEPTED' else |
| 75 | + intended_validity == 'invalid' if record['state'] == 'SYNTAX_REJECTED' else None) |
| 76 | + record.update(id=case['id'], sql=case['sql'], action=action, executed_sql=sql, |
| 77 | + returncode=result.returncode, stdout=result.stdout.rstrip('\n'), |
| 78 | + stderr=result.stderr.rstrip('\n')) |
| 79 | + report['cases'].append(record) |
| 80 | + finally: |
| 81 | + cleanup = query(f'DROP DATABASE `{schema}`;') |
| 82 | + report['schema_cleaned_up'] = cleanup.returncode == 0 |
| 83 | + if cleanup.returncode: |
| 84 | + raise RuntimeError(f'Native schema cleanup failed for {schema}: ' + cleanup.stderr.strip()) |
| 85 | + return report |
0 commit comments