4.5 调用MongoDB

4.5.1 简介

store/mongox 包是对 go.mongodb.org/mongo-driver/mongo 进行二次封装。

4.5.2 配置规范

配置说明4.5 调用MongoDB - 图1 (opens new window)

4.5.3 用法

访问mongo示例4.5 调用MongoDB - 图2 (opens new window)

  1. // run: go run main.go --config=config.toml
  3. package main
  5. import (
  6.     "context"
  7.     "fmt"
  8.     "log"
  9.     "time"
  11.     "github.com/douyu/jupiter"
  12.     "github.com/douyu/jupiter/pkg/store/mongox"
  13.     "github.com/douyu/jupiter/pkg/xlog"
  14.     "go.mongodb.org/mongo-driver/mongo"
  15.     "gopkg.in/mgo.v2/bson"
  16. )
  18. // run: go run main.go -config=config.toml
  19. type Engine struct {
  20.     jupiter.Application
  21. }
  23. func NewEngine() *Engine {
  24.     eng := &Engine{}
  25.     if err := eng.Startup(
  26.         eng.exampleMongo,
  27.     ); err != nil {
  28.         xlog.Panic("startup", xlog.Any("err", err))
  29.     }
  30.     return eng
  31. }
  33. func main() {
  34.     app := NewEngine()
  35.     if err := app.Run(); err != nil {
  36.         panic(err)
  37.     }
  38. }
  40. func (eng *Engine) exampleMongo() (err error) {
  41.     client := mongox.StdConfig("test").Build()
  43.     write(client)
  44.     read(client)
  46.     return
  47. }
  49. func write(client *mongo.Client) {
  50.     collection := client.Database("test").Collection("test")
  52.     ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  53.     defer cancel()
  55.     _, err := collection.InsertOne(ctx, bson.M{"rid": 888, "dateline": time.Now().Unix()})
  56.     if err != nil {
  57.         panic(err)
  58.     }
  59. }
  61. func read(client *mongo.Client) {
  63.     collection := client.Database("test").Collection("test")
  65.     ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  66.     defer cancel()
  68.     cur, err := collection.Find(ctx, bson.M{"rid": 888})
  69.     if err != nil {
  70.         log.Fatal(err)
  71.     }
  72.     defer cur.Close(ctx)
  73.     for cur.Next(ctx) {
  74.         var result bson.M
  75.         err := cur.Decode(&result)
  76.         if err != nil {
  77.             xlog.Fatal("exampleMongo", xlog.Any("err", err.Error()))
  78.         }
  79.         fmt.Println("result...", result)
  81.         // do something with result....
  82.     }
  83. }

执行 go run main.go —config=config.toml,可以看到如下图结果 image