session_plus.go 22 KB

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