mongodb.go 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780
  1. package mongodb
  2. import (
  3. "context"
  4. "encoding/json"
  5. "fmt"
  6. "log"
  7. "math/big"
  8. "reflect"
  9. "runtime"
  10. "strconv"
  11. "strings"
  12. "time"
  13. "go.mongodb.org/mongo-driver/bson"
  14. "go.mongodb.org/mongo-driver/bson/primitive"
  15. "go.mongodb.org/mongo-driver/mongo"
  16. "go.mongodb.org/mongo-driver/mongo/options"
  17. )
  18. func NewMgo(addr, db string, size int) *MongodbSim {
  19. mgo := &MongodbSim{
  20. MongodbAddr: addr,
  21. Size: size,
  22. DbName: db,
  23. }
  24. mgo.InitPool()
  25. return mgo
  26. }
  27. type Bluk struct {
  28. ms *MgoSess
  29. writes []mongo.WriteModel
  30. }
  31. func (b *Bluk) Insert(doc interface{}) {
  32. write := mongo.NewInsertOneModel()
  33. write.SetDocument(doc)
  34. b.writes = append(b.writes, write)
  35. }
  36. func (b *Bluk) Update(doc ...interface{}) {
  37. write := mongo.NewUpdateOneModel()
  38. write.SetFilter(doc[0])
  39. ue := ObjToM(doc[1])
  40. autoUpdateTime(b.ms.db, b.ms.coll, ue)
  41. write.SetUpdate(ue)
  42. write.SetUpsert(false)
  43. b.writes = append(b.writes, write)
  44. }
  45. func (b *Bluk) UpdateAll(doc ...interface{}) {
  46. write := mongo.NewUpdateManyModel()
  47. write.SetFilter(doc[0])
  48. ue := ObjToM(doc[1])
  49. autoUpdateTime(b.ms.db, b.ms.coll, ue)
  50. write.SetUpdate(ue)
  51. write.SetUpsert(false)
  52. b.writes = append(b.writes, write)
  53. }
  54. func (b *Bluk) Upsert(doc ...interface{}) {
  55. write := mongo.NewUpdateOneModel()
  56. write.SetFilter(doc[0])
  57. ue := ObjToM(doc[1])
  58. autoUpdateTime(b.ms.db, b.ms.coll, ue)
  59. write.SetUpdate(ue)
  60. write.SetUpsert(true)
  61. b.writes = append(b.writes, write)
  62. }
  63. func (b *Bluk) Remove(doc interface{}) {
  64. write := mongo.NewDeleteOneModel()
  65. write.SetFilter(doc)
  66. b.writes = append(b.writes, write)
  67. }
  68. func (b *Bluk) RemoveAll(doc interface{}) {
  69. write := mongo.NewDeleteManyModel()
  70. write.SetFilter(doc)
  71. b.writes = append(b.writes, write)
  72. }
  73. func (b *Bluk) Run() (*mongo.BulkWriteResult, error) {
  74. return b.ms.M.C.Database(b.ms.db).Collection(b.ms.coll).BulkWrite(b.ms.M.Ctx, b.writes)
  75. }
  76. //
  77. type MgoIter struct {
  78. Cursor *mongo.Cursor
  79. Ctx context.Context
  80. }
  81. func (mt *MgoIter) Next(result interface{}) bool {
  82. if mt.Cursor != nil {
  83. if mt.Cursor.Next(mt.Ctx) {
  84. rType := reflect.TypeOf(result)
  85. rVal := reflect.ValueOf(result)
  86. if rType.Kind() == reflect.Ptr {
  87. rType = rType.Elem()
  88. rVal = rVal.Elem()
  89. }
  90. var err error
  91. if rType.Kind() == reflect.Map {
  92. r := make(map[string]interface{})
  93. err = mt.Cursor.Decode(&r)
  94. if rVal.CanSet() {
  95. rVal.Set(reflect.ValueOf(r))
  96. } else {
  97. for it := rVal.MapRange(); it.Next(); {
  98. rVal.SetMapIndex(it.Key(), reflect.Value{})
  99. }
  100. for it := reflect.ValueOf(r).MapRange(); it.Next(); {
  101. rVal.SetMapIndex(it.Key(), it.Value())
  102. }
  103. }
  104. } else {
  105. err = mt.Cursor.Decode(&result)
  106. }
  107. if err != nil {
  108. log.Println("mgo cur err", err.Error())
  109. mt.Cursor.Close(mt.Ctx)
  110. return false
  111. }
  112. return true
  113. } else {
  114. mt.Cursor.Close(mt.Ctx)
  115. return false
  116. }
  117. } else {
  118. return false
  119. }
  120. }
  121. //
  122. type MgoSess struct {
  123. db string
  124. coll string
  125. query interface{}
  126. sorts []string
  127. fields interface{}
  128. limit int64
  129. skip int64
  130. pipe []map[string]interface{}
  131. all interface{}
  132. M *MongodbSim
  133. }
  134. func (ms *MgoSess) DB(name string) *MgoSess {
  135. ms.db = name
  136. return ms
  137. }
  138. func (ms *MgoSess) C(name string) *MgoSess {
  139. ms.coll = name
  140. return ms
  141. }
  142. func (ms *MgoSess) Bulk() *Bluk {
  143. return &Bluk{ms: ms}
  144. }
  145. func (ms *MgoSess) Find(q interface{}) *MgoSess {
  146. if q == nil {
  147. q = map[string]interface{}{}
  148. }
  149. ms.query = q
  150. return ms
  151. }
  152. func (ms *MgoSess) FindId(_id interface{}) *MgoSess {
  153. ms.query = map[string]interface{}{"_id": _id}
  154. return ms
  155. }
  156. func (ms *MgoSess) Select(fields interface{}) *MgoSess {
  157. ms.fields = fields
  158. return ms
  159. }
  160. func (ms *MgoSess) Limit(limit int64) *MgoSess {
  161. ms.limit = limit
  162. return ms
  163. }
  164. func (ms *MgoSess) Skip(skip int64) *MgoSess {
  165. ms.skip = skip
  166. return ms
  167. }
  168. func (ms *MgoSess) Sort(sorts ...string) *MgoSess {
  169. ms.sorts = sorts
  170. return ms
  171. }
  172. func (ms *MgoSess) Pipe(p []map[string]interface{}) *MgoSess {
  173. ms.pipe = p
  174. return ms
  175. }
  176. func (ms *MgoSess) Insert(doc interface{}) error {
  177. _, err := ms.M.C.Database(ms.db).Collection(ms.coll).InsertOne(ms.M.Ctx, doc)
  178. return err
  179. }
  180. func (ms *MgoSess) Remove(filter interface{}) error {
  181. _, err := ms.M.C.Database(ms.db).Collection(ms.coll).DeleteOne(ms.M.Ctx, filter)
  182. return err
  183. }
  184. func (ms *MgoSess) RemoveId(_id interface{}) error {
  185. _, err := ms.M.C.Database(ms.db).Collection(ms.coll).DeleteOne(ms.M.Ctx, map[string]interface{}{"_id": _id})
  186. return err
  187. }
  188. func (ms *MgoSess) RemoveAll(filter interface{}) (*mongo.DeleteResult, error) {
  189. return ms.M.C.Database(ms.db).Collection(ms.coll).DeleteMany(ms.M.Ctx, filter)
  190. }
  191. func (ms *MgoSess) Upsert(filter, update interface{}) (*mongo.UpdateResult, error) {
  192. ct := options.Update()
  193. ct.SetUpsert(true)
  194. ue := ObjToM(update)
  195. autoUpdateTime(ms.db, ms.coll, ue)
  196. return ms.M.C.Database(ms.db).Collection(ms.coll).UpdateOne(ms.M.Ctx, filter, ue, ct)
  197. }
  198. func (ms *MgoSess) UpsertId(filter, update interface{}) (*mongo.UpdateResult, error) {
  199. ct := options.Update()
  200. ct.SetUpsert(true)
  201. ue := ObjToM(update)
  202. autoUpdateTime(ms.db, ms.coll, ue)
  203. return ms.M.C.Database(ms.db).Collection(ms.coll).UpdateOne(ms.M.Ctx, map[string]interface{}{"_id": filter}, ue, ct)
  204. }
  205. func (ms *MgoSess) UpdateId(filter, update interface{}) error {
  206. ue := ObjToM(update)
  207. autoUpdateTime(ms.db, ms.coll, ue)
  208. _, err := ms.M.C.Database(ms.db).Collection(ms.coll).UpdateOne(ms.M.Ctx, map[string]interface{}{"_id": filter}, ue)
  209. return err
  210. }
  211. func (ms *MgoSess) Update(filter, update interface{}) error {
  212. ue := ObjToM(update)
  213. autoUpdateTime(ms.db, ms.coll, ue)
  214. _, err := ms.M.C.Database(ms.db).Collection(ms.coll).UpdateOne(ms.M.Ctx, filter, ue)
  215. return err
  216. }
  217. func (ms *MgoSess) Count() (int64, error) {
  218. return ms.M.C.Database(ms.db).Collection(ms.coll).CountDocuments(ms.M.Ctx, ms.query)
  219. }
  220. func (ms *MgoSess) One(v *map[string]interface{}) {
  221. of := options.FindOne()
  222. of.SetProjection(ms.fields)
  223. sr := ms.M.C.Database(ms.db).Collection(ms.coll).FindOne(ms.M.Ctx, ms.query, of)
  224. if sr.Err() == nil {
  225. sr.Decode(&v)
  226. }
  227. }
  228. func (ms *MgoSess) All(v *[]map[string]interface{}) {
  229. cur, err := ms.M.C.Database(ms.db).Collection(ms.coll).Aggregate(ms.M.Ctx, ms.pipe)
  230. if err == nil && cur.Err() == nil {
  231. cur.All(ms.M.Ctx, v)
  232. }
  233. }
  234. func (ms *MgoSess) Iter() *MgoIter {
  235. it := &MgoIter{}
  236. find := options.Find()
  237. if ms.skip > 0 {
  238. find.SetSkip(ms.skip)
  239. }
  240. if ms.limit > 0 {
  241. find.SetLimit(ms.limit)
  242. }
  243. find.SetBatchSize(100)
  244. if len(ms.sorts) > 0 {
  245. sort := bson.D{}
  246. for _, k := range ms.sorts {
  247. switch k[:1] {
  248. case "-":
  249. sort = append(sort, bson.E{k[1:], -1})
  250. case "+":
  251. sort = append(sort, bson.E{k[1:], 1})
  252. default:
  253. sort = append(sort, bson.E{k, 1})
  254. }
  255. }
  256. find.SetSort(sort)
  257. }
  258. if ms.fields != nil {
  259. find.SetProjection(ms.fields)
  260. }
  261. cur, err := ms.M.C.Database(ms.db).Collection(ms.coll).Find(ms.M.Ctx, ms.query, find)
  262. if err != nil {
  263. log.Println("mgo find err", err.Error())
  264. } else {
  265. it.Cursor = cur
  266. it.Ctx = ms.M.Ctx
  267. }
  268. return it
  269. }
  270. type MongodbSim struct {
  271. MongodbAddr string
  272. Size int
  273. // MinSize int
  274. DbName string
  275. C *mongo.Client
  276. Ctx context.Context
  277. ShortCtx context.Context
  278. pool chan bool
  279. UserName string
  280. Password string
  281. ReplSet string
  282. }
  283. func (m *MongodbSim) GetMgoConn() *MgoSess {
  284. //m.Open()
  285. ms := &MgoSess{}
  286. ms.M = m
  287. return ms
  288. }
  289. func (m *MongodbSim) DestoryMongoConn(ms *MgoSess) {
  290. //m.Close()
  291. ms.M = nil
  292. ms = nil
  293. }
  294. func (m *MongodbSim) Destroy() {
  295. //m.Close()
  296. m.C.Disconnect(nil)
  297. m.C = nil
  298. }
  299. func (m *MongodbSim) InitPool() {
  300. opts := options.Client()
  301. registry := bson.NewRegistryBuilder().RegisterTypeMapEntry(bson.TypeArray, reflect.TypeOf([]interface{}{})).Build()
  302. opts.SetRegistry(registry)
  303. opts.SetConnectTimeout(3 * time.Second)
  304. opts.SetHosts(strings.Split(m.MongodbAddr, ","))
  305. //opts.ApplyURI("mongodb://" + m.MongodbAddr)
  306. opts.SetMaxPoolSize(uint64(m.Size))
  307. if m.UserName != "" && m.Password != "" {
  308. cre := options.Credential{
  309. Username: m.UserName,
  310. Password: m.Password,
  311. }
  312. opts.SetAuth(cre)
  313. }
  314. /*ms := strings.Split(m.MongodbAddr, ",")
  315. if m.ReplSet == "" && len(ms) > 1 {
  316. m.ReplSet = "qfws"
  317. }*/
  318. if m.ReplSet != "" {
  319. opts.SetReplicaSet(m.ReplSet)
  320. opts.SetDirect(false)
  321. }
  322. m.pool = make(chan bool, m.Size)
  323. opts.SetMaxConnIdleTime(2 * time.Hour)
  324. m.Ctx, _ = context.WithTimeout(context.Background(), 99999*time.Hour)
  325. m.ShortCtx, _ = context.WithTimeout(context.Background(), 1*time.Minute)
  326. client, err := mongo.Connect(m.ShortCtx, opts)
  327. if err != nil {
  328. log.Println("mgo init error:", err.Error())
  329. } else {
  330. m.C = client
  331. }
  332. }
  333. func (m *MongodbSim) Open() {
  334. m.pool <- true
  335. }
  336. func (m *MongodbSim) Close() {
  337. <-m.pool
  338. }
  339. func (m *MongodbSim) Save(c string, doc interface{}) string {
  340. defer catch()
  341. m.Open()
  342. defer m.Close()
  343. coll := m.C.Database(m.DbName).Collection(c)
  344. obj := ObjToM(doc)
  345. id := primitive.NewObjectID()
  346. (*obj)["_id"] = id
  347. _, err := coll.InsertOne(m.Ctx, obj)
  348. if nil != err {
  349. log.Println("SaveError", err)
  350. return ""
  351. }
  352. return id.Hex()
  353. }
  354. //原_id不变
  355. func (m *MongodbSim) SaveByOriID(c string, doc interface{}) bool {
  356. defer catch()
  357. m.Open()
  358. defer m.Close()
  359. coll := m.C.Database(m.DbName).Collection(c)
  360. _, err := coll.InsertOne(m.Ctx, ObjToM(doc))
  361. if nil != err {
  362. log.Println("SaveByOriIDError", err)
  363. return false
  364. }
  365. return true
  366. }
  367. //批量插入
  368. func (m *MongodbSim) SaveBulk(c string, doc ...map[string]interface{}) bool {
  369. defer catch()
  370. m.Open()
  371. defer m.Close()
  372. coll := m.C.Database(m.DbName).Collection(c)
  373. var writes []mongo.WriteModel
  374. for _, d := range doc {
  375. write := mongo.NewInsertOneModel()
  376. write.SetDocument(d)
  377. writes = append(writes, write)
  378. }
  379. br, e := coll.BulkWrite(m.Ctx, writes)
  380. if e != nil {
  381. b := strings.Index(e.Error(), "duplicate") > -1
  382. log.Println("mgo savebulk error:", e.Error())
  383. if br != nil {
  384. log.Println("mgo savebulk size:", br.InsertedCount)
  385. }
  386. return b
  387. }
  388. return true
  389. }
  390. //批量插入
  391. func (m *MongodbSim) SaveBulkInterface(c string, doc ...interface{}) bool {
  392. defer catch()
  393. m.Open()
  394. defer m.Close()
  395. coll := m.C.Database(m.DbName).Collection(c)
  396. var writes []mongo.WriteModel
  397. for _, d := range doc {
  398. write := mongo.NewInsertOneModel()
  399. write.SetDocument(d)
  400. writes = append(writes, write)
  401. }
  402. br, e := coll.BulkWrite(m.Ctx, writes)
  403. if e != nil {
  404. b := strings.Index(e.Error(), "duplicate") > -1
  405. log.Println("mgo SaveBulkInterface error:", e.Error())
  406. if br != nil {
  407. log.Println("mgo SaveBulkInterface size:", br.InsertedCount)
  408. }
  409. return b
  410. }
  411. return true
  412. }
  413. //按条件统计
  414. func (m *MongodbSim) Count(c string, q interface{}) int {
  415. r, _ := m.CountByErr(c, q)
  416. return r
  417. }
  418. //统计
  419. func (m *MongodbSim) CountByErr(c string, q interface{}) (int, error) {
  420. defer catch()
  421. m.Open()
  422. defer m.Close()
  423. var res int64
  424. var err error
  425. if filter := ObjToM(q); filter != nil && len(*filter) > 0 {
  426. res, err = m.C.Database(m.DbName).Collection(c).CountDocuments(m.Ctx, filter)
  427. } else {
  428. res, err = m.C.Database(m.DbName).Collection(c).EstimatedDocumentCount(m.Ctx)
  429. }
  430. if err != nil {
  431. log.Println("统计错误", err.Error())
  432. return 0, err
  433. } else {
  434. return int(res), nil
  435. }
  436. }
  437. //按条件删除
  438. func (m *MongodbSim) Delete(c string, q interface{}) int64 {
  439. defer catch()
  440. m.Open()
  441. defer m.Close()
  442. res, err := m.C.Database(m.DbName).Collection(c).DeleteMany(m.Ctx, ObjToM(q))
  443. if err != nil && res == nil {
  444. log.Println("删除错误", err.Error())
  445. }
  446. return res.DeletedCount
  447. }
  448. //删除对象
  449. func (m *MongodbSim) Del(c string, q interface{}) bool {
  450. defer catch()
  451. m.Open()
  452. defer m.Close()
  453. _, err := m.C.Database(m.DbName).Collection(c).DeleteMany(m.Ctx, ObjToM(q))
  454. if err != nil {
  455. log.Println("删除错误", err.Error())
  456. return false
  457. }
  458. return true
  459. }
  460. //按条件更新
  461. func (m *MongodbSim) Update(c string, q, u interface{}, upsert bool, multi bool) bool {
  462. defer catch()
  463. m.Open()
  464. defer m.Close()
  465. ct := options.Update()
  466. if upsert {
  467. ct.SetUpsert(true)
  468. }
  469. coll := m.C.Database(m.DbName).Collection(c)
  470. ue := ObjToM(u)
  471. autoUpdateTime(m.DbName, c, ue)
  472. var err error
  473. if multi {
  474. _, err = coll.UpdateMany(m.Ctx, ObjToM(q), ue, ct)
  475. } else {
  476. _, err = coll.UpdateOne(m.Ctx, ObjToM(q), ue, ct)
  477. }
  478. if err != nil {
  479. log.Println("更新错误", err.Error())
  480. return false
  481. }
  482. return true
  483. }
  484. func (m *MongodbSim) UpdateById(c string, id interface{}, set interface{}) bool {
  485. defer catch()
  486. m.Open()
  487. defer m.Close()
  488. q := make(map[string]interface{})
  489. if sid, ok := id.(string); ok {
  490. q["_id"], _ = primitive.ObjectIDFromHex(sid)
  491. } else {
  492. q["_id"] = id
  493. }
  494. ue := ObjToM(set)
  495. autoUpdateTime(m.DbName, c, ue)
  496. _, err := m.C.Database(m.DbName).Collection(c).UpdateOne(m.Ctx, q, ue)
  497. if nil != err {
  498. log.Println("UpdateByIdError", err)
  499. return false
  500. }
  501. return true
  502. }
  503. //批量更新
  504. func (m *MongodbSim) UpdateBulkAll(db, c string, doc ...[]map[string]interface{}) bool {
  505. return m.NewUpdateBulk(db, c, false, false, doc...)
  506. }
  507. func (m *MongodbSim) UpdateBulk(c string, doc ...[]map[string]interface{}) bool {
  508. return m.UpdateBulkAll(m.DbName, c, doc...)
  509. }
  510. //批量插入
  511. func (m *MongodbSim) UpSertBulk(c string, doc ...[]map[string]interface{}) bool {
  512. return m.NewUpdateBulk(m.DbName, c, true, false, doc...)
  513. }
  514. //批量插入
  515. func (m *MongodbSim) UpSertMultiBulk(c string, upsert, multi bool, doc ...[]map[string]interface{}) bool {
  516. return m.NewUpdateBulk(m.DbName, c, upsert, multi, doc...)
  517. }
  518. //批量插入
  519. func (m *MongodbSim) NewUpdateBulk(db, c string, upsert, multi bool, doc ...[]map[string]interface{}) bool {
  520. defer catch()
  521. m.Open()
  522. defer m.Close()
  523. coll := m.C.Database(db).Collection(c)
  524. var writes []mongo.WriteModel
  525. for _, d := range doc {
  526. if multi {
  527. write := mongo.NewUpdateManyModel()
  528. write.SetFilter(d[0])
  529. ue := ObjToM(d[1])
  530. autoUpdateTime(m.DbName, c, ue)
  531. write.SetUpdate(ue)
  532. write.SetUpsert(upsert)
  533. writes = append(writes, write)
  534. } else {
  535. write := mongo.NewUpdateOneModel()
  536. write.SetFilter(d[0])
  537. ue := ObjToM(d[1])
  538. autoUpdateTime(m.DbName, c, ue)
  539. write.SetUpdate(ue)
  540. write.SetUpsert(upsert)
  541. writes = append(writes, write)
  542. }
  543. }
  544. br, e := coll.BulkWrite(m.Ctx, writes)
  545. if e != nil {
  546. log.Println("mgo upsert error:", e.Error())
  547. return br == nil || br.UpsertedCount == 0
  548. }
  549. // else {
  550. // if r.UpsertedCount != int64(len(doc)) {
  551. // log.Println("mgo upsert uncomplete:uc/dc", r.UpsertedCount, len(doc))
  552. // }
  553. // return true
  554. // }
  555. return true
  556. }
  557. //查询单条对象
  558. func (m *MongodbSim) FindOne(c string, query interface{}) (*map[string]interface{}, bool) {
  559. return m.FindOneByField(c, query, nil)
  560. }
  561. //查询单条对象
  562. func (m *MongodbSim) FindOneByField(c string, query interface{}, fields interface{}) (*map[string]interface{}, bool) {
  563. defer catch()
  564. res, ok := m.Find(c, query, nil, fields, true, -1, -1)
  565. if nil != res && len(*res) > 0 {
  566. return &((*res)[0]), ok
  567. }
  568. return nil, ok
  569. }
  570. //查询单条对象
  571. func (m *MongodbSim) FindById(c string, query string, fields interface{}) (*map[string]interface{}, bool) {
  572. defer catch()
  573. m.Open()
  574. defer m.Close()
  575. of := options.FindOne()
  576. of.SetProjection(ObjToOth(fields))
  577. res := make(map[string]interface{})
  578. _id, err := primitive.ObjectIDFromHex(query)
  579. if err != nil {
  580. log.Println("_id error", err)
  581. return &res, true
  582. }
  583. sr := m.C.Database(m.DbName).Collection(c).FindOne(m.Ctx, map[string]interface{}{"_id": _id}, of)
  584. if sr.Err() == nil {
  585. sr.Decode(&res)
  586. }
  587. return &res, true
  588. }
  589. //底层查询方法
  590. func (m *MongodbSim) Find(c string, query interface{}, order interface{}, fields interface{}, single bool, start int, limit int) (*[]map[string]interface{}, bool) {
  591. defer catch()
  592. m.Open()
  593. defer m.Close()
  594. var res []map[string]interface{}
  595. coll := m.C.Database(m.DbName).Collection(c)
  596. if single {
  597. res = make([]map[string]interface{}, 1)
  598. of := options.FindOne()
  599. of.SetProjection(ObjToOth(fields))
  600. of.SetSort(ObjToM(order))
  601. if sr := coll.FindOne(m.Ctx, ObjToM(query), of); sr.Err() == nil {
  602. sr.Decode(&res[0])
  603. }
  604. } else {
  605. res = []map[string]interface{}{}
  606. of := options.Find()
  607. of.SetProjection(ObjToOth(fields))
  608. of.SetSort(ObjToM(order))
  609. if start > -1 {
  610. of.SetSkip(int64(start))
  611. of.SetLimit(int64(limit))
  612. }
  613. cur, err := coll.Find(m.Ctx, ObjToM(query), of)
  614. if err == nil && cur.Err() == nil {
  615. cur.All(m.Ctx, &res)
  616. }
  617. }
  618. return &res, true
  619. }
  620. func ObjToOth(query interface{}) *bson.M {
  621. return ObjToMQ(query, false)
  622. }
  623. func ObjToM(query interface{}) *bson.M {
  624. return ObjToMQ(query, true)
  625. }
  626. //obj(string,M)转M,查询用到
  627. func ObjToMQ(query interface{}, isQuery bool) *bson.M {
  628. data := make(bson.M)
  629. defer catch()
  630. if s2, ok2 := query.(*map[string]interface{}); ok2 {
  631. data = bson.M(*s2)
  632. } else if s3, ok3 := query.(*bson.M); ok3 {
  633. return s3
  634. } else if s3, ok3 := query.(*primitive.M); ok3 {
  635. return s3
  636. } else if s, ok := query.(string); ok {
  637. json.Unmarshal([]byte(strings.Replace(s, "'", "\"", -1)), &data)
  638. if ss, oks := data["_id"]; oks && isQuery {
  639. switch ss.(type) {
  640. case string:
  641. data["_id"], _ = primitive.ObjectIDFromHex(ss.(string))
  642. case map[string]interface{}:
  643. tmp := ss.(map[string]interface{})
  644. for k, v := range tmp {
  645. tmp[k], _ = primitive.ObjectIDFromHex(v.(string))
  646. }
  647. data["_id"] = tmp
  648. }
  649. }
  650. } else if s1, ok1 := query.(map[string]interface{}); ok1 {
  651. data = s1
  652. } else if s4, ok4 := query.(bson.M); ok4 {
  653. data = s4
  654. } else if s4, ok4 := query.(primitive.M); ok4 {
  655. data = s4
  656. } else {
  657. data = nil
  658. }
  659. return &data
  660. }
  661. func intAllDef(num interface{}, defaultNum int) int {
  662. if i, ok := num.(int); ok {
  663. return int(i)
  664. } else if i0, ok0 := num.(int32); ok0 {
  665. return int(i0)
  666. } else if i1, ok1 := num.(float64); ok1 {
  667. return int(i1)
  668. } else if i2, ok2 := num.(int64); ok2 {
  669. return int(i2)
  670. } else if i3, ok3 := num.(float32); ok3 {
  671. return int(i3)
  672. } else if i4, ok4 := num.(string); ok4 {
  673. in, _ := strconv.Atoi(i4)
  674. return int(in)
  675. } else if i5, ok5 := num.(int16); ok5 {
  676. return int(i5)
  677. } else if i6, ok6 := num.(int8); ok6 {
  678. return int(i6)
  679. } else if i7, ok7 := num.(*big.Int); ok7 {
  680. in, _ := strconv.Atoi(fmt.Sprint(i7))
  681. return int(in)
  682. } else if i8, ok8 := num.(*big.Float); ok8 {
  683. in, _ := strconv.Atoi(fmt.Sprint(i8))
  684. return int(in)
  685. } else {
  686. return defaultNum
  687. }
  688. }
  689. //出错拦截
  690. func catch() {
  691. if r := recover(); r != nil {
  692. log.Println(r)
  693. for skip := 0; ; skip++ {
  694. _, file, line, ok := runtime.Caller(skip)
  695. if !ok {
  696. break
  697. }
  698. go log.Printf("%v,%v\n", file, line)
  699. }
  700. }
  701. }
  702. //根据bsonID转string
  703. func BsonIdToSId(uid interface{}) string {
  704. if uid == nil {
  705. return ""
  706. } else if u, ok := uid.(string); ok {
  707. return u
  708. } else if u, ok := uid.(primitive.ObjectID); ok {
  709. return u.Hex()
  710. } else {
  711. return ""
  712. }
  713. }
  714. func StringTOBsonId(id string) (bid primitive.ObjectID) {
  715. defer catch()
  716. if id != "" {
  717. bid, _ = primitive.ObjectIDFromHex(id)
  718. }
  719. return
  720. }
  721. func ToObjectIds(ids []string) []primitive.ObjectID {
  722. _ids := []primitive.ObjectID{}
  723. for _, v := range ids {
  724. _id, _ := primitive.ObjectIDFromHex(v)
  725. _ids = append(_ids, _id)
  726. }
  727. return _ids
  728. }
  729. //自动添加更新时间
  730. func autoUpdateTime(db, coll string, ue *bson.M) {
  731. if db == "qfw" && coll == "user" {
  732. set := ObjToM((*ue)["$set"])
  733. if *set == nil {
  734. set = &bson.M{}
  735. }
  736. (*set)["auto_updatetime"] = time.Now().Unix()
  737. (*ue)["$set"] = set
  738. }
  739. }