engineplus.go 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166
  1. package xorm
  2. import (
  3. "encoding/json"
  4. "github.com/Chronokeeper/anyxml"
  5. )
  6. func (engine *Engine) SqlMapClient(sqlTagName string, args ...interface{}) *Session {
  7. session := engine.NewSession()
  8. session.IsAutoClose = true
  9. return session.Sql(engine.SqlMap.Sql[sqlTagName], args...)
  10. }
  11. func (engine *Engine) SqlTemplateClient(sqlTagName string, args ...interface{}) *Session {
  12. session := engine.NewSession()
  13. session.IsAutoClose = true
  14. map1:=args[0].(map[string]interface{})
  15. if engine.SqlTemplate.Template[sqlTagName]==nil{
  16. return session.Sql("", &map1)
  17. }
  18. sql, err := engine.SqlTemplate.Template[sqlTagName].Execute(map1)
  19. if err != nil {
  20. engine.Logger.Err(err)
  21. }
  22. return session.Sql(sql, &map1)
  23. }
  24. // Get retrieve one record from table, bean's non-empty fields
  25. // are conditions
  26. func (engine *Engine) GetFirst(bean interface{}) ResultBean {
  27. session := engine.NewSession()
  28. defer session.Close()
  29. return session.GetFirst(bean)
  30. }
  31. // Exec a raw sql and return records as []map[string]interface{}
  32. func (engine *Engine) QueryAll(sql string, paramStr ...interface{}) (resultsSlice []map[string]interface{}, err error) {
  33. session := engine.NewSession()
  34. defer session.Close()
  35. return session.queryAll(sql, paramStr...)
  36. }
  37. // Exec a raw sql and return records as []map[string]interface{}
  38. func (engine *Engine) QueryAllByMap(sql string, paramMap interface{}) (resultsSlice []map[string]interface{}, err error) {
  39. session := engine.NewSession()
  40. defer session.Close()
  41. return session.queryAllByMap(sql, paramMap)
  42. }
  43. func (engine *Engine) QueryAllByMapToJsonString(sql string, paramMap interface{}) (string, error) {
  44. session := engine.NewSession()
  45. defer session.Close()
  46. results, err := session.queryAllByMap(sql, paramMap)
  47. if err != nil {
  48. return "", err
  49. }
  50. return JSONString(results, true)
  51. }
  52. func (engine *Engine) QueryAllByMapToJsonStringWithDateFormat(dateFormat string, sql string, paramMap interface{}) (string, error) {
  53. session := engine.NewSession()
  54. defer session.Close()
  55. results, err := session.queryAllByMapWithDateFormat(dateFormat, sql, paramMap)
  56. if err != nil {
  57. return "", err
  58. }
  59. return JSONString(results, true)
  60. }
  61. func (engine *Engine) QueryAllToJsonString(sql string, paramStr ...interface{}) (string, error) {
  62. session := engine.NewSession()
  63. defer session.Close()
  64. results, err := session.queryAll(sql, paramStr...)
  65. if err != nil {
  66. return "", err
  67. }
  68. return JSONString(results, true)
  69. }
  70. func (engine *Engine) QueryAllToJsonStringWithDateFormat(dateFormat string, sql string, paramStr ...interface{}) (string, error) {
  71. session := engine.NewSession()
  72. defer session.Close()
  73. results, err := session.queryAllWithDateFormat(dateFormat, sql, paramStr...)
  74. if err != nil {
  75. return "", err
  76. }
  77. return JSONString(results, true)
  78. }
  79. func (engine *Engine) queryAllToXmlString(sql string, paramStr ...interface{}) (string, error) {
  80. session := engine.NewSession()
  81. defer session.Close()
  82. resultSlice, err := session.queryAll(sql, paramStr...)
  83. if err != nil {
  84. return "", err
  85. }
  86. results, err := anyxml.Xml(resultSlice, "result")
  87. if err != nil {
  88. return "", err
  89. }
  90. return string(results), nil
  91. }
  92. func (engine *Engine) queryAllToXmlIndentString(sql string, prefix string, indent string, paramStr ...interface{}) (string, error) {
  93. session := engine.NewSession()
  94. defer session.Close()
  95. resultSlice, err := session.queryAll(sql, paramStr...)
  96. if err != nil {
  97. return "", err
  98. }
  99. results, err := anyxml.XmlIndent(resultSlice, prefix, indent, "result")
  100. if err != nil {
  101. return "", err
  102. }
  103. return string(results), nil
  104. }
  105. func (engine *Engine) queryAllToXmlStringWithDateFormat(dateFormat string, sql string, paramStr ...interface{}) (string, error) {
  106. session := engine.NewSession()
  107. defer session.Close()
  108. resultSlice, err := session.queryAll(sql, paramStr...)
  109. if err != nil {
  110. return "", err
  111. }
  112. results, err := anyxml.XmlWithDateFormat(dateFormat, resultSlice)
  113. if err != nil {
  114. return "", err
  115. }
  116. return string(results), nil
  117. }
  118. func (engine *Engine) QueryAllToXmlIndentStringWithDateFormat(dateFormat string, sql string, prefix string, indent string, paramStr ...interface{}) (string, error) {
  119. session := engine.NewSession()
  120. defer session.Close()
  121. resultSlice, err := session.queryAll(sql, paramStr...)
  122. if err != nil {
  123. return "", err
  124. }
  125. results, err := anyxml.XmlIndentWithDateFormat(dateFormat, resultSlice, "", " ", "results")
  126. if err != nil {
  127. return "", err
  128. }
  129. return string(results), nil
  130. }
  131. func JSONString(v interface{}, IndentJSON bool) (string, error) {
  132. var result []byte
  133. var err error
  134. if IndentJSON {
  135. result, err = json.MarshalIndent(v, "", " ")
  136. } else {
  137. result, err = json.Marshal(v)
  138. }
  139. if err != nil {
  140. return "", err
  141. }
  142. if string(result)=="null"{
  143. return "", nil
  144. }
  145. return string(result), nil
  146. }