xugu_utils.go 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221
  1. package xugu
  2. import (
  3. "fmt"
  4. "strings"
  5. )
  6. func parseDSN(dsn string) dsnConfig {
  7. // Initialize a dsnConfig struct
  8. var config dsnConfig
  9. // Split the string by semicolons
  10. pairs := strings.Split(dsn, ";")
  11. // Iterate over the pairs and map them to the struct fields
  12. for _, pair := range pairs {
  13. // Split each pair by the equals sign
  14. kv := strings.SplitN(pair, "=", 2)
  15. if len(kv) != 2 {
  16. continue
  17. }
  18. key, value := strings.TrimSpace(kv[0]), strings.Trim(strings.TrimSpace(kv[1]), "'")
  19. keyL := strings.ToLower(key)
  20. // Map the key to the appropriate struct field
  21. switch keyL {
  22. case "ip":
  23. config.IP = value
  24. case "port":
  25. config.Port = value
  26. case "db":
  27. config.Database = value
  28. case "user":
  29. config.User = value
  30. case "pwd":
  31. config.Password = value
  32. case "encryptor":
  33. config.Encryptor = value
  34. case "char_set":
  35. config.CharSet = value
  36. case "time_zone":
  37. config.TimeZone = value
  38. case "iso_level":
  39. config.IsoLevel = value
  40. case "lock_timeout":
  41. config.LockTimeout = value
  42. case "auto_commit":
  43. config.AutoCommit = value
  44. case "strict_commit":
  45. config.StrictCommit = value
  46. case "result":
  47. config.Result = value
  48. case "return_schema":
  49. config.ReturnSchema = value
  50. case "return_cursor_id":
  51. config.ReturnCursorID = value
  52. case "lob_ret":
  53. config.LobRet = value
  54. case "return_rowid":
  55. config.ReturnRowid = value
  56. case "version":
  57. config.Version = value
  58. }
  59. }
  60. return config
  61. }
  62. func generateLoginString(config dsnConfig) string {
  63. baseString := "login database = '%s' user = '%s' password = '%s' "
  64. additionalParams := ""
  65. if config.Encryptor != "" {
  66. additionalParams += fmt.Sprintf(" encryptor='%s'", config.Encryptor)
  67. }
  68. if config.CharSet != "" {
  69. additionalParams += fmt.Sprintf(" char_set='%s'", config.CharSet)
  70. }
  71. if config.TimeZone != "" {
  72. additionalParams += fmt.Sprintf(" time_zone='%s'", config.TimeZone)
  73. }
  74. if config.IsoLevel != "" {
  75. additionalParams += fmt.Sprintf(" iso_level='%s'", config.IsoLevel)
  76. }
  77. if config.LockTimeout != "" {
  78. additionalParams += fmt.Sprintf(" lock_timeout='%s'", config.LockTimeout)
  79. }
  80. if config.AutoCommit != "" {
  81. additionalParams += fmt.Sprintf(" auto_commit='%s'", config.AutoCommit)
  82. }
  83. if config.StrictCommit != "" {
  84. additionalParams += fmt.Sprintf(" strict_commit='%s'", config.StrictCommit)
  85. }
  86. if config.Result != "" {
  87. additionalParams += fmt.Sprintf(" result='%s'", config.Result)
  88. }
  89. if config.ReturnSchema != "" {
  90. additionalParams += fmt.Sprintf(" return_schema='%s'", config.ReturnSchema)
  91. }
  92. if config.ReturnCursorID != "" {
  93. additionalParams += fmt.Sprintf(" return_cursor_id='%s'", config.ReturnCursorID)
  94. }
  95. if config.LobRet != "" {
  96. additionalParams += fmt.Sprintf(" lob_ret='%s'", config.LobRet)
  97. }
  98. if config.ReturnRowid != "" {
  99. additionalParams += fmt.Sprintf(" return_rowid='%s'", config.ReturnRowid)
  100. }
  101. if config.Version != "" {
  102. additionalParams += fmt.Sprintf(" version='%s'", config.Version)
  103. } else {
  104. additionalParams += " version='201'"
  105. }
  106. finalString := fmt.Sprintf(baseString, config.Database, config.User, config.Password)
  107. if additionalParams != "" {
  108. finalString += additionalParams
  109. }
  110. //finalString += " version='201'"
  111. return finalString
  112. }
  113. // cgo_xgc_sql_type -> fun_sql_type
  114. // 判断sql类型
  115. func switchSQLType(sql string) int {
  116. // 去掉首尾的空格、换行符和回车符
  117. sql = strings.TrimSpace(sql)
  118. if len(sql) < 6 {
  119. return SQL_OTHER
  120. }
  121. // 取前6个字符并转为大写
  122. kstr := strings.ToUpper(sql[:6])
  123. // 根据SQL语句前缀判断类型
  124. switch {
  125. case strings.HasPrefix(kstr, "SELECT"):
  126. if strings.Contains(sql, ";") && len(sql[strings.Index(sql, ";"):]) > 5 {
  127. return SQL_OTHER // 多结果集
  128. }
  129. return SQL_SELECT
  130. case strings.HasPrefix(kstr, "INSERT"):
  131. return SQL_INSERT
  132. case strings.HasPrefix(kstr, "UPDATE"):
  133. return SQL_UPDATE
  134. case strings.HasPrefix(kstr, "DELETE"):
  135. return SQL_DELETE
  136. case strings.HasPrefix(kstr, "CREATE"):
  137. return SQL_CREATE
  138. case strings.HasPrefix(kstr, "ALTER "):
  139. return SQL_ALTER
  140. case strings.HasPrefix(kstr, "EXEC "):
  141. return SQL_PROCEDURE
  142. case strings.HasPrefix(kstr, "EXECUT"):
  143. return SQL_PROCEDURE
  144. case strings.HasPrefix(kstr, "STC"):
  145. return SQL_SELECT
  146. default:
  147. return SQL_OTHER
  148. }
  149. }
  150. func getSQLCType(typeID uint32) int32 {
  151. switch typeID {
  152. case TYPE_BOOL:
  153. return XG_C_BOOL
  154. case TYPE_I1:
  155. return XG_C_TINYINT
  156. case TYPE_I2:
  157. return XG_C_SHORT
  158. case TYPE_I4:
  159. return XG_C_INTEGER
  160. case TYPE_I8:
  161. return XG_C_BIGINT
  162. case TYPE_R4:
  163. return XG_C_FLOAT
  164. case TYPE_R8:
  165. return XG_C_DOUBLE
  166. case TYPE_NUM:
  167. return XG_C_NUMERIC
  168. case TYPE_DATETIME:
  169. return XG_C_DATETIME
  170. case TYPE_DATETIME_TZ:
  171. return XG_C_DATETIME_TZ
  172. case TYPE_INTERVAL_Y2M:
  173. return XG_C_INTERVAL_YEAR_TO_MONTH
  174. case TYPE_INTERVAL_D2S:
  175. return XG_C_INTERVAL_DAY_TO_SECOND
  176. case TYPE_GUID:
  177. return TYPE_GUID + 40
  178. case TYPE_TIME_TZ:
  179. return XG_C_TIME_TZ
  180. case TYPE_DATE:
  181. return XG_C_DATE
  182. case TYPE_TIME:
  183. return XG_C_TIME
  184. case TYPE_BINARY:
  185. return XG_C_BINARY
  186. case TYPE_CLOB:
  187. return XG_C_CLOB
  188. case TYPE_BLOB:
  189. return XG_C_BLOB
  190. case TYPE_INTERVAL_Y, TYPE_INTERVAL_M, TYPE_INTERVAL_D, TYPE_INTERVAL_H,
  191. TYPE_INTERVAL_MI, TYPE_INTERVAL_S, TYPE_INTERVAL_D2H, TYPE_INTERVAL_D2M,
  192. TYPE_INTERVAL_H2M, TYPE_INTERVAL_H2S, TYPE_INTERVAL_M2S:
  193. return XG_C_INTERVAL
  194. default:
  195. return XG_C_CHAR
  196. }
  197. }
  198. func needCopyV(xgtype int) bool {
  199. switch xgtype {
  200. case XG_C_DOUBLE, XG_C_FLOAT, DATETIME_ASLONG, XG_C_BIGINT, XG_C_INTEGER, XG_C_SHORT, XG_C_TINYINT, XG_C_CHARN1, XG_C_BOOL, XG_C_NUMERIC:
  201. return true
  202. default:
  203. return false
  204. }
  205. }