osbzr
/
com
项目红包剩余9514，下一贡献奖励5%，即 476个红包
Watch 15
Star 7
Wiki 捐赠
本站源代码
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branch: gamification
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'gamification'
${ noResults }
com/vendor/xorm.io/xorm/dialect_postgres.go

1254 lines
45KB
Raw Permalink Blame History 

  1. // Copyright 2015 The Xorm Authors. All rights reserved.

  2. // Use of this source code is governed by a BSD-style

  3. // license that can be found in the LICENSE file.

  4. 

  5. package xorm

  6. 

  7. import (

  8. 	"errors"

  9. 	"fmt"

  10. 	"net/url"

  11. 	"strconv"

  12. 	"strings"

  13. 

  14. 	"xorm.io/core"

  15. )

  16. 

  17. // from http://www.postgresql.org/docs/current/static/sql-keywords-appendix.html

  18. var (

  19. 	postgresReservedWords = map[string]bool{

  20. 		"A":                                true,

  21. 		"ABORT":                            true,

  22. 		"ABS":                              true,

  23. 		"ABSENT":                           true,

  24. 		"ABSOLUTE":                         true,

  25. 		"ACCESS":                           true,

  26. 		"ACCORDING":                        true,

  27. 		"ACTION":                           true,

  28. 		"ADA":                              true,

  29. 		"ADD":                              true,

  30. 		"ADMIN":                            true,

  31. 		"AFTER":                            true,

  32. 		"AGGREGATE":                        true,

  33. 		"ALL":                              true,

  34. 		"ALLOCATE":                         true,

  35. 		"ALSO":                             true,

  36. 		"ALTER":                            true,

  37. 		"ALWAYS":                           true,

  38. 		"ANALYSE":                          true,

  39. 		"ANALYZE":                          true,

  40. 		"AND":                              true,

  41. 		"ANY":                              true,

  42. 		"ARE":                              true,

  43. 		"ARRAY":                            true,

  44. 		"ARRAY_AGG":                        true,

  45. 		"ARRAY_MAX_CARDINALITY":            true,

  46. 		"AS":                               true,

  47. 		"ASC":                              true,

  48. 		"ASENSITIVE":                       true,

  49. 		"ASSERTION":                        true,

  50. 		"ASSIGNMENT":                       true,

  51. 		"ASYMMETRIC":                       true,

  52. 		"AT":                               true,

  53. 		"ATOMIC":                           true,

  54. 		"ATTRIBUTE":                        true,

  55. 		"ATTRIBUTES":                       true,

  56. 		"AUTHORIZATION":                    true,

  57. 		"AVG":                              true,

  58. 		"BACKWARD":                         true,

  59. 		"BASE64":                           true,

  60. 		"BEFORE":                           true,

  61. 		"BEGIN":                            true,

  62. 		"BEGIN_FRAME":                      true,

  63. 		"BEGIN_PARTITION":                  true,

  64. 		"BERNOULLI":                        true,

  65. 		"BETWEEN":                          true,

  66. 		"BIGINT":                           true,

  67. 		"BINARY":                           true,

  68. 		"BIT":                              true,

  69. 		"BIT_LENGTH":                       true,

  70. 		"BLOB":                             true,

  71. 		"BLOCKED":                          true,

  72. 		"BOM":                              true,

  73. 		"BOOLEAN":                          true,

  74. 		"BOTH":                             true,

  75. 		"BREADTH":                          true,

  76. 		"BY":                               true,

  77. 		"C":                                true,

  78. 		"CACHE":                            true,

  79. 		"CALL":                             true,

  80. 		"CALLED":                           true,

  81. 		"CARDINALITY":                      true,

  82. 		"CASCADE":                          true,

  83. 		"CASCADED":                         true,

  84. 		"CASE":                             true,

  85. 		"CAST":                             true,

  86. 		"CATALOG":                          true,

  87. 		"CATALOG_NAME":                     true,

  88. 		"CEIL":                             true,

  89. 		"CEILING":                          true,

  90. 		"CHAIN":                            true,

  91. 		"CHAR":                             true,

  92. 		"CHARACTER":                        true,

  93. 		"CHARACTERISTICS":                  true,

  94. 		"CHARACTERS":                       true,

  95. 		"CHARACTER_LENGTH":                 true,

  96. 		"CHARACTER_SET_CATALOG":            true,

  97. 		"CHARACTER_SET_NAME":               true,

  98. 		"CHARACTER_SET_SCHEMA":             true,

  99. 		"CHAR_LENGTH":                      true,

  100. 		"CHECK":                            true,

  101. 		"CHECKPOINT":                       true,

  102. 		"CLASS":                            true,

  103. 		"CLASS_ORIGIN":                     true,

  104. 		"CLOB":                             true,

  105. 		"CLOSE":                            true,

  106. 		"CLUSTER":                          true,

  107. 		"COALESCE":                         true,

  108. 		"COBOL":                            true,

  109. 		"COLLATE":                          true,

  110. 		"COLLATION":                        true,

  111. 		"COLLATION_CATALOG":                true,

  112. 		"COLLATION_NAME":                   true,

  113. 		"COLLATION_SCHEMA":                 true,

  114. 		"COLLECT":                          true,

  115. 		"COLUMN":                           true,

  116. 		"COLUMNS":                          true,

  117. 		"COLUMN_NAME":                      true,

  118. 		"COMMAND_FUNCTION":                 true,

  119. 		"COMMAND_FUNCTION_CODE":            true,

  120. 		"COMMENT":                          true,

  121. 		"COMMENTS":                         true,

  122. 		"COMMIT":                           true,

  123. 		"COMMITTED":                        true,

  124. 		"CONCURRENTLY":                     true,

  125. 		"CONDITION":                        true,

  126. 		"CONDITION_NUMBER":                 true,

  127. 		"CONFIGURATION":                    true,

  128. 		"CONNECT":                          true,

  129. 		"CONNECTION":                       true,

  130. 		"CONNECTION_NAME":                  true,

  131. 		"CONSTRAINT":                       true,

  132. 		"CONSTRAINTS":                      true,

  133. 		"CONSTRAINT_CATALOG":               true,

  134. 		"CONSTRAINT_NAME":                  true,

  135. 		"CONSTRAINT_SCHEMA":                true,

  136. 		"CONSTRUCTOR":                      true,

  137. 		"CONTAINS":                         true,

  138. 		"CONTENT":                          true,

  139. 		"CONTINUE":                         true,

  140. 		"CONTROL":                          true,

  141. 		"CONVERSION":                       true,

  142. 		"CONVERT":                          true,

  143. 		"COPY":                             true,

  144. 		"CORR":                             true,

  145. 		"CORRESPONDING":                    true,

  146. 		"COST":                             true,

  147. 		"COUNT":                            true,

  148. 		"COVAR_POP":                        true,

  149. 		"COVAR_SAMP":                       true,

  150. 		"CREATE":                           true,

  151. 		"CROSS":                            true,

  152. 		"CSV":                              true,

  153. 		"CUBE":                             true,

  154. 		"CUME_DIST":                        true,

  155. 		"CURRENT":                          true,

  156. 		"CURRENT_CATALOG":                  true,

  157. 		"CURRENT_DATE":                     true,

  158. 		"CURRENT_DEFAULT_TRANSFORM_GROUP":  true,

  159. 		"CURRENT_PATH":                     true,

  160. 		"CURRENT_ROLE":                     true,

  161. 		"CURRENT_ROW":                      true,

  162. 		"CURRENT_SCHEMA":                   true,

  163. 		"CURRENT_TIME":                     true,

  164. 		"CURRENT_TIMESTAMP":                true,

  165. 		"CURRENT_TRANSFORM_GROUP_FOR_TYPE": true,

  166. 		"CURRENT_USER":                     true,

  167. 		"CURSOR":                           true,

  168. 		"CURSOR_NAME":                      true,

  169. 		"CYCLE":                            true,

  170. 		"DATA":                             true,

  171. 		"DATABASE":                         true,

  172. 		"DATALINK":                         true,

  173. 		"DATE":                             true,

  174. 		"DATETIME_INTERVAL_CODE":           true,

  175. 		"DATETIME_INTERVAL_PRECISION":      true,

  176. 		"DAY":                              true,

  177. 		"DB":                               true,

  178. 		"DEALLOCATE":                       true,

  179. 		"DEC":                              true,

  180. 		"DECIMAL":                          true,

  181. 		"DECLARE":                          true,

  182. 		"DEFAULT":                          true,

  183. 		"DEFAULTS":                         true,

  184. 		"DEFERRABLE":                       true,

  185. 		"DEFERRED":                         true,

  186. 		"DEFINED":                          true,

  187. 		"DEFINER":                          true,

  188. 		"DEGREE":                           true,

  189. 		"DELETE":                           true,

  190. 		"DELIMITER":                        true,

  191. 		"DELIMITERS":                       true,

  192. 		"DENSE_RANK":                       true,

  193. 		"DEPTH":                            true,

  194. 		"DEREF":                            true,

  195. 		"DERIVED":                          true,

  196. 		"DESC":                             true,

  197. 		"DESCRIBE":                         true,

  198. 		"DESCRIPTOR":                       true,

  199. 		"DETERMINISTIC":                    true,

  200. 		"DIAGNOSTICS":                      true,

  201. 		"DICTIONARY":                       true,

  202. 		"DISABLE":                          true,

  203. 		"DISCARD":                          true,

  204. 		"DISCONNECT":                       true,

  205. 		"DISPATCH":                         true,

  206. 		"DISTINCT":                         true,

  207. 		"DLNEWCOPY":                        true,

  208. 		"DLPREVIOUSCOPY":                   true,

  209. 		"DLURLCOMPLETE":                    true,

  210. 		"DLURLCOMPLETEONLY":                true,

  211. 		"DLURLCOMPLETEWRITE":               true,

  212. 		"DLURLPATH":                        true,

  213. 		"DLURLPATHONLY":                    true,

  214. 		"DLURLPATHWRITE":                   true,

  215. 		"DLURLSCHEME":                      true,

  216. 		"DLURLSERVER":                      true,

  217. 		"DLVALUE":                          true,

  218. 		"DO":                               true,

  219. 		"DOCUMENT":                         true,

  220. 		"DOMAIN":                           true,

  221. 		"DOUBLE":                           true,

  222. 		"DROP":                             true,

  223. 		"DYNAMIC":                          true,

  224. 		"DYNAMIC_FUNCTION":                 true,

  225. 		"DYNAMIC_FUNCTION_CODE":            true,

  226. 		"EACH":                             true,

  227. 		"ELEMENT":                          true,

  228. 		"ELSE":                             true,

  229. 		"EMPTY":                            true,

  230. 		"ENABLE":                           true,

  231. 		"ENCODING":                         true,

  232. 		"ENCRYPTED":                        true,

  233. 		"END":                              true,

  234. 		"END-EXEC":                         true,

  235. 		"END_FRAME":                        true,

  236. 		"END_PARTITION":                    true,

  237. 		"ENFORCED":                         true,

  238. 		"ENUM":                             true,

  239. 		"EQUALS":                           true,

  240. 		"ESCAPE":                           true,

  241. 		"EVENT":                            true,

  242. 		"EVERY":                            true,

  243. 		"EXCEPT":                           true,

  244. 		"EXCEPTION":                        true,

  245. 		"EXCLUDE":                          true,

  246. 		"EXCLUDING":                        true,

  247. 		"EXCLUSIVE":                        true,

  248. 		"EXEC":                             true,

  249. 		"EXECUTE":                          true,

  250. 		"EXISTS":                           true,

  251. 		"EXP":                              true,

  252. 		"EXPLAIN":                          true,

  253. 		"EXPRESSION":                       true,

  254. 		"EXTENSION":                        true,

  255. 		"EXTERNAL":                         true,

  256. 		"EXTRACT":                          true,

  257. 		"FALSE":                            true,

  258. 		"FAMILY":                           true,

  259. 		"FETCH":                            true,

  260. 		"FILE":                             true,

  261. 		"FILTER":                           true,

  262. 		"FINAL":                            true,

  263. 		"FIRST":                            true,

  264. 		"FIRST_VALUE":                      true,

  265. 		"FLAG":                             true,

  266. 		"FLOAT":                            true,

  267. 		"FLOOR":                            true,

  268. 		"FOLLOWING":                        true,

  269. 		"FOR":                              true,

  270. 		"FORCE":                            true,

  271. 		"FOREIGN":                          true,

  272. 		"FORTRAN":                          true,

  273. 		"FORWARD":                          true,

  274. 		"FOUND":                            true,

  275. 		"FRAME_ROW":                        true,

  276. 		"FREE":                             true,

  277. 		"FREEZE":                           true,

  278. 		"FROM":                             true,

  279. 		"FS":                               true,

  280. 		"FULL":                             true,

  281. 		"FUNCTION":                         true,

  282. 		"FUNCTIONS":                        true,

  283. 		"FUSION":                           true,

  284. 		"G":                                true,

  285. 		"GENERAL":                          true,

  286. 		"GENERATED":                        true,

  287. 		"GET":                              true,

  288. 		"GLOBAL":                           true,

  289. 		"GO":                               true,

  290. 		"GOTO":                             true,

  291. 		"GRANT":                            true,

  292. 		"GRANTED":                          true,

  293. 		"GREATEST":                         true,

  294. 		"GROUP":                            true,

  295. 		"GROUPING":                         true,

  296. 		"GROUPS":                           true,

  297. 		"HANDLER":                          true,

  298. 		"HAVING":                           true,

  299. 		"HEADER":                           true,

  300. 		"HEX":                              true,

  301. 		"HIERARCHY":                        true,

  302. 		"HOLD":                             true,

  303. 		"HOUR":                             true,

  304. 		"ID":                               true,

  305. 		"IDENTITY":                         true,

  306. 		"IF":                               true,

  307. 		"IGNORE":                           true,

  308. 		"ILIKE":                            true,

  309. 		"IMMEDIATE":                        true,

  310. 		"IMMEDIATELY":                      true,

  311. 		"IMMUTABLE":                        true,

  312. 		"IMPLEMENTATION":                   true,

  313. 		"IMPLICIT":                         true,

  314. 		"IMPORT":                           true,

  315. 		"IN":                               true,

  316. 		"INCLUDING":                        true,

  317. 		"INCREMENT":                        true,

  318. 		"INDENT":                           true,

  319. 		"INDEX":                            true,

  320. 		"INDEXES":                          true,

  321. 		"INDICATOR":                        true,

  322. 		"INHERIT":                          true,

  323. 		"INHERITS":                         true,

  324. 		"INITIALLY":                        true,

  325. 		"INLINE":                           true,

  326. 		"INNER":                            true,

  327. 		"INOUT":                            true,

  328. 		"INPUT":                            true,

  329. 		"INSENSITIVE":                      true,

  330. 		"INSERT":                           true,

  331. 		"INSTANCE":                         true,

  332. 		"INSTANTIABLE":                     true,

  333. 		"INSTEAD":                          true,

  334. 		"INT":                              true,

  335. 		"INTEGER":                          true,

  336. 		"INTEGRITY":                        true,

  337. 		"INTERSECT":                        true,

  338. 		"INTERSECTION":                     true,

  339. 		"INTERVAL":                         true,

  340. 		"INTO":                             true,

  341. 		"INVOKER":                          true,

  342. 		"IS":                               true,

  343. 		"ISNULL":                           true,

  344. 		"ISOLATION":                        true,

  345. 		"JOIN":                             true,

  346. 		"K":                                true,

  347. 		"KEY":                              true,

  348. 		"KEY_MEMBER":                       true,

  349. 		"KEY_TYPE":                         true,

  350. 		"LABEL":                            true,

  351. 		"LAG":                              true,

  352. 		"LANGUAGE":                         true,

  353. 		"LARGE":                            true,

  354. 		"LAST":                             true,

  355. 		"LAST_VALUE":                       true,

  356. 		"LATERAL":                          true,

  357. 		"LC_COLLATE":                       true,

  358. 		"LC_CTYPE":                         true,

  359. 		"LEAD":                             true,

  360. 		"LEADING":                          true,

  361. 		"LEAKPROOF":                        true,

  362. 		"LEAST":                            true,

  363. 		"LEFT":                             true,

  364. 		"LENGTH":                           true,

  365. 		"LEVEL":                            true,

  366. 		"LIBRARY":                          true,

  367. 		"LIKE":                             true,

  368. 		"LIKE_REGEX":                       true,

  369. 		"LIMIT":                            true,

  370. 		"LINK":                             true,

  371. 		"LISTEN":                           true,

  372. 		"LN":                               true,

  373. 		"LOAD":                             true,

  374. 		"LOCAL":                            true,

  375. 		"LOCALTIME":                        true,

  376. 		"LOCALTIMESTAMP":                   true,

  377. 		"LOCATION":                         true,

  378. 		"LOCATOR":                          true,

  379. 		"LOCK":                             true,

  380. 		"LOWER":                            true,

  381. 		"M":                                true,

  382. 		"MAP":                              true,

  383. 		"MAPPING":                          true,

  384. 		"MATCH":                            true,

  385. 		"MATCHED":                          true,

  386. 		"MATERIALIZED":                     true,

  387. 		"MAX":                              true,

  388. 		"MAXVALUE":                         true,

  389. 		"MAX_CARDINALITY":                  true,

  390. 		"MEMBER":                           true,

  391. 		"MERGE":                            true,

  392. 		"MESSAGE_LENGTH":                   true,

  393. 		"MESSAGE_OCTET_LENGTH":             true,

  394. 		"MESSAGE_TEXT":                     true,

  395. 		"METHOD":                           true,

  396. 		"MIN":                              true,

  397. 		"MINUTE":                           true,

  398. 		"MINVALUE":                         true,

  399. 		"MOD":                              true,

  400. 		"MODE":                             true,

  401. 		"MODIFIES":                         true,

  402. 		"MODULE":                           true,

  403. 		"MONTH":                            true,

  404. 		"MORE":                             true,

  405. 		"MOVE":                             true,

  406. 		"MULTISET":                         true,

  407. 		"MUMPS":                            true,

  408. 		"NAME":                             true,

  409. 		"NAMES":                            true,

  410. 		"NAMESPACE":                        true,

  411. 		"NATIONAL":                         true,

  412. 		"NATURAL":                          true,

  413. 		"NCHAR":                            true,

  414. 		"NCLOB":                            true,

  415. 		"NESTING":                          true,

  416. 		"NEW":                              true,

  417. 		"NEXT":                             true,

  418. 		"NFC":                              true,

  419. 		"NFD":                              true,

  420. 		"NFKC":                             true,

  421. 		"NFKD":                             true,

  422. 		"NIL":                              true,

  423. 		"NO":                               true,

  424. 		"NONE":                             true,

  425. 		"NORMALIZE":                        true,

  426. 		"NORMALIZED":                       true,

  427. 		"NOT":                              true,

  428. 		"NOTHING":                          true,

  429. 		"NOTIFY":                           true,

  430. 		"NOTNULL":                          true,

  431. 		"NOWAIT":                           true,

  432. 		"NTH_VALUE":                        true,

  433. 		"NTILE":                            true,

  434. 		"NULL":                             true,

  435. 		"NULLABLE":                         true,

  436. 		"NULLIF":                           true,

  437. 		"NULLS":                            true,

  438. 		"NUMBER":                           true,

  439. 		"NUMERIC":                          true,

  440. 		"OBJECT":                           true,

  441. 		"OCCURRENCES_REGEX":                true,

  442. 		"OCTETS":                           true,

  443. 		"OCTET_LENGTH":                     true,

  444. 		"OF":                               true,

  445. 		"OFF":                              true,

  446. 		"OFFSET":                           true,

  447. 		"OIDS":                             true,

  448. 		"OLD":                              true,

  449. 		"ON":                               true,

  450. 		"ONLY":                             true,

  451. 		"OPEN":                             true,

  452. 		"OPERATOR":                         true,

  453. 		"OPTION":                           true,

  454. 		"OPTIONS":                          true,

  455. 		"OR":                               true,

  456. 		"ORDER":                            true,

  457. 		"ORDERING":                         true,

  458. 		"ORDINALITY":                       true,

  459. 		"OTHERS":                           true,

  460. 		"OUT":                              true,

  461. 		"OUTER":                            true,

  462. 		"OUTPUT":                           true,

  463. 		"OVER":                             true,

  464. 		"OVERLAPS":                         true,

  465. 		"OVERLAY":                          true,

  466. 		"OVERRIDING":                       true,

  467. 		"OWNED":                            true,

  468. 		"OWNER":                            true,

  469. 		"P":                                true,

  470. 		"PAD":                              true,

  471. 		"PARAMETER":                        true,

  472. 		"PARAMETER_MODE":                   true,

  473. 		"PARAMETER_NAME":                   true,

  474. 		"PARAMETER_ORDINAL_POSITION":       true,

  475. 		"PARAMETER_SPECIFIC_CATALOG":       true,

  476. 		"PARAMETER_SPECIFIC_NAME":          true,

  477. 		"PARAMETER_SPECIFIC_SCHEMA":        true,

  478. 		"PARSER":                           true,

  479. 		"PARTIAL":                          true,

  480. 		"PARTITION":                        true,

  481. 		"PASCAL":                           true,

  482. 		"PASSING":                          true,

  483. 		"PASSTHROUGH":                      true,

  484. 		"PASSWORD":                         true,

  485. 		"PATH":                             true,

  486. 		"PERCENT":                          true,

  487. 		"PERCENTILE_CONT":                  true,

  488. 		"PERCENTILE_DISC":                  true,

  489. 		"PERCENT_RANK":                     true,

  490. 		"PERIOD":                           true,

  491. 		"PERMISSION":                       true,

  492. 		"PLACING":                          true,

  493. 		"PLANS":                            true,

  494. 		"PLI":                              true,

  495. 		"PORTION":                          true,

  496. 		"POSITION":                         true,

  497. 		"POSITION_REGEX":                   true,

  498. 		"POWER":                            true,

  499. 		"PRECEDES":                         true,

  500. 		"PRECEDING":                        true,

  501. 		"PRECISION":                        true,

  502. 		"PREPARE":                          true,

  503. 		"PREPARED":                         true,

  504. 		"PRESERVE":                         true,

  505. 		"PRIMARY":                          true,

  506. 		"PRIOR":                            true,

  507. 		"PRIVILEGES":                       true,

  508. 		"PROCEDURAL":                       true,

  509. 		"PROCEDURE":                        true,

  510. 		"PROGRAM":                          true,

  511. 		"PUBLIC":                           true,

  512. 		"QUOTE":                            true,

  513. 		"RANGE":                            true,

  514. 		"RANK":                             true,

  515. 		"READ":                             true,

  516. 		"READS":                            true,

  517. 		"REAL":                             true,

  518. 		"REASSIGN":                         true,

  519. 		"RECHECK":                          true,

  520. 		"RECOVERY":                         true,

  521. 		"RECURSIVE":                        true,

  522. 		"REF":                              true,

  523. 		"REFERENCES":                       true,

  524. 		"REFERENCING":                      true,

  525. 		"REFRESH":                          true,

  526. 		"REGR_AVGX":                        true,

  527. 		"REGR_AVGY":                        true,

  528. 		"REGR_COUNT":                       true,

  529. 		"REGR_INTERCEPT":                   true,

  530. 		"REGR_R2":                          true,

  531. 		"REGR_SLOPE":                       true,

  532. 		"REGR_SXX":                         true,

  533. 		"REGR_SXY":                         true,

  534. 		"REGR_SYY":                         true,

  535. 		"REINDEX":                          true,

  536. 		"RELATIVE":                         true,

  537. 		"RELEASE":                          true,

  538. 		"RENAME":                           true,

  539. 		"REPEATABLE":                       true,

  540. 		"REPLACE":                          true,

  541. 		"REPLICA":                          true,

  542. 		"REQUIRING":                        true,

  543. 		"RESET":                            true,

  544. 		"RESPECT":                          true,

  545. 		"RESTART":                          true,

  546. 		"RESTORE":                          true,

  547. 		"RESTRICT":                         true,

  548. 		"RESULT":                           true,

  549. 		"RETURN":                           true,

  550. 		"RETURNED_CARDINALITY":             true,

  551. 		"RETURNED_LENGTH":                  true,

  552. 		"RETURNED_OCTET_LENGTH":            true,

  553. 		"RETURNED_SQLSTATE":                true,

  554. 		"RETURNING":                        true,

  555. 		"RETURNS":                          true,

  556. 		"REVOKE":                           true,

  557. 		"RIGHT":                            true,

  558. 		"ROLE":                             true,

  559. 		"ROLLBACK":                         true,

  560. 		"ROLLUP":                           true,

  561. 		"ROUTINE":                          true,

  562. 		"ROUTINE_CATALOG":                  true,

  563. 		"ROUTINE_NAME":                     true,

  564. 		"ROUTINE_SCHEMA":                   true,

  565. 		"ROW":                              true,

  566. 		"ROWS":                             true,

  567. 		"ROW_COUNT":                        true,

  568. 		"ROW_NUMBER":                       true,

  569. 		"RULE":                             true,

  570. 		"SAVEPOINT":                        true,

  571. 		"SCALE":                            true,

  572. 		"SCHEMA":                           true,

  573. 		"SCHEMA_NAME":                      true,

  574. 		"SCOPE":                            true,

  575. 		"SCOPE_CATALOG":                    true,

  576. 		"SCOPE_NAME":                       true,

  577. 		"SCOPE_SCHEMA":                     true,

  578. 		"SCROLL":                           true,

  579. 		"SEARCH":                           true,

  580. 		"SECOND":                           true,

  581. 		"SECTION":                          true,

  582. 		"SECURITY":                         true,

  583. 		"SELECT":                           true,

  584. 		"SELECTIVE":                        true,

  585. 		"SELF":                             true,

  586. 		"SENSITIVE":                        true,

  587. 		"SEQUENCE":                         true,

  588. 		"SEQUENCES":                        true,

  589. 		"SERIALIZABLE":                     true,

  590. 		"SERVER":                           true,

  591. 		"SERVER_NAME":                      true,

  592. 		"SESSION":                          true,

  593. 		"SESSION_USER":                     true,

  594. 		"SET":                              true,

  595. 		"SETOF":                            true,

  596. 		"SETS":                             true,

  597. 		"SHARE":                            true,

  598. 		"SHOW":                             true,

  599. 		"SIMILAR":                          true,

  600. 		"SIMPLE":                           true,

  601. 		"SIZE":                             true,

  602. 		"SMALLINT":                         true,

  603. 		"SNAPSHOT":                         true,

  604. 		"SOME":                             true,

  605. 		"SOURCE":                           true,

  606. 		"SPACE":                            true,

  607. 		"SPECIFIC":                         true,

  608. 		"SPECIFICTYPE":                     true,

  609. 		"SPECIFIC_NAME":                    true,

  610. 		"SQL":                              true,

  611. 		"SQLCODE":                          true,

  612. 		"SQLERROR":                         true,

  613. 		"SQLEXCEPTION":                     true,

  614. 		"SQLSTATE":                         true,

  615. 		"SQLWARNING":                       true,

  616. 		"SQRT":                             true,

  617. 		"STABLE":                           true,

  618. 		"STANDALONE":                       true,

  619. 		"START":                            true,

  620. 		"STATE":                            true,

  621. 		"STATEMENT":                        true,

  622. 		"STATIC":                           true,

  623. 		"STATISTICS":                       true,

  624. 		"STDDEV_POP":                       true,

  625. 		"STDDEV_SAMP":                      true,

  626. 		"STDIN":                            true,

  627. 		"STDOUT":                           true,

  628. 		"STORAGE":                          true,

  629. 		"STRICT":                           true,

  630. 		"STRIP":                            true,

  631. 		"STRUCTURE":                        true,

  632. 		"STYLE":                            true,

  633. 		"SUBCLASS_ORIGIN":                  true,

  634. 		"SUBMULTISET":                      true,

  635. 		"SUBSTRING":                        true,

  636. 		"SUBSTRING_REGEX":                  true,

  637. 		"SUCCEEDS":                         true,

  638. 		"SUM":                              true,

  639. 		"SYMMETRIC":                        true,

  640. 		"SYSID":                            true,

  641. 		"SYSTEM":                           true,

  642. 		"SYSTEM_TIME":                      true,

  643. 		"SYSTEM_USER":                      true,

  644. 		"T":                                true,

  645. 		"TABLE":                            true,

  646. 		"TABLES":                           true,

  647. 		"TABLESAMPLE":                      true,

  648. 		"TABLESPACE":                       true,

  649. 		"TABLE_NAME":                       true,

  650. 		"TEMP":                             true,

  651. 		"TEMPLATE":                         true,

  652. 		"TEMPORARY":                        true,

  653. 		"TEXT":                             true,

  654. 		"THEN":                             true,

  655. 		"TIES":                             true,

  656. 		"TIME":                             true,

  657. 		"TIMESTAMP":                        true,

  658. 		"TIMEZONE_HOUR":                    true,

  659. 		"TIMEZONE_MINUTE":                  true,

  660. 		"TO":                               true,

  661. 		"TOKEN":                            true,

  662. 		"TOP_LEVEL_COUNT":                  true,

  663. 		"TRAILING":                         true,

  664. 		"TRANSACTION":                      true,

  665. 		"TRANSACTIONS_COMMITTED":           true,

  666. 		"TRANSACTIONS_ROLLED_BACK":         true,

  667. 		"TRANSACTION_ACTIVE":               true,

  668. 		"TRANSFORM":                        true,

  669. 		"TRANSFORMS":                       true,

  670. 		"TRANSLATE":                        true,

  671. 		"TRANSLATE_REGEX":                  true,

  672. 		"TRANSLATION":                      true,

  673. 		"TREAT":                            true,

  674. 		"TRIGGER":                          true,

  675. 		"TRIGGER_CATALOG":                  true,

  676. 		"TRIGGER_NAME":                     true,

  677. 		"TRIGGER_SCHEMA":                   true,

  678. 		"TRIM":                             true,

  679. 		"TRIM_ARRAY":                       true,

  680. 		"TRUE":                             true,

  681. 		"TRUNCATE":                         true,

  682. 		"TRUSTED":                          true,

  683. 		"TYPE":                             true,

  684. 		"TYPES":                            true,

  685. 		"UESCAPE":                          true,

  686. 		"UNBOUNDED":                        true,

  687. 		"UNCOMMITTED":                      true,

  688. 		"UNDER":                            true,

  689. 		"UNENCRYPTED":                      true,

  690. 		"UNION":                            true,

  691. 		"UNIQUE":                           true,

  692. 		"UNKNOWN":                          true,

  693. 		"UNLINK":                           true,

  694. 		"UNLISTEN":                         true,

  695. 		"UNLOGGED":                         true,

  696. 		"UNNAMED":                          true,

  697. 		"UNNEST":                           true,

  698. 		"UNTIL":                            true,

  699. 		"UNTYPED":                          true,

  700. 		"UPDATE":                           true,

  701. 		"UPPER":                            true,

  702. 		"URI":                              true,

  703. 		"USAGE":                            true,

  704. 		"USER":                             true,

  705. 		"USER_DEFINED_TYPE_CATALOG":        true,

  706. 		"USER_DEFINED_TYPE_CODE":           true,

  707. 		"USER_DEFINED_TYPE_NAME":           true,

  708. 		"USER_DEFINED_TYPE_SCHEMA":         true,

  709. 		"USING":                            true,

  710. 		"VACUUM":                           true,

  711. 		"VALID":                            true,

  712. 		"VALIDATE":                         true,

  713. 		"VALIDATOR":                        true,

  714. 		"VALUE":                            true,

  715. 		"VALUES":                           true,

  716. 		"VALUE_OF":                         true,

  717. 		"VARBINARY":                        true,

  718. 		"VARCHAR":                          true,

  719. 		"VARIADIC":                         true,

  720. 		"VARYING":                          true,

  721. 		"VAR_POP":                          true,

  722. 		"VAR_SAMP":                         true,

  723. 		"VERBOSE":                          true,

  724. 		"VERSION":                          true,

  725. 		"VERSIONING":                       true,

  726. 		"VIEW":                             true,

  727. 		"VOLATILE":                         true,

  728. 		"WHEN":                             true,

  729. 		"WHENEVER":                         true,

  730. 		"WHERE":                            true,

  731. 		"WHITESPACE":                       true,

  732. 		"WIDTH_BUCKET":                     true,

  733. 		"WINDOW":                           true,

  734. 		"WITH":                             true,

  735. 		"WITHIN":                           true,

  736. 		"WITHOUT":                          true,

  737. 		"WORK":                             true,

  738. 		"WRAPPER":                          true,

  739. 		"WRITE":                            true,

  740. 		"XML":                              true,

  741. 		"XMLAGG":                           true,

  742. 		"XMLATTRIBUTES":                    true,

  743. 		"XMLBINARY":                        true,

  744. 		"XMLCAST":                          true,

  745. 		"XMLCOMMENT":                       true,

  746. 		"XMLCONCAT":                        true,

  747. 		"XMLDECLARATION":                   true,

  748. 		"XMLDOCUMENT":                      true,

  749. 		"XMLELEMENT":                       true,

  750. 		"XMLEXISTS":                        true,

  751. 		"XMLFOREST":                        true,

  752. 		"XMLITERATE":                       true,

  753. 		"XMLNAMESPACES":                    true,

  754. 		"XMLPARSE":                         true,

  755. 		"XMLPI":                            true,

  756. 		"XMLQUERY":                         true,

  757. 		"XMLROOT":                          true,

  758. 		"XMLSCHEMA":                        true,

  759. 		"XMLSERIALIZE":                     true,

  760. 		"XMLTABLE":                         true,

  761. 		"XMLTEXT":                          true,

  762. 		"XMLVALIDATE":                      true,

  763. 		"YEAR":                             true,

  764. 		"YES":                              true,

  765. 		"ZONE":                             true,

  766. 	}

  767. 

  768. 	// DefaultPostgresSchema default postgres schema

  769. 	DefaultPostgresSchema = "public"

  770. )

  771. 

  772. const postgresPublicSchema = "public"

  773. 

  774. type postgres struct {

  775. 	core.Base

  776. }

  777. 

  778. func (db *postgres) Init(d *core.DB, uri *core.Uri, drivername, dataSourceName string) error {

  779. 	err := db.Base.Init(d, db, uri, drivername, dataSourceName)

  780. 	if err != nil {

  781. 		return err

  782. 	}

  783. 	if db.Schema == "" {

  784. 		db.Schema = DefaultPostgresSchema

  785. 	}

  786. 	return nil

  787. }

  788. 

  789. func (db *postgres) SqlType(c *core.Column) string {

  790. 	var res string

  791. 	switch t := c.SQLType.Name; t {

  792. 	case core.TinyInt:

  793. 		res = core.SmallInt

  794. 		return res

  795. 	case core.Bit:

  796. 		res = core.Boolean

  797. 		return res

  798. 	case core.MediumInt, core.Int, core.Integer:

  799. 		if c.IsAutoIncrement {

  800. 			return core.Serial

  801. 		}

  802. 		return core.Integer

  803. 	case core.BigInt:

  804. 		if c.IsAutoIncrement {

  805. 			return core.BigSerial

  806. 		}

  807. 		return core.BigInt

  808. 	case core.Serial, core.BigSerial:

  809. 		c.IsAutoIncrement = true

  810. 		c.Nullable = false

  811. 		res = t

  812. 	case core.Binary, core.VarBinary:

  813. 		return core.Bytea

  814. 	case core.DateTime:

  815. 		res = core.TimeStamp

  816. 	case core.TimeStampz:

  817. 		return "timestamp with time zone"

  818. 	case core.Float:

  819. 		res = core.Real

  820. 	case core.TinyText, core.MediumText, core.LongText:

  821. 		res = core.Text

  822. 	case core.NVarchar:

  823. 		res = core.Varchar

  824. 	case core.Uuid:

  825. 		return core.Uuid

  826. 	case core.Blob, core.TinyBlob, core.MediumBlob, core.LongBlob:

  827. 		return core.Bytea

  828. 	case core.Double:

  829. 		return "DOUBLE PRECISION"

  830. 	default:

  831. 		if c.IsAutoIncrement {

  832. 			return core.Serial

  833. 		}

  834. 		res = t

  835. 	}

  836. 

  837. 	if strings.EqualFold(res, "bool") {

  838. 		// for bool, we don't need length information

  839. 		return res

  840. 	}

  841. 	hasLen1 := (c.Length > 0)

  842. 	hasLen2 := (c.Length2 > 0)

  843. 

  844. 	if hasLen2 {

  845. 		res += "(" + strconv.Itoa(c.Length) + "," + strconv.Itoa(c.Length2) + ")"

  846. 	} else if hasLen1 {

  847. 		res += "(" + strconv.Itoa(c.Length) + ")"

  848. 	}

  849. 	return res

  850. }

  851. 

  852. func (db *postgres) SupportInsertMany() bool {

  853. 	return true

  854. }

  855. 

  856. func (db *postgres) IsReserved(name string) bool {

  857. 	_, ok := postgresReservedWords[name]

  858. 	return ok

  859. }

  860. 

  861. func (db *postgres) Quote(name string) string {

  862. 	name = strings.Replace(name, ".", `"."`, -1)

  863. 	return "\"" + name + "\""

  864. }

  865. 

  866. func (db *postgres) AutoIncrStr() string {

  867. 	return ""

  868. }

  869. 

  870. func (db *postgres) SupportEngine() bool {

  871. 	return false

  872. }

  873. 

  874. func (db *postgres) SupportCharset() bool {

  875. 	return false

  876. }

  877. 

  878. func (db *postgres) IndexOnTable() bool {

  879. 	return false

  880. }

  881. 

  882. func (db *postgres) IndexCheckSql(tableName, idxName string) (string, []interface{}) {

  883. 	if len(db.Schema) == 0 {

  884. 		args := []interface{}{tableName, idxName}

  885. 		return `SELECT indexname FROM pg_indexes WHERE tablename = ? AND indexname = ?`, args

  886. 	}

  887. 

  888. 	args := []interface{}{db.Schema, tableName, idxName}

  889. 	return `SELECT indexname FROM pg_indexes ` +

  890. 		`WHERE schemaname = ? AND tablename = ? AND indexname = ?`, args

  891. }

  892. 

  893. func (db *postgres) TableCheckSql(tableName string) (string, []interface{}) {

  894. 	if len(db.Schema) == 0 {

  895. 		args := []interface{}{tableName}

  896. 		return `SELECT tablename FROM pg_tables WHERE tablename = ?`, args

  897. 	}

  898. 

  899. 	args := []interface{}{db.Schema, tableName}

  900. 	return `SELECT tablename FROM pg_tables WHERE schemaname = ? AND tablename = ?`, args

  901. }

  902. 

  903. func (db *postgres) ModifyColumnSql(tableName string, col *core.Column) string {

  904. 	if len(db.Schema) == 0 {

  905. 		return fmt.Sprintf("alter table %s ALTER COLUMN %s TYPE %s",

  906. 			tableName, col.Name, db.SqlType(col))

  907. 	}

  908. 	return fmt.Sprintf("alter table %s.%s ALTER COLUMN %s TYPE %s",

  909. 		db.Schema, tableName, col.Name, db.SqlType(col))

  910. }

  911. 

  912. func (db *postgres) DropIndexSql(tableName string, index *core.Index) string {

  913. 	quote := db.Quote

  914. 	idxName := index.Name

  915. 

  916. 	tableName = strings.Replace(tableName, `"`, "", -1)

  917. 	tableName = strings.Replace(tableName, `.`, "_", -1)

  918. 

  919. 	if !strings.HasPrefix(idxName, "UQE_") &&

  920. 		!strings.HasPrefix(idxName, "IDX_") {

  921. 		if index.Type == core.UniqueType {

  922. 			idxName = fmt.Sprintf("UQE_%v_%v", tableName, index.Name)

  923. 		} else {

  924. 			idxName = fmt.Sprintf("IDX_%v_%v", tableName, index.Name)

  925. 		}

  926. 	}

  927. 	if db.Uri.Schema != "" {

  928. 		idxName = db.Uri.Schema + "." + idxName

  929. 	}

  930. 	return fmt.Sprintf("DROP INDEX %v", quote(idxName))

  931. }

  932. 

  933. func (db *postgres) IsColumnExist(tableName, colName string) (bool, error) {

  934. 	args := []interface{}{db.Schema, tableName, colName}

  935. 	query := "SELECT column_name FROM INFORMATION_SCHEMA.COLUMNS WHERE table_schema = $1 AND table_name = $2" +

  936. 		" AND column_name = $3"

  937. 	if len(db.Schema) == 0 {

  938. 		args = []interface{}{tableName, colName}

  939. 		query = "SELECT column_name FROM INFORMATION_SCHEMA.COLUMNS WHERE table_name = $1" +

  940. 			" AND column_name = $2"

  941. 	}

  942. 	db.LogSQL(query, args)

  943. 

  944. 	rows, err := db.DB().Query(query, args...)

  945. 	if err != nil {

  946. 		return false, err

  947. 	}

  948. 	defer rows.Close()

  949. 

  950. 	return rows.Next(), nil

  951. }

  952. 

  953. func (db *postgres) GetColumns(tableName string) ([]string, map[string]*core.Column, error) {

  954. 	args := []interface{}{tableName}

  955. 	s := `SELECT column_name, column_default, is_nullable, data_type, character_maximum_length,

  956.     CASE WHEN p.contype = 'p' THEN true ELSE false END AS primarykey,

  957.     CASE WHEN p.contype = 'u' THEN true ELSE false END AS uniquekey

  958. FROM pg_attribute f

  959.     JOIN pg_class c ON c.oid = f.attrelid JOIN pg_type t ON t.oid = f.atttypid

  960.     LEFT JOIN pg_attrdef d ON d.adrelid = c.oid AND d.adnum = f.attnum

  961.     LEFT JOIN pg_namespace n ON n.oid = c.relnamespace

  962.     LEFT JOIN pg_constraint p ON p.conrelid = c.oid AND f.attnum = ANY (p.conkey)

  963.     LEFT JOIN pg_class AS g ON p.confrelid = g.oid

  964.     LEFT JOIN INFORMATION_SCHEMA.COLUMNS s ON s.column_name=f.attname AND c.relname=s.table_name

  965. WHERE c.relkind = 'r'::char AND c.relname = $1%s AND f.attnum > 0 ORDER BY f.attnum;`

  966. 

  967. 	var f string

  968. 	if len(db.Schema) != 0 {

  969. 		args = append(args, db.Schema)

  970. 		f = " AND s.table_schema = $2"

  971. 	}

  972. 	s = fmt.Sprintf(s, f)

  973. 

  974. 	db.LogSQL(s, args)

  975. 

  976. 	rows, err := db.DB().Query(s, args...)

  977. 	if err != nil {

  978. 		return nil, nil, err

  979. 	}

  980. 	defer rows.Close()

  981. 

  982. 	cols := make(map[string]*core.Column)

  983. 	colSeq := make([]string, 0)

  984. 

  985. 	for rows.Next() {

  986. 		col := new(core.Column)

  987. 		col.Indexes = make(map[string]int)

  988. 

  989. 		var colName, isNullable, dataType string

  990. 		var maxLenStr, colDefault *string

  991. 		var isPK, isUnique bool

  992. 		err = rows.Scan(&colName, &colDefault, &isNullable, &dataType, &maxLenStr, &isPK, &isUnique)

  993. 		if err != nil {

  994. 			return nil, nil, err

  995. 		}

  996. 

  997. 		// fmt.Println(args, colName, isNullable, dataType, maxLenStr, colDefault, isPK, isUnique)

  998. 		var maxLen int

  999. 		if maxLenStr != nil {

  1000. 			maxLen, err = strconv.Atoi(*maxLenStr)

  1001. 			if err != nil {

  1002. 				return nil, nil, err

  1003. 			}

  1004. 		}

  1005. 

  1006. 		col.Name = strings.Trim(colName, `" `)

  1007. 

  1008. 		if colDefault != nil {

  1009. 			col.Default = *colDefault

  1010. 			col.DefaultIsEmpty = false

  1011. 			if strings.HasPrefix(col.Default, "nextval(") {

  1012. 				col.IsAutoIncrement = true

  1013. 			}

  1014. 		} else {

  1015. 			col.DefaultIsEmpty = true

  1016. 		}

  1017. 

  1018. 		if isPK {

  1019. 			col.IsPrimaryKey = true

  1020. 		}

  1021. 

  1022. 		col.Nullable = (isNullable == "YES")

  1023. 

  1024. 		switch dataType {

  1025. 		case "character varying", "character":

  1026. 			col.SQLType = core.SQLType{Name: core.Varchar, DefaultLength: 0, DefaultLength2: 0}

  1027. 		case "timestamp without time zone":

  1028. 			col.SQLType = core.SQLType{Name: core.DateTime, DefaultLength: 0, DefaultLength2: 0}

  1029. 		case "timestamp with time zone":

  1030. 			col.SQLType = core.SQLType{Name: core.TimeStampz, DefaultLength: 0, DefaultLength2: 0}

  1031. 		case "double precision":

  1032. 			col.SQLType = core.SQLType{Name: core.Double, DefaultLength: 0, DefaultLength2: 0}

  1033. 		case "boolean":

  1034. 			col.SQLType = core.SQLType{Name: core.Bool, DefaultLength: 0, DefaultLength2: 0}

  1035. 		case "time without time zone":

  1036. 			col.SQLType = core.SQLType{Name: core.Time, DefaultLength: 0, DefaultLength2: 0}

  1037. 		case "oid":

  1038. 			col.SQLType = core.SQLType{Name: core.BigInt, DefaultLength: 0, DefaultLength2: 0}

  1039. 		default:

  1040. 			col.SQLType = core.SQLType{Name: strings.ToUpper(dataType), DefaultLength: 0, DefaultLength2: 0}

  1041. 		}

  1042. 		if _, ok := core.SqlTypes[col.SQLType.Name]; !ok {

  1043. 			return nil, nil, fmt.Errorf("Unknown colType: %v", dataType)

  1044. 		}

  1045. 

  1046. 		col.Length = maxLen

  1047. 

  1048. 		if !col.DefaultIsEmpty {

  1049. 			if col.SQLType.IsText() {

  1050. 				if strings.HasSuffix(col.Default, "::character varying") {

  1051. 					col.Default = strings.TrimRight(col.Default, "::character varying")

  1052. 				} else if !strings.HasPrefix(col.Default, "'") {

  1053. 					col.Default = "'" + col.Default + "'"

  1054. 				}

  1055. 			} else if col.SQLType.IsTime() {

  1056. 				if strings.HasSuffix(col.Default, "::timestamp without time zone") {

  1057. 					col.Default = strings.TrimRight(col.Default, "::timestamp without time zone")

  1058. 				}

  1059. 			}

  1060. 		}

  1061. 		cols[col.Name] = col

  1062. 		colSeq = append(colSeq, col.Name)

  1063. 	}

  1064. 

  1065. 	return colSeq, cols, nil

  1066. }

  1067. 

  1068. func (db *postgres) GetTables() ([]*core.Table, error) {

  1069. 	args := []interface{}{}

  1070. 	s := "SELECT tablename FROM pg_tables"

  1071. 	if len(db.Schema) != 0 {

  1072. 		args = append(args, db.Schema)

  1073. 		s = s + " WHERE schemaname = $1"

  1074. 	}

  1075. 

  1076. 	db.LogSQL(s, args)

  1077. 

  1078. 	rows, err := db.DB().Query(s, args...)

  1079. 	if err != nil {

  1080. 		return nil, err

  1081. 	}

  1082. 	defer rows.Close()

  1083. 

  1084. 	tables := make([]*core.Table, 0)

  1085. 	for rows.Next() {

  1086. 		table := core.NewEmptyTable()

  1087. 		var name string

  1088. 		err = rows.Scan(&name)

  1089. 		if err != nil {

  1090. 			return nil, err

  1091. 		}

  1092. 		table.Name = name

  1093. 		tables = append(tables, table)

  1094. 	}

  1095. 	return tables, nil

  1096. }

  1097. 

  1098. func getIndexColName(indexdef string) []string {

  1099. 	var colNames []string

  1100. 

  1101. 	cs := strings.Split(indexdef, "(")

  1102. 	for _, v := range strings.Split(strings.Split(cs[1], ")")[0], ",") {

  1103. 		colNames = append(colNames, strings.Split(strings.TrimLeft(v, " "), " ")[0])

  1104. 	}

  1105. 

  1106. 	return colNames

  1107. }

  1108. 

  1109. func (db *postgres) GetIndexes(tableName string) (map[string]*core.Index, error) {

  1110. 	args := []interface{}{tableName}

  1111. 	s := fmt.Sprintf("SELECT indexname, indexdef FROM pg_indexes WHERE tablename=$1")

  1112. 	if len(db.Schema) != 0 {

  1113. 		args = append(args, db.Schema)

  1114. 		s = s + " AND schemaname=$2"

  1115. 	}

  1116. 	db.LogSQL(s, args)

  1117. 

  1118. 	rows, err := db.DB().Query(s, args...)

  1119. 	if err != nil {

  1120. 		return nil, err

  1121. 	}

  1122. 	defer rows.Close()

  1123. 

  1124. 	indexes := make(map[string]*core.Index, 0)

  1125. 	for rows.Next() {

  1126. 		var indexType int

  1127. 		var indexName, indexdef string

  1128. 		var colNames []string

  1129. 		err = rows.Scan(&indexName, &indexdef)

  1130. 		if err != nil {

  1131. 			return nil, err

  1132. 		}

  1133. 		indexName = strings.Trim(indexName, `" `)

  1134. 		if strings.HasSuffix(indexName, "_pkey") {

  1135. 			continue

  1136. 		}

  1137. 		if strings.HasPrefix(indexdef, "CREATE UNIQUE INDEX") {

  1138. 			indexType = core.UniqueType

  1139. 		} else {

  1140. 			indexType = core.IndexType

  1141. 		}

  1142. 		colNames = getIndexColName(indexdef)

  1143. 		var isRegular bool

  1144. 		if strings.HasPrefix(indexName, "IDX_"+tableName) || strings.HasPrefix(indexName, "UQE_"+tableName) {

  1145. 			newIdxName := indexName[5+len(tableName):]

  1146. 			isRegular = true

  1147. 			if newIdxName != "" {

  1148. 				indexName = newIdxName

  1149. 			}

  1150. 		}

  1151. 

  1152. 		index := &core.Index{Name: indexName, Type: indexType, Cols: make([]string, 0)}

  1153. 		for _, colName := range colNames {

  1154. 			index.Cols = append(index.Cols, strings.Trim(colName, `" `))

  1155. 		}

  1156. 		index.IsRegular = isRegular

  1157. 		indexes[index.Name] = index

  1158. 	}

  1159. 	return indexes, nil

  1160. }

  1161. 

  1162. func (db *postgres) Filters() []core.Filter {

  1163. 	return []core.Filter{&core.IdFilter{}, &core.QuoteFilter{}, &core.SeqFilter{Prefix: "$", Start: 1}}

  1164. }

  1165. 

  1166. type pqDriver struct {

  1167. }

  1168. 

  1169. type values map[string]string

  1170. 

  1171. func (vs values) Set(k, v string) {

  1172. 	vs[k] = v

  1173. }

  1174. 

  1175. func (vs values) Get(k string) (v string) {

  1176. 	return vs[k]

  1177. }

  1178. 

  1179. func parseURL(connstr string) (string, error) {

  1180. 	u, err := url.Parse(connstr)

  1181. 	if err != nil {

  1182. 		return "", err

  1183. 	}

  1184. 

  1185. 	if u.Scheme != "postgresql" && u.Scheme != "postgres" {

  1186. 		return "", fmt.Errorf("invalid connection protocol: %s", u.Scheme)

  1187. 	}

  1188. 

  1189. 	escaper := strings.NewReplacer(` `, `\ `, `'`, `\'`, `\`, `\\`)

  1190. 

  1191. 	if u.Path != "" {

  1192. 		return escaper.Replace(u.Path[1:]), nil

  1193. 	}

  1194. 

  1195. 	return "", nil

  1196. }

  1197. 

  1198. func parseOpts(name string, o values) error {

  1199. 	if len(name) == 0 {

  1200. 		return fmt.Errorf("invalid options: %s", name)

  1201. 	}

  1202. 

  1203. 	name = strings.TrimSpace(name)

  1204. 

  1205. 	ps := strings.Split(name, " ")

  1206. 	for _, p := range ps {

  1207. 		kv := strings.Split(p, "=")

  1208. 		if len(kv) < 2 {

  1209. 			return fmt.Errorf("invalid option: %q", p)

  1210. 		}

  1211. 		o.Set(kv[0], kv[1])

  1212. 	}

  1213. 

  1214. 	return nil

  1215. }

  1216. 

  1217. func (p *pqDriver) Parse(driverName, dataSourceName string) (*core.Uri, error) {

  1218. 	db := &core.Uri{DbType: core.POSTGRES}

  1219. 	var err error

  1220. 

  1221. 	if strings.HasPrefix(dataSourceName, "postgresql://") || strings.HasPrefix(dataSourceName, "postgres://") {

  1222. 		db.DbName, err = parseURL(dataSourceName)

  1223. 		if err != nil {

  1224. 			return nil, err

  1225. 		}

  1226. 	} else {

  1227. 		o := make(values)

  1228. 		err = parseOpts(dataSourceName, o)

  1229. 		if err != nil {

  1230. 			return nil, err

  1231. 		}

  1232. 

  1233. 		db.DbName = o.Get("dbname")

  1234. 	}

  1235. 

  1236. 	if db.DbName == "" {

  1237. 		return nil, errors.New("dbname is empty")

  1238. 	}

  1239. 

  1240. 	return db, nil

  1241. }

  1242. 

  1243. type pqDriverPgx struct {

  1244. 	pqDriver

  1245. }

  1246. 

  1247. func (pgx *pqDriverPgx) Parse(driverName, dataSourceName string) (*core.Uri, error) {

  1248. 	// Remove the leading characters for driver to work

  1249. 	if len(dataSourceName) >= 9 && dataSourceName[0] == 0 {

  1250. 		dataSourceName = dataSourceName[9:]

  1251. 	}

  1252. 	return pgx.pqDriver.Parse(driverName, dataSourceName)

  1253. }
上海开阖软件有限公司 沪ICP备12045867号-1