session_plus.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887
  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. package xorm
  5. import (
  6. "database/sql"
  7. "encoding/json"
  8. "fmt"
  9. "os"
  10. "reflect"
  11. "regexp"
  12. "strings"
  13. "github.com/Chronokeeper/anyxml"
  14. "github.com/xormplus/core"
  15. // "gopkg.in/flosch/pongo2.v3"
  16. )
  17. type ResultBean struct {
  18. Has bool
  19. Result interface{}
  20. Error error
  21. }
  22. func (resultBean *ResultBean) Json() (bool, string, error) {
  23. if resultBean.Error != nil {
  24. return resultBean.Has, "", resultBean.Error
  25. }
  26. if !resultBean.Has {
  27. return resultBean.Has, "", nil
  28. }
  29. result, err := JSONString(resultBean.Result, true)
  30. return resultBean.Has, result, err
  31. }
  32. func (resultBean *ResultBean) GetResult() (bool, interface{}, error) {
  33. return resultBean.Has, resultBean.Result, resultBean.Error
  34. }
  35. func (session *Session) GetFirst(bean interface{}) *ResultBean {
  36. has, err := session.Get(bean)
  37. r := &ResultBean{Has: has, Result: bean, Error: err}
  38. return r
  39. }
  40. func (resultBean *ResultBean) Xml() (bool, string, error) {
  41. if resultBean.Error != nil {
  42. return false, "", resultBean.Error
  43. }
  44. if !resultBean.Has {
  45. return resultBean.Has, "", nil
  46. }
  47. has, result, err := resultBean.Json()
  48. if err != nil {
  49. return false, "", err
  50. }
  51. if !has {
  52. return has, "", nil
  53. }
  54. var anydata = []byte(result)
  55. var i interface{}
  56. err = json.Unmarshal(anydata, &i)
  57. if err != nil {
  58. return false, "", err
  59. }
  60. resultByte, err := anyxml.Xml(i)
  61. if err != nil {
  62. return false, "", err
  63. }
  64. return resultBean.Has, string(resultByte), err
  65. }
  66. func (resultBean *ResultBean) XmlIndent(prefix string, indent string, recordTag string) (bool, string, error) {
  67. if resultBean.Error != nil {
  68. return false, "", resultBean.Error
  69. }
  70. if !resultBean.Has {
  71. return resultBean.Has, "", nil
  72. }
  73. has, result, err := resultBean.Json()
  74. if err != nil {
  75. return false, "", err
  76. }
  77. if !has {
  78. return has, "", nil
  79. }
  80. var anydata = []byte(result)
  81. var i interface{}
  82. err = json.Unmarshal(anydata, &i)
  83. if err != nil {
  84. return false, "", err
  85. }
  86. resultByte, err := anyxml.XmlIndent(i, prefix, indent, recordTag)
  87. if err != nil {
  88. return false, "", err
  89. }
  90. return resultBean.Has, string(resultByte), err
  91. }
  92. type ResultMap struct {
  93. Results []map[string]interface{}
  94. Error error
  95. }
  96. func (resultMap *ResultMap) List() ([]map[string]interface{}, error) {
  97. return resultMap.Results, resultMap.Error
  98. }
  99. func (resultMap *ResultMap) Count() (int, error) {
  100. if resultMap.Error != nil {
  101. return 0, resultMap.Error
  102. }
  103. if resultMap.Results == nil {
  104. return 0, nil
  105. }
  106. return len(resultMap.Results), nil
  107. }
  108. func (resultMap *ResultMap) ListPage(firstResult int, maxResults int) ([]map[string]interface{}, error) {
  109. if resultMap.Error != nil {
  110. return nil, resultMap.Error
  111. }
  112. if resultMap.Results == nil {
  113. return nil, nil
  114. }
  115. if firstResult > maxResults {
  116. return nil, ErrParamsFormat
  117. }
  118. if firstResult < 0 {
  119. return nil, ErrParamsFormat
  120. }
  121. if maxResults < 0 {
  122. return nil, ErrParamsFormat
  123. }
  124. if maxResults > len(resultMap.Results) {
  125. return nil, ErrParamsFormat
  126. }
  127. return resultMap.Results[(firstResult - 1):maxResults], resultMap.Error
  128. }
  129. func (resultMap *ResultMap) Json() (string, error) {
  130. if resultMap.Error != nil {
  131. return "", resultMap.Error
  132. }
  133. return JSONString(resultMap.Results, true)
  134. }
  135. func (resultMap *ResultMap) Xml() (string, error) {
  136. if resultMap.Error != nil {
  137. return "", resultMap.Error
  138. }
  139. results, err := anyxml.Xml(resultMap.Results)
  140. if err != nil {
  141. return "", err
  142. }
  143. return string(results), nil
  144. }
  145. func (resultMap *ResultMap) XmlIndent(prefix string, indent string, recordTag string) (string, error) {
  146. if resultMap.Error != nil {
  147. return "", resultMap.Error
  148. }
  149. results, err := anyxml.XmlIndent(resultMap.Results, prefix, indent, recordTag)
  150. if err != nil {
  151. return "", err
  152. }
  153. return string(results), nil
  154. }
  155. func (resultMap *ResultMap) SaveAsCSV(filename string, headers []string, perm os.FileMode) error {
  156. if resultMap.Error != nil {
  157. return resultMap.Error
  158. }
  159. dataset, err := NewDatasetWithData(headers, resultMap.Results, true)
  160. if err != nil {
  161. return err
  162. }
  163. csv, err := dataset.CSV()
  164. if err != nil {
  165. return err
  166. }
  167. return csv.WriteFile(filename, perm)
  168. }
  169. func (resultMap *ResultMap) SaveAsTSV(filename string, headers []string, perm os.FileMode) error {
  170. if resultMap.Error != nil {
  171. return resultMap.Error
  172. }
  173. dataset, err := NewDatasetWithData(headers, resultMap.Results, true)
  174. if err != nil {
  175. return err
  176. }
  177. tsv, err := dataset.TSV()
  178. if err != nil {
  179. return err
  180. }
  181. return tsv.WriteFile(filename, perm)
  182. }
  183. func (resultMap *ResultMap) SaveAsHTML(filename string, headers []string, perm os.FileMode) error {
  184. if resultMap.Error != nil {
  185. return resultMap.Error
  186. }
  187. dataset, err := NewDatasetWithData(headers, resultMap.Results, true)
  188. if err != nil {
  189. return err
  190. }
  191. html := dataset.HTML()
  192. return html.WriteFile(filename, perm)
  193. }
  194. func (resultMap *ResultMap) SaveAsXML(filename string, headers []string, perm os.FileMode) error {
  195. if resultMap.Error != nil {
  196. return resultMap.Error
  197. }
  198. dataset, err := NewDatasetWithData(headers, resultMap.Results, false)
  199. if err != nil {
  200. return err
  201. }
  202. xml, err := dataset.XML()
  203. if err != nil {
  204. return err
  205. }
  206. return xml.WriteFile(filename, perm)
  207. }
  208. func (resultMap *ResultMap) SaveAsXMLWithTagNamePrefixIndent(tagName string, prifix string, indent string, filename string, headers []string, perm os.FileMode) error {
  209. if resultMap.Error != nil {
  210. return resultMap.Error
  211. }
  212. dataset, err := NewDatasetWithData(headers, resultMap.Results, false)
  213. if err != nil {
  214. return err
  215. }
  216. xml, err := dataset.XMLWithTagNamePrefixIndent(tagName, prifix, indent)
  217. if err != nil {
  218. return err
  219. }
  220. return xml.WriteFile(filename, perm)
  221. }
  222. func (resultMap *ResultMap) SaveAsYAML(filename string, headers []string, perm os.FileMode) error {
  223. if resultMap.Error != nil {
  224. return resultMap.Error
  225. }
  226. dataset, err := NewDatasetWithData(headers, resultMap.Results, false)
  227. if err != nil {
  228. return err
  229. }
  230. yaml, err := dataset.YAML()
  231. if err != nil {
  232. return err
  233. }
  234. return yaml.WriteFile(filename, perm)
  235. }
  236. func (resultMap *ResultMap) SaveAsJSON(filename string, headers []string, perm os.FileMode) error {
  237. if resultMap.Error != nil {
  238. return resultMap.Error
  239. }
  240. dataset, err := NewDatasetWithData(headers, resultMap.Results, false)
  241. if err != nil {
  242. return err
  243. }
  244. json, err := dataset.JSON()
  245. if err != nil {
  246. return err
  247. }
  248. return json.WriteFile(filename, perm)
  249. }
  250. func (resultMap *ResultMap) SaveAsXLSX(filename string, headers []string, perm os.FileMode) error {
  251. if resultMap.Error != nil {
  252. return resultMap.Error
  253. }
  254. dataset, err := NewDatasetWithData(headers, resultMap.Results, true)
  255. if err != nil {
  256. return err
  257. }
  258. xlsx, err := dataset.XLSX()
  259. if err != nil {
  260. return err
  261. }
  262. return xlsx.WriteFile(filename, perm)
  263. }
  264. type ResultStructs struct {
  265. Result interface{}
  266. Error error
  267. }
  268. func (resultStructs *ResultStructs) Json() (string, error) {
  269. if resultStructs.Error != nil {
  270. return "", resultStructs.Error
  271. }
  272. return JSONString(resultStructs.Result, true)
  273. }
  274. func (resultStructs *ResultStructs) Xml() (string, error) {
  275. if resultStructs.Error != nil {
  276. return "", resultStructs.Error
  277. }
  278. result, err := resultStructs.Json()
  279. if err != nil {
  280. return "", err
  281. }
  282. var anydata = []byte(result)
  283. var i interface{}
  284. err = json.Unmarshal(anydata, &i)
  285. if err != nil {
  286. return "", err
  287. }
  288. resultByte, err := anyxml.Xml(i)
  289. if err != nil {
  290. return "", err
  291. }
  292. return string(resultByte), nil
  293. }
  294. func (resultStructs *ResultStructs) XmlIndent(prefix string, indent string, recordTag string) (string, error) {
  295. if resultStructs.Error != nil {
  296. return "", resultStructs.Error
  297. }
  298. result, err := resultStructs.Json()
  299. if err != nil {
  300. return "", err
  301. }
  302. var anydata = []byte(result)
  303. var i interface{}
  304. err = json.Unmarshal(anydata, &i)
  305. if err != nil {
  306. return "", err
  307. }
  308. resultByte, err := anyxml.XmlIndent(i, prefix, indent, recordTag)
  309. if err != nil {
  310. return "", err
  311. }
  312. return string(resultByte), nil
  313. }
  314. func (session *Session) SqlMapClient(sqlTagName string, args ...interface{}) *Session {
  315. return session.Sql(session.engine.SqlMap.Sql[sqlTagName], args...)
  316. }
  317. //func (session *Session) SqlTemplateClient(sqlTagName string, args ...interface{}) *Session {
  318. // session.isSqlFunc = true
  319. // if session.engine.sqlTemplate.Template[sqlTagName] == nil {
  320. // if len(args) == 0 {
  321. // return session.Sql("")
  322. // } else {
  323. // map1 := args[0].(*map[string]interface{})
  324. // return session.Sql("", map1)
  325. // }
  326. // }
  327. // if len(args) == 0 {
  328. // parmap := &pongo2.Context{"1": 1}
  329. // sql, err := session.engine.sqlTemplate.Template[sqlTagName].Execute(*parmap)
  330. // if err != nil {
  331. // session.engine.logger.Error(err)
  332. // }
  333. // return session.Sql(sql)
  334. // } else {
  335. // map1 := args[0].(*map[string]interface{})
  336. // sql, err := session.engine.sqlTemplate.Template[sqlTagName].Execute(*map1)
  337. // if err != nil {
  338. // session.engine.logger.Error(err)
  339. // }
  340. // return session.Sql(sql, map1)
  341. // }
  342. //}
  343. func (session *Session) SqlTemplateClient(sqlTagName string, args ...interface{}) *Session {
  344. session.isSqlFunc = true
  345. sql, err := session.engine.SqlTemplate.Execute(sqlTagName, args...)
  346. if err != nil {
  347. session.engine.logger.Error(err)
  348. }
  349. if len(args) == 0 {
  350. return session.Sql(sql)
  351. } else {
  352. map1 := args[0].(*map[string]interface{})
  353. return session.Sql(sql, map1)
  354. }
  355. }
  356. func (session *Session) Search(rowsSlicePtr interface{}, condiBean ...interface{}) *ResultStructs {
  357. err := session.Find(rowsSlicePtr, condiBean...)
  358. r := &ResultStructs{Result: rowsSlicePtr, Error: err}
  359. return r
  360. }
  361. func (session *Session) genSelectSql(dialect core.Dialect, rownumber string) string {
  362. var sql = session.statement.RawSQL
  363. var orderBys = session.statement.OrderStr
  364. if dialect.DBType() != core.MSSQL && dialect.DBType() != core.ORACLE {
  365. if session.statement.Start > 0 {
  366. sql = fmt.Sprintf("%v LIMIT %v OFFSET %v", sql, session.statement.LimitN, session.statement.Start)
  367. } else if session.statement.LimitN > 0 {
  368. sql = fmt.Sprintf("%v LIMIT %v", sql, session.statement.LimitN)
  369. }
  370. } else if dialect.DBType() == core.ORACLE {
  371. if session.statement.Start != 0 || session.statement.LimitN != 0 {
  372. sql = fmt.Sprintf("SELECT aat.* FROM (SELECT at.*,ROWNUM %v FROM (%v) at WHERE ROWNUM <= %d) aat WHERE %v > %d",
  373. rownumber, sql, session.statement.Start+session.statement.LimitN, rownumber, session.statement.Start)
  374. }
  375. } else {
  376. keepSelect := false
  377. var fullQuery string
  378. if session.statement.Start > 0 {
  379. fullQuery = fmt.Sprintf("SELECT sq.* FROM (SELECT ROW_NUMBER() OVER (ORDER BY %v) AS %v,", orderBys, rownumber)
  380. } else if session.statement.LimitN > 0 {
  381. fullQuery = fmt.Sprintf("SELECT TOP %d", session.statement.LimitN)
  382. } else {
  383. keepSelect = true
  384. }
  385. if !keepSelect {
  386. expr := `^\s*SELECT\s*`
  387. reg, err := regexp.Compile(expr)
  388. if err != nil {
  389. fmt.Println(err)
  390. }
  391. sql = strings.ToUpper(sql)
  392. if reg.MatchString(sql) {
  393. str := reg.FindAllString(sql, -1)
  394. fullQuery = fmt.Sprintf("%v %v", fullQuery, sql[len(str[0]):])
  395. }
  396. }
  397. if session.statement.Start > 0 {
  398. // T-SQL offset starts with 1, not like MySQL with 0;
  399. if session.statement.LimitN > 0 {
  400. fullQuery = fmt.Sprintf("%v) AS sq WHERE %v BETWEEN %d AND %d", fullQuery, rownumber, session.statement.Start+1, session.statement.Start+session.statement.LimitN)
  401. } else {
  402. fullQuery = fmt.Sprintf("%v) AS sq WHERE %v >= %d", fullQuery, rownumber, session.statement.Start+1)
  403. }
  404. } else {
  405. fullQuery = fmt.Sprintf("%v ORDER BY %v", fullQuery, orderBys)
  406. }
  407. if keepSelect {
  408. if len(orderBys) > 0 {
  409. sql = fmt.Sprintf("%v ORDER BY %v", sql, orderBys)
  410. }
  411. } else {
  412. sql = fullQuery
  413. }
  414. }
  415. return sql
  416. }
  417. // Exec a raw sql and return records as ResultMap
  418. func (session *Session) Query() *ResultMap {
  419. defer session.resetStatement()
  420. if session.isAutoClose {
  421. defer session.Close()
  422. }
  423. var dialect = session.statement.Engine.Dialect()
  424. rownumber := "xorm" + NewShortUUID().String()
  425. sql := session.genSelectSql(dialect, rownumber)
  426. params := session.statement.RawParams
  427. i := len(params)
  428. var result []map[string]interface{}
  429. var err error
  430. if i == 1 {
  431. vv := reflect.ValueOf(params[0])
  432. if vv.Kind() != reflect.Ptr || vv.Elem().Kind() != reflect.Map {
  433. result, err = session.queryAll(sql, params...)
  434. } else {
  435. result, err = session.queryAllByMap(sql, params[0])
  436. }
  437. } else {
  438. result, err = session.queryAll(sql, params...)
  439. }
  440. if dialect.DBType() == core.MSSQL {
  441. if session.statement.Start > 0 {
  442. for i, _ := range result {
  443. delete(result[i], rownumber)
  444. }
  445. }
  446. } else if dialect.DBType() == core.ORACLE {
  447. if session.statement.Start != 0 || session.statement.LimitN != 0 {
  448. for i, _ := range result {
  449. delete(result[i], rownumber)
  450. }
  451. }
  452. }
  453. r := &ResultMap{Results: result, Error: err}
  454. return r
  455. }
  456. // Exec a raw sql and return records as ResultMap
  457. func (session *Session) QueryWithDateFormat(dateFormat string) *ResultMap {
  458. defer session.resetStatement()
  459. if session.isAutoClose {
  460. defer session.Close()
  461. }
  462. var dialect = session.statement.Engine.Dialect()
  463. rownumber := "xorm" + NewShortUUID().String()
  464. sql := session.genSelectSql(dialect, rownumber)
  465. params := session.statement.RawParams
  466. i := len(params)
  467. var result []map[string]interface{}
  468. var err error
  469. if i == 1 {
  470. vv := reflect.ValueOf(params[0])
  471. if vv.Kind() != reflect.Ptr || vv.Elem().Kind() != reflect.Map {
  472. result, err = session.queryAllWithDateFormat(dateFormat, sql, params...)
  473. } else {
  474. result, err = session.queryAllByMapWithDateFormat(dateFormat, sql, params[0])
  475. }
  476. } else {
  477. result, err = session.queryAllWithDateFormat(dateFormat, sql, params...)
  478. }
  479. if dialect.DBType() == core.MSSQL {
  480. if session.statement.Start > 0 {
  481. for i, _ := range result {
  482. delete(result[i], rownumber)
  483. }
  484. }
  485. } else if dialect.DBType() == core.ORACLE {
  486. if session.statement.Start != 0 || session.statement.LimitN != 0 {
  487. for i, _ := range result {
  488. delete(result[i], rownumber)
  489. }
  490. }
  491. }
  492. r := &ResultMap{Results: result, Error: err}
  493. return r
  494. }
  495. // Execute raw sql
  496. func (session *Session) Execute() (sql.Result, error) {
  497. defer session.resetStatement()
  498. if session.isAutoClose {
  499. defer session.Close()
  500. }
  501. sqlStr := session.statement.RawSQL
  502. params := session.statement.RawParams
  503. i := len(params)
  504. if i == 1 {
  505. vv := reflect.ValueOf(params[0])
  506. if vv.Kind() != reflect.Ptr || vv.Elem().Kind() != reflect.Map {
  507. return session.exec(sqlStr, params...)
  508. } else {
  509. sqlStr1, args, _ := core.MapToSlice(sqlStr, params[0])
  510. return session.exec(sqlStr1, args...)
  511. }
  512. } else {
  513. return session.exec(sqlStr, params...)
  514. }
  515. }
  516. // =============================
  517. // for Object
  518. // =============================
  519. func (session *Session) queryAll(sqlStr string, paramStr ...interface{}) (resultsSlice []map[string]interface{}, err error) {
  520. session.queryPreprocess(&sqlStr, paramStr...)
  521. if session.isAutoCommit {
  522. return query3(session.DB(), sqlStr, paramStr...)
  523. }
  524. return txQuery3(session.tx, sqlStr, paramStr...)
  525. }
  526. func (session *Session) queryAllByMap(sqlStr string, paramMap interface{}) (resultsSlice []map[string]interface{}, err error) {
  527. sqlStr1, param, _ := core.MapToSlice(sqlStr, paramMap)
  528. session.queryPreprocess(&sqlStr1, param...)
  529. if session.isAutoCommit {
  530. return query3(session.DB(), sqlStr1, param...)
  531. }
  532. return txQuery3(session.tx, sqlStr1, param...)
  533. }
  534. func (session *Session) queryAllByMapWithDateFormat(dateFormat string, sqlStr string, paramMap interface{}) (resultsSlice []map[string]interface{}, err error) {
  535. sqlStr1, param, _ := core.MapToSlice(sqlStr, paramMap)
  536. session.queryPreprocess(&sqlStr1, param...)
  537. if session.isAutoCommit {
  538. return query3WithDateFormat(session.DB(), dateFormat, sqlStr1, param...)
  539. }
  540. return txQuery3WithDateFormat(session.tx, dateFormat, sqlStr1, param...)
  541. }
  542. func (session *Session) queryAllWithDateFormat(dateFormat string, sqlStr string, paramStr ...interface{}) (resultsSlice []map[string]interface{}, err error) {
  543. session.queryPreprocess(&sqlStr, paramStr...)
  544. if session.isAutoCommit {
  545. return query3WithDateFormat(session.DB(), dateFormat, sqlStr, paramStr...)
  546. }
  547. return txQuery3WithDateFormat(session.tx, dateFormat, sqlStr, paramStr...)
  548. }
  549. func (session *Session) queryAllToJsonString(sql string, paramStr ...interface{}) (string, error) {
  550. results, err := session.queryAll(sql, paramStr...)
  551. if err != nil {
  552. return "", err
  553. }
  554. return JSONString(results, true)
  555. }
  556. func (session *Session) queryAllToXmlString(sql string, paramStr ...interface{}) (string, error) {
  557. resultMap, err := session.queryAll(sql, paramStr...)
  558. if err != nil {
  559. return "", err
  560. }
  561. results, err := anyxml.Xml(resultMap)
  562. if err != nil {
  563. return "", err
  564. }
  565. return string(results), nil
  566. }
  567. func (session *Session) queryAllToXmlIndentString(sql string, prefix string, indent string, paramStr ...interface{}) (string, error) {
  568. resultSlice, err := session.queryAll(sql, paramStr...)
  569. if err != nil {
  570. return "", err
  571. }
  572. results, err := anyxml.XmlIndent(resultSlice, prefix, indent, "result")
  573. if err != nil {
  574. return "", err
  575. }
  576. return string(results), nil
  577. }
  578. func (session *Session) queryAllToXmlStringWithDateFormat(dateFormat string, sql string, paramStr ...interface{}) (string, error) {
  579. resultSlice, err := session.queryAll(sql, paramStr...)
  580. if err != nil {
  581. return "", err
  582. }
  583. results, err := anyxml.XmlWithDateFormat(dateFormat, resultSlice)
  584. if err != nil {
  585. return "", err
  586. }
  587. return string(results), nil
  588. }
  589. func (session *Session) queryAllToXmlIndentStringWithDateFormat(dateFormat string, sql string, prefix string, indent string, paramStr ...interface{}) (string, error) {
  590. resultSlice, err := session.queryAll(sql, paramStr...)
  591. if err != nil {
  592. return "", err
  593. }
  594. results, err := anyxml.XmlIndentWithDateFormat(dateFormat, resultSlice, prefix, indent, "results")
  595. if err != nil {
  596. return "", err
  597. }
  598. return string(results), nil
  599. }
  600. func (session *Session) queryAllByMapToJsonString(sql string, paramMap interface{}) (string, error) {
  601. results, err := session.queryAllByMap(sql, paramMap)
  602. if err != nil {
  603. return "", err
  604. }
  605. return JSONString(results, true)
  606. }
  607. func (session *Session) queryAllByMapToJsonStringWithDateFormat(dateFormat string, sql string, paramMap interface{}) (string, error) {
  608. results, err := session.queryAllByMapWithDateFormat(dateFormat, sql, paramMap)
  609. if err != nil {
  610. return "", err
  611. }
  612. return JSONString(results, true)
  613. }
  614. func (session *Session) queryAllToJsonStringWithDateFormat(dateFormat string, sql string, paramStr ...interface{}) (string, error) {
  615. results, err := session.queryAllWithDateFormat(dateFormat, sql, paramStr...)
  616. if err != nil {
  617. return "", err
  618. }
  619. return JSONString(results, true)
  620. }
  621. func (session *Session) queryPreprocessByMap(sqlStr *string, paramMap interface{}) {
  622. re := regexp.MustCompile(`[?](\w+)`)
  623. query := *sqlStr
  624. names := make(map[string]int)
  625. var i int
  626. query = re.ReplaceAllStringFunc(query, func(src string) string {
  627. names[src[1:]] = i
  628. i += 1
  629. return "?"
  630. })
  631. for _, filter := range session.engine.dialect.Filters() {
  632. query = filter.Do(query, session.engine.dialect, session.statement.RefTable)
  633. }
  634. *sqlStr = query
  635. session.engine.logSQL(*sqlStr, paramMap)
  636. }
  637. func (session *Session) Sqls(sqls interface{}, parmas ...interface{}) *SqlsExecutor {
  638. sqlsExecutor := new(SqlsExecutor)
  639. switch sqls.(type) {
  640. case string:
  641. sqlsExecutor.sqls = sqls.(string)
  642. case []string:
  643. sqlsExecutor.sqls = sqls.([]string)
  644. case map[string]string:
  645. sqlsExecutor.sqls = sqls.(map[string]string)
  646. default:
  647. sqlsExecutor.sqls = nil
  648. sqlsExecutor.err = ErrParamsType
  649. }
  650. if len(parmas) == 0 {
  651. sqlsExecutor.parmas = nil
  652. }
  653. if len(parmas) > 1 {
  654. sqlsExecutor.parmas = nil
  655. sqlsExecutor.err = ErrParamsType
  656. }
  657. if len(parmas) == 1 {
  658. switch parmas[0].(type) {
  659. case map[string]interface{}:
  660. sqlsExecutor.parmas = parmas[0].(map[string]interface{})
  661. case []map[string]interface{}:
  662. sqlsExecutor.parmas = parmas[0].([]map[string]interface{})
  663. case map[string]map[string]interface{}:
  664. sqlsExecutor.parmas = parmas[0].(map[string]map[string]interface{})
  665. default:
  666. sqlsExecutor.parmas = nil
  667. sqlsExecutor.err = ErrParamsType
  668. }
  669. }
  670. sqlsExecutor.session = session
  671. return sqlsExecutor
  672. }
  673. func (session *Session) SqlMapsClient(sqlkeys interface{}, parmas ...interface{}) *SqlMapsExecutor {
  674. sqlMapsExecutor := new(SqlMapsExecutor)
  675. switch sqlkeys.(type) {
  676. case string:
  677. sqlMapsExecutor.sqlkeys = sqlkeys.(string)
  678. case []string:
  679. sqlMapsExecutor.sqlkeys = sqlkeys.([]string)
  680. case map[string]string:
  681. sqlMapsExecutor.sqlkeys = sqlkeys.(map[string]string)
  682. default:
  683. sqlMapsExecutor.sqlkeys = nil
  684. sqlMapsExecutor.err = ErrParamsType
  685. }
  686. if len(parmas) == 0 {
  687. sqlMapsExecutor.parmas = nil
  688. }
  689. if len(parmas) > 1 {
  690. sqlMapsExecutor.parmas = nil
  691. sqlMapsExecutor.err = ErrParamsType
  692. }
  693. if len(parmas) == 1 {
  694. switch parmas[0].(type) {
  695. case map[string]interface{}:
  696. sqlMapsExecutor.parmas = parmas[0].(map[string]interface{})
  697. case []map[string]interface{}:
  698. sqlMapsExecutor.parmas = parmas[0].([]map[string]interface{})
  699. case map[string]map[string]interface{}:
  700. sqlMapsExecutor.parmas = parmas[0].(map[string]map[string]interface{})
  701. default:
  702. sqlMapsExecutor.parmas = nil
  703. sqlMapsExecutor.err = ErrParamsType
  704. }
  705. }
  706. sqlMapsExecutor.session = session
  707. return sqlMapsExecutor
  708. }
  709. func (session *Session) SqlTemplatesClient(sqlkeys interface{}, parmas ...interface{}) *SqlTemplatesExecutor {
  710. sqlTemplatesExecutor := new(SqlTemplatesExecutor)
  711. switch sqlkeys.(type) {
  712. case string:
  713. sqlTemplatesExecutor.sqlkeys = sqlkeys.(string)
  714. case []string:
  715. sqlTemplatesExecutor.sqlkeys = sqlkeys.([]string)
  716. case map[string]string:
  717. sqlTemplatesExecutor.sqlkeys = sqlkeys.(map[string]string)
  718. default:
  719. sqlTemplatesExecutor.sqlkeys = nil
  720. sqlTemplatesExecutor.err = ErrParamsType
  721. }
  722. if len(parmas) == 0 {
  723. sqlTemplatesExecutor.parmas = nil
  724. }
  725. if len(parmas) > 1 {
  726. sqlTemplatesExecutor.parmas = nil
  727. sqlTemplatesExecutor.err = ErrParamsType
  728. }
  729. if len(parmas) == 1 {
  730. switch parmas[0].(type) {
  731. case map[string]interface{}:
  732. sqlTemplatesExecutor.parmas = parmas[0].(map[string]interface{})
  733. case []map[string]interface{}:
  734. sqlTemplatesExecutor.parmas = parmas[0].([]map[string]interface{})
  735. case map[string]map[string]interface{}:
  736. sqlTemplatesExecutor.parmas = parmas[0].(map[string]map[string]interface{})
  737. default:
  738. sqlTemplatesExecutor.parmas = nil
  739. sqlTemplatesExecutor.err = ErrParamsType
  740. }
  741. }
  742. sqlTemplatesExecutor.session = session
  743. return sqlTemplatesExecutor
  744. }