value.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834
  1. package migrate
  2. import (
  3. "TsDB2/DB/core/types"
  4. "fmt"
  5. "math"
  6. "math/big"
  7. "strconv"
  8. "strings"
  9. "time"
  10. "unicode/utf8"
  11. )
  12. // ============================ 值转换 ============================
  13. //
  14. // 两端拿到的值形态不同:
  15. // VRTX 侧 (deserializeRow 的产出): uint8 / int16 / int32 / uint64 / float64 / []byte
  16. // SQL 侧 (database/sql 的驱动值): int64 / float64 / []byte / string / nil / time.Time
  17. //
  18. // 中间表示不做统一 (会把值搬两遍), 而是让目标端按自己的列类型把来源值"收编":
  19. // 走到 VRTX 用 coerceForVRTX, 走到 SQL 用 coerceForSQL。
  20. // zeroForVRTX 某种 VRTX 列类型的零值。
  21. //
  22. // VRTX 没有 NULL 这个概念 (变长列的行内引用 PageID==0 就是"空"), 所以 SQL 的 NULL
  23. // 落到这里只能是零值 —— 这是单向的信息损失, 文档里明说。
  24. func zeroForVRTX(ft types.FieldType) any {
  25. switch ft {
  26. case types.FieldType_Uint8:
  27. return uint8(0)
  28. case types.FieldType_Uint16:
  29. return int16(0)
  30. case types.FieldType_Uint32:
  31. return int32(0)
  32. case types.FieldType_Uint64:
  33. return uint64(0)
  34. case types.FieldType_Int8:
  35. return int8(0)
  36. case types.FieldType_Int16:
  37. return int16(0)
  38. case types.FieldType_Int32:
  39. return int32(0)
  40. case types.FieldType_Int64:
  41. return int64(0)
  42. case types.FieldType_Float64:
  43. return float64(0)
  44. case types.FieldType_Float32:
  45. return float32(0)
  46. case types.FieldType_Bool:
  47. return false
  48. case types.FieldType_Decimal:
  49. return types.Decimal{}
  50. case types.FieldType_UUID:
  51. return types.UUID{}
  52. case types.FieldType_Year:
  53. return int16(0)
  54. case types.FieldType_Int128:
  55. return new(big.Int)
  56. default:
  57. return []byte{}
  58. }
  59. }
  60. // coerceForVRTX 把来源值规范成 core.writeFieldValue 认得的确切 Go 类型。
  61. //
  62. // 必须逐一对齐 (uint8→uint8, uint16→int16, uint32→int32, uint64→uint64,
  63. // Int8→int8, Int16→int16, Int32→int32, Int64→int64,
  64. // float64→float64, 定长/变长→[]byte): writeFieldValue 内部是类型断言,
  65. // 给错类型不是"写进去一个错值", 而是直接 panic。
  66. //
  67. // param 是列级参数 (Decimal 的 scale): 定点列必须按它对齐小数位, 否则同一列里
  68. // 混进另一种 scale 的行, B+ 树的字节序比较会静默出错。
  69. //
  70. // nullable 决定 nil 怎么落地: 可空列写 NULL (原样把 nil 交给 core, 由它置 NULL
  71. // 位图的位), 不可空列才退化成零值。
  72. func coerceForVRTX(ft types.FieldType, param byte, v any, col string, strict bool, nullable bool) (any, error) {
  73. if v == nil {
  74. // SQL 侧的 NULL 读出来就是 nil (见 SQL.Scan 的 *any)。可空列必须把它写成
  75. // NULL —— 落成零值等于"把没有值"变成"值是 0 / 空串", 而且不可逆: 导回去
  76. // 分不出这行的 0 原本是 NULL 还是真 0。
  77. if nullable {
  78. return nil, nil
  79. }
  80. return zeroForVRTX(ft), nil
  81. }
  82. switch ft {
  83. case types.FieldType_Uint8:
  84. n, err := asInt64(v)
  85. if err != nil {
  86. return nil, fmt.Errorf("列 %s: %w", col, err)
  87. }
  88. if n < 0 || n > math.MaxUint8 {
  89. if strict {
  90. return nil, fmt.Errorf("列 %s (uint8): 值 %d 超出 0..255", col, n)
  91. }
  92. n &= 0xFF
  93. }
  94. return uint8(n), nil
  95. case types.FieldType_Uint16:
  96. n, err := asInt64(v)
  97. if err != nil {
  98. return nil, fmt.Errorf("列 %s: %w", col, err)
  99. }
  100. // 无符号档盘上存的是 16 位位模式, 读取载体是 int16 (见 codec.DecodeValue 与
  101. // writeFieldValue 里的 value.(int16))。所以"装得下"的判据是这 16 位放得下,
  102. // 而不是"int16 的正数区放得下":
  103. // 0..65535 无符号来源的原样值 (MySQL SMALLINT UNSIGNED 的 33400 就在这一档);
  104. // -32768..-1 回读 / SQLite INTEGER 给出的负位模式 (0x8278 读出来是 -32136)。
  105. // 按 int16 区间判会把 32768..65535 这半个无符号区误判成越界: 位模式明明装得下,
  106. // 报错拦掉的却是完全合法的数据。
  107. if strict && (n < math.MinInt16 || n > math.MaxUint16) {
  108. return nil, fmt.Errorf("列 %s (uint16): 值 %d 超出 16 位位模式可表达的 -32768..65535", col, n)
  109. }
  110. return int16(uint16(n)), nil
  111. case types.FieldType_Uint32:
  112. n, err := asInt64(v)
  113. if err != nil {
  114. return nil, fmt.Errorf("列 %s: %w", col, err)
  115. }
  116. // 同 uint16: 32 位位模式, 载体 int32。0..4294967295 是无符号来源的原样值,
  117. // -2147483648..-1 是回读给出的负位模式。
  118. if strict && (n < math.MinInt32 || n > math.MaxUint32) {
  119. return nil, fmt.Errorf("列 %s (uint32): 值 %d 超出 32 位位模式可表达的 -2147483648..4294967295", col, n)
  120. }
  121. return int32(uint32(n)), nil
  122. case types.FieldType_Uint64:
  123. // 有符号来源按"64 位位模式"原样搬: int64(-1) → uint64(0xFFFF...)。
  124. // 导出回 SQL 时再用 int64(u) 写, 于是 MySQL 的 -1 ↔ VRTX 的 0xFFFF...
  125. // 往返无损 —— VRTX 的 uint64 本身不分正负, 同一个位模式 Go 侧既可当
  126. // uint64 也可当 int64 读, 正是 ORM 让 int64 字段能存负数的原理。
  127. u, err := asUint64(v)
  128. if err != nil {
  129. return nil, fmt.Errorf("列 %s: %w", col, err)
  130. }
  131. return u, nil
  132. case types.FieldType_Int8:
  133. n, err := asInt64(v)
  134. if err != nil {
  135. return nil, fmt.Errorf("列 %s: %w", col, err)
  136. }
  137. if strict && (n < math.MinInt8 || n > math.MaxInt8) {
  138. return nil, fmt.Errorf("列 %s (int8): 值 %d 超出 -128..127", col, n)
  139. }
  140. return int8(n), nil
  141. case types.FieldType_Int16:
  142. n, err := asInt64(v)
  143. if err != nil {
  144. return nil, fmt.Errorf("列 %s: %w", col, err)
  145. }
  146. if strict && (n < math.MinInt16 || n > math.MaxInt16) {
  147. return nil, fmt.Errorf("列 %s (int16): 值 %d 超出 int16", col, n)
  148. }
  149. return int16(n), nil
  150. case types.FieldType_Int32:
  151. n, err := asInt64(v)
  152. if err != nil {
  153. return nil, fmt.Errorf("列 %s: %w", col, err)
  154. }
  155. if strict && (n < math.MinInt32 || n > math.MaxInt32) {
  156. return nil, fmt.Errorf("列 %s (int32): 值 %d 超出 int32", col, n)
  157. }
  158. return int32(n), nil
  159. case types.FieldType_Int64:
  160. n, err := asInt64(v)
  161. if err != nil {
  162. return nil, fmt.Errorf("列 %s: %w", col, err)
  163. }
  164. return n, nil
  165. case types.FieldType_Float64:
  166. f, err := asFloat64(v)
  167. if err != nil {
  168. return nil, fmt.Errorf("列 %s: %w", col, err)
  169. }
  170. return f, nil
  171. // ===== 扩展类型: 一律归到 core 的标准解码载体 =====
  172. // (Bool→bool, Float32→float32, Decimal→types.Decimal, UUID→types.UUID,
  173. // Year→int16, Int128→*big.Int) —— 与 core/types.DecodeValue 反着走一遍。
  174. case types.FieldType_Float32:
  175. return coerceFloat32(v, col, strict)
  176. case types.FieldType_Bool:
  177. n, err := asInt64(v)
  178. if err != nil {
  179. return nil, fmt.Errorf("列 %s: %w", col, err)
  180. }
  181. if strict && n != 0 && n != 1 {
  182. return nil, fmt.Errorf("列 %s (bool): 值 %d 不是 0/1 —— "+
  183. "bool 只认 0/1, 否则等值与索引会给出多套答案", col, n)
  184. }
  185. return n != 0, nil
  186. case types.FieldType_Decimal:
  187. d, err := types.DecimalFromValue(v)
  188. if err != nil {
  189. return nil, fmt.Errorf("列 %s: %w", col, err)
  190. }
  191. if uint16(param) != d.Scale() {
  192. d, err = d.Rescale(uint16(param))
  193. if err != nil {
  194. return nil, fmt.Errorf("列 %s: %w", col, err)
  195. }
  196. }
  197. return d, nil
  198. case types.FieldType_UUID:
  199. u, err := asUUID(v)
  200. if err != nil {
  201. return nil, fmt.Errorf("列 %s: %w", col, err)
  202. }
  203. return u, nil
  204. case types.FieldType_Year:
  205. n, err := asInt64(v)
  206. if err != nil {
  207. return nil, fmt.Errorf("列 %s: %w", col, err)
  208. }
  209. if strict && (n < 0 || n > math.MaxInt16) {
  210. return nil, fmt.Errorf("列 %s (year): 值 %d 超出 0..32767", col, n)
  211. }
  212. return int16(n), nil
  213. case types.FieldType_Int128:
  214. bi, err := asBigInt(v)
  215. if err != nil {
  216. return nil, fmt.Errorf("列 %s: %w", col, err)
  217. }
  218. return bi, nil
  219. case types.FieldType_Blob:
  220. // Blob 列: 中间表示里给的是【内容本身】(VRTX.Scan 已把行内的 8B 引用解成字节),
  221. // 这里原样交给 core —— 行内那 8 字节链引用由 core 的 Table.blobFieldRef 落链页得到
  222. // (见 types/blob.go: "把 io.Reader / []byte 写进 FieldType_Blob 列, 引擎自动走链页")。
  223. //
  224. // 刻意不走下面 default 的定长容量闸门: types.FieldSize(Blob) 只有 8 —— 那是行内引用的
  225. // 宽度, 不是容量。拿它当上限会把任何超过 8 字节的 blob 判成"超过定长列容量"。单值上限
  226. // 由链长决定 (约 4GB), 超了 core 会返 ErrBlobTooLarge。
  227. // 也不补零 / 不裁零: blob 是二进制, 0x00 是内容的一部分。
  228. b, err := asBytes(v)
  229. if err != nil {
  230. return nil, fmt.Errorf("列 %s: %w", col, err)
  231. }
  232. return b, nil
  233. default:
  234. // 定长字符串 / 变长列
  235. b, err := asBytes(v)
  236. if err != nil {
  237. return nil, fmt.Errorf("列 %s: %w", col, err)
  238. }
  239. if !types.IsDynamic(ft) {
  240. // 容量取列的真实宽度 (宽度码), 不是类型默认档: 对声明 1024 字节的列按 256
  241. // 判断会把合法值判成超长。
  242. size := int(types.FieldSize(ft, param))
  243. if len(b) > size {
  244. if strict {
  245. return nil, fmt.Errorf("列 %s (%s): 值 %d 字节超过定长列容量 %d 字节, "+
  246. "导入会截断 —— 请把该列改成 dynamic 或放宽源值", col, TypeName(ft), len(b), size)
  247. }
  248. b = b[:size]
  249. }
  250. }
  251. return b, nil
  252. }
  253. }
  254. // coerceFloat32 收编到单精度。来源已是 float32 时原样; float64 只在 strict 下要求
  255. // 无损 —— 让"迁移会丢精度"这件事在严格档下当场暴露, 而不是事后对不上数。
  256. func coerceFloat32(v any, col string, strict bool) (float32, error) {
  257. if f32, ok := v.(float32); ok {
  258. return f32, nil
  259. }
  260. f, err := asFloat64(v)
  261. if err != nil {
  262. return 0, fmt.Errorf("列 %s: %w", col, err)
  263. }
  264. fb := float32(f)
  265. if strict && float64(fb) != f {
  266. return 0, fmt.Errorf("列 %s (float32): 值 %v 用 float32 表示会损失精度", col, f)
  267. }
  268. return fb, nil
  269. }
  270. // asUUID 把来源值收成 UUID: UUID 值 / 16 字节 / 文本 (带或不带连字符)。
  271. func asUUID(v any) (types.UUID, error) {
  272. var u types.UUID
  273. switch x := v.(type) {
  274. case types.UUID:
  275. return x, nil
  276. case []byte:
  277. if len(x) != types.UUIDSize {
  278. return u, fmt.Errorf("UUID 需要 %d 字节, 得到 %d 字节", types.UUIDSize, len(x))
  279. }
  280. copy(u[:], x)
  281. return u, nil
  282. case string:
  283. return types.ParseUUID(x)
  284. default:
  285. return u, fmt.Errorf("无法把 %T 当 UUID", v)
  286. }
  287. }
  288. // asBigInt 把来源值收成 *big.Int (128 位整数的标准载体)。
  289. func asBigInt(v any) (*big.Int, error) {
  290. switch x := v.(type) {
  291. case *big.Int:
  292. if x == nil {
  293. return new(big.Int), nil
  294. }
  295. return new(big.Int).Set(x), nil
  296. case big.Int:
  297. return new(big.Int).Set(&x), nil
  298. case types.Decimal:
  299. if x.Scale() != 0 {
  300. return nil, fmt.Errorf("int128 列不能接收带小数的值 %s", x.String())
  301. }
  302. return x.Coefficient(), nil
  303. case []byte:
  304. return parseInt128(string(x))
  305. case string:
  306. return parseInt128(x)
  307. case uint64:
  308. return new(big.Int).SetUint64(x), nil
  309. case int64:
  310. return big.NewInt(x), nil
  311. default:
  312. n, err := asInt64(v)
  313. if err != nil {
  314. return nil, err
  315. }
  316. return big.NewInt(n), nil
  317. }
  318. }
  319. // parseInt128 解析十进制/十六进制文本的 128 位整数 (base 0 支持 0x 前缀)。
  320. func parseInt128(s string) (*big.Int, error) {
  321. s = strings.TrimSpace(s)
  322. bi, ok := new(big.Int).SetString(s, 0)
  323. if !ok {
  324. return nil, fmt.Errorf("无法把 %q 当 128 位整数", s)
  325. }
  326. return bi, nil
  327. }
  328. // coerceForSQL 把来源值规范成 database/sql 能绑定的类型。
  329. //
  330. // 三个约定:
  331. // - uint64 必须按方言选 Go 类型 (mysql 参数), 因为两个方言的声明本就不同,
  332. // 绑定类型得跟着走 —— 见下面 FieldType_Uint64 分支。
  333. // - 字节类列 (定长串 / 变长) 若是合法 UTF-8 就绑成 string, 否则绑 []byte。
  334. // 前者让 SQLite 把它存成 TEXT (BLOB 亲和性的列不会自动转换),
  335. // 后者保证二进制列不被改坏。定长列导出时去掉尾部补零。
  336. // - 时间列 (TimeKind 非空) 一律还原成时间文本, 不写纳秒整数。
  337. //
  338. // mysql 来自目标方言 (Sink), 与 sqlTypeName 的第二个参数同源。
  339. func coerceForSQL(c Column, v any, mysql bool) (any, error) {
  340. if v == nil {
  341. // 可空列: 原样写 NULL。以前所有列都不可空, 这里一路把 nil 落成零值 ——
  342. // 在可空列上那是【静默改数据】: VRTX 的 NULL 到了 SQL 变成 0 / 空串 /
  343. // 1970-01-01, 而且不可逆, 再导回来分不出它原本是 NULL。
  344. if c.Nullable {
  345. return nil, nil
  346. }
  347. if c.IsDynamic() {
  348. return []byte{}, nil
  349. }
  350. if c.IsTime() {
  351. // 时间列的零值给 Unix 0, 不是数字 0 —— 后者塞进 DATETIME 列不是
  352. // "1970" 而是让 MySQL 报错 (严格模式) 或存成 0000-00-00。
  353. return formatSQLTime(0, c.TimeKind), nil
  354. }
  355. return zeroForSQL(c), nil
  356. }
  357. switch c.Type {
  358. case types.FieldType_Uint8:
  359. n, err := asInt64(v)
  360. if err != nil {
  361. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  362. }
  363. return int64(uint8(n)), nil
  364. case types.FieldType_Uint16:
  365. n, err := asInt64(v)
  366. if err != nil {
  367. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  368. }
  369. // 出库一侧给的是【无符号数值】, 不是 int64 原样 —— 与 uint8 那条同形。
  370. // 位模式 0x8278 读出来是 -32136, 原样绑 int64 送进 MySQL 的 SMALLINT UNSIGNED
  371. // (sqlTypeName 就是这么声明的) 会被判成 Out of range: 严格模式直接报错, 非严格
  372. // 模式截成 0 —— 那是静默改数据。取回无符号值 33400 才是这一列真实存的东西。
  373. // SQLite 侧声明是 INTEGER, 没有无符号概念, 但 0..65535 它整个装得下, 同样给
  374. // 无符号值即可 (只有 uint64 才必须按位模式降级, 因为 SQLite 的 INTEGER 装不下
  375. // 2^63 以上)。
  376. return int64(uint16(n)), nil
  377. case types.FieldType_Uint32:
  378. n, err := asInt64(v)
  379. if err != nil {
  380. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  381. }
  382. // 同 uint16: 给无符号数值, 别把 4294967295 当成 -1 送给 MySQL 的 INT UNSIGNED。
  383. return int64(uint32(n)), nil
  384. case types.FieldType_Uint64:
  385. u, err := asUint64(v)
  386. if err != nil {
  387. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  388. }
  389. if c.IsTime() {
  390. // 时间列: 盘上是 Unix 纳秒, 回 SQL 得还原成时间文本 —— 否则 MySQL 的
  391. // DATETIME 列会收到一个 1.7e18 的整数 (驱动不会替我们转)。
  392. return formatSQLTime(int64(u), c.TimeKind), nil
  393. }
  394. // 绑定的 Go 类型必须跟目标列的声明一致, 而两个方言的声明不同:
  395. // - MySQL: sqlTypeName 声明成 BIGINT UNSIGNED (uint64 是无符号档),
  396. // 所以绑 uint64 —— 整个 2^64 区间都写得进去。以前一律绑 int64(u),
  397. // 高位为 1 的值 (包括 VRTX 从 int64 负值搬过来的位模式) 在严格模式下
  398. // 会被 MySQL 判成 "Out of range value" 直接报错, 非严格模式则被静默改写。
  399. // - SQLite: 只有有符号的 INTEGER, 绑 int64 —— 位模式原样存进 64 位整数,
  400. // 读回来还是同一个 8 字节。
  401. // 两条路共同的保证是【位模式不变】, 这正是 VRTX uint64 的语义 (不分正负)。
  402. if mysql {
  403. return u, nil
  404. }
  405. return int64(u), nil
  406. case types.FieldType_Int8, types.FieldType_Int16, types.FieldType_Int32, types.FieldType_Int64:
  407. // 有符号档存的就是数值本身 (盘上的保序编码由 core 负责), 直接绑 int64
  408. n, err := asInt64(v)
  409. if err != nil {
  410. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  411. }
  412. return n, nil
  413. case types.FieldType_Float64:
  414. f, err := asFloat64(v)
  415. if err != nil {
  416. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  417. }
  418. return f, nil
  419. // ===== 扩展类型 =====
  420. // Decimal / Int128 一律绑文本: 目标列分别是 DECIMAL(38,s)/VARCHAR(40) (MySQL) 与
  421. // TEXT (SQLite), 文本既能精确表达 128 位整数, 也不会被驱动转成浮点。
  422. // UUID 绑 16 字节 (BINARY(16)/BLOB)。Bool 绑 bool, Year 绑 int64。
  423. case types.FieldType_Float32:
  424. f, err := asFloat64(v)
  425. if err != nil {
  426. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  427. }
  428. return float32(f), nil
  429. case types.FieldType_Bool:
  430. n, err := asInt64(v)
  431. if err != nil {
  432. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  433. }
  434. return n != 0, nil
  435. case types.FieldType_Decimal:
  436. d, err := types.DecimalFromValue(v)
  437. if err != nil {
  438. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  439. }
  440. if uint16(c.Param) != d.Scale() {
  441. d, err = d.Rescale(uint16(c.Param))
  442. if err != nil {
  443. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  444. }
  445. }
  446. return d.String(), nil
  447. case types.FieldType_UUID:
  448. u, err := asUUID(v)
  449. if err != nil {
  450. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  451. }
  452. return append([]byte(nil), u[:]...), nil
  453. case types.FieldType_Year:
  454. n, err := asInt64(v)
  455. if err != nil {
  456. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  457. }
  458. return n, nil
  459. case types.FieldType_Int128:
  460. bi, err := asBigInt(v)
  461. if err != nil {
  462. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  463. }
  464. return bi.String(), nil
  465. case types.FieldType_Blob:
  466. // Blob 列: 中间表示里已是内容本身 (VRTX.Scan 解过引用), 原样绑给 SQL。
  467. //
  468. // 不做 UTF-8 判定也不去尾零 —— 那是"字符串列"的规矩 (见 default 分支)。blob 是二进制,
  469. // 把末尾的 0x00 当补位裁掉就是静默改数据; 绑成 string 还会让 SQLite 按 TEXT 亲和性存,
  470. // 真正的二进制或非 UTF-8 内容会被写坏。
  471. b, err := asBytes(v)
  472. if err != nil {
  473. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  474. }
  475. return b, nil
  476. default:
  477. b, err := asBytes(v)
  478. if err != nil {
  479. return nil, fmt.Errorf("列 %s: %w", c.Name, err)
  480. }
  481. if !c.IsDynamic() {
  482. b = trimTrailingNUL(b)
  483. }
  484. if utf8.Valid(b) {
  485. return string(b), nil
  486. }
  487. return b, nil
  488. }
  489. }
  490. func zeroForSQL(c Column) any {
  491. switch c.Type {
  492. case types.FieldType_Float64:
  493. return float64(0)
  494. case types.FieldType_Float32:
  495. return float32(0)
  496. case types.FieldType_Bool:
  497. return false
  498. case types.FieldType_Decimal:
  499. // 零值也要按列 scale 输出定点文本 ("0.00"), 否则 DECIMAL 列会收到 "0"。
  500. return types.NewDecimalInt64(0, uint16(c.Param)).String()
  501. case types.FieldType_UUID:
  502. return make([]byte, types.UUIDSize)
  503. case types.FieldType_Year:
  504. return int64(0)
  505. case types.FieldType_Int128:
  506. return "0"
  507. case types.FieldType_Blob:
  508. // blob 列的零值是"空内容"(长度 0), 不是行内那 8 字节链引用 —— core 收到空 []byte
  509. // 会给出空引用 (不分配链页)。
  510. return []byte{}
  511. case types.FieldType_Uint8, types.FieldType_Uint16, types.FieldType_Uint32, types.FieldType_Uint64,
  512. types.FieldType_Int8, types.FieldType_Int16, types.FieldType_Int32, types.FieldType_Int64:
  513. return int64(0)
  514. default:
  515. return ""
  516. }
  517. }
  518. // trimTrailingNUL 去掉定长列右补的 \x00。
  519. // VRTX 定长字符串是"右补零到固定宽度", SQL 侧不该带上这些补位 (否则 varchar
  520. // 里会多出一堆看不见的字符, 也没法用等值条件查)。
  521. func trimTrailingNUL(b []byte) []byte {
  522. end := len(b)
  523. for end > 0 && b[end-1] == 0 {
  524. end--
  525. }
  526. return b[:end]
  527. }
  528. // ============================ 底层取值 ============================
  529. func asInt64(v any) (int64, error) {
  530. switch x := v.(type) {
  531. case int64:
  532. return x, nil
  533. case int:
  534. return int64(x), nil
  535. case int32:
  536. return int64(x), nil
  537. case int16:
  538. return int64(x), nil
  539. case int8:
  540. return int64(x), nil
  541. case uint64:
  542. if x > math.MaxInt64 {
  543. return 0, fmt.Errorf("值 %d 超出 int64", x)
  544. }
  545. return int64(x), nil
  546. case uint32:
  547. return int64(x), nil
  548. case uint16:
  549. return int64(x), nil
  550. case uint8:
  551. return int64(x), nil
  552. case uint:
  553. return int64(x), nil
  554. case float64:
  555. return roundFloatToInt(x)
  556. case float32:
  557. return roundFloatToInt(float64(x))
  558. case bool:
  559. if x {
  560. return 1, nil
  561. }
  562. return 0, nil
  563. case []byte:
  564. return parseIntOrTime(string(x))
  565. case string:
  566. return parseIntOrTime(x)
  567. case time.Time:
  568. return x.UnixNano(), nil
  569. default:
  570. return 0, fmt.Errorf("无法把 %T 当整数", v)
  571. }
  572. }
  573. func roundFloatToInt(f float64) (int64, error) {
  574. if math.IsNaN(f) || math.IsInf(f, 0) {
  575. return 0, fmt.Errorf("值 %v 不是有限数", f)
  576. }
  577. // 上界必须是 >= 而不是 >: math.MaxInt64 (2^63-1) 在 float64 里存不下, 会被舍入
  578. // 到 2^63 —— 于是 f == 2^63 时 "f > MaxInt64" 是假, 放过去之后 int64(f) 越界,
  579. // Go 对越界的 float→int 转换是未定义结果 (amd64 上给 MinInt64), 静默存成反数。
  580. // 下界不受这个影响: MinInt64 = -2^63 在 float64 里可以精确表示。
  581. if f >= math.MaxInt64 || f < math.MinInt64 {
  582. return 0, fmt.Errorf("值 %v 超出 int64", f)
  583. }
  584. return int64(math.Round(f)), nil
  585. }
  586. // asUint64 与 asInt64 的差别只有一处: 有符号的负值不报错, 而是按位模式搬进 uint64。
  587. func asUint64(v any) (uint64, error) {
  588. switch x := v.(type) {
  589. case uint64:
  590. return x, nil
  591. case uint32:
  592. return uint64(x), nil
  593. case uint16:
  594. return uint64(x), nil
  595. case uint8:
  596. return uint64(x), nil
  597. case uint:
  598. return uint64(x), nil
  599. case int64:
  600. return uint64(x), nil // 位模式, 负数 → 高位全 1
  601. case int:
  602. return uint64(x), nil
  603. case int32:
  604. return uint64(x), nil
  605. case int16:
  606. return uint64(x), nil
  607. case int8:
  608. return uint64(x), nil
  609. case float64:
  610. n, err := roundFloatToInt(x)
  611. if err != nil {
  612. return 0, err
  613. }
  614. return uint64(n), nil
  615. case float32:
  616. n, err := roundFloatToInt(float64(x))
  617. if err != nil {
  618. return 0, err
  619. }
  620. return uint64(n), nil
  621. case bool:
  622. if x {
  623. return 1, nil
  624. }
  625. return 0, nil
  626. case []byte:
  627. return parseUintOrTime(string(x))
  628. case string:
  629. return parseUintOrTime(x)
  630. case time.Time:
  631. return uint64(x.UnixNano()), nil
  632. default:
  633. return 0, fmt.Errorf("无法把 %T 当整数", v)
  634. }
  635. }
  636. // ============================ 时间列 ============================
  637. //
  638. // VRTX 的时间表示只有一种: 8 字节整数 (Unix 纳秒时间戳)。SQL 的时间列迁进来
  639. // 要换算成它, 迁回去要换算回时间文本 —— 这一节就是这两个方向。
  640. // sqlTimeLayouts SQL 时间文本的常见写法。MySQL 的 DATETIME / TIMESTAMP 打出来
  641. // 就是第一个; SQLite 存的是 TEXT, 可能带 T 和时区; DATE 只有日期部分。
  642. var sqlTimeLayouts = []string{
  643. "2006-01-02 15:04:05.999999999",
  644. "2006-01-02 15:04:05",
  645. "2006-01-02T15:04:05.999999999Z07:00",
  646. "2006-01-02T15:04:05",
  647. "2006-01-02",
  648. }
  649. // parseSQLTime 把 SQL 时间文本解析成 Unix 纳秒。
  650. //
  651. // 一律按 UTC 解释 (除非文本自带偏移): MySQL 的 DATETIME 本身不带时区, 若用本地
  652. // 时区去解释, 同一份数据在不同机器上会迁出不同的时间戳 —— 迁移工具不该引入
  653. // 这种漂移。往返一致性优先于"看着像本地时间"。
  654. func parseSQLTime(s string) (int64, bool) {
  655. for _, layout := range sqlTimeLayouts {
  656. if tm, err := time.ParseInLocation(layout, s, time.UTC); err == nil {
  657. return tm.UnixNano(), true
  658. }
  659. }
  660. return 0, false
  661. }
  662. // parseIntOrTime 先按整数读, 读不动再按时间文本读。
  663. //
  664. // 时间列映射成 uint64 后, 值可能是 time.Time (MySQL 驱动开了 parseTime),
  665. // 也可能是文本 (parseTime=false 的默认档, 以及 SQLite 的 DATETIME 列) ——
  666. // 后者必须在这里换算, 否则写进 uint64 列时会被当成"不是数字"直接报错。
  667. func parseIntOrTime(s string) (int64, error) {
  668. s = strings.TrimSpace(s)
  669. if n, err := strconv.ParseInt(s, 10, 64); err == nil {
  670. return n, nil
  671. }
  672. if ns, ok := parseSQLTime(s); ok {
  673. return ns, nil
  674. }
  675. return 0, fmt.Errorf("无法把 %q 当整数 (也不是能识别的时间文本)", s)
  676. }
  677. // parseUintOrTime 同 parseIntOrTime, 只是落在无符号上。
  678. func parseUintOrTime(s string) (uint64, error) {
  679. s = strings.TrimSpace(s)
  680. if n, err := strconv.ParseUint(s, 10, 64); err == nil {
  681. return n, nil
  682. }
  683. if ns, ok := parseSQLTime(s); ok {
  684. return uint64(ns), nil
  685. }
  686. return 0, fmt.Errorf("无法把 %q 当整数 (也不是能识别的时间文本)", s)
  687. }
  688. // formatSQLTime Unix 纳秒 → SQL 时间文本。
  689. //
  690. // DATE 只留日期; DATETIME / TIMESTAMP 留到微秒 (纳秒到微秒是 SQL 时间类型本身
  691. // 的限制, MySQL 最细 DATETIME(6), 不是工具丢的)。
  692. func formatSQLTime(ns int64, kind string) string {
  693. tm := time.Unix(0, ns).UTC()
  694. if kind == "date" {
  695. return tm.Format("2006-01-02")
  696. }
  697. if ns%int64(time.Second) == 0 {
  698. return tm.Format("2006-01-02 15:04:05") // 整秒别拖一串 .000000
  699. }
  700. return tm.Format("2006-01-02 15:04:05.000000")
  701. }
  702. func asFloat64(v any) (float64, error) {
  703. switch x := v.(type) {
  704. case float64:
  705. return x, nil
  706. case float32:
  707. return float64(x), nil
  708. case int64:
  709. return float64(x), nil
  710. case int:
  711. return float64(x), nil
  712. case int32:
  713. return float64(x), nil
  714. case int16:
  715. return float64(x), nil
  716. case int8:
  717. return float64(x), nil
  718. case uint64:
  719. return float64(x), nil
  720. case uint32:
  721. return float64(x), nil
  722. case uint16:
  723. return float64(x), nil
  724. case uint8:
  725. return float64(x), nil
  726. case []byte:
  727. f, err := strconv.ParseFloat(strings.TrimSpace(string(x)), 64)
  728. if err != nil {
  729. return 0, fmt.Errorf("无法把 %q 当浮点", x)
  730. }
  731. return f, nil
  732. case string:
  733. f, err := strconv.ParseFloat(strings.TrimSpace(x), 64)
  734. if err != nil {
  735. return 0, fmt.Errorf("无法把 %q 当浮点", x)
  736. }
  737. return f, nil
  738. default:
  739. return 0, fmt.Errorf("无法把 %T 当浮点", v)
  740. }
  741. }
  742. func asBytes(v any) ([]byte, error) {
  743. switch x := v.(type) {
  744. case []byte:
  745. return x, nil
  746. case string:
  747. return []byte(x), nil
  748. case nil:
  749. return []byte{}, nil
  750. case time.Time:
  751. return []byte(x.Format("2006-01-02 15:04:05.999999999")), nil
  752. case bool:
  753. if x {
  754. return []byte{1}, nil
  755. }
  756. return []byte{0}, nil
  757. case int64:
  758. return []byte(strconv.FormatInt(x, 10)), nil
  759. case int:
  760. return []byte(strconv.Itoa(x)), nil
  761. case int32:
  762. return []byte(strconv.FormatInt(int64(x), 10)), nil
  763. case int16:
  764. return []byte(strconv.FormatInt(int64(x), 10)), nil
  765. case int8:
  766. return []byte(strconv.FormatInt(int64(x), 10)), nil
  767. case uint64:
  768. return []byte(strconv.FormatUint(x, 10)), nil
  769. case uint32:
  770. return []byte(strconv.FormatUint(uint64(x), 10)), nil
  771. case uint16:
  772. return []byte(strconv.FormatUint(uint64(x), 10)), nil
  773. case uint8:
  774. return []byte(strconv.FormatUint(uint64(x), 10)), nil
  775. case float64:
  776. return []byte(strconv.FormatFloat(x, 'g', -1, 64)), nil
  777. case float32:
  778. return []byte(strconv.FormatFloat(float64(x), 'g', -1, 32)), nil
  779. default:
  780. return nil, fmt.Errorf("无法把 %T 当字节串", v)
  781. }
  782. }