join.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023
  1. package vrtxsql
  2. // JOIN 查询执行: 多表嵌套循环 + 右表索引点查, 结果物化后立即放锁。
  3. // 仅 SELECT ... FROM t1 JOIN t2 ON ... 走本路径 (单表查询仍走流式路径,
  4. // 见 exec.go queryStatement)。
  5. //
  6. // 执行策略:
  7. // - 左表全扫描 (本表 WHERE 条件下推), 每行存 别名(小写) → 列(小写) → driver.Value;
  8. // - 逐 JOIN: 取组合行里 ON 左列的值, 对右表做索引点查
  9. // (tbl.Query().Where(core.Eq(...)) + 右表本表条件), 命中行做笛卡尔组合;
  10. // INNER 无匹配丢行, LEFT 无匹配时右表列全 NULL 保留行;
  11. // - WHERE 按列归属拆分: 叶子条件 (AND 结构) 下推到所属表, 跨表 OR/NOT/列比较
  12. // 作为残余谓词在组合行上用 Go 值比较过滤 (= != < <= > >=);
  13. // - 投影 (*/t.*/t.col/col 混用, 重名列用 别名_列名 去重) → ORDER BY 内存
  14. // 稳定排序 (NULL 视为最小) → LIMIT/OFFSET, 返回物化 *rows。
  15. import (
  16. "database/sql/driver"
  17. "fmt"
  18. "sort"
  19. "strconv"
  20. "strings"
  21. "time"
  22. "git.ycl.pub/yl32940/TsDB2/DB/core"
  23. "git.ycl.pub/yl32940/TsDB2/DB/core/types"
  24. )
  25. // =============================================================================
  26. // 限定名列引用的解析辅助 (单表路径解限定 + JOIN 路径归属判定共用)
  27. // =============================================================================
  28. // splitQualCol 把列引用拆成 (限定词, 列名); 无 '.' 时 qual 为空。
  29. func splitQualCol(ref string) (qual, col string) {
  30. if i := strings.IndexByte(ref, '.'); i >= 0 {
  31. return ref[:i], ref[i+1:]
  32. }
  33. return "", ref
  34. }
  35. // dequalifyColRef 校验并剥掉列引用的限定词 ("t.col" → "col"); 非限定名原样返回。
  36. func dequalifyColRef(ref string, quals ...string) (string, error) {
  37. qual, col := splitQualCol(ref)
  38. if qual == "" {
  39. return ref, nil
  40. }
  41. for _, q := range quals {
  42. if strings.EqualFold(q, qual) {
  43. return col, nil
  44. }
  45. }
  46. return "", fmt.Errorf("列限定 %q 与 FROM 表 (%s) 不匹配", qual, strings.Join(quals, "/"))
  47. }
  48. // dequalifyExpr 剥掉表达式里所有列引用的限定词 (单表 SELECT 用; 值原样保留)。
  49. func dequalifyExpr(e expr, quals ...string) (expr, error) {
  50. if e == nil {
  51. return nil, nil
  52. }
  53. switch t := e.(type) {
  54. case *andExpr:
  55. l, err := dequalifyExpr(t.l, quals...)
  56. if err != nil {
  57. return nil, err
  58. }
  59. r, err := dequalifyExpr(t.r, quals...)
  60. if err != nil {
  61. return nil, err
  62. }
  63. return &andExpr{l: l, r: r}, nil
  64. case *orExpr:
  65. l, err := dequalifyExpr(t.l, quals...)
  66. if err != nil {
  67. return nil, err
  68. }
  69. r, err := dequalifyExpr(t.r, quals...)
  70. if err != nil {
  71. return nil, err
  72. }
  73. return &orExpr{l: l, r: r}, nil
  74. case *notExpr:
  75. inner, err := dequalifyExpr(t.e, quals...)
  76. if err != nil {
  77. return nil, err
  78. }
  79. return &notExpr{e: inner}, nil
  80. case *cmpExpr:
  81. col, err := dequalifyColRef(t.col, quals...)
  82. if err != nil {
  83. return nil, err
  84. }
  85. return &cmpExpr{col: col, op: t.op, val: t.val}, nil
  86. case *colCmpExpr:
  87. l, err := dequalifyColRef(t.left, quals...)
  88. if err != nil {
  89. return nil, err
  90. }
  91. r, err := dequalifyColRef(t.right, quals...)
  92. if err != nil {
  93. return nil, err
  94. }
  95. return &colCmpExpr{left: l, op: t.op, right: r}, nil
  96. case *inExpr:
  97. col, err := dequalifyColRef(t.col, quals...)
  98. if err != nil {
  99. return nil, err
  100. }
  101. return &inExpr{col: col, vals: t.vals, neg: t.neg}, nil
  102. case *isNullExpr:
  103. col, err := dequalifyColRef(t.col, quals...)
  104. if err != nil {
  105. return nil, err
  106. }
  107. return &isNullExpr{col: col, neg: t.neg}, nil
  108. case *likeExpr:
  109. col, err := dequalifyColRef(t.col, quals...)
  110. if err != nil {
  111. return nil, err
  112. }
  113. return &likeExpr{col: col, pattern: t.pattern, neg: t.neg}, nil
  114. }
  115. return nil, fmt.Errorf("无法处理的 WHERE 表达式")
  116. }
  117. // selectColsSingle 单表 SELECT 的列清单展开: 空 / "*" / "t.*" → 全部活跃列;
  118. // "t.col" → 校验限定词后取列; 列存在性校验与旧行为一致 (ErrFieldNotFound)。
  119. func selectColsSingle(tbl *core.Table, sel *selectStmt, quals ...string) ([]string, error) {
  120. if len(sel.columns) == 0 {
  121. return activeColumnNames(tbl), nil
  122. }
  123. out := make([]string, 0, len(sel.columns))
  124. for _, c := range sel.columns {
  125. if c == "*" {
  126. out = append(out, activeColumnNames(tbl)...)
  127. continue
  128. }
  129. if strings.HasSuffix(c, ".*") {
  130. if _, err := dequalifyColRef(c, quals...); err != nil {
  131. return nil, err
  132. }
  133. out = append(out, activeColumnNames(tbl)...)
  134. continue
  135. }
  136. col, err := dequalifyColRef(c, quals...)
  137. if err != nil {
  138. return nil, err
  139. }
  140. // 投影列名统一改用引擎规范名: 结果集表头与引擎列名一致 (MySQL 语义),
  141. // 也避免下游按用户原文 (大小写混写) 做精确匹配时落空。
  142. fi := findField(tbl, col)
  143. if fi == nil {
  144. return nil, fmt.Errorf("%w: %q", core.ErrFieldNotFound, c)
  145. }
  146. out = append(out, string(fi.FieldName))
  147. }
  148. return out, nil
  149. }
  150. // =============================================================================
  151. // JOIN 执行
  152. // =============================================================================
  153. // joinTable 参与 JOIN 的一张表: 引擎表 + 归一化别名 + 活跃列。
  154. type joinTable struct {
  155. idx int
  156. name string
  157. alias string // 限定名归一化结果 (alias 缺省 = 表名), 比较时大小写不敏感
  158. tbl *core.Table
  159. cols []string // 活跃列名 (原始大小写, 投影展示用)
  160. ft map[string]types.FieldType // 小写列名 → 列类型 (typed 值转换用)
  161. }
  162. func buildJoinTable(idx int, ref tableRef, tbl *core.Table) *joinTable {
  163. alias := ref.alias
  164. if alias == "" {
  165. alias = ref.name
  166. }
  167. jt := &joinTable{idx: idx, name: ref.name, alias: alias, tbl: tbl, ft: map[string]types.FieldType{}}
  168. for i := range tbl.Field {
  169. f := &tbl.Field[i]
  170. if f.IsDropped() {
  171. continue
  172. }
  173. cn := string(f.FieldName)
  174. jt.cols = append(jt.cols, cn)
  175. jt.ft[strings.ToLower(cn)] = f.FieldType
  176. }
  177. return jt
  178. }
  179. // joinRow 组合行: 表别名(小写) → 列名(小写) → driver.Value (未匹配侧为 NULL)。
  180. type joinRow map[string]map[string]driver.Value
  181. func (r joinRow) get(jt *joinTable, col string) driver.Value {
  182. return r[strings.ToLower(jt.alias)][strings.ToLower(col)]
  183. }
  184. func cloneJoinRow(r joinRow) joinRow {
  185. nr := make(joinRow, len(r)+1)
  186. for k, v := range r {
  187. nr[k] = v
  188. }
  189. return nr
  190. }
  191. // nullJoinRow 右表无匹配时 (LEFT JOIN) 的全 NULL 行。
  192. func nullJoinRow(jt *joinTable) map[string]driver.Value {
  193. m := make(map[string]driver.Value, len(jt.cols))
  194. for _, cn := range jt.cols {
  195. m[strings.ToLower(cn)] = nil
  196. }
  197. return m
  198. }
  199. // findQual 按限定词找表 (别名或表名, 大小写不敏感)。
  200. func findQual(ts []*joinTable, qual string) *joinTable {
  201. for _, jt := range ts {
  202. if strings.EqualFold(jt.alias, qual) || strings.EqualFold(jt.name, qual) {
  203. return jt
  204. }
  205. }
  206. return nil
  207. }
  208. // ownerOfCol 解析列引用归属的表: 限定名直查; 非限定名要求在恰好一张表上存在
  209. // (存在于多张报"列不明确", 没有报列不存在)。返回归属表与剥掉限定后的列名。
  210. func ownerOfCol(ts []*joinTable, ref string) (*joinTable, string, error) {
  211. qual, col := splitQualCol(ref)
  212. if qual != "" {
  213. jt := findQual(ts, qual)
  214. if jt == nil {
  215. return nil, "", fmt.Errorf("未知的表限定 %q (列 %q)", qual, ref)
  216. }
  217. if findField(jt.tbl, col) == nil {
  218. return nil, "", fmt.Errorf("%w: %q", core.ErrFieldNotFound, ref)
  219. }
  220. return jt, col, nil
  221. }
  222. var hit *joinTable
  223. for _, jt := range ts {
  224. if findField(jt.tbl, col) != nil {
  225. if hit != nil {
  226. return nil, "", fmt.Errorf("列 %q 不明确 (同时存在于 %q 和 %q)", col, hit.alias, jt.alias)
  227. }
  228. hit = jt
  229. }
  230. }
  231. if hit == nil {
  232. return nil, "", fmt.Errorf("%w: %q", core.ErrFieldNotFound, ref)
  233. }
  234. return hit, col, nil
  235. }
  236. // scanJoinTable 扫一张表 (应用下推条件), 每行 → 小写列名 → driver.Value。
  237. func scanJoinTable(jt *joinTable, cond core.Condition) ([]map[string]driver.Value, error) {
  238. q := jt.tbl.Query()
  239. if cond != nil {
  240. q = q.Where(cond)
  241. }
  242. var out []map[string]driver.Value
  243. var convErr error
  244. readBlob := blobReaderOf(jt.tbl) // Blob 列行内 8B 引用 → 内容字节 (持锁期间调)
  245. err := q.Scan(func(full []types.FieldValue) bool {
  246. m := make(map[string]driver.Value, len(full))
  247. for _, fv := range full {
  248. key := strings.ToLower(string(fv.FieldName))
  249. raw := fv.Value
  250. if ref, isRef := raw.(types.BlobRef); isRef {
  251. data, err := readBlob(ref)
  252. if err != nil {
  253. convErr = fmt.Errorf("读取 BLOB 列 %q: %w", fv.FieldName, err)
  254. return false
  255. }
  256. m[key] = data
  257. continue
  258. }
  259. dv, err := fieldValueToDriverTyped(jt.ft[key], raw)
  260. if err != nil {
  261. convErr = err
  262. return false
  263. }
  264. m[key] = joinTypedValue(jt.ft[key], dv)
  265. }
  266. out = append(out, m)
  267. return true
  268. })
  269. if convErr != nil {
  270. return nil, convErr
  271. }
  272. if err != nil {
  273. return nil, err
  274. }
  275. return out, nil
  276. }
  277. // joinTypedValue 把整型/浮点列的文本化值归一回数值: fieldValueToDriver 对
  278. // int32 等小整型走 fmt.Sprint 兜底成 []byte, 残余谓词/排序会因此变成字符串
  279. // 比较 ("100" < "25"), 这里按列类型归一, 与 BIGINT 的输出形态 (int64) 对齐。
  280. func joinTypedValue(ft types.FieldType, v driver.Value) driver.Value {
  281. if v == nil {
  282. return nil
  283. }
  284. var s string
  285. switch t := v.(type) {
  286. case []byte:
  287. s = string(t)
  288. case string:
  289. s = t
  290. default:
  291. return v
  292. }
  293. if isIntType(ft) {
  294. if n, err := strconv.ParseInt(strings.TrimSpace(s), 10, 64); err == nil {
  295. return n
  296. }
  297. }
  298. switch ft {
  299. case types.FieldType_Float64, types.FieldType_Float32:
  300. if f, err := strconv.ParseFloat(strings.TrimSpace(s), 64); err == nil {
  301. return f
  302. }
  303. }
  304. return v
  305. }
  306. // =============================================================================
  307. // WHERE 按归属拆分: 下推条件 + 残余谓词
  308. // =============================================================================
  309. // joinWherePlan WHERE 拆分结果: perTable[i] 下推到第 i 张表扫描的 AND 条件,
  310. // residual 为不可下推的表达式 (跨表 OR/NOT、列与列比较), 组合行上过滤。
  311. type joinWherePlan struct {
  312. perTable []core.Condition
  313. residual expr
  314. }
  315. func planJoinWhere(ts []*joinTable, where expr, args []driver.NamedValue) (*joinWherePlan, error) {
  316. plan := &joinWherePlan{perTable: make([]core.Condition, len(ts))}
  317. if where == nil {
  318. return plan, nil
  319. }
  320. if err := splitJoinWhere(ts, where, args, plan); err != nil {
  321. return nil, err
  322. }
  323. return plan, nil
  324. }
  325. // splitJoinWhere 递归拆分 AND 结构; 叶子条件按列归属下推, 其余整体留作残余。
  326. func splitJoinWhere(ts []*joinTable, e expr, args []driver.NamedValue, plan *joinWherePlan) error {
  327. switch t := e.(type) {
  328. case *andExpr:
  329. if err := splitJoinWhere(ts, t.l, args, plan); err != nil {
  330. return err
  331. }
  332. return splitJoinWhere(ts, t.r, args, plan)
  333. case *orExpr, *notExpr:
  334. // 结构条件: 整棵子树只涉及一张表仍可下推该表, 跨表则留作残余谓词
  335. owners := map[*joinTable]bool{}
  336. if err := exprOwners(ts, e, owners); err != nil {
  337. return err
  338. }
  339. if len(owners) == 1 {
  340. for jt := range owners {
  341. return pushJoinWhere(jt, e, args, plan)
  342. }
  343. }
  344. plan.residual = andResidual(plan.residual, e)
  345. return nil
  346. case *colCmpExpr:
  347. // 列与列比较: 引擎 Condition 只支持列与值, 一律留作残余 (列归属校验在此做)
  348. if _, _, err := ownerOfCol(ts, t.left); err != nil {
  349. return err
  350. }
  351. if _, _, err := ownerOfCol(ts, t.right); err != nil {
  352. return err
  353. }
  354. plan.residual = andResidual(plan.residual, e)
  355. return nil
  356. case *cmpExpr:
  357. return pushJoinLeaf(ts, t.col, e, args, plan)
  358. case *inExpr:
  359. return pushJoinLeaf(ts, t.col, e, args, plan)
  360. case *isNullExpr:
  361. return pushJoinLeaf(ts, t.col, e, args, plan)
  362. case *likeExpr:
  363. return pushJoinLeaf(ts, t.col, e, args, plan)
  364. }
  365. return fmt.Errorf("无法拆分的 WHERE 表达式")
  366. }
  367. // pushJoinLeaf 叶子条件按列归属下推到对应表。
  368. func pushJoinLeaf(ts []*joinTable, ref string, e expr, args []driver.NamedValue, plan *joinWherePlan) error {
  369. jt, _, err := ownerOfCol(ts, ref)
  370. if err != nil {
  371. return err
  372. }
  373. return pushJoinWhere(jt, e, args, plan)
  374. }
  375. // pushJoinWhere 把表达式翻译后 AND 进对应表的下推条件 (先剥限定词再走标准翻译)。
  376. func pushJoinWhere(jt *joinTable, e expr, args []driver.NamedValue, plan *joinWherePlan) error {
  377. dq, err := dequalifyExpr(e, jt.alias, jt.name)
  378. if err != nil {
  379. return err
  380. }
  381. cond, err := translateWhere(jt.tbl, dq, args)
  382. if err != nil {
  383. return err
  384. }
  385. if cond == nil {
  386. return nil
  387. }
  388. if plan.perTable[jt.idx] == nil {
  389. plan.perTable[jt.idx] = cond
  390. } else {
  391. plan.perTable[jt.idx] = core.And(plan.perTable[jt.idx], cond)
  392. }
  393. return nil
  394. }
  395. // andResidual 把不可下推的条件用 AND 串起来。
  396. func andResidual(a, b expr) expr {
  397. if a == nil {
  398. return b
  399. }
  400. return &andExpr{l: a, r: b}
  401. }
  402. // exprOwners 收集表达式引用的全部表 (按指针去重); 列归属解析错误直接透出。
  403. func exprOwners(ts []*joinTable, e expr, owners map[*joinTable]bool) error {
  404. mark := func(ref string) error {
  405. jt, _, err := ownerOfCol(ts, ref)
  406. if err != nil {
  407. return err
  408. }
  409. owners[jt] = true
  410. return nil
  411. }
  412. switch t := e.(type) {
  413. case *andExpr:
  414. if err := exprOwners(ts, t.l, owners); err != nil {
  415. return err
  416. }
  417. return exprOwners(ts, t.r, owners)
  418. case *orExpr:
  419. if err := exprOwners(ts, t.l, owners); err != nil {
  420. return err
  421. }
  422. return exprOwners(ts, t.r, owners)
  423. case *notExpr:
  424. return exprOwners(ts, t.e, owners)
  425. case *cmpExpr:
  426. return mark(t.col)
  427. case *colCmpExpr:
  428. if err := mark(t.left); err != nil {
  429. return err
  430. }
  431. return mark(t.right)
  432. case *inExpr:
  433. return mark(t.col)
  434. case *isNullExpr:
  435. return mark(t.col)
  436. case *likeExpr:
  437. return mark(t.col)
  438. }
  439. return nil
  440. }
  441. // =============================================================================
  442. // 残余谓词求值: 组合行上的 Go 值比较
  443. // =============================================================================
  444. // evalJoinExpr 在组合行上求值残余谓词 (跨表 WHERE 条件)。
  445. func evalJoinExpr(e expr, ts []*joinTable, r joinRow, args []driver.NamedValue) (bool, error) {
  446. switch t := e.(type) {
  447. case *andExpr:
  448. l, err := evalJoinExpr(t.l, ts, r, args)
  449. if err != nil || !l {
  450. return l, err
  451. }
  452. return evalJoinExpr(t.r, ts, r, args)
  453. case *orExpr:
  454. l, err := evalJoinExpr(t.l, ts, r, args)
  455. if err != nil || l {
  456. return l, err
  457. }
  458. return evalJoinExpr(t.r, ts, r, args)
  459. case *notExpr:
  460. v, err := evalJoinExpr(t.e, ts, r, args)
  461. if err != nil {
  462. return false, err
  463. }
  464. return !v, nil
  465. case *cmpExpr:
  466. jt, col, err := ownerOfCol(ts, t.col)
  467. if err != nil {
  468. return false, err
  469. }
  470. fi := findField(jt.tbl, col)
  471. raw, err := resolveValue(fi, t.val, args)
  472. if err != nil {
  473. return false, err
  474. }
  475. rv, err := fieldValueToDriverTyped(fi.FieldType, raw)
  476. if err != nil {
  477. return false, err
  478. }
  479. return compareOp(r.get(jt, col), rv, t.op), nil
  480. case *colCmpExpr:
  481. ljt, lcol, err := ownerOfCol(ts, t.left)
  482. if err != nil {
  483. return false, err
  484. }
  485. rjt, rcol, err := ownerOfCol(ts, t.right)
  486. if err != nil {
  487. return false, err
  488. }
  489. return compareOp(r.get(ljt, lcol), r.get(rjt, rcol), t.op), nil
  490. case *inExpr:
  491. jt, col, err := ownerOfCol(ts, t.col)
  492. if err != nil {
  493. return false, err
  494. }
  495. fi := findField(jt.tbl, col)
  496. lv := r.get(jt, col)
  497. hit := false
  498. for _, tv := range t.vals {
  499. raw, err := resolveValue(fi, tv, args)
  500. if err != nil {
  501. return false, err
  502. }
  503. ev, err := fieldValueToDriverTyped(fi.FieldType, raw)
  504. if err != nil {
  505. return false, err
  506. }
  507. if c, ok := compareDriverValues(lv, ev); ok && c == 0 {
  508. hit = true
  509. break
  510. }
  511. }
  512. return hit != t.neg, nil
  513. case *isNullExpr:
  514. jt, col, err := ownerOfCol(ts, t.col)
  515. if err != nil {
  516. return false, err
  517. }
  518. return (r.get(jt, col) == nil) != t.neg, nil
  519. case *likeExpr:
  520. jt, col, err := ownerOfCol(ts, t.col)
  521. if err != nil {
  522. return false, err
  523. }
  524. pat, err := resolveLikePattern(t.pattern, args)
  525. if err != nil {
  526. return false, err
  527. }
  528. s, ok := toCmpString(r.get(jt, col))
  529. if !ok {
  530. // NULL / 非文本: LIKE 不匹配 (NOT LIKE 相应为真)
  531. return t.neg, nil
  532. }
  533. return likeMatch(s, pat) != t.neg, nil
  534. }
  535. return false, fmt.Errorf("无法求值的 WHERE 表达式")
  536. }
  537. // compareOp 按比较符求值; 类型不可比视为不匹配。
  538. func compareOp(l, r driver.Value, op string) bool {
  539. c, ok := compareDriverValues(l, r)
  540. if !ok {
  541. return false
  542. }
  543. switch op {
  544. case "=":
  545. return c == 0
  546. case "!=":
  547. return c != 0
  548. case "<":
  549. return c < 0
  550. case "<=":
  551. return c <= 0
  552. case ">":
  553. return c > 0
  554. case ">=":
  555. return c >= 0
  556. }
  557. return false
  558. }
  559. // compareDriverValues 归一化比较两个 driver.Value, 返回 -1/0/1; 类型不可比返回
  560. // ok=false。NULL 视为最小 (两 NULL 相等); 数值按数值比, 文本按字典序比。
  561. func compareDriverValues(a, b driver.Value) (int, bool) {
  562. if a == nil || b == nil {
  563. switch {
  564. case a == nil && b == nil:
  565. return 0, true
  566. case a == nil:
  567. return -1, true
  568. default:
  569. return 1, true
  570. }
  571. }
  572. if av, ok := toCmpFloat(a); ok {
  573. bv, ok := toCmpFloat(b)
  574. if !ok {
  575. return 0, false
  576. }
  577. switch {
  578. case av < bv:
  579. return -1, true
  580. case av > bv:
  581. return 1, true
  582. default:
  583. return 0, true
  584. }
  585. }
  586. if av, ok := toCmpString(a); ok {
  587. bv, ok := toCmpString(b)
  588. if !ok {
  589. return 0, false
  590. }
  591. return strings.Compare(av, bv), true
  592. }
  593. if av, ok := a.(time.Time); ok {
  594. bv, ok := b.(time.Time)
  595. if !ok {
  596. return 0, false
  597. }
  598. switch {
  599. case av.Before(bv):
  600. return -1, true
  601. case av.After(bv):
  602. return 1, true
  603. default:
  604. return 0, true
  605. }
  606. }
  607. return 0, false
  608. }
  609. // toCmpFloat 数值型 driver.Value → float64 (bool 按 0/1)。
  610. func toCmpFloat(v driver.Value) (float64, bool) {
  611. switch t := v.(type) {
  612. case int64:
  613. return float64(t), true
  614. case float64:
  615. return t, true
  616. case bool:
  617. if t {
  618. return 1, true
  619. }
  620. return 0, true
  621. }
  622. return 0, false
  623. }
  624. // toCmpString 文本型 driver.Value → string。
  625. func toCmpString(v driver.Value) (string, bool) {
  626. switch t := v.(type) {
  627. case string:
  628. return t, true
  629. case []byte:
  630. return string(t), true
  631. }
  632. return "", false
  633. }
  634. // likeMatch SQL LIKE 匹配: % 任意长度任意字符, _ 单字符, 其余原样。
  635. func likeMatch(s, pat string) bool {
  636. sr, pr := []rune(s), []rune(pat)
  637. si, pi := 0, 0
  638. starP, starS := -1, 0
  639. for si < len(sr) {
  640. switch {
  641. case pi < len(pr) && (pr[pi] == '_' || pr[pi] == sr[si]):
  642. si++
  643. pi++
  644. case pi < len(pr) && pr[pi] == '%':
  645. starP, starS = pi, si
  646. pi++
  647. case starP != -1:
  648. pi = starP + 1
  649. starS++
  650. si = starS
  651. default:
  652. return false
  653. }
  654. }
  655. for pi < len(pr) && pr[pi] == '%' {
  656. pi++
  657. }
  658. return pi == len(pr)
  659. }
  660. // =============================================================================
  661. // ON 解析 / 投影 / 排序
  662. // =============================================================================
  663. // resolvedJoinOn 定好左右的 ON 条件: right 指向本 JOIN 的新表, left 指向其左侧
  664. // 已连接的表 (允许 "b.y = a.x" 写法, 解析期交换回来)。
  665. type resolvedJoinOn struct {
  666. leftIdx int
  667. leftCol string
  668. rightIdx int
  669. rightCol string
  670. }
  671. func resolveJoinOn(ts []*joinTable, newIdx int, jc *joinClause) (resolvedJoinOn, error) {
  672. li, lcol, err := qualIndexCol(ts, jc.onL)
  673. if err != nil {
  674. return resolvedJoinOn{}, err
  675. }
  676. ri, rcol, err := qualIndexCol(ts, jc.onR)
  677. if err != nil {
  678. return resolvedJoinOn{}, err
  679. }
  680. if li == newIdx && ri < newIdx {
  681. li, lcol, ri, rcol = ri, rcol, li, lcol
  682. }
  683. if ri != newIdx || li >= newIdx {
  684. return resolvedJoinOn{}, fmt.Errorf("JOIN ON 条件 %q = %q 必须连接新表 %q 与其左侧的表", jc.onL, jc.onR, ts[newIdx].alias)
  685. }
  686. return resolvedJoinOn{leftIdx: li, leftCol: lcol, rightIdx: ri, rightCol: rcol}, nil
  687. }
  688. // qualIndexCol 解析 ON 里的限定列名为 (表下标, 列名); 必须是限定名且列存在。
  689. func qualIndexCol(ts []*joinTable, ref string) (int, string, error) {
  690. qual, col := splitQualCol(ref)
  691. if qual == "" {
  692. return -1, "", fmt.Errorf("JOIN ON 列 %q 必须是限定名 (别名.列)", ref)
  693. }
  694. for i, jt := range ts {
  695. if strings.EqualFold(jt.alias, qual) || strings.EqualFold(jt.name, qual) {
  696. if findField(jt.tbl, col) == nil {
  697. return -1, "", fmt.Errorf("%w: %q", core.ErrFieldNotFound, ref)
  698. }
  699. return i, col, nil
  700. }
  701. }
  702. return -1, "", fmt.Errorf("未知的表限定 %q (列 %q)", qual, ref)
  703. }
  704. // joinProjCol 投影列: 归属表 + 小写列名 (行内取值) + 列元信息 + 展示名 (已去重)。
  705. type joinProjCol struct {
  706. jt *joinTable
  707. col string
  708. fi *types.FieldInfo
  709. name string
  710. }
  711. // buildJoinProjection 展开 SELECT 清单 (*/t.*/t.col/col 混用); 展示名为列名,
  712. // 重名列用 "别名_列名" 去重。
  713. func buildJoinProjection(ts []*joinTable, columns []string) ([]joinProjCol, error) {
  714. var out []joinProjCol
  715. seen := map[string]bool{}
  716. add := func(jt *joinTable, col string) error {
  717. fi := findField(jt.tbl, col)
  718. if fi == nil {
  719. return fmt.Errorf("%w: %q", core.ErrFieldNotFound, col)
  720. }
  721. name := string(fi.FieldName)
  722. if seen[strings.ToLower(name)] {
  723. name = jt.alias + "_" + name
  724. for seen[strings.ToLower(name)] {
  725. name += "_"
  726. }
  727. }
  728. seen[strings.ToLower(name)] = true
  729. out = append(out, joinProjCol{jt: jt, col: strings.ToLower(col), fi: fi, name: name})
  730. return nil
  731. }
  732. expand := func(jt *joinTable) error {
  733. for _, cn := range jt.cols {
  734. if err := add(jt, cn); err != nil {
  735. return err
  736. }
  737. }
  738. return nil
  739. }
  740. expandAll := func() error {
  741. for _, jt := range ts {
  742. if err := expand(jt); err != nil {
  743. return err
  744. }
  745. }
  746. return nil
  747. }
  748. if len(columns) == 0 {
  749. return out, expandAll()
  750. }
  751. for _, c := range columns {
  752. switch {
  753. case c == "*":
  754. if err := expandAll(); err != nil {
  755. return nil, err
  756. }
  757. case strings.HasSuffix(c, ".*"):
  758. jt := findQual(ts, c[:len(c)-2])
  759. if jt == nil {
  760. return nil, fmt.Errorf("未知的表限定 %q (列 %q)", c[:len(c)-2], c)
  761. }
  762. if err := expand(jt); err != nil {
  763. return nil, err
  764. }
  765. default:
  766. jt, col, err := ownerOfCol(ts, c) // 限定名/非限定名统一走归属解析
  767. if err != nil {
  768. return nil, err
  769. }
  770. if err := add(jt, col); err != nil {
  771. return nil, err
  772. }
  773. }
  774. }
  775. return out, nil
  776. }
  777. // joinOrderKey 排序键: 归属表 + 列 + 方向。
  778. type joinOrderKey struct {
  779. jt *joinTable
  780. col string
  781. desc bool
  782. }
  783. // resolveJoinOrderKeys 解析 ORDER BY 键 (限定名直查, 非限定名按唯一归属)。
  784. func resolveJoinOrderKeys(ts []*joinTable, specs []orderSpec) ([]joinOrderKey, error) {
  785. keys := make([]joinOrderKey, len(specs))
  786. for i, o := range specs {
  787. jt, col, err := ownerOfCol(ts, o.col)
  788. if err != nil {
  789. return nil, err
  790. }
  791. keys[i] = joinOrderKey{jt: jt, col: strings.ToLower(col), desc: o.desc}
  792. }
  793. return keys, nil
  794. }
  795. // =============================================================================
  796. // queryJoin: JOIN 查询主流程
  797. // =============================================================================
  798. func queryJoin(db *core.DB, sel *selectStmt, args []driver.NamedValue, release func()) (driver.Rows, error) {
  799. if release != nil {
  800. defer release() // 物化结果集, 算完即放锁
  801. }
  802. if sel.agg != nil {
  803. return nil, fmt.Errorf("JOIN 与聚合函数暂不支持同时使用")
  804. }
  805. // a. 解析所有表并归一化别名 (别名冲突提前报错)
  806. refs := make([]tableRef, 0, len(sel.joins)+1)
  807. refs = append(refs, sel.from)
  808. for _, j := range sel.joins {
  809. refs = append(refs, j.table)
  810. }
  811. ts := make([]*joinTable, len(refs))
  812. seenAlias := map[string]bool{}
  813. for i, ref := range refs {
  814. tbl, err := findTable(db, ref.name)
  815. if err != nil {
  816. return nil, err
  817. }
  818. ts[i] = buildJoinTable(i, ref, tbl)
  819. key := strings.ToLower(ts[i].alias)
  820. if seenAlias[key] {
  821. return nil, fmt.Errorf("JOIN 表别名 %q 重复", ts[i].alias)
  822. }
  823. seenAlias[key] = true
  824. }
  825. // b. WHERE 按归属拆分: 单表条件下推, 跨表条件留作残余谓词
  826. plan, err := planJoinWhere(ts, sel.where, args)
  827. if err != nil {
  828. return nil, err
  829. }
  830. // c. 左表扫描 (本表条件下推), 每行包一层别名空间
  831. leftRows, err := scanJoinTable(ts[0], plan.perTable[0])
  832. if err != nil {
  833. return nil, err
  834. }
  835. rows := make([]joinRow, 0, len(leftRows))
  836. for _, m := range leftRows {
  837. rows = append(rows, joinRow{strings.ToLower(ts[0].alias): m})
  838. }
  839. // d. 逐 JOIN: 左行 ON 列值 → 右表索引点查 (+右表条件下推), 组合笛卡尔行
  840. for i := range sel.joins {
  841. jc := &sel.joins[i]
  842. jt := ts[i+1]
  843. on, err := resolveJoinOn(ts, i+1, jc)
  844. if err != nil {
  845. return nil, err
  846. }
  847. onFi := findField(jt.tbl, on.rightCol)
  848. next := make([]joinRow, 0, len(rows))
  849. for _, lr := range rows {
  850. onVal := lr.get(ts[on.leftIdx], on.leftCol)
  851. var matched []map[string]driver.Value
  852. if onVal != nil {
  853. // NULL 与任何值都不相等, 右表点查直接跳过 (INNER 丢行 / LEFT 补 NULL)
  854. ev, err := driverToEngineValue(onFi, onVal)
  855. if err != nil {
  856. return nil, fmt.Errorf("JOIN ON 列 %q: %w", on.rightCol, err)
  857. }
  858. // 引擎条件按精确字节匹配列名, 改用引擎规范名
  859. // (ON 右列经 qualIndexCol 存在性校验, 这里取规范名构造点查条件)。
  860. cond := core.Eq(string(onFi.FieldName), ev, onFi.FieldType)
  861. if plan.perTable[i+1] != nil {
  862. cond = core.And(cond, plan.perTable[i+1])
  863. }
  864. matched, err = scanJoinTable(jt, cond)
  865. if err != nil {
  866. return nil, err
  867. }
  868. }
  869. if len(matched) == 0 {
  870. if jc.left {
  871. nr := cloneJoinRow(lr)
  872. nr[strings.ToLower(jt.alias)] = nullJoinRow(jt)
  873. next = append(next, nr)
  874. }
  875. continue // INNER JOIN: 无匹配丢行
  876. }
  877. for _, rm := range matched {
  878. nr := cloneJoinRow(lr)
  879. nr[strings.ToLower(jt.alias)] = rm
  880. next = append(next, nr)
  881. }
  882. }
  883. rows = next
  884. }
  885. // e. 残余谓词: 组合行上过滤 (跨表比较)
  886. if plan.residual != nil {
  887. kept := make([]joinRow, 0, len(rows))
  888. for _, r := range rows {
  889. ok, err := evalJoinExpr(plan.residual, ts, r, args)
  890. if err != nil {
  891. return nil, err
  892. }
  893. if ok {
  894. kept = append(kept, r)
  895. }
  896. }
  897. rows = kept
  898. }
  899. // f. 投影列清单 (*/t.*/t.col/col 混用, 重名列 别名_列名 去重)
  900. proj, err := buildJoinProjection(ts, sel.columns)
  901. if err != nil {
  902. return nil, err
  903. }
  904. // g. ORDER BY: 投影前对组合行内存稳定排序 (NULL 视为最小), 再 LIMIT/OFFSET
  905. if len(sel.orderBy) > 0 {
  906. keys, err := resolveJoinOrderKeys(ts, sel.orderBy)
  907. if err != nil {
  908. return nil, err
  909. }
  910. sort.SliceStable(rows, func(x, y int) bool {
  911. for _, k := range keys {
  912. c, ok := compareDriverValues(rows[x].get(k.jt, k.col), rows[y].get(k.jt, k.col))
  913. if !ok || c == 0 {
  914. continue
  915. }
  916. if k.desc {
  917. return c > 0
  918. }
  919. return c < 0
  920. }
  921. return false
  922. })
  923. }
  924. if sel.offset != nil {
  925. o, err := resolveLimitValue("OFFSET", sel.offset, args)
  926. if err != nil {
  927. return nil, err
  928. }
  929. if int(o) >= len(rows) {
  930. rows = nil
  931. } else {
  932. rows = rows[int(o):]
  933. }
  934. }
  935. if sel.limit != nil {
  936. l, err := resolveLimitValue("LIMIT", sel.limit, args)
  937. if err != nil {
  938. return nil, err
  939. }
  940. if int(l) < len(rows) {
  941. rows = rows[:int(l)]
  942. }
  943. }
  944. // 物化投影行 (DISTINCT 按投影值去重)
  945. colNames := make([]string, len(proj))
  946. metas := make([]colMeta, len(proj))
  947. for i, pc := range proj {
  948. colNames[i] = pc.name
  949. metas[i] = colMetaFor(pc.fi)
  950. }
  951. data := make([][]driver.Value, 0, len(rows))
  952. var seen map[string]bool
  953. if sel.distinct {
  954. seen = map[string]bool{}
  955. }
  956. for _, r := range rows {
  957. vals := make([]driver.Value, len(proj))
  958. for i, pc := range proj {
  959. vals[i] = r.get(pc.jt, pc.col) // LEFT JOIN 未匹配侧 → NULL
  960. }
  961. if seen != nil {
  962. key := distinctKey(vals)
  963. if seen[key] {
  964. continue
  965. }
  966. seen[key] = true
  967. }
  968. data = append(data, vals)
  969. }
  970. return materialRows(colNames, metas, data), nil
  971. }