sessionplus.go 26 KB

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