sessionplus.go 24 KB

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