sessionplus.go 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011
  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. "errors"
  9. "fmt"
  10. "reflect"
  11. "regexp"
  12. "strconv"
  13. "strings"
  14. "time"
  15. // "unsafe"
  16. "runtime"
  17. "github.com/Chronokeeper/anyxml"
  18. "github.com/xormplus/core"
  19. )
  20. type ResultBean struct {
  21. Has bool
  22. Result interface{}
  23. Error error
  24. }
  25. func (resultBean ResultBean) Json() (bool, string, error) {
  26. if resultBean.Error != nil {
  27. return resultBean.Has, "", resultBean.Error
  28. }
  29. if !resultBean.Has {
  30. return resultBean.Has, "", nil
  31. }
  32. result, err := JSONString(resultBean.Result, true)
  33. return resultBean.Has, result, err
  34. }
  35. func (resultBean ResultBean) GetResult() (bool, interface{}, error) {
  36. return resultBean.Has, resultBean.Result, resultBean.Error
  37. }
  38. func (session *Session) GetFirst(bean interface{}) ResultBean {
  39. has, err := session.Get(bean)
  40. r := ResultBean{Has: has, Result: bean, Error: err}
  41. return r
  42. }
  43. func (resultBean ResultBean) Xml() (bool, string, error) {
  44. if resultBean.Error != nil {
  45. return false, "", resultBean.Error
  46. }
  47. if !resultBean.Has {
  48. return resultBean.Has, "", nil
  49. }
  50. has, result, err := resultBean.Json()
  51. if err != nil {
  52. return false, "", err
  53. }
  54. if !has {
  55. return has, "", nil
  56. }
  57. var anydata = []byte(result) //str2byte(result)
  58. var i interface{}
  59. err = json.Unmarshal(anydata, &i)
  60. if err != nil {
  61. return false, "", err
  62. }
  63. resultByte, err := anyxml.Xml(i)
  64. if err != nil {
  65. return false, "", err
  66. }
  67. return resultBean.Has, string(resultByte), err
  68. }
  69. //func str2byte(s string) []byte {
  70. // x := (*[2]uintptr)(unsafe.Pointer(&s))
  71. // h := [3]uintptr{x[0], x[1], x[1]}
  72. // return *(*[]byte)((unsafe.Pointer(&h)))
  73. //}
  74. func (resultBean ResultBean) XmlIndent(prefix string, indent string, recordTag string) (bool, string, error) {
  75. if resultBean.Error != nil {
  76. return false, "", resultBean.Error
  77. }
  78. if !resultBean.Has {
  79. return resultBean.Has, "", nil
  80. }
  81. has, result, err := resultBean.Json()
  82. if err != nil {
  83. return false, "", err
  84. }
  85. if !has {
  86. return has, "", nil
  87. }
  88. var anydata = []byte(result)
  89. var i interface{}
  90. err = json.Unmarshal(anydata, &i)
  91. if err != nil {
  92. return false, "", err
  93. }
  94. resultByte, err := anyxml.XmlIndent(i, prefix, indent, recordTag)
  95. if err != nil {
  96. return false, "", err
  97. }
  98. return resultBean.Has, string(resultByte), err
  99. }
  100. type ResultMap struct {
  101. Results []map[string]interface{}
  102. Error error
  103. }
  104. func (resultMap ResultMap) GetResults() ([]map[string]interface{}, error) {
  105. return resultMap.Results, resultMap.Error
  106. }
  107. func (resultMap ResultMap) Json() (string, error) {
  108. if resultMap.Error != nil {
  109. return "", resultMap.Error
  110. }
  111. return JSONString(resultMap.Results, true)
  112. }
  113. func (resultMap ResultMap) Xml() (string, error) {
  114. if resultMap.Error != nil {
  115. return "", resultMap.Error
  116. }
  117. results, err := anyxml.Xml(resultMap.Results)
  118. if err != nil {
  119. return "", err
  120. }
  121. return string(results), nil
  122. }
  123. func (resultMap ResultMap) XmlIndent(prefix string, indent string, recordTag string) (string, error) {
  124. if resultMap.Error != nil {
  125. return "", resultMap.Error
  126. }
  127. results, err := anyxml.XmlIndent(resultMap.Results, prefix, indent, recordTag)
  128. if err != nil {
  129. return "", err
  130. }
  131. return string(results), nil
  132. }
  133. type ResultStructs struct {
  134. Result interface{}
  135. Error error
  136. }
  137. func (resultStructs ResultStructs) Json() (string, error) {
  138. if resultStructs.Error != nil {
  139. return "", resultStructs.Error
  140. }
  141. return JSONString(resultStructs.Result, true)
  142. }
  143. func (resultStructs ResultStructs) Xml() (string, error) {
  144. if resultStructs.Error != nil {
  145. return "", resultStructs.Error
  146. }
  147. result, err := resultStructs.Json()
  148. if err != nil {
  149. return "", err
  150. }
  151. var anydata = []byte(result)
  152. var i interface{}
  153. err = json.Unmarshal(anydata, &i)
  154. if err != nil {
  155. return "", err
  156. }
  157. resultByte, err := anyxml.Xml(i)
  158. if err != nil {
  159. return "", err
  160. }
  161. return string(resultByte), nil
  162. }
  163. func (resultStructs ResultStructs) XmlIndent(prefix string, indent string, recordTag string) (string, error) {
  164. if resultStructs.Error != nil {
  165. return "", resultStructs.Error
  166. }
  167. result, err := resultStructs.Json()
  168. if err != nil {
  169. return "", err
  170. }
  171. var anydata = []byte(result)
  172. var i interface{}
  173. err = json.Unmarshal(anydata, &i)
  174. if err != nil {
  175. return "", err
  176. }
  177. resultByte, err := anyxml.XmlIndent(i, prefix, indent, recordTag)
  178. if err != nil {
  179. return "", err
  180. }
  181. return string(resultByte), nil
  182. }
  183. func (session *Session) SqlMapClient(sqlTagName string, args ...interface{}) *Session {
  184. return session.Sql(session.Engine.sqlMap.Sql[sqlTagName], args...)
  185. }
  186. func (session *Session) SqlTemplateClient(sqlTagName string, args ...interface{}) *Session {
  187. map1 := args[0].(map[string]interface{})
  188. if session.Engine.sqlTemplate.Template[sqlTagName] == nil {
  189. return session.Sql("", &map1)
  190. }
  191. sql, err := session.Engine.sqlTemplate.Template[sqlTagName].Execute(map1)
  192. if err != nil {
  193. session.Engine.logger.Error(err)
  194. }
  195. return session.Sql(sql, &map1)
  196. }
  197. func (session *Session) Search(rowsSlicePtr interface{}, condiBean ...interface{}) ResultStructs {
  198. err := session.Find(rowsSlicePtr, condiBean...)
  199. r := ResultStructs{Result: rowsSlicePtr, Error: err}
  200. return r
  201. }
  202. // Exec a raw sql and return records as []map[string]interface{}
  203. func (session *Session) Query() ResultMap {
  204. sql := session.Statement.RawSQL
  205. params := session.Statement.RawParams
  206. i := len(params)
  207. var result []map[string]interface{}
  208. var err error
  209. if i == 1 {
  210. vv := reflect.ValueOf(params[0])
  211. if vv.Kind() != reflect.Ptr || vv.Elem().Kind() != reflect.Map {
  212. result, err = session.queryAll(sql, params...)
  213. } else {
  214. result, err = session.queryAllByMap(sql, params[0])
  215. }
  216. } else {
  217. result, err = session.queryAll(sql, params...)
  218. }
  219. r := ResultMap{Results: result, Error: err}
  220. return r
  221. }
  222. // Exec a raw sql and return records as []map[string]interface{}
  223. func (session *Session) QueryWithDateFormat(dateFormat string) ResultMap {
  224. sql := session.Statement.RawSQL
  225. params := session.Statement.RawParams
  226. i := len(params)
  227. var result []map[string]interface{}
  228. var err error
  229. if i == 1 {
  230. vv := reflect.ValueOf(params[0])
  231. if vv.Kind() != reflect.Ptr || vv.Elem().Kind() != reflect.Map {
  232. result, err = session.queryAllWithDateFormat(dateFormat, sql, params...)
  233. } else {
  234. result, err = session.queryAllByMapWithDateFormat(dateFormat, sql, params[0])
  235. }
  236. } else {
  237. result, err = session.queryAllWithDateFormat(dateFormat, sql, params...)
  238. }
  239. r := ResultMap{Results: result, Error: err}
  240. return r
  241. }
  242. // Execute raw sql
  243. func (session *Session) Execute() (sql.Result, error) {
  244. sqlStr := session.Statement.RawSQL
  245. params := session.Statement.RawParams
  246. defer session.resetStatement()
  247. if session.IsAutoClose {
  248. defer session.Close()
  249. }
  250. i := len(params)
  251. if i == 1 {
  252. vv := reflect.ValueOf(params[0])
  253. if vv.Kind() != reflect.Ptr || vv.Elem().Kind() != reflect.Map {
  254. return session.exec(sqlStr, params...)
  255. } else {
  256. sqlStr1, args, _ := core.MapToSlice(sqlStr, params[0])
  257. return session.exec(sqlStr1, args...)
  258. }
  259. } else {
  260. return session.exec(sqlStr, params...)
  261. }
  262. }
  263. // =============================
  264. // for Object
  265. // =============================
  266. func (session *Session) queryAll(sqlStr string, paramStr ...interface{}) (resultsSlice []map[string]interface{}, err error) {
  267. session.queryPreprocess(&sqlStr, paramStr...)
  268. if session.IsAutoCommit {
  269. return query3(session.DB(), sqlStr, paramStr...)
  270. }
  271. return txQuery3(session.Tx, sqlStr, paramStr...)
  272. }
  273. func (session *Session) queryAllByMap(sqlStr string, paramMap interface{}) (resultsSlice []map[string]interface{}, err error) {
  274. sqlStr1, param, _ := core.MapToSlice(sqlStr, paramMap)
  275. session.queryPreprocess(&sqlStr1, param...)
  276. if session.IsAutoCommit {
  277. return query3(session.DB(), sqlStr1, param...)
  278. }
  279. return txQuery3(session.Tx, sqlStr1, param...)
  280. }
  281. func (session *Session) queryAllByMapWithDateFormat(dateFormat string, sqlStr string, paramMap interface{}) (resultsSlice []map[string]interface{}, err error) {
  282. sqlStr1, param, _ := core.MapToSlice(sqlStr, paramMap)
  283. session.queryPreprocess(&sqlStr1, param...)
  284. if session.IsAutoCommit {
  285. return query3WithDateFormat(session.DB(), dateFormat, sqlStr1, param...)
  286. }
  287. return txQuery3WithDateFormat(session.Tx, dateFormat, sqlStr1, param...)
  288. }
  289. func (session *Session) queryAllWithDateFormat(dateFormat string, sqlStr string, paramStr ...interface{}) (resultsSlice []map[string]interface{}, err error) {
  290. session.queryPreprocess(&sqlStr, paramStr...)
  291. if session.IsAutoCommit {
  292. return query3WithDateFormat(session.DB(), dateFormat, sqlStr, paramStr...)
  293. }
  294. return txQuery3WithDateFormat(session.Tx, dateFormat, sqlStr, paramStr...)
  295. }
  296. func (session *Session) queryAllToJsonString(sql string, paramStr ...interface{}) (string, error) {
  297. results, err := session.queryAll(sql, paramStr...)
  298. if err != nil {
  299. return "", err
  300. }
  301. return JSONString(results, true)
  302. }
  303. func (session *Session) queryAllToXmlString(sql string, paramStr ...interface{}) (string, error) {
  304. resultMap, err := session.queryAll(sql, paramStr...)
  305. if err != nil {
  306. return "", err
  307. }
  308. results, err := anyxml.Xml(resultMap)
  309. if err != nil {
  310. return "", err
  311. }
  312. return string(results), nil
  313. }
  314. func (session *Session) queryAllToXmlIndentString(sql string, prefix string, indent string, paramStr ...interface{}) (string, error) {
  315. resultSlice, err := session.queryAll(sql, paramStr...)
  316. if err != nil {
  317. return "", err
  318. }
  319. results, err := anyxml.XmlIndent(resultSlice, prefix, indent, "result")
  320. if err != nil {
  321. return "", err
  322. }
  323. return string(results), nil
  324. }
  325. func (session *Session) queryAllToXmlStringWithDateFormat(dateFormat string, sql string, paramStr ...interface{}) (string, error) {
  326. resultSlice, err := session.queryAll(sql, paramStr...)
  327. if err != nil {
  328. return "", err
  329. }
  330. results, err := anyxml.XmlWithDateFormat(dateFormat, resultSlice)
  331. if err != nil {
  332. return "", err
  333. }
  334. return string(results), nil
  335. }
  336. func (session *Session) queryAllToXmlIndentStringWithDateFormat(dateFormat string, sql string, prefix string, indent string, paramStr ...interface{}) (string, error) {
  337. resultSlice, err := session.queryAll(sql, paramStr...)
  338. if err != nil {
  339. return "", err
  340. }
  341. results, err := anyxml.XmlIndentWithDateFormat(dateFormat, resultSlice, prefix, indent, "results")
  342. if err != nil {
  343. return "", err
  344. }
  345. return string(results), nil
  346. }
  347. func (session *Session) queryAllByMapToJsonString(sql string, paramMap interface{}) (string, error) {
  348. results, err := session.queryAllByMap(sql, paramMap)
  349. if err != nil {
  350. return "", err
  351. }
  352. return JSONString(results, true)
  353. }
  354. func (session *Session) queryAllByMapToJsonStringWithDateFormat(dateFormat string, sql string, paramMap interface{}) (string, error) {
  355. results, err := session.queryAllByMapWithDateFormat(dateFormat, sql, paramMap)
  356. if err != nil {
  357. return "", err
  358. }
  359. return JSONString(results, true)
  360. }
  361. func (session *Session) queryAllToJsonStringWithDateFormat(dateFormat string, sql string, paramStr ...interface{}) (string, error) {
  362. results, err := session.queryAllWithDateFormat(dateFormat, sql, paramStr...)
  363. if err != nil {
  364. return "", err
  365. }
  366. return JSONString(results, true)
  367. }
  368. func (session *Session) row2BeanWithDateFormat(dateFormat string, rows *core.Rows, fields []string, fieldsCount int, bean interface{}) error {
  369. dataStruct := rValue(bean)
  370. if dataStruct.Kind() != reflect.Struct {
  371. return errors.New("Expected a pointer to a struct")
  372. }
  373. table := session.Engine.autoMapType(dataStruct)
  374. return session._row2BeanWithDateFormat(dateFormat, rows, fields, fieldsCount, bean, &dataStruct, table)
  375. }
  376. func (session *Session) _row2BeanWithDateFormat(dateFormat string, rows *core.Rows, fields []string, fieldsCount int, bean interface{}, dataStruct *reflect.Value, table *core.Table) error {
  377. scanResults := make([]interface{}, fieldsCount)
  378. for i := 0; i < len(fields); i++ {
  379. var cell interface{}
  380. scanResults[i] = &cell
  381. }
  382. if err := rows.Scan(scanResults...); err != nil {
  383. return err
  384. }
  385. if b, hasBeforeSet := bean.(BeforeSetProcessor); hasBeforeSet {
  386. for ii, key := range fields {
  387. b.BeforeSet(key, Cell(scanResults[ii].(*interface{})))
  388. }
  389. }
  390. defer func() {
  391. if b, hasAfterSet := bean.(AfterSetProcessor); hasAfterSet {
  392. for ii, key := range fields {
  393. b.AfterSet(key, Cell(scanResults[ii].(*interface{})))
  394. }
  395. }
  396. }()
  397. var tempMap = make(map[string]int)
  398. for ii, key := range fields {
  399. var idx int
  400. var ok bool
  401. var lKey = strings.ToLower(key)
  402. if idx, ok = tempMap[lKey]; !ok {
  403. idx = 0
  404. } else {
  405. idx = idx + 1
  406. }
  407. tempMap[lKey] = idx
  408. if fieldValue := session.getField(dataStruct, key, table, idx); fieldValue != nil {
  409. rawValue := reflect.Indirect(reflect.ValueOf(scanResults[ii]))
  410. // if row is null then ignore
  411. if rawValue.Interface() == nil {
  412. continue
  413. }
  414. if fieldValue.CanAddr() {
  415. if structConvert, ok := fieldValue.Addr().Interface().(core.Conversion); ok {
  416. if data, err := value2Bytes(&rawValue); err == nil {
  417. structConvert.FromDB(data)
  418. } else {
  419. session.Engine.logger.Error(err)
  420. }
  421. continue
  422. }
  423. }
  424. if _, ok := fieldValue.Interface().(core.Conversion); ok {
  425. if data, err := value2Bytes(&rawValue); err == nil {
  426. if fieldValue.Kind() == reflect.Ptr && fieldValue.IsNil() {
  427. fieldValue.Set(reflect.New(fieldValue.Type().Elem()))
  428. }
  429. fieldValue.Interface().(core.Conversion).FromDB(data)
  430. } else {
  431. session.Engine.logger.Error(err)
  432. }
  433. continue
  434. }
  435. rawValueType := reflect.TypeOf(rawValue.Interface())
  436. vv := reflect.ValueOf(rawValue.Interface())
  437. fieldType := fieldValue.Type()
  438. hasAssigned := false
  439. col := table.GetColumnIdx(key, idx)
  440. if col.SQLType.IsJson() {
  441. var bs []byte
  442. if rawValueType.Kind() == reflect.String {
  443. bs = []byte(vv.String())
  444. } else if rawValueType.ConvertibleTo(core.BytesType) {
  445. bs = vv.Bytes()
  446. } else {
  447. return fmt.Errorf("unsupported database data type: %s %v", key, rawValueType.Kind())
  448. }
  449. hasAssigned = true
  450. if len(bs) > 0 {
  451. if fieldValue.CanAddr() {
  452. err := json.Unmarshal(bs, fieldValue.Addr().Interface())
  453. if err != nil {
  454. session.Engine.logger.Error(key, err)
  455. return err
  456. }
  457. } else {
  458. x := reflect.New(fieldType)
  459. err := json.Unmarshal(bs, x.Interface())
  460. if err != nil {
  461. session.Engine.logger.Error(key, err)
  462. return err
  463. }
  464. fieldValue.Set(x.Elem())
  465. }
  466. }
  467. continue
  468. }
  469. switch fieldType.Kind() {
  470. case reflect.Complex64, reflect.Complex128:
  471. // TODO: reimplement this
  472. var bs []byte
  473. if rawValueType.Kind() == reflect.String {
  474. bs = []byte(vv.String())
  475. } else if rawValueType.ConvertibleTo(core.BytesType) {
  476. bs = vv.Bytes()
  477. }
  478. hasAssigned = true
  479. if len(bs) > 0 {
  480. if fieldValue.CanAddr() {
  481. err := json.Unmarshal(bs, fieldValue.Addr().Interface())
  482. if err != nil {
  483. session.Engine.logger.Error(err)
  484. return err
  485. }
  486. } else {
  487. x := reflect.New(fieldType)
  488. err := json.Unmarshal(bs, x.Interface())
  489. if err != nil {
  490. session.Engine.logger.Error(err)
  491. return err
  492. }
  493. fieldValue.Set(x.Elem())
  494. }
  495. }
  496. case reflect.Slice, reflect.Array:
  497. switch rawValueType.Kind() {
  498. case reflect.Slice, reflect.Array:
  499. switch rawValueType.Elem().Kind() {
  500. case reflect.Uint8:
  501. if fieldType.Elem().Kind() == reflect.Uint8 {
  502. hasAssigned = true
  503. fieldValue.Set(vv)
  504. }
  505. }
  506. }
  507. case reflect.String:
  508. if rawValueType.Kind() == reflect.String {
  509. hasAssigned = true
  510. fieldValue.SetString(vv.String())
  511. }
  512. case reflect.Bool:
  513. if rawValueType.Kind() == reflect.Bool {
  514. hasAssigned = true
  515. fieldValue.SetBool(vv.Bool())
  516. }
  517. case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
  518. switch rawValueType.Kind() {
  519. case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
  520. hasAssigned = true
  521. fieldValue.SetInt(vv.Int())
  522. }
  523. case reflect.Float32, reflect.Float64:
  524. switch rawValueType.Kind() {
  525. case reflect.Float32, reflect.Float64:
  526. hasAssigned = true
  527. fieldValue.SetFloat(vv.Float())
  528. }
  529. case reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uint:
  530. switch rawValueType.Kind() {
  531. case reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uint:
  532. hasAssigned = true
  533. fieldValue.SetUint(vv.Uint())
  534. case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
  535. hasAssigned = true
  536. fieldValue.SetUint(uint64(vv.Int()))
  537. }
  538. case reflect.Struct:
  539. if fieldType.ConvertibleTo(core.TimeType) {
  540. if rawValueType == core.TimeType {
  541. hasAssigned = true
  542. t := vv.Convert(core.TimeType).Interface().(time.Time)
  543. z, _ := t.Zone()
  544. if len(z) == 0 || t.Year() == 0 {
  545. // !nashtsai! HACK tmp work around for lib/pq doesn't properly time with location
  546. session.Engine.logger.Debugf("empty zone key[%v] : %v | zone: %v | location: %+v\n", key, t, z, *t.Location())
  547. t = time.Date(t.Year(), t.Month(), t.Day(), t.Hour(),
  548. t.Minute(), t.Second(), t.Nanosecond(), time.Local)
  549. }
  550. // !nashtsai! convert to engine location
  551. if col.TimeZone == nil {
  552. t = t.In(session.Engine.TZLocation)
  553. } else {
  554. t = t.In(col.TimeZone)
  555. }
  556. // dateFormat to string
  557. loc, _ := time.LoadLocation("Local") //重要:获取时区 rawValue.Interface().(time.Time).Format(dateFormat)
  558. t, _ = time.ParseInLocation(dateFormat, t.Format(dateFormat), loc)
  559. fieldValue.Set(reflect.ValueOf(t).Convert(fieldType))
  560. } else if rawValueType == core.IntType || rawValueType == core.Int64Type ||
  561. rawValueType == core.Int32Type {
  562. hasAssigned = true
  563. var tz *time.Location
  564. if col.TimeZone == nil {
  565. tz = session.Engine.TZLocation
  566. } else {
  567. tz = col.TimeZone
  568. }
  569. t := time.Unix(vv.Int(), 0).In(tz)
  570. //vv = reflect.ValueOf(t)
  571. fieldValue.Set(reflect.ValueOf(t).Convert(fieldType))
  572. } else {
  573. if d, ok := vv.Interface().([]uint8); ok {
  574. hasAssigned = true
  575. t, err := session.byte2Time(col, d)
  576. if err != nil {
  577. session.Engine.logger.Error("byte2Time error:", err.Error())
  578. hasAssigned = false
  579. } else {
  580. fieldValue.Set(reflect.ValueOf(t).Convert(fieldType))
  581. }
  582. } else if d, ok := vv.Interface().(string); ok {
  583. hasAssigned = true
  584. t, err := session.str2Time(col, d)
  585. if err != nil {
  586. session.Engine.logger.Error("byte2Time error:", err.Error())
  587. hasAssigned = false
  588. } else {
  589. fieldValue.Set(reflect.ValueOf(t).Convert(fieldType))
  590. }
  591. } else {
  592. panic(fmt.Sprintf("rawValueType is %v, value is %v", rawValueType, vv.Interface()))
  593. }
  594. }
  595. } else if nulVal, ok := fieldValue.Addr().Interface().(sql.Scanner); ok {
  596. // !<winxxp>! 增加支持sql.Scanner接口的结构,如sql.NullString
  597. hasAssigned = true
  598. if err := nulVal.Scan(vv.Interface()); err != nil {
  599. //fmt.Println("sql.Sanner error:", err.Error())
  600. session.Engine.logger.Error("sql.Sanner error:", err.Error())
  601. hasAssigned = false
  602. }
  603. } else if col.SQLType.IsJson() {
  604. if rawValueType.Kind() == reflect.String {
  605. hasAssigned = true
  606. x := reflect.New(fieldType)
  607. if len([]byte(vv.String())) > 0 {
  608. err := json.Unmarshal([]byte(vv.String()), x.Interface())
  609. if err != nil {
  610. session.Engine.logger.Error(err)
  611. return err
  612. }
  613. fieldValue.Set(x.Elem())
  614. }
  615. } else if rawValueType.Kind() == reflect.Slice {
  616. hasAssigned = true
  617. x := reflect.New(fieldType)
  618. if len(vv.Bytes()) > 0 {
  619. err := json.Unmarshal(vv.Bytes(), x.Interface())
  620. if err != nil {
  621. session.Engine.logger.Error(err)
  622. return err
  623. }
  624. fieldValue.Set(x.Elem())
  625. }
  626. }
  627. } else if session.Statement.UseCascade {
  628. table := session.Engine.autoMapType(*fieldValue)
  629. if table != nil {
  630. if len(table.PrimaryKeys) != 1 {
  631. panic("unsupported non or composited primary key cascade")
  632. }
  633. var pk = make(core.PK, len(table.PrimaryKeys))
  634. switch rawValueType.Kind() {
  635. case reflect.Int64:
  636. pk[0] = vv.Int()
  637. case reflect.Int:
  638. pk[0] = int(vv.Int())
  639. case reflect.Int32:
  640. pk[0] = int32(vv.Int())
  641. case reflect.Int16:
  642. pk[0] = int16(vv.Int())
  643. case reflect.Int8:
  644. pk[0] = int8(vv.Int())
  645. case reflect.Uint64:
  646. pk[0] = vv.Uint()
  647. case reflect.Uint:
  648. pk[0] = uint(vv.Uint())
  649. case reflect.Uint32:
  650. pk[0] = uint32(vv.Uint())
  651. case reflect.Uint16:
  652. pk[0] = uint16(vv.Uint())
  653. case reflect.Uint8:
  654. pk[0] = uint8(vv.Uint())
  655. case reflect.String:
  656. pk[0] = vv.String()
  657. case reflect.Slice:
  658. pk[0], _ = strconv.ParseInt(string(rawValue.Interface().([]byte)), 10, 64)
  659. default:
  660. panic(fmt.Sprintf("unsupported primary key type: %v, %v", rawValueType, fieldValue))
  661. }
  662. if !isPKZero(pk) {
  663. // !nashtsai! TODO for hasOne relationship, it's preferred to use join query for eager fetch
  664. // however, also need to consider adding a 'lazy' attribute to xorm tag which allow hasOne
  665. // property to be fetched lazily
  666. structInter := reflect.New(fieldValue.Type())
  667. newsession := session.Engine.NewSession()
  668. defer newsession.Close()
  669. has, err := newsession.Id(pk).NoCascade().Get(structInter.Interface())
  670. if err != nil {
  671. return err
  672. }
  673. if has {
  674. v := structInter.Elem().Interface()
  675. fieldValue.Set(reflect.ValueOf(v))
  676. } else {
  677. return errors.New("cascade obj is not exist!")
  678. }
  679. }
  680. } else {
  681. session.Engine.logger.Error("unsupported struct type in Scan: ", fieldValue.Type().String())
  682. }
  683. }
  684. case reflect.Ptr:
  685. // !nashtsai! TODO merge duplicated codes above
  686. //typeStr := fieldType.String()
  687. switch fieldType {
  688. // following types case matching ptr's native type, therefore assign ptr directly
  689. case core.PtrStringType:
  690. if rawValueType.Kind() == reflect.String {
  691. x := vv.String()
  692. hasAssigned = true
  693. fieldValue.Set(reflect.ValueOf(&x))
  694. }
  695. case core.PtrBoolType:
  696. if rawValueType.Kind() == reflect.Bool {
  697. x := vv.Bool()
  698. hasAssigned = true
  699. fieldValue.Set(reflect.ValueOf(&x))
  700. }
  701. case core.PtrTimeType:
  702. if rawValueType == core.PtrTimeType {
  703. hasAssigned = true
  704. var x = rawValue.Interface().(time.Time)
  705. fieldValue.Set(reflect.ValueOf(&x))
  706. }
  707. case core.PtrFloat64Type:
  708. if rawValueType.Kind() == reflect.Float64 {
  709. x := vv.Float()
  710. hasAssigned = true
  711. fieldValue.Set(reflect.ValueOf(&x))
  712. }
  713. case core.PtrUint64Type:
  714. if rawValueType.Kind() == reflect.Int64 {
  715. var x = uint64(vv.Int())
  716. hasAssigned = true
  717. fieldValue.Set(reflect.ValueOf(&x))
  718. }
  719. case core.PtrInt64Type:
  720. if rawValueType.Kind() == reflect.Int64 {
  721. x := vv.Int()
  722. hasAssigned = true
  723. fieldValue.Set(reflect.ValueOf(&x))
  724. }
  725. case core.PtrFloat32Type:
  726. if rawValueType.Kind() == reflect.Float64 {
  727. var x = float32(vv.Float())
  728. hasAssigned = true
  729. fieldValue.Set(reflect.ValueOf(&x))
  730. }
  731. case core.PtrIntType:
  732. if rawValueType.Kind() == reflect.Int64 {
  733. var x = int(vv.Int())
  734. hasAssigned = true
  735. fieldValue.Set(reflect.ValueOf(&x))
  736. }
  737. case core.PtrInt32Type:
  738. if rawValueType.Kind() == reflect.Int64 {
  739. var x = int32(vv.Int())
  740. hasAssigned = true
  741. fieldValue.Set(reflect.ValueOf(&x))
  742. }
  743. case core.PtrInt8Type:
  744. if rawValueType.Kind() == reflect.Int64 {
  745. var x = int8(vv.Int())
  746. hasAssigned = true
  747. fieldValue.Set(reflect.ValueOf(&x))
  748. }
  749. case core.PtrInt16Type:
  750. if rawValueType.Kind() == reflect.Int64 {
  751. var x = int16(vv.Int())
  752. hasAssigned = true
  753. fieldValue.Set(reflect.ValueOf(&x))
  754. }
  755. case core.PtrUintType:
  756. if rawValueType.Kind() == reflect.Int64 {
  757. var x = uint(vv.Int())
  758. hasAssigned = true
  759. fieldValue.Set(reflect.ValueOf(&x))
  760. }
  761. case core.PtrUint32Type:
  762. if rawValueType.Kind() == reflect.Int64 {
  763. var x = uint32(vv.Int())
  764. hasAssigned = true
  765. fieldValue.Set(reflect.ValueOf(&x))
  766. }
  767. case core.Uint8Type:
  768. if rawValueType.Kind() == reflect.Int64 {
  769. var x = uint8(vv.Int())
  770. hasAssigned = true
  771. fieldValue.Set(reflect.ValueOf(&x))
  772. }
  773. case core.Uint16Type:
  774. if rawValueType.Kind() == reflect.Int64 {
  775. var x = uint16(vv.Int())
  776. hasAssigned = true
  777. fieldValue.Set(reflect.ValueOf(&x))
  778. }
  779. case core.Complex64Type:
  780. var x complex64
  781. if len([]byte(vv.String())) > 0 {
  782. err := json.Unmarshal([]byte(vv.String()), &x)
  783. if err != nil {
  784. session.Engine.logger.Error(err)
  785. } else {
  786. fieldValue.Set(reflect.ValueOf(&x))
  787. }
  788. }
  789. hasAssigned = true
  790. case core.Complex128Type:
  791. var x complex128
  792. if len([]byte(vv.String())) > 0 {
  793. err := json.Unmarshal([]byte(vv.String()), &x)
  794. if err != nil {
  795. session.Engine.logger.Error(err)
  796. } else {
  797. fieldValue.Set(reflect.ValueOf(&x))
  798. }
  799. }
  800. hasAssigned = true
  801. } // switch fieldType
  802. // default:
  803. // session.Engine.LogError("unsupported type in Scan: ", reflect.TypeOf(v).String())
  804. } // switch fieldType.Kind()
  805. // !nashtsai! for value can't be assigned directly fallback to convert to []byte then back to value
  806. if !hasAssigned {
  807. data, err := value2Bytes(&rawValue)
  808. if err == nil {
  809. session.bytes2Value(col, fieldValue, data)
  810. } else {
  811. session.Engine.logger.Error(err.Error())
  812. }
  813. }
  814. }
  815. }
  816. return nil
  817. }
  818. func (session *Session) queryPreprocessByMap(sqlStr *string, paramMap interface{}) {
  819. re := regexp.MustCompile(`[?](\w+)`)
  820. query := *sqlStr
  821. names := make(map[string]int)
  822. var i int
  823. query = re.ReplaceAllStringFunc(query, func(src string) string {
  824. names[src[1:]] = i
  825. i += 1
  826. return "?"
  827. })
  828. for _, filter := range session.Engine.dialect.Filters() {
  829. query = filter.Do(query, session.Engine.dialect, session.Statement.RefTable)
  830. }
  831. *sqlStr = query
  832. session.Engine.logSQL(*sqlStr, paramMap)
  833. }
  834. func (session *Session) BatchSql() *Session {
  835. return session
  836. }
  837. func (session *Session) BatchExecute() interface{} {
  838. session.Engine.logger.Info(runtime.Caller(0))
  839. session.Engine.logger.Info(runtime.Caller(1))
  840. session.Engine.logger.Info(runtime.Caller(2))
  841. return []string{"GetSqlMap", "sql_2_1", "sql1"}
  842. }
  843. func (session *Session) Sqls(sqls interface{}, parmas ...interface{}) *SqlsExecutor {
  844. sqlsExecutor := new(SqlsExecutor)
  845. switch sqls.(type) {
  846. case string:
  847. sqlsExecutor.sqls = sqls.(string)
  848. case []string:
  849. sqlsExecutor.sqls = sqls.([]string)
  850. case map[string]string:
  851. sqlsExecutor.sqls = sqls.(map[string]string)
  852. }
  853. if len(parmas) == 0 {
  854. sqlsExecutor.parmas = nil
  855. }
  856. if len(parmas) > 1 {
  857. sqlsExecutor.parmas = nil
  858. sqlsExecutor.err = ErrParamsType
  859. }
  860. if len(parmas) == 1 {
  861. switch parmas[0].(type) {
  862. case map[string]interface{}:
  863. sqlsExecutor.parmas = parmas[0].(map[string]interface{})
  864. case []map[string]interface{}:
  865. sqlsExecutor.parmas = parmas[0].([]map[string]interface{})
  866. case map[string]map[string]interface{}:
  867. sqlsExecutor.parmas = parmas[0].(map[string]map[string]interface{})
  868. }
  869. }
  870. sqlsExecutor.session = session
  871. return sqlsExecutor
  872. }
  873. func (session *Session) SqlMapsClient(sqlkeys interface{}, parmas ...interface{}) *SqlMapsExecutor {
  874. sqlMapsExecutor := new(SqlMapsExecutor)
  875. switch sqlkeys.(type) {
  876. case string:
  877. sqlMapsExecutor.sqlkeys = sqlkeys.(string)
  878. case []string:
  879. sqlMapsExecutor.sqlkeys = sqlkeys.([]string)
  880. case map[string]string:
  881. sqlMapsExecutor.sqlkeys = sqlkeys.(map[string]string)
  882. }
  883. if len(parmas) == 0 {
  884. sqlMapsExecutor.parmas = nil
  885. }
  886. if len(parmas) > 1 {
  887. sqlMapsExecutor.parmas = nil
  888. sqlMapsExecutor.err = ErrParamsType
  889. }
  890. if len(parmas) == 1 {
  891. switch parmas[0].(type) {
  892. case map[string]interface{}:
  893. sqlMapsExecutor.parmas = parmas[0].(map[string]interface{})
  894. case []map[string]interface{}:
  895. sqlMapsExecutor.parmas = parmas[0].([]map[string]interface{})
  896. case map[string]map[string]interface{}:
  897. sqlMapsExecutor.parmas = parmas[0].(map[string]map[string]interface{})
  898. }
  899. }
  900. sqlMapsExecutor.session = session
  901. return sqlMapsExecutor
  902. }