mongodb_test.go 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. package mongodb
  2. import (
  3. "log"
  4. "testing"
  5. "go.mongodb.org/mongo-driver/bson"
  6. )
  7. func Test_add(t *testing.T) {
  8. m := &MongodbSim{
  9. MongodbAddr: "192.168.3.128:27090",
  10. Size: 5,
  11. DbName: "wcj",
  12. UserName: "admin",
  13. Password: "123456",
  14. }
  15. m.InitPool()
  16. // log.Println(m.Save("test", map[string]interface{}{
  17. // "name": "张三",
  18. // "age": 12,
  19. // }))
  20. // log.Println(m.SaveByOriID("test", map[string]interface{}{
  21. // "name": "张三",
  22. // "age": 25,
  23. // }))
  24. log.Println(m.SaveBulkInterface("test", []interface{}{
  25. map[string]interface{}{
  26. "name": "张三1",
  27. "age": 1,
  28. },
  29. map[string]interface{}{
  30. "name": "张三2",
  31. "age": 2,
  32. },
  33. }...))
  34. }
  35. func Test_find(t *testing.T) {
  36. m := &MongodbSim{
  37. MongodbAddr: "192.168.3.128:27080",
  38. Size: 5,
  39. DbName: "qfw",
  40. }
  41. m.InitPool()
  42. list, _ := m.Find("test", map[string]interface{}{
  43. "name": "张三",
  44. }, map[string]interface{}{"age": -1}, map[string]interface{}{"age": 1, "name": 1, "_id": 0}, false, -1, -1)
  45. log.Println(len(*list))
  46. }
  47. func Test_update(t *testing.T) {
  48. m := &MongodbSim{
  49. MongodbAddr: "192.168.3.128:27080",
  50. Size: 5,
  51. DbName: "qfw",
  52. //UserName: "admin",
  53. //Password: "123456",
  54. }
  55. m.InitPool()
  56. log.Println(m.Update("user", map[string]interface{}{
  57. "s_phone": "15037870765",
  58. }, map[string]interface{}{
  59. "$unset": map[string]interface{}{
  60. "s_name": "",
  61. },
  62. }, false, false))
  63. return
  64. s := [][]map[string]interface{}{
  65. []map[string]interface{}{
  66. map[string]interface{}{"name": "李四111"},
  67. map[string]interface{}{"$set": map[string]interface{}{"type": 1}},
  68. },
  69. []map[string]interface{}{
  70. map[string]interface{}{"name": "张三111"},
  71. map[string]interface{}{"$set": map[string]interface{}{"type": 2}},
  72. },
  73. }
  74. one := m.UpdateBulk("test", s...)
  75. log.Println(one)
  76. }
  77. func Test_count(t *testing.T) {
  78. m := &MongodbSim{
  79. MongodbAddr: "192.168.3.128:27090",
  80. Size: 5,
  81. DbName: "wcj",
  82. UserName: "admin",
  83. Password: "123456",
  84. }
  85. m.InitPool()
  86. one := m.Count("test", bson.M{
  87. "name": "张三",
  88. })
  89. log.Println(one)
  90. }
  91. func Test_del(t *testing.T) {
  92. m := &MongodbSim{
  93. MongodbAddr: "192.168.3.128:27090",
  94. Size: 5,
  95. DbName: "wcj",
  96. UserName: "admin",
  97. Password: "123456",
  98. }
  99. m.InitPool()
  100. one := m.Del("test", nil)
  101. log.Println(one)
  102. }
  103. func Test_itor(t *testing.T) {
  104. m := &MongodbSim{
  105. MongodbAddr: "192.168.3.128:27080",
  106. Size: 5,
  107. DbName: "wcj",
  108. }
  109. m.InitPool()
  110. sess := m.GetMgoConn()
  111. it := sess.DB("wcj").C("jyopenplatform_user").Find(map[string]interface{}{
  112. "i_status": 1,
  113. "i_freeze": 1,
  114. }).Select(map[string]interface{}{
  115. "a_power": 1,
  116. "s_m_openid": 1,
  117. "o_rulepro": 1,
  118. "s_appid": 1,
  119. "s_enterprisename": 1,
  120. }).Sort("_id").Iter()
  121. for user := make(map[string]interface{}); it.Next(&user); {
  122. log.Println(user["_id"], user["s_appid"])
  123. }
  124. }