repository.go 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186
  1. package orm
  2. import (
  3. "fmt"
  4. "reflect"
  5. "strings"
  6. "git.giaever.org/bnb.hosting/orm/conn"
  7. "github.com/huandu/go-sqlbuilder"
  8. "github.com/jmoiron/sqlx"
  9. )
  10. type repository struct {
  11. dbc map[string]MappableInterface
  12. }
  13. var r *repository = nil
  14. func Repository() *repository {
  15. if r == nil {
  16. r = &repository{
  17. make(map[string]MappableInterface),
  18. }
  19. }
  20. return r
  21. }
  22. func (r *repository) FetchFirst(db *conn.DB, i MappableInterface) (MappableInterface, error) {
  23. res, err := r.Fetch(db, i, (Cond{Method: Limit}).SetVar(1))
  24. if len(res) == 1 {
  25. return res[0], err
  26. }
  27. return nil, err
  28. }
  29. func (r *repository) Fetch(db *conn.DB, i MappableInterface, conds ...Cond) ([]MappableInterface, error) {
  30. tbl := ctx.getTbl(i)
  31. b := getSelectBuilder(tbl)._select(tbl)._where(tbl, i)
  32. b._extra(tbl, conds...)
  33. return r.query(db, tbl, b)
  34. }
  35. func (r *repository) FetchRelated(i MappableInterface, n string, conds ...Cond) (interface{}, error) {
  36. tbl := ctx.getTbl(i)
  37. for rtype, rels := range tbl.getRelations() {
  38. for _, rel := range rels {
  39. if rel.f.getFieldName() == n {
  40. switch rtype {
  41. case hasMany, belongsTo:
  42. b := getSelectBuilder(tbl)
  43. b._select(rel)
  44. b._join(tbl, rel)
  45. b._wherePrimaryOrElse(tbl, i)
  46. b._extra(tbl, conds...)
  47. switch rel.f.MakeType().Kind() {
  48. case reflect.Map, reflect.Slice:
  49. // No limit?
  50. default:
  51. b.Limit(1)
  52. }
  53. /*if n == "parentUnitOption" {
  54. fmt.Println(b.Build())
  55. }*/
  56. res, err := r.query(i.GetDB(), rel.table, b)
  57. if err != nil {
  58. return nil, err
  59. }
  60. return r.mapScanResultToInterface(i, tbl, rel, res), nil
  61. case hasOne:
  62. b := getSelectBuilder(tbl)
  63. b._select(rel)
  64. b._join(tbl, rel, sqlbuilder.RightJoin)
  65. b._wherePrimaryOrElse(tbl, i)
  66. b.Limit(1)
  67. res, err := r.query(i.GetDB(), rel.table, b)
  68. if err != nil || len(res) == 0 {
  69. return nil, err
  70. }
  71. return res[0], nil
  72. }
  73. }
  74. }
  75. }
  76. return nil, nil
  77. }
  78. func (r *repository) mapScanResultToInterface(i MappableInterface, tbl *table, rel relation, res []MappableInterface) interface{} {
  79. m := rel.f.Make()
  80. switch m.Kind() {
  81. case reflect.Map:
  82. for midx, rel := range res {
  83. if val, err := tbl.CallMethod(i, "GetId"); err != nil {
  84. switch val[0].(type) {
  85. case int, uint, int8, uint8, int16, uint16, int32, uint32, int64, uint64:
  86. m.SetMapIndex(reflect.ValueOf(val[0].(uint64)), reflect.ValueOf(rel))
  87. }
  88. } else {
  89. m.SetMapIndex(reflect.ValueOf(uint64(midx)), reflect.ValueOf(rel))
  90. }
  91. }
  92. case reflect.Slice:
  93. for _, rel := range res {
  94. m = reflect.Append(m, reflect.ValueOf(rel))
  95. }
  96. default:
  97. if len(res) != 0 {
  98. m.Set(reflect.ValueOf(res[0]))
  99. }
  100. }
  101. return m.Interface()
  102. }
  103. func (r *repository) mapScan(i MappableInterface, rows *sqlx.Rows, tbl *table) (MappableInterface, error) {
  104. res := make(map[string]interface{})
  105. if err := rows.MapScan(res); err != nil {
  106. return nil, err
  107. }
  108. for n, s := range res {
  109. if fn := tbl.hasColumn(n).getFieldName(); len(fn) != 0 {
  110. if s != nil {
  111. if _, err := tbl.CallMethod(i, "Set"+strings.Title(fn), s); err != nil {
  112. fmt.Println(err)
  113. //return nil, err
  114. }
  115. }
  116. } else {
  117. fmt.Println("MISSING FIELD NAME", n, s)
  118. }
  119. }
  120. if col := tbl.getPrimaryKey(); col != nil {
  121. if v, err := tbl.CallMethod(i, "Get"+strings.Title(col.getFieldName())); err == nil {
  122. idx := fmt.Sprintf("%s_%#v", tbl.getStructName(), v[0])
  123. if ni, ok := r.dbc[idx]; ok {
  124. return ni, nil
  125. } else {
  126. r.dbc[idx] = i
  127. }
  128. }
  129. }
  130. return i, nil
  131. }
  132. func (r *repository) query(db *conn.DB, tbl *table, sql sqlbuilder.Builder) ([]MappableInterface, error) {
  133. var ret []MappableInterface
  134. q, args := sql.Build()
  135. rows, err := db.Queryx(q, args...)
  136. for err != nil {
  137. return nil, err
  138. }
  139. //defer rows.Close()
  140. for rows.Next() {
  141. i, err := r.mapScan(tbl.Make(), rows, tbl)
  142. if err == nil {
  143. i.SetDB(db)
  144. ret = append(ret, i)
  145. } else {
  146. fmt.Println(err)
  147. return nil, err
  148. }
  149. }
  150. return ret, nil
  151. }