sessionplus.go 25 KB

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