mongodb.go 849 B

1234567891011121314151617181920212223242526272829303132
  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. url = "mongodb://admin:admin@124.223.73.12:27017"
  16. } else {
  17. url = "mongodb://admin:admin@localhost:27017"
  18. }
  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. }