数据库操作
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.

967 lines
21 KiB

3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
  1. package dbquery
  2. import (
  3. "database/sql"
  4. "log"
  5. "strconv"
  6. "errors"
  7. "strings"
  8. "time"
  9. "git.tetele.net/tgo/helper"
  10. _ "github.com/go-sql-driver/mysql"
  11. )
  12. /**
  13. * 创建数据
  14. */
  15. func Insert(dbName, table string, data map[string]string) (int64, error) {
  16. var insertId int64
  17. var err error
  18. if dbName == "" && table == "" {
  19. return insertId, errors.New("没有数据表")
  20. }
  21. if strings.Contains(table, "select ") {
  22. dbName = table
  23. } else {
  24. dbName = getTableName(dbName, table)
  25. }
  26. if len(data) < 1 {
  27. return insertId, errors.New("没有要写入的数据")
  28. }
  29. keyList := make([]string, len(data))
  30. keyStr := make([]string, len(data))
  31. valueList := make([]interface{}, len(data))
  32. var i int = 0
  33. for key, value := range data {
  34. keyList[i] = key
  35. keyStr[i] = "?"
  36. valueList[i] = value
  37. i++
  38. }
  39. result, err := DB.Exec("insert into "+dbName+" ("+strings.Join(keyList, ",")+") value("+strings.Join(keyStr, ",")+")", valueList...)
  40. if err != nil {
  41. log.Println("ERROR|插入", dbName, "数据失败,", err)
  42. return insertId, err
  43. } else {
  44. insertId, _ = result.LastInsertId()
  45. time.Sleep(time.Second * 2)
  46. return insertId, nil
  47. }
  48. }
  49. /**
  50. * 修改数据
  51. */
  52. func Update(dbName, table string, data map[string]string, where map[string]string) (int64, error) {
  53. var rowsAffected int64
  54. var err error
  55. if dbName == "" && table == "" {
  56. return rowsAffected, errors.New("没有数据表")
  57. }
  58. if strings.Contains(table, "select ") {
  59. dbName = table
  60. } else {
  61. dbName = getTableName(dbName, table)
  62. }
  63. if len(data) < 1 {
  64. return rowsAffected, errors.New("同有更新的数据")
  65. }
  66. if len(where) < 1 {
  67. return rowsAffected, errors.New("没有更新条件")
  68. }
  69. keyList := make([]string, len(data))
  70. valueList := make([]interface{}, len(data), len(data)+len(where))
  71. whereStr := make([]string, len(where))
  72. var i int = 0
  73. empty := false
  74. for key, value := range data {
  75. keyList[i] = key + "=?"
  76. valueList[i] = value
  77. i++
  78. }
  79. i = 0
  80. for key, value := range where {
  81. if value == "" {
  82. empty = true
  83. break
  84. }
  85. whereStr[i] = key + "=?"
  86. valueList = append(valueList, value)
  87. i++
  88. }
  89. if empty {
  90. log.Println("ERROR|修改数据表", dbName, "时条件中有空数据,条件:", where, "数据:", data)
  91. return rowsAffected, errors.New("条件中有空数据")
  92. }
  93. result, err := DB.Exec("update "+dbName+" set "+strings.Join(keyList, " , ")+" where "+strings.Join(whereStr, " and "), valueList...)
  94. if err != nil {
  95. log.Println("ERROR|修改", dbName, "数据失败,", err)
  96. return rowsAffected, err
  97. } else {
  98. rowsAffected, _ = result.RowsAffected()
  99. return rowsAffected, nil
  100. }
  101. }
  102. /**
  103. * 删除数据
  104. * @param count 删除数量
  105. */
  106. func Delete(dbName, table string, data map[string]string, del_count ...string) (int64, error) {
  107. var count int64
  108. var err error
  109. if dbName == "" && table == "" {
  110. return count, errors.New("没有数据表")
  111. }
  112. if strings.Contains(table, "select ") {
  113. dbName = table
  114. } else {
  115. dbName = getTableName(dbName, table)
  116. }
  117. if len(data) < 1 {
  118. return count, errors.New("没有要删除的数据")
  119. }
  120. keyList := make([]string, len(data))
  121. valueList := make([]interface{}, len(data))
  122. var i int = 0
  123. empty := false
  124. for key, value := range data {
  125. if value == "" {
  126. empty = true
  127. break
  128. }
  129. keyList[i] = key + "=?"
  130. valueList[i] = value
  131. i++
  132. }
  133. if empty {
  134. log.Println("ERROR|删除数据表", dbName, "时条件中有空数据,条件:", data)
  135. return count, errors.New("条件中有空数据")
  136. }
  137. var limitStr string = ""
  138. if len(del_count) > 0 {
  139. limitStr = " limit " + del_count[0]
  140. }
  141. result, err := DB.Exec("delete from "+dbName+" where "+strings.Join(keyList, " and ")+limitStr, valueList...)
  142. if err != nil {
  143. log.Println("ERROR|删除", dbName, "数据失败,", err)
  144. return count, err
  145. } else {
  146. count, _ = result.RowsAffected()
  147. return count, nil
  148. }
  149. }
  150. /**
  151. * 查找一条记录
  152. * @param dbName 数据表名
  153. * @param title 查询字段名
  154. */
  155. func GetData(dbName, table string, title string, where map[string]string, limit map[string]string) (int, map[string]string, error) {
  156. var count int = 0
  157. info := make(map[string]string)
  158. if dbName == "" && table == "" {
  159. return count, info, errors.New("没有数据表")
  160. }
  161. dbName = getTableName(dbName, table)
  162. if len(title) < 1 {
  163. return count, info, errors.New("没有指定查询内容")
  164. }
  165. var limitStr string = ""
  166. if limit != nil && len(limit) > 0 {
  167. var from string = "0" //开始
  168. if _, ok := limit["order"]; ok {
  169. limitStr += " order by " + limit["order"]
  170. }
  171. if _, ok := limit["from"]; ok {
  172. from = limit["from"]
  173. }
  174. limitStr += " limit " + from + ",1"
  175. } else {
  176. limitStr = " limit 1"
  177. }
  178. if len(where) < 1 {
  179. return count, info, errors.New("Query condition is empty")
  180. }
  181. keyList := make([]string, len(where))
  182. valueList := make([]interface{}, len(where))
  183. var i int = 0
  184. empty := false
  185. for key, value := range where {
  186. if value == "" {
  187. empty = true
  188. break
  189. }
  190. keyList[i] = key + " = ? "
  191. valueList[i] = value
  192. i++
  193. }
  194. if empty {
  195. return count, info, errors.New("Query condition is empty")
  196. }
  197. var rows *sql.Rows
  198. var err error
  199. var queryNum int = 0
  200. for queryNum < 3 { //如发生错误,继续查询3次,防止数据库连接断开问题
  201. rows, err = DB.Query("SELECT "+title+" FROM "+dbName+" where "+strings.Join(keyList, " and ")+" "+limitStr, valueList...)
  202. if err == nil {
  203. break
  204. } else {
  205. log.Println("ERROR", "Query from", dbName, "failed,", err, "| SELECT "+title+" FROM "+dbName+" where "+strings.Join(keyList, " and ")+" "+limitStr)
  206. time.Sleep(time.Millisecond * 500)
  207. }
  208. queryNum++
  209. }
  210. defer rows.Close()
  211. if err != nil {
  212. return count, info, err
  213. }
  214. columns, _ := rows.Columns()
  215. scanArgs := make([]interface{}, len(columns))
  216. values := make([]interface{}, len(columns))
  217. for i := range values {
  218. scanArgs[i] = &values[i]
  219. }
  220. var index string
  221. var rowerr error
  222. for rows.Next() {
  223. rowerr = rows.Scan(scanArgs...)
  224. if rowerr == nil {
  225. for i, col := range values {
  226. if col != nil {
  227. index = helper.StrFirstToUpper(columns[i])
  228. info[index] = helper.ToString(col)
  229. }
  230. }
  231. count++
  232. } else {
  233. log.Println("ERROR", "rows scan error", rowerr, dbName, keyList, valueList)
  234. }
  235. }
  236. if rowerr != nil {
  237. return count, info, rowerr
  238. }
  239. return count, info, nil
  240. }
  241. /**
  242. * 查找一条记录
  243. * @param dbName 数据表名
  244. * @param title 查询字段名
  245. */
  246. func GetRow(dbName, table_name, alias string, titles string, join [][]string, where, where_or []string, valueList []interface{}, orderby, groupby, having string, debug bool) (int, map[string]string, error) {
  247. var count int = 0
  248. info := make(map[string]string)
  249. if dbName == "" && table_name == "" {
  250. return count, info, errors.New("没有数据表")
  251. }
  252. table := ""
  253. if strings.Contains(table_name, "select ") {
  254. table = table_name
  255. } else {
  256. table = getTableName(dbName, table_name)
  257. }
  258. var sql_str, title string
  259. if titles != "" {
  260. title = titles
  261. } else {
  262. title = "*"
  263. }
  264. if DB_PROVIDER == "TencentDB" {
  265. sql_str = helper.StringJoin("/*slave*/ select ", title)
  266. } else {
  267. sql_str = helper.StringJoin("select ", title)
  268. }
  269. if alias != "" {
  270. table = helper.StringJoin(table, " as ", alias)
  271. }
  272. sql_str = helper.StringJoin(sql_str, " from ", table)
  273. if len(join) > 0 {
  274. for _, joinitem := range join {
  275. if len(joinitem) < 2 {
  276. continue
  277. }
  278. if len(joinitem) == 3 {
  279. sql_str = helper.StringJoin(sql_str, " ", joinitem[2], " join ", getTableName(dbName, joinitem[0]), " on ", joinitem[1])
  280. } else { //默认左连接
  281. sql_str = helper.StringJoin(sql_str, " left join ", getTableName(dbName, joinitem[0]), " on ", joinitem[1])
  282. }
  283. }
  284. }
  285. if len(where) > 0 || len(where_or) > 0 {
  286. sql_str = helper.StringJoin(sql_str, " where ")
  287. }
  288. if len(where) > 0 {
  289. sql_str = helper.StringJoin(sql_str, " (", strings.Join(where, " and "), " ) ")
  290. }
  291. if len(where_or) > 0 {
  292. if len(where) > 0 {
  293. sql_str = helper.StringJoin(sql_str, " or ", strings.Join(where_or, " or "))
  294. } else {
  295. sql_str = helper.StringJoin(sql_str, strings.Join(where_or, " or "))
  296. }
  297. }
  298. if groupby != "" {
  299. sql_str = helper.StringJoin(sql_str, " group by ", groupby)
  300. }
  301. if having != "" {
  302. sql_str = helper.StringJoin(sql_str, " having ", having)
  303. }
  304. if orderby != "" {
  305. sql_str = helper.StringJoin(sql_str, " order by ", orderby)
  306. }
  307. if debug {
  308. log.Println("query sql:", sql_str, valueList)
  309. }
  310. condition_len := 0 //所有条件数
  311. for _, ch2 := range sql_str {
  312. if string(ch2) == "?" {
  313. condition_len++
  314. }
  315. }
  316. if condition_len != len(valueList) {
  317. return 0, nil, errors.New("参数错误,条件值错误")
  318. }
  319. var rows *sql.Rows
  320. var err error
  321. var queryNum int = 0
  322. sql_str = helper.StringJoin(sql_str, " limit 1")
  323. var db *sql.DB
  324. if SLAVER_DB != nil {
  325. db = SLAVER_DB
  326. } else {
  327. db = DB
  328. }
  329. for queryNum < 2 { //如发生错误,继续查询2次,防止数据库连接断开问题
  330. rows, err = db.Query(sql_str, valueList...)
  331. if err == nil {
  332. break
  333. } else {
  334. log.Println(err)
  335. time.Sleep(time.Millisecond * 500)
  336. }
  337. queryNum++
  338. }
  339. if err != nil {
  340. rows.Close()
  341. return count, info, err
  342. }
  343. columns, _ := rows.Columns()
  344. scanArgs := make([]interface{}, len(columns))
  345. values := make([]interface{}, len(columns))
  346. for i := range values {
  347. scanArgs[i] = &values[i]
  348. }
  349. var index string
  350. var rowerr error
  351. for rows.Next() {
  352. rowerr = rows.Scan(scanArgs...)
  353. if rowerr == nil {
  354. for i, col := range values {
  355. if col != nil {
  356. index = helper.StrFirstToUpper(columns[i])
  357. info[index] = helper.ToString(col)
  358. }
  359. }
  360. count++
  361. } else {
  362. log.Println("ERROR", rowerr)
  363. }
  364. }
  365. rows.Close()
  366. if rowerr != nil {
  367. return count, info, rowerr
  368. }
  369. return count, info, nil
  370. }
  371. /**
  372. * 查找多条记录
  373. * @param dbName 数据表名
  374. * @param title 查询字段名
  375. */
  376. func FetchRows(dbName, table_name, alias string, titles string, join [][]string, where, where_or []string, valueList []interface{}, orderby, groupby, having string, page int, page_size int, debug bool) (int, []map[string]string, error) {
  377. var count int = 0
  378. list := make([]map[string]string, 0)
  379. if dbName == "" && table_name == "" {
  380. return count, list, errors.New("没有数据表")
  381. }
  382. table := ""
  383. if strings.Contains(table_name, "select ") {
  384. table = table_name
  385. } else {
  386. table = getTableName(dbName, table_name)
  387. }
  388. var sql_str, title string
  389. if titles != "" {
  390. title = titles
  391. } else {
  392. title = "*"
  393. }
  394. if DB_PROVIDER == "TencentDB" {
  395. sql_str = helper.StringJoin("/*slave*/ select ", title)
  396. } else {
  397. sql_str = helper.StringJoin("select ", title)
  398. }
  399. if alias != "" {
  400. table = helper.StringJoin(table, " as ", alias)
  401. }
  402. sql_str = helper.StringJoin(sql_str, " from ", table)
  403. if len(join) > 0 {
  404. for _, joinitem := range join {
  405. if len(joinitem) < 2 {
  406. continue
  407. }
  408. if len(joinitem) == 3 {
  409. sql_str = helper.StringJoin(sql_str, " ", joinitem[2], " join ", getTableName(dbName, joinitem[0]), " on ", joinitem[1])
  410. } else { //默认左连接
  411. sql_str = helper.StringJoin(sql_str, " left join ", getTableName(dbName, joinitem[0]), " on ", joinitem[1])
  412. }
  413. }
  414. }
  415. if len(where) > 0 || len(where_or) > 0 {
  416. sql_str = helper.StringJoin(sql_str, " where ")
  417. }
  418. if len(where) > 0 {
  419. sql_str = helper.StringJoin(sql_str, " (", strings.Join(where, " and "), " ) ")
  420. }
  421. if len(where_or) > 0 {
  422. if len(where) > 0 {
  423. sql_str = helper.StringJoin(sql_str, " or ", strings.Join(where_or, " or "))
  424. } else {
  425. sql_str = helper.StringJoin(sql_str, strings.Join(where_or, " or "))
  426. }
  427. }
  428. if groupby != "" {
  429. sql_str = helper.StringJoin(sql_str, " group by ", groupby)
  430. }
  431. if having != "" {
  432. sql_str = helper.StringJoin(sql_str, " HAVING ", having)
  433. }
  434. if orderby != "" {
  435. sql_str = helper.StringJoin(sql_str, " order by ", orderby)
  436. }
  437. if page > 0 || page_size > 0 {
  438. if page < 1 {
  439. page = 1
  440. }
  441. if page_size < 1 {
  442. page_size = 10
  443. }
  444. from := strconv.Itoa((page - 1) * page_size)
  445. offset := strconv.Itoa(page_size)
  446. if from != "" && offset != "" {
  447. sql_str = helper.StringJoin(sql_str, " limit ", from, " , ", offset)
  448. }
  449. }
  450. if debug {
  451. log.Println("query sql:", sql_str, valueList)
  452. }
  453. condition_len := 0 //所有条件数
  454. for _, ch2 := range sql_str {
  455. if string(ch2) == "?" {
  456. condition_len++
  457. }
  458. }
  459. if condition_len != len(valueList) {
  460. return 0, list, errors.New("参数错误,条件值错误")
  461. }
  462. var db *sql.DB
  463. if SLAVER_DB != nil {
  464. db = SLAVER_DB
  465. } else {
  466. db = DB
  467. }
  468. var rows *sql.Rows
  469. var err error
  470. var queryNum int = 0
  471. for queryNum < 2 { //如发生错误,继续查询2次,防止数据库连接断开问题
  472. rows, err = db.Query(sql_str, valueList...)
  473. if err == nil {
  474. break
  475. } else {
  476. log.Println(err)
  477. time.Sleep(time.Millisecond * 500)
  478. }
  479. queryNum++
  480. }
  481. if err != nil {
  482. rows.Close()
  483. return 0, list, err
  484. }
  485. columns, _ := rows.Columns()
  486. scanArgs := make([]interface{}, len(columns))
  487. values := make([]interface{}, len(columns))
  488. for i := range values {
  489. scanArgs[i] = &values[i]
  490. }
  491. var index string
  492. var rowerr error
  493. var info map[string]string
  494. for rows.Next() {
  495. rowerr = rows.Scan(scanArgs...)
  496. info = make(map[string]string)
  497. if rowerr == nil {
  498. for i, col := range values {
  499. if col != nil {
  500. index = helper.StrFirstToUpper(columns[i])
  501. info[index] = helper.ToString(col)
  502. }
  503. }
  504. count++
  505. } else {
  506. log.Println("ERROR", rowerr)
  507. }
  508. if len(info) > 0 {
  509. list = append(list, info)
  510. }
  511. }
  512. rows.Close()
  513. return count, list, nil
  514. }
  515. func GetInfo(dbName, table string, title string, where map[string]string) (map[string]string, error) {
  516. count, info, gzErr := GetData(dbName, table, title, where, nil)
  517. if gzErr != nil {
  518. return info, gzErr
  519. } else {
  520. if count < 1 {
  521. return info, errors.New("No data")
  522. }
  523. return info, nil
  524. }
  525. }
  526. /**
  527. * 查询列表
  528. * 2018/04/19
  529. */
  530. func GetList(dbName, table string, title string, where map[string]string, limit map[string]string) ([]map[string]string, error) {
  531. var list []map[string]string
  532. if dbName == "" && table == "" {
  533. return list, errors.New("没有数据表")
  534. }
  535. if strings.Contains(table, "select ") {
  536. dbName = table
  537. } else {
  538. dbName = getTableName(dbName, table)
  539. }
  540. var rows *sql.Rows
  541. var err error
  542. var queryNum int = 0
  543. var limitStr string = ""
  544. if len(limit) > 0 {
  545. var offset string = "0" //偏移量,个数
  546. var from string = "" //开始
  547. if _, ok := limit["order"]; ok {
  548. limitStr += " order by " + limit["order"]
  549. }
  550. if _, ok := limit["offset"]; ok {
  551. offset = limit["offset"]
  552. }
  553. if _, ok := limit["from"]; ok {
  554. from = limit["from"]
  555. }
  556. if offset != "0" && from != "" {
  557. limitStr += " limit " + from + "," + offset
  558. }
  559. }
  560. if len(where) > 0 {
  561. valueList := make([]interface{}, len(where))
  562. whereStr := make([]string, len(where))
  563. i := 0
  564. var keys []string
  565. for key, value := range where {
  566. key = strings.Trim(key, " ")
  567. value = strings.Trim(value, " ")
  568. if value == "" || key == "" {
  569. continue
  570. // return list, errors.New("Query condition is empty")
  571. }
  572. if strings.Contains(key, " ") {
  573. //key中包含空格,判断是否不等于条件
  574. keys = strings.Split(key, " ")
  575. if helper.IsInStringArray(judg(), strings.Trim(keys[1], " ")) {
  576. whereStr[i] = strings.Trim(keys[0], " ") + " " + strings.Trim(keys[1], " ") + " ?"
  577. }
  578. } else {
  579. whereStr[i] = key + " = ?"
  580. valueList[i] = value
  581. }
  582. i++
  583. }
  584. if len(whereStr) == 0 || len(valueList) == 0 {
  585. return list, errors.New("Query condition is empty")
  586. }
  587. for queryNum < 5 { //如发生错误,继续查询5次,防止数据库连接断开问题
  588. rows, err = DB.Query("select "+title+" from "+dbName+" where "+strings.Join(whereStr, " and ")+" "+limitStr, valueList...)
  589. if err == nil {
  590. break
  591. } else {
  592. time.Sleep(time.Millisecond * 500)
  593. }
  594. queryNum++
  595. }
  596. } else {
  597. for queryNum < 5 { //如发生错误,继续查询5次,防止数据库连接断开问题
  598. rows, err = DB.Query("select " + title + " from " + dbName + " " + limitStr)
  599. if err == nil {
  600. break
  601. } else {
  602. time.Sleep(time.Millisecond * 500)
  603. }
  604. queryNum++
  605. }
  606. }
  607. if err != nil {
  608. return list, err
  609. }
  610. defer rows.Close()
  611. columns, _ := rows.Columns()
  612. scanArgs := make([]interface{}, len(columns))
  613. values := make([]interface{}, len(columns))
  614. for i := range values {
  615. scanArgs[i] = &values[i]
  616. }
  617. var record map[string]string
  618. var index string
  619. for rows.Next() {
  620. //将行数据保存到record字典
  621. err = rows.Scan(scanArgs...)
  622. record = make(map[string]string)
  623. for i, col := range values {
  624. if col != nil {
  625. index = helper.StrFirstToUpper(columns[i])
  626. record[index] = helper.ToString(col)
  627. }
  628. }
  629. list = append(list, record)
  630. }
  631. return list, nil
  632. }
  633. /**
  634. * 查询总数
  635. * 2018/04/19
  636. */
  637. func GetTotal(dbName, table string, args ...string) (total int) {
  638. if dbName == "" && table == "" {
  639. return
  640. }
  641. if strings.Contains(table, "select ") {
  642. dbName = table
  643. } else {
  644. dbName = getTableName(dbName, table)
  645. }
  646. var title string = "*"
  647. if len(args) > 0 {
  648. title = args[0]
  649. }
  650. var rows *sql.Rows
  651. var err error
  652. var queryNum int = 0
  653. for queryNum < 5 { //如发生错误,继续查询5次,防止数据库连接断开问题
  654. rows, err = DB.Query("select count(" + title + ") as count from " + dbName + " limit 1")
  655. if err == nil {
  656. break
  657. } else {
  658. time.Sleep(time.Millisecond * 500)
  659. }
  660. queryNum++
  661. }
  662. if err != nil {
  663. log.Println("ERROR|get", dbName, "total error", err)
  664. return
  665. }
  666. defer rows.Close()
  667. var count int
  668. for rows.Next() {
  669. //将行数据保存到record字典
  670. err = rows.Scan(&count)
  671. if err != nil {
  672. log.Println("ERROR|get", dbName, "total error", err)
  673. } else {
  674. total = count
  675. }
  676. }
  677. return
  678. }
  679. /**
  680. * 查询总数
  681. * 2020/06/04
  682. */
  683. func GetCount(dbName, table string, where map[string]string, args ...string) (total int) {
  684. if dbName == "" && table == "" {
  685. return
  686. }
  687. if strings.Contains(table, "select ") {
  688. dbName = table
  689. } else {
  690. dbName = getTableName(dbName, table)
  691. }
  692. var title string = "*"
  693. if len(title) > 0 {
  694. title = args[0]
  695. }
  696. var rows *sql.Rows
  697. var err error
  698. var queryNum int = 0
  699. if len(where) > 0 {
  700. valueList := make([]interface{}, len(where))
  701. whereStr := make([]string, len(where))
  702. i := 0
  703. var keys []string
  704. for key, value := range where {
  705. key = strings.Trim(key, " ")
  706. value = strings.Trim(value, " ")
  707. if value == "" || key == "" {
  708. continue
  709. // return list, errors.New("Query condition is empty")
  710. }
  711. if strings.Contains(key, " ") {
  712. //key中包含空格,判断是否不等于条件
  713. keys = strings.Split(key, " ")
  714. if helper.IsInStringArray(judg(), strings.Trim(keys[1], " ")) {
  715. whereStr[i] = strings.Trim(keys[0], " ") + " " + strings.Trim(keys[1], " ") + " ?"
  716. }
  717. } else {
  718. whereStr[i] = key + " = ?"
  719. valueList[i] = value
  720. }
  721. i++
  722. }
  723. for queryNum < 5 { //如发生错误,继续查询5次,防止数据库连接断开问题
  724. rows, err = DB.Query("select count("+title+") as count from "+dbName+" where "+strings.Join(whereStr, " and ")+" limit 1", valueList...)
  725. if err == nil {
  726. break
  727. } else {
  728. time.Sleep(time.Millisecond * 500)
  729. }
  730. queryNum++
  731. }
  732. } else {
  733. for queryNum < 5 { //如发生错误,继续查询5次,防止数据库连接断开问题
  734. rows, err = DB.Query("select count(" + title + ") as count from " + dbName + " limit 1")
  735. if err == nil {
  736. break
  737. } else {
  738. time.Sleep(time.Millisecond * 500)
  739. }
  740. queryNum++
  741. }
  742. }
  743. if err != nil {
  744. log.Println("ERROR|get", dbName, "count error", err)
  745. return
  746. }
  747. defer rows.Close()
  748. var count int
  749. for rows.Next() {
  750. //将行数据保存到record字典
  751. err = rows.Scan(&count)
  752. if err != nil {
  753. log.Println("ERROR|get", dbName, "count error", err)
  754. } else {
  755. total = count
  756. }
  757. }
  758. return
  759. }
  760. func DoQuery(args ...interface{}) ([]map[string]string, error) {
  761. var list []map[string]string
  762. if len(args) < 1 {
  763. return list, errors.New("Query condition is empty")
  764. }
  765. queryStr := helper.ToString(args[0])
  766. if queryStr == "" {
  767. return list, errors.New("Query condition is empty")
  768. }
  769. var rows *sql.Rows
  770. var err error
  771. var queryNum int = 0
  772. for queryNum < 3 { //如发生错误,继续查询5次,防止数据库连接断开问题
  773. if len(args) > 1 {
  774. rows, err = DB.Query(queryStr, args[1:]...) //strings.Join(args[1:], ",")
  775. if err != nil {
  776. log.Println("ERROR|DoQuery error:", err)
  777. }
  778. } else {
  779. rows, err = DB.Query(queryStr)
  780. if err != nil {
  781. log.Println("ERROR|DoQuery error:", err)
  782. }
  783. }
  784. if err == nil {
  785. break
  786. } else {
  787. time.Sleep(time.Millisecond * 500)
  788. }
  789. queryNum++
  790. }
  791. if err != nil {
  792. return list, err
  793. }
  794. defer rows.Close()
  795. columns, _ := rows.Columns()
  796. scanArgs := make([]interface{}, len(columns))
  797. values := make([]interface{}, len(columns))
  798. for i := range values {
  799. scanArgs[i] = &values[i]
  800. }
  801. var record map[string]string
  802. var index string
  803. for rows.Next() {
  804. //将行数据保存到record字典
  805. err = rows.Scan(scanArgs...)
  806. record = make(map[string]string)
  807. for i, col := range values {
  808. if col != nil {
  809. index = helper.StrFirstToUpper(columns[i])
  810. record[index] = helper.ToString(col)
  811. }
  812. }
  813. list = append(list, record)
  814. }
  815. return list, nil
  816. }