shop-java/tools/compare_live.py

# -*- coding: utf-8 -*-
"""实时对比：同时请求线上 PHP 与本地 Java，逐字段 diff。

与 compare_batch.py 的区别：不依赖 golden 快照，每次都拉线上最新数据，
适合带参数/随数据变化的接口（商品基础信息、订单发货、设置等）。

用法：python compare_live.py [用例名...]
"""
import sys
import re
import json
import os
import requests
from pathlib import Path

PHP_BASE = os.environ.get("YOSHOP_PHP_BASE", "https://shop.qinys.cn/index.php?s=")
JAVA_BASE = "http://127.0.0.1:18080"
TOKEN = "testsession123"

SESSION = requests.Session()
SESSION.trust_env = False

# (用例名, Java 路径, PHP 路径, 参数)
CASES = [
    ("goods_basic_10004", "/store/goods/basic", "/store/goods/basic", {"goodsId": 10004}),
    ("goods_basic_10005", "/store/goods/basic", "/store/goods/basic", {"goodsId": 10005}),
    ("goods_list_by_ids", "/store/goods/listByIds", "/store/goods/listByIds", {"goodsIds[]": [10004, 10002]}),
    ("goods_spec_list", "/store/goods.spec/list", "/store/goods.spec/list", {"goodsId": 10004}),
    ("goods_detail_multi", "/store/goods/detail", "/store/goods/detail", {"goodsId": 10004}),
    ("goods_detail_multi2", "/store/goods/detail", "/store/goods/detail", {"goodsId": 10009}),
    ("goods_detail_single", "/store/goods/detail", "/store/goods/detail", {"goodsId": 10005}),
]


def php_token() -> str:
    """复用上传商品脚本里的真实后台 token（PHP 黄金样本需要线上登录态）"""
    src = Path(r"F:\import goods\upload_goods.py").read_text(encoding="utf-8")
    m = re.search(r'TOKEN\s*=\s*"([^"]+)"', src)
    return m.group(1) if m else TOKEN


def diff(a, b, path="$", out=None, limit=15):
    out = out if out is not None else []
    if len(out) >= limit:
        return out
    if type(a) != type(b):
        if isinstance(a, (int, float)) and isinstance(b, (int, float)) \
                and not isinstance(a, bool) and not isinstance(b, bool):
            if float(a) != float(b):
                out.append(f"{path}: php={a!r} java={b!r}")
            return out
        out.append(f"{path}: 类型不同 php={type(a).__name__} java={type(b).__name__} (php={a!r} java={b!r})")
        return out
    if isinstance(a, dict):
        for k in a:
            if k not in b:
                out.append(f"{path}.{k}: Java 缺失 (php={a[k]!r})")
            else:
                diff(a[k], b[k], f"{path}.{k}", out, limit)
        for k in b:
            if k not in a:
                out.append(f"{path}.{k}: Java 多出 (java={b[k]!r})")
        return out
    if isinstance(a, list):
        if len(a) != len(b):
            out.append(f"{path}: 长度不同 php={len(a)} java={len(b)}")
        for i in range(min(len(a), len(b))):
            diff(a[i], b[i], f"{path}[{i}]", out, limit)
        return out
    if a != b:
        out.append(f"{path}: php={a!r} java={b!r}")
    return out


def fetch_php(path, params, token):
    r = SESSION.get(PHP_BASE + path, headers={"Access-Token": token}, params=params, timeout=60)
    return r.json()


def fetch_java(path, params):
    # 数组参数名已带 []（与前端 axios 序列化一致），直接透传
    r = SESSION.get(JAVA_BASE + path, headers={"Access-Token": TOKEN}, params=params or {}, timeout=60)
    return r.json()


def main():
    only = set(sys.argv[1:])
    token = php_token()
    ok = fail = 0
    for name, java_path, php_path, params in CASES:
        if only and name not in only:
            continue
        try:
            php = fetch_php(php_path, params, token)
        except Exception as e:
            print(f"[skip] {name} 拉取 PHP 失败: {e}")
            continue
        try:
            java = fetch_java(java_path, params)
        except Exception as e:
            print(f"[FAIL] {name} 拉取 Java 失败: {e}")
            fail += 1
            continue
        if php.get("status") != 200:
            print(f"[skip] {name} PHP 返回 {php.get('status')} {php.get('message')}")
            continue
        if java.get("status") != 200:
            print(f"[FAIL] {name} Java 返回 {java.get('status')} {java.get('message')}")
            fail += 1
            continue
        diffs = diff(php.get("data"), java.get("data"))
        if diffs:
            fail += 1
            print(f"[DIFF] {name} ({len(diffs)}+ 处)")
            for d in diffs:
                print("   ", d)
        else:
            ok += 1
            print(f"[PASS] {name}")
    print(f"\n通过 {ok} / 差异 {fail}")


if __name__ == "__main__":
    main()