mongodb.go 847 B

123456789101112131415161718192021222324252627282930313233
  1. package bootstrap
  2. import (
  3. "context"
  4. "designs/config"
  5. "designs/global"
  6. "go.mongodb.org/mongo-driver/v2/mongo"
  7. "go.mongodb.org/mongo-driver/v2/mongo/readpref"
  8. "go.uber.org/zap"
  9. )
  10. import mongoOption "go.mongodb.org/mongo-driver/v2/mongo/options"
  11. func InitializeMongo() *mongo.Client {
  12. var url string
  13. if config.Get("app.local") == "local" {
  14. url = "mongodb://localhost:27017"
  15. } else {
  16. url = "mongodb://admin:admin@localhost:27017"
  17. }
  18. url = "mongodb://admin:admin@124.223.73.12:27017"
  19. client, _ := mongo.Connect(mongoOption.Client().ApplyURI(url))
  20. //client, _ := mongo.Connect(mongoOption.Client().ApplyURI("mongodb://localhost:27017"))
  21. err := client.Ping(context.Background(), readpref.Primary())
  22. if err != nil {
  23. global.App.Log.Error("mongoDB connect ping failed, err:", zap.Any("err", err))
  24. return nil
  25. }
  26. return client
  27. }