sql_utils.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988
  1. package restful
  2. import (
  3. "database/sql"
  4. "errors"
  5. "fmt"
  6. "math"
  7. "strconv"
  8. "strings"
  9. "time"
  10. "unsafe"
  11. "../../pkg/setting"
  12. _ "github.com/go-sql-driver/mysql"
  13. "github.com/patrickmn/go-cache"
  14. "github.com/tealeg/xlsx"
  15. "github.com/xormplus/xorm"
  16. )
  17. var (
  18. Engine *xorm.Engine
  19. Apisql_cache *cache.Cache
  20. )
  21. func SQLT() *xorm.Session {
  22. tx := Engine.NewSession()
  23. return tx
  24. }
  25. // SQLInit 初始化数据库操作句柄,这里要提供:
  26. // driverName string: 数据库类型,例如mysql、sqlite等,参考github.com/go-sql-driver/mysql官方介绍
  27. // dataSourceName string: 数据库地址,参考github.com/go-sql-driver/mysql官方介绍
  28. // MaxOpenConns int: 最大缓存连接数,这个数值包含了MaxIdleConns
  29. // MaxIdleConns int:预备的最大空闲连接数
  30. func SQLInit(driverName, dataSourceName string, maxOpenConns, maxIdleConns int, showlog bool) error {
  31. if Engine == nil {
  32. var err error
  33. if Engine, err = xorm.NewEngine(driverName, dataSourceName); err != nil {
  34. return err
  35. }
  36. if showlog {
  37. Engine.ShowSQL(true)
  38. }
  39. Engine.SetMaxIdleConns(maxIdleConns)
  40. Engine.SetMaxOpenConns(maxOpenConns)
  41. }
  42. if Apisql_cache == nil {
  43. Apisql_cache = cache.New(600*time.Second, 10*time.Second)
  44. SetCacheSql()
  45. }
  46. return nil
  47. }
  48. func SetCacheSql() {
  49. valuesMap, err := Engine.SQL("SELECT sqlname, CONCAT(sqlstr,'|',IFNULL(params,'')) AS sqlstr FROM apisql WHERE ENABLE>0 ").QueryString()
  50. if err == nil && len(valuesMap) > 0 {
  51. for _, v := range valuesMap {
  52. Apisql_cache.Set(v["sqlname"], v["sqlstr"], cache.DefaultExpiration)
  53. }
  54. }
  55. }
  56. func sqlCheckParam(param string) error {
  57. /*if strings.Contains(param, "where") {
  58. return errors.New("can not have where")
  59. }
  60. if strings.Contains(param, "and") {
  61. return errors.New("can not have and")
  62. }
  63. if strings.Contains(param, "or") {
  64. return errors.New("can not have or")
  65. }
  66. if strings.Contains(param, "=") {
  67. return errors.New("can not have =")
  68. }*/
  69. if strings.Contains(param, ";") {
  70. return errors.New("can not have ;")
  71. }
  72. return nil
  73. }
  74. func GetSqlByName(name string, tx *xorm.Session) (string, string) {
  75. sqls := ""
  76. param := ""
  77. if sql, ok := Apisql_cache.Get(name); ok {
  78. tempstrb := strings.IndexAny(sql.(string), "|")
  79. sqls = sql.(string)[0:tempstrb]
  80. param = sql.(string)[tempstrb+1:]
  81. } else {
  82. valuesMap, err := tx.SQL("SELECT sqlstr, params FROM apisql WHERE sqlname = ? AND ENABLE>0 ", name).QueryString()
  83. if err == nil && len(valuesMap) > 0 {
  84. sqls = valuesMap[0]["sqlstr"]
  85. param = valuesMap[0]["params"]
  86. Apisql_cache.Set(name, (sqls + "|" + param), cache.DefaultExpiration)
  87. }
  88. }
  89. if setting.DatabaseSetting.ShowGetSqllog {
  90. fmt.Println("apisql:" + sqls)
  91. fmt.Println("param:" + param)
  92. }
  93. return sqls, param
  94. }
  95. func GetSqlByNameDB(name string) (string, string) {
  96. sqls := ""
  97. param := ""
  98. if sql, ok := Apisql_cache.Get(name); ok && setting.DatabaseSetting.CacheApiSql {
  99. tempstrb := strings.IndexAny(sql.(string), "|")
  100. sqls = sql.(string)[0:tempstrb]
  101. param = sql.(string)[tempstrb+1:]
  102. } else {
  103. valuesMap, err := Engine.SQL("SELECT sqlstr,params FROM apisql WHERE sqlname = ? AND ENABLE>0 ", name).QueryString()
  104. if err == nil && len(valuesMap) > 0 {
  105. sqls = valuesMap[0]["sqlstr"]
  106. param = valuesMap[0]["params"]
  107. Apisql_cache.Set(name, (sqls + "|" + param), cache.DefaultExpiration)
  108. }
  109. }
  110. if setting.DatabaseSetting.ShowGetSqllog {
  111. fmt.Println("apisql:" + sqls)
  112. fmt.Println("param:" + param)
  113. }
  114. return sqls, param
  115. }
  116. func GetSqlByNameDBT(name string, tx *xorm.Session) (string, string) {
  117. sqls := ""
  118. param := ""
  119. if sql, ok := Apisql_cache.Get(name); ok && setting.DatabaseSetting.CacheApiSql {
  120. tempstrb := strings.IndexAny(sql.(string), "|")
  121. sqls = sql.(string)[0:tempstrb]
  122. param = sql.(string)[tempstrb+1:]
  123. } else {
  124. valuesMap, err := tx.SQL("SELECT sqlstr,params FROM apisql WHERE sqlname = ? AND ENABLE>0 ", name).QueryString()
  125. if err == nil && len(valuesMap) > 0 {
  126. sqls = valuesMap[0]["sqlstr"]
  127. param = valuesMap[0]["params"]
  128. Apisql_cache.Set(name, (sqls + "|" + param), cache.DefaultExpiration)
  129. }
  130. }
  131. if setting.DatabaseSetting.ShowGetSqllog {
  132. fmt.Println("apisql:" + sqls)
  133. fmt.Println("param:" + param)
  134. }
  135. return sqls, param
  136. }
  137. // sqlGetValues 根据结构体中指向实际数据的指针获取出数据,并存储到另一张表中返回
  138. func sqlGetValues(pvs []interface{}, columnsType []*sql.ColumnType, columnsLen int) map[string]interface{} {
  139. result := make(map[string]interface{}, columnsLen)
  140. for a := 0; a < columnsLen; a++ {
  141. switch s := pvs[a].(type) {
  142. case *int8:
  143. result[columnsType[a].Name()] = *s
  144. case *int16:
  145. result[columnsType[a].Name()] = *s
  146. case *int32:
  147. result[columnsType[a].Name()] = *s
  148. case *int64:
  149. result[columnsType[a].Name()] = *s
  150. case *float32:
  151. result[columnsType[a].Name()] = *s
  152. case *float64:
  153. result[columnsType[a].Name()] = *s
  154. case *string:
  155. result[columnsType[a].Name()] = *s
  156. case *sql.NullInt64:
  157. result[columnsType[a].Name()] = *s
  158. case *sql.NullString:
  159. result[columnsType[a].Name()] = *s
  160. }
  161. }
  162. return result
  163. }
  164. // 这里返回的是原始数组的基础上加上了字段名标识
  165. func sqlQuery(columnsType []*sql.ColumnType, columnsLen int, queryData [][]interface{}, queryCount int) ([]map[string]interface{}, error) {
  166. jsondata := make([]map[string]interface{}, queryCount)
  167. for k1, v1 := range queryData {
  168. if k1 >= queryCount {
  169. break
  170. }
  171. jsondata[k1] = sqlGetValues(v1, columnsType, columnsLen)
  172. }
  173. return jsondata, nil
  174. }
  175. func sqlQueryByTinyIntMap(columnName string, columnsType []*sql.ColumnType, columnsLen int, queryData [][]interface{}, queryCount int) (map[int8]map[string]interface{}, error) {
  176. jsondata := make(map[int8]map[string]interface{}, queryCount)
  177. for k1, v1 := range queryData {
  178. if k1 >= queryCount {
  179. break
  180. }
  181. for a := 0; a < columnsLen; a++ {
  182. if columnsType[a].Name() == columnName {
  183. if value, ok := v1[a].(*int8); ok {
  184. jsondata[*value] = sqlGetValues(v1, columnsType, columnsLen)
  185. }
  186. break
  187. }
  188. }
  189. }
  190. return jsondata, nil
  191. }
  192. func sqlQueryBySmallIntMap(columnName string, columnsType []*sql.ColumnType, columnsLen int, queryData [][]interface{}, queryCount int) (map[int16]map[string]interface{}, error) {
  193. jsondata := make(map[int16]map[string]interface{}, queryCount)
  194. for k1, v1 := range queryData {
  195. if k1 >= queryCount {
  196. break
  197. }
  198. for a := 0; a < columnsLen; a++ {
  199. if columnsType[a].Name() == columnName {
  200. if value, ok := v1[a].(*int16); ok {
  201. jsondata[*value] = sqlGetValues(v1, columnsType, columnsLen)
  202. }
  203. break
  204. }
  205. }
  206. }
  207. return jsondata, nil
  208. }
  209. func sqlQueryByIntMap(columnName string, columnsType []*sql.ColumnType, columnsLen int, queryData [][]interface{}, queryCount int) (map[int32]map[string]interface{}, error) {
  210. jsondata := make(map[int32]map[string]interface{}, queryCount)
  211. for k1, v1 := range queryData {
  212. if k1 >= queryCount {
  213. break
  214. }
  215. for a := 0; a < columnsLen; a++ {
  216. if columnsType[a].Name() == columnName {
  217. if value, ok := v1[a].(*int32); ok {
  218. jsondata[*value] = sqlGetValues(v1, columnsType, columnsLen)
  219. }
  220. break
  221. }
  222. }
  223. }
  224. return jsondata, nil
  225. }
  226. func sqlQueryByBigIntMap(columnName string, columnsType []*sql.ColumnType, columnsLen int, queryData [][]interface{}, queryCount int) (map[int64]map[string]interface{}, error) {
  227. jsondata := make(map[int64]map[string]interface{}, queryCount)
  228. for k1, v1 := range queryData {
  229. if k1 >= queryCount {
  230. break
  231. }
  232. for a := 0; a < columnsLen; a++ {
  233. if columnsType[a].Name() == columnName {
  234. if value, ok := v1[a].(*int64); ok {
  235. jsondata[*value] = sqlGetValues(v1, columnsType, columnsLen)
  236. }
  237. break
  238. }
  239. }
  240. }
  241. return jsondata, nil
  242. }
  243. func sqlQueryByFloatIntMap(columnName string, columnsType []*sql.ColumnType, columnsLen int, queryData [][]interface{}, queryCount int) (map[float32]map[string]interface{}, error) {
  244. jsondata := make(map[float32]map[string]interface{}, queryCount)
  245. for k1, v1 := range queryData {
  246. if k1 >= queryCount {
  247. break
  248. }
  249. for a := 0; a < columnsLen; a++ {
  250. if columnsType[a].Name() == columnName {
  251. if value, ok := v1[a].(*float32); ok {
  252. jsondata[*value] = sqlGetValues(v1, columnsType, columnsLen)
  253. }
  254. break
  255. }
  256. }
  257. }
  258. return jsondata, nil
  259. }
  260. func sqlQueryByDoubleMap(columnName string, columnsType []*sql.ColumnType, columnsLen int, queryData [][]interface{}, queryCount int) (map[float64]map[string]interface{}, error) {
  261. jsondata := make(map[float64]map[string]interface{}, queryCount)
  262. for k1, v1 := range queryData {
  263. if k1 >= queryCount {
  264. break
  265. }
  266. for a := 0; a < columnsLen; a++ {
  267. if columnsType[a].Name() == columnName {
  268. if value, ok := v1[a].(*float64); ok {
  269. jsondata[*value] = sqlGetValues(v1, columnsType, columnsLen)
  270. }
  271. break
  272. }
  273. }
  274. }
  275. return jsondata, nil
  276. }
  277. func sqlQueryByStringMap(columnName string, columnsType []*sql.ColumnType, columnsLen int, queryData [][]interface{}, queryCount int) (map[string]map[string]interface{}, error) {
  278. jsondata := make(map[string]map[string]interface{}, queryCount)
  279. for k1, v1 := range queryData {
  280. if k1 >= queryCount {
  281. break
  282. }
  283. for a := 0; a < columnsLen; a++ {
  284. if columnsType[a].Name() == columnName {
  285. if value, ok := v1[a].(*string); ok {
  286. jsondata[*value] = sqlGetValues(v1, columnsType, columnsLen)
  287. }
  288. break
  289. }
  290. }
  291. }
  292. return jsondata, nil
  293. }
  294. func sqlGetColumnType(columnsType []*sql.ColumnType, columnsLen int, valueName string) string {
  295. for a := 0; a < columnsLen; a++ {
  296. if columnsType[a].Name() == valueName {
  297. return columnsType[a].DatabaseTypeName()
  298. }
  299. }
  300. return ""
  301. }
  302. func round(num float64) int {
  303. return int(num + math.Copysign(0.5, num))
  304. }
  305. func toFixed(num float64, precision int) float64 {
  306. output := math.Pow(10, float64(precision))
  307. return float64(round(num*output)) / output
  308. }
  309. // QueryByMap 将查询到的数据,按照指定字段的值做为索引构建map并返回
  310. // columnName string: 作为索引的字段名称
  311. // sql string: 查询需要获取哪些字段的值,就是select后面跟着的部分,一般用"*"
  312. // where string: 过滤条件,就是where后面跟着的部分
  313. // order string: 排序条件,就是order by后面跟着的部分。默认是ASC排序,除非"-"开头则DESC排序
  314. // offset string: limit后面逗号相隔的两个数值,前者就是offset,后者就是count
  315. // count string: limit后面逗号相隔的两个数值,前者就是offset,后者就是count
  316. func QueryByMap(sqlstr string, offset, count int, params []interface{}) (interface{}, error) {
  317. if err := sqlCheckParam(sqlstr); err != nil {
  318. return 0, err
  319. }
  320. recordcount := int64(0)
  321. if count > 1 {
  322. tt, err := Engine.SQL("SELECT COUNT(*) as count FROM ("+sqlstr+") a", params...).Query().List()
  323. if err != nil {
  324. fmt.Println("s1", err)
  325. return nil, err
  326. }
  327. if tt == nil {
  328. recordcount = 0
  329. } else {
  330. recordcount = tt[0]["count"].(int64)
  331. }
  332. }
  333. queryData := make([]map[string]interface{}, 1)
  334. var err error
  335. if count == 0 {
  336. queryData, err = Engine.SQL(sqlstr, params...).Query().List()
  337. } else {
  338. if count > 0 && offset > 0 {
  339. offset = offset * count
  340. }
  341. queryData, err = Engine.SQL(sqlstr, params...).Limit(count, offset).Query().List()
  342. }
  343. //fmt.Println("count, offset==================",count, offset)
  344. if err != nil {
  345. return nil, err
  346. }
  347. if recordcount == 0 {
  348. recordcount = int64(len(queryData))
  349. }
  350. data := make(map[string]interface{})
  351. data["list"] = queryData
  352. data["total"] = recordcount
  353. data["code"] = 0
  354. data["message"] = "ok"
  355. if count > 0 {
  356. data["pages"] = math.Ceil(float64(recordcount) / float64(count))
  357. data["pageNum"] = math.Ceil(float64(offset+1) / float64(count))
  358. }
  359. data["pageSize"] = count
  360. return data, nil
  361. }
  362. func QueryByMapT(sqlstr string, offset, count int, params []interface{}, tx *xorm.Session) (interface{}, error) {
  363. if err := sqlCheckParam(sqlstr); err != nil {
  364. return 0, err
  365. }
  366. recordcount := int64(0)
  367. if count > 1 {
  368. tt, err := tx.SQL("SELECT COUNT(*) as count FROM ("+sqlstr+") a", params...).Query().List()
  369. if err != nil {
  370. fmt.Println("s1", err)
  371. return nil, err
  372. }
  373. if tt == nil {
  374. recordcount = 0
  375. } else {
  376. switch tt[0]["count"].(type) {
  377. case int64:
  378. recordcount = tt[0]["count"].(int64)
  379. case string:
  380. temp, _ := strconv.Atoi(tt[0]["count"].(string))
  381. recordcount = int64(temp)
  382. }
  383. }
  384. }
  385. queryData := make([]map[string]interface{}, 1)
  386. var err error
  387. if count == 0 {
  388. queryData, err = tx.SQL(sqlstr, params...).Query().List()
  389. } else {
  390. if count > 0 && offset > 0 {
  391. offset = offset * count
  392. }
  393. queryData, err = tx.SQL(sqlstr, params...).Limit(count, offset).Query().List()
  394. }
  395. //fmt.Println("count, offset==================",count, offset)
  396. for _, item := range queryData {
  397. fmt.Println(item)
  398. }
  399. if err != nil {
  400. return nil, err
  401. }
  402. if recordcount == 0 {
  403. recordcount = int64(len(queryData))
  404. }
  405. data := make(map[string]interface{})
  406. data["list"] = queryData
  407. data["total"] = recordcount
  408. data["code"] = 0
  409. data["message"] = "ok"
  410. if count > 0 {
  411. data["pages"] = math.Ceil(float64(recordcount) / float64(count))
  412. data["pageNum"] = math.Ceil(float64(offset+1) / float64(count))
  413. }
  414. data["pageSize"] = count
  415. return data, nil
  416. }
  417. // QueryByXlsx 将查询到的数据,按照指定字段的值做为索引构建map并返回
  418. // sql string: 查询需要获取哪些字段的值,就是select后面跟着的部分,一般用"*"
  419. // params string: 过滤条件,就是where后面跟着的部分
  420. func QueryByXlsx(sqlstr string, params []interface{}) (*xlsx.File, error) {
  421. if err := sqlCheckParam(sqlstr); err != nil {
  422. return nil, err
  423. }
  424. rows, err := Engine.SQL(sqlstr, params...).QueryRows()
  425. if err != nil {
  426. return nil, err
  427. }
  428. defer rows.Close()
  429. columnsType, _ := rows.ColumnTypes()
  430. columnsLen := len(columnsType)
  431. columns, _ := rows.Columns()
  432. values := make([]interface{}, columnsLen)
  433. valuePtrs := make([]interface{}, columnsLen)
  434. var sheet *xlsx.Sheet
  435. var row *xlsx.Row
  436. var cell *xlsx.Cell
  437. file := xlsx.NewFile()
  438. sheet, err = file.AddSheet("Sheet1")
  439. if err != nil {
  440. fmt.Printf(err.Error())
  441. }
  442. row = sheet.AddRow()
  443. for _, columnname := range columns {
  444. cell = row.AddCell()
  445. cell.Value = columnname
  446. }
  447. for rows.Next() {
  448. for a := 0; a < columnsLen; a++ {
  449. valuePtrs[a] = &values[a]
  450. }
  451. rows.Scan(valuePtrs...)
  452. row = sheet.AddRow()
  453. for i, _ := range columns {
  454. var v interface{}
  455. val := values[i]
  456. b, ok := val.([]byte)
  457. cell = row.AddCell()
  458. if ok {
  459. v1 := string(b)
  460. switch columnsType[i].DatabaseTypeName() {
  461. case "TINYINT", "SMALLINT", "MEDIUMINT", "INT", "INTEGER", "BIGINT":
  462. {
  463. v, _ = strconv.ParseInt(v1, 10, 32)
  464. cell.Value = fmt.Sprintf("%v", v)
  465. }
  466. case "FLOAT", "DOUBLE", "DECIMAL":
  467. {
  468. cindex := len(val.([]byte)) - strings.Index(v1, ".") - 1
  469. v2, _ := strconv.ParseFloat(v1, 32)
  470. v = toFixed(v2, cindex)
  471. cell.Value = fmt.Sprintf("%v", v)
  472. }
  473. case "BOOL":
  474. {
  475. v, _ = strconv.ParseBool(v1)
  476. cell.Value = fmt.Sprintf("%v", v)
  477. }
  478. default:
  479. {
  480. v = v1
  481. cell.Value = fmt.Sprintf("%v", v)
  482. }
  483. }
  484. } else {
  485. v = val
  486. cell.Value = fmt.Sprintf("%v", v)
  487. }
  488. if cell.Value == "<nil>" {
  489. cell.Value = ""
  490. }
  491. }
  492. }
  493. return file, nil
  494. }
  495. // QueryByMap 将查询到的数据,按照指定字段的值做为索引构建map并返回
  496. // columnName string: 作为索引的字段名称
  497. // sql string: 查询需要获取哪些字段的值,就是select后面跟着的部分,一般用"*"
  498. // where string: 过滤条件,就是where后面跟着的部分
  499. // order string: 排序条件,就是order by后面跟着的部分。默认是ASC排序,除非"-"开头则DESC排序
  500. // offset string: limit后面逗号相隔的两个数值,前者就是offset,后者就是count
  501. // count string: limit后面逗号相隔的两个数值,前者就是offset,后者就是count
  502. func QueryByMapMenu(sqlstr string, offset, count int, params []interface{}) ([]map[string]interface{}, error) {
  503. if err := sqlCheckParam(sqlstr); err != nil {
  504. return nil, err
  505. }
  506. if offset < 0 {
  507. offset = 0
  508. }
  509. if count <= 0 {
  510. count = 0
  511. }
  512. if count > 0 {
  513. sqlstr += " limit " + strconv.Itoa(offset) + "," + strconv.Itoa(count)
  514. }
  515. rows, err := Engine.SQL(sqlstr, params...).QueryRows()
  516. if err != nil {
  517. return nil, err
  518. }
  519. defer rows.Close()
  520. columnsType, _ := rows.ColumnTypes()
  521. columnsLen := len(columnsType)
  522. columns, _ := rows.Columns()
  523. queryData := make([]map[string]interface{}, 0)
  524. values := make([]interface{}, columnsLen)
  525. valuePtrs := make([]interface{}, columnsLen)
  526. for rows.Next() {
  527. for a := 0; a < columnsLen; a++ {
  528. valuePtrs[a] = &values[a]
  529. }
  530. rows.Scan(valuePtrs...)
  531. entry := make(map[string]interface{})
  532. for i, col := range columns {
  533. entry[col] = getValue(values[i], columnsType[i].DatabaseTypeName())
  534. }
  535. queryData = append(queryData, entry)
  536. }
  537. data := queryData
  538. return data, nil
  539. //return Engine.SQL(sqlstr, params...).Query().List()
  540. }
  541. func QueryByList(sqlstr string, offset, count int, params []interface{}) (interface{}, error) {
  542. if err := sqlCheckParam(sqlstr); err != nil {
  543. return 0, err
  544. }
  545. if count == 0 {
  546. count = 1000
  547. }
  548. recordcount := int64(0)
  549. if count > 1 {
  550. tt, err := Engine.SQL("SELECT COUNT(*) as count FROM ("+sqlstr+") a", params...).Query().List()
  551. if err != nil {
  552. return nil, err
  553. }
  554. recordcount = tt[0]["count"].(int64)
  555. }
  556. queryData := make([]map[string]interface{}, 1)
  557. var err error
  558. if count == 0 {
  559. queryData, err = Engine.SQL(sqlstr, params...).Query().List()
  560. } else {
  561. if count > 0 && offset > 0 {
  562. offset = offset * count
  563. }
  564. queryData, err = Engine.SQL(sqlstr, params...).Limit(count, offset).Query().List()
  565. }
  566. if err != nil {
  567. return nil, err
  568. }
  569. columnsname := make([]string, 0)
  570. listdata := make(map[string][]interface{})
  571. if len(queryData) > 0 {
  572. for i, _ := range queryData {
  573. if i == 0 {
  574. for i, _ := range queryData[0] {
  575. columnsname = append(columnsname, i)
  576. }
  577. }
  578. for _, v := range columnsname {
  579. listdata[v] = append(listdata[v], queryData[i][v])
  580. }
  581. }
  582. recordcount = int64(len(queryData))
  583. }
  584. data := make(map[string]interface{})
  585. data["lists"] = listdata
  586. //data["columnsname"] = columnsname
  587. data["total"] = recordcount
  588. data["code"] = 0
  589. data["message"] = "ok"
  590. if count > 0 {
  591. data["pages"] = math.Ceil(float64(recordcount) / float64(count))
  592. data["pageNum"] = math.Ceil(float64(offset+1) / float64(count))
  593. }
  594. data["pageSize"] = count
  595. return data, nil
  596. }
  597. func QueryByListList(sqlstr string, offset, count int, params []interface{}) (interface{}, error) {
  598. if err := sqlCheckParam(sqlstr); err != nil {
  599. return 0, err
  600. }
  601. if count == 0 {
  602. count = 1000
  603. }
  604. recordcount := int64(0)
  605. if count > 1 {
  606. tt, err := Engine.SQL("SELECT COUNT(*) as count FROM ("+sqlstr+") a", params...).Query().List()
  607. if err != nil {
  608. return nil, err
  609. }
  610. recordcount = tt[0]["count"].(int64)
  611. }
  612. queryData := make([]map[string]interface{}, 1)
  613. var err error
  614. if count == 0 {
  615. queryData, err = Engine.SQL(sqlstr, params...).Query().List()
  616. } else {
  617. if count > 0 && offset > 0 {
  618. offset = offset * count
  619. }
  620. queryData, err = Engine.SQL(sqlstr, params...).Limit(count, offset).Query().List()
  621. }
  622. if err != nil {
  623. return nil, err
  624. }
  625. columnsname := make([]string, 0)
  626. listdata := make([]interface{}, 0)
  627. if len(queryData) > 0 {
  628. for i, _ := range queryData {
  629. if i == 0 {
  630. for i, _ := range queryData[0] {
  631. columnsname = append(columnsname, i)
  632. }
  633. }
  634. listdatarow := make([]interface{}, 0)
  635. for _, v := range columnsname {
  636. listdatarow = append(listdatarow, queryData[i][v])
  637. }
  638. listdata = append(listdata, listdatarow)
  639. }
  640. recordcount = int64(len(queryData))
  641. }
  642. data := make(map[string]interface{})
  643. data["list"] = listdata
  644. data["columnsname"] = columnsname
  645. data["total"] = recordcount
  646. data["code"] = 0
  647. data["message"] = "ok"
  648. if count > 0 {
  649. data["pages"] = math.Ceil(float64(recordcount) / float64(count))
  650. data["pageNum"] = math.Ceil(float64(offset+1) / float64(count))
  651. }
  652. data["pageSize"] = count
  653. return data, nil
  654. }
  655. func QueryByListT(sqlstr string, offset, count int, params []interface{}, tx *xorm.Session) (interface{}, error) {
  656. if err := sqlCheckParam(sqlstr); err != nil {
  657. return 0, err
  658. }
  659. if count == 0 {
  660. count = 1000
  661. }
  662. recordcount := int64(0)
  663. if count > 1 {
  664. tt, err := tx.SQL("SELECT COUNT(*) as count FROM ("+sqlstr+") a", params...).Query().List()
  665. if err != nil {
  666. return nil, err
  667. }
  668. switch tt[0]["count"].(type) {
  669. case int64:
  670. recordcount = tt[0]["count"].(int64)
  671. case string:
  672. temp, _ := strconv.Atoi(tt[0]["count"].(string))
  673. recordcount = int64(temp)
  674. }
  675. }
  676. queryData := make([]map[string]interface{}, 1)
  677. var err error
  678. if count == 0 {
  679. queryData, err = tx.SQL(sqlstr, params...).Query().List()
  680. } else {
  681. if count > 0 && offset > 0 {
  682. offset = offset * count
  683. }
  684. queryData, err = tx.SQL(sqlstr, params...).Limit(count, offset).Query().List()
  685. }
  686. if err != nil {
  687. return nil, err
  688. }
  689. columnsname := make([]string, 0)
  690. listdata := make(map[string][]interface{})
  691. if len(queryData) > 0 {
  692. for i, _ := range queryData {
  693. if i == 0 {
  694. for i, _ := range queryData[0] {
  695. columnsname = append(columnsname, i)
  696. }
  697. }
  698. for _, v := range columnsname {
  699. listdata[v] = append(listdata[v], queryData[i][v])
  700. }
  701. }
  702. recordcount = int64(len(queryData))
  703. }
  704. data := make(map[string]interface{})
  705. data["lists"] = listdata
  706. //data["columnsname"] = columnsname
  707. data["total"] = recordcount
  708. data["code"] = 0
  709. data["message"] = "ok"
  710. if count > 0 {
  711. data["pages"] = math.Ceil(float64(recordcount) / float64(count))
  712. data["pageNum"] = math.Ceil(float64(offset+1) / float64(count))
  713. }
  714. data["pageSize"] = count
  715. return data, nil
  716. }
  717. func QueryByListListT(sqlstr string, offset, count int, params []interface{}, tx *xorm.Session) (interface{}, error) {
  718. if err := sqlCheckParam(sqlstr); err != nil {
  719. return 0, err
  720. }
  721. if count == 0 {
  722. count = 1000
  723. }
  724. recordcount := int64(0)
  725. if count > 1 {
  726. tt, err := tx.SQL("SELECT COUNT(*) as count FROM ("+sqlstr+") a", params...).Query().List()
  727. if err != nil {
  728. return nil, err
  729. }
  730. switch tt[0]["count"].(type) {
  731. case int64:
  732. recordcount = tt[0]["count"].(int64)
  733. case string:
  734. temp, _ := strconv.Atoi(tt[0]["count"].(string))
  735. recordcount = int64(temp)
  736. }
  737. }
  738. queryData := make([]map[string]interface{}, 1)
  739. var err error
  740. if count == 0 {
  741. queryData, err = tx.SQL(sqlstr, params...).Query().List()
  742. } else {
  743. if count > 0 && offset > 0 {
  744. offset = offset * count
  745. }
  746. queryData, err = tx.SQL(sqlstr, params...).Limit(count, offset).Query().List()
  747. }
  748. if err != nil {
  749. return nil, err
  750. }
  751. columnsname := make([]string, 0)
  752. listdata := make([]interface{}, 0)
  753. if len(queryData) > 0 {
  754. for i, _ := range queryData {
  755. if i == 0 {
  756. for i, _ := range queryData[0] {
  757. columnsname = append(columnsname, i)
  758. }
  759. }
  760. listdatarow := make([]interface{}, 0)
  761. for _, v := range columnsname {
  762. listdatarow = append(listdatarow, queryData[i][v])
  763. }
  764. listdata = append(listdata, listdatarow)
  765. }
  766. recordcount = int64(len(queryData))
  767. }
  768. data := make(map[string]interface{})
  769. data["list"] = listdata
  770. data["columnsname"] = columnsname
  771. data["total"] = recordcount
  772. data["code"] = 0
  773. data["message"] = "ok"
  774. if count > 0 {
  775. data["pages"] = math.Ceil(float64(recordcount) / float64(count))
  776. data["pageNum"] = math.Ceil(float64(offset+1) / float64(count))
  777. }
  778. data["pageSize"] = count
  779. return data, nil
  780. }
  781. func ExecQuery(sqlstr string, params []interface{}) (interface{}, error) {
  782. if err := sqlCheckParam(sqlstr); err != nil {
  783. return 0, err
  784. }
  785. rows, err := Engine.SQL(sqlstr, params...).Execute()
  786. if err != nil {
  787. return nil, err
  788. }
  789. data := make(map[string]interface{})
  790. LastInsertId, _ := rows.LastInsertId()
  791. idPointer := (*int)(unsafe.Pointer(&LastInsertId))
  792. idd16 := *idPointer
  793. RowsAffected, _ := rows.RowsAffected()
  794. _idPointer := (*int)(unsafe.Pointer(&RowsAffected))
  795. _idd16 := *_idPointer
  796. data["result"] = rows
  797. data["LastInsertId"] = idd16
  798. data["RowsAffected"] = _idd16
  799. return data, nil
  800. }
  801. func ExecQueryT(sqlstr string, params []interface{}, tx *xorm.Session) (interface{}, error) {
  802. if err := sqlCheckParam(sqlstr); err != nil {
  803. return 0, err
  804. }
  805. rows, err := tx.SQL(sqlstr, params...).Execute()
  806. if err != nil {
  807. fmt.Println("exe", err)
  808. return nil, err
  809. }
  810. LastInsertId, _ := rows.LastInsertId()
  811. idPointer := (*int)(unsafe.Pointer(&LastInsertId))
  812. idd16 := *idPointer
  813. data := make(map[string]interface{})
  814. RowsAffected, _ := rows.RowsAffected()
  815. _idPointer := (*int)(unsafe.Pointer(&RowsAffected))
  816. _idd16 := *_idPointer
  817. data["LastInsertId"] = strconv.Itoa(idd16)
  818. data["LastInsertIdStr"] = idd16
  819. data["RowsAffected"] = _idd16
  820. data["result"] = rows
  821. return data, nil
  822. }
  823. func ExecQuerys(sqlstr string, params []interface{}) (interface{}, error) {
  824. if err := sqlCheckParam(sqlstr); err != nil {
  825. return 0, err
  826. }
  827. /*for i, v := range params{
  828. _, err := dbs.Exec(sqlstr, v[i]...)
  829. if err != nil {
  830. return nil, err
  831. }
  832. }
  833. */
  834. data := make(map[string]interface{})
  835. data["result"] = nil
  836. return data, nil
  837. }
  838. func getValue(value interface{}, CT string) interface{} {
  839. var v interface{}
  840. val := value
  841. b, ok := val.([]byte)
  842. if ok {
  843. v1 := string(b)
  844. switch CT {
  845. case "TINYINT", "SMALLINT", "MEDIUMINT", "INT", "INTEGER", "BIGINT":
  846. {
  847. v, _ = strconv.ParseInt(v1, 10, 32)
  848. }
  849. case "FLOAT", "DOUBLE", "DECIMAL":
  850. {
  851. cindex := len(val.([]byte)) - strings.Index(v1, ".") - 1
  852. v2, _ := strconv.ParseFloat(v1, 32)
  853. v = toFixed(v2, cindex)
  854. }
  855. case "BOOL":
  856. {
  857. v, _ = strconv.ParseBool(v1)
  858. }
  859. default:
  860. {
  861. v = v1
  862. }
  863. }
  864. } else {
  865. v = val
  866. }
  867. return v
  868. }