switch to mongodb uri in config
This commit is contained in:
parent
a76c0412a3
commit
c96b6ed943
|
@ -19,13 +19,18 @@ type MongoDBStorage struct {
|
||||||
|
|
||||||
func NewMongoDBStorage(cfg *viper.Viper) (MongoDBStorage, error) {
|
func NewMongoDBStorage(cfg *viper.Viper) (MongoDBStorage, error) {
|
||||||
var (
|
var (
|
||||||
|
mongodb_uri = cfg.GetString("storage.db.mongodb.uri")
|
||||||
mongodb_host = cfg.GetString("storage.db.mongodb.host")
|
mongodb_host = cfg.GetString("storage.db.mongodb.host")
|
||||||
mongodb_port = cfg.GetString("storage.db.mongodb.port")
|
mongodb_port = cfg.GetString("storage.db.mongodb.port")
|
||||||
mongodb_dbname = cfg.GetString("storage.db.mongodb.db_name")
|
mongodb_dbname = cfg.GetString("storage.db.mongodb.db_name")
|
||||||
mongodb_users = cfg.GetString("storage.db.mongodb.collections.users")
|
mongodb_users = cfg.GetString("storage.db.mongodb.collections.users")
|
||||||
)
|
)
|
||||||
|
|
||||||
client, err := mongo.NewClient(options.Client().ApplyURI("mongodb://" + mongodb_host + ":" + mongodb_port))
|
if mongodb_uri == "" {
|
||||||
|
mongodb_uri = fmt.Sprintf("mongodb://%s:%s/%s", mongodb_host, mongodb_port, mongodb_dbname)
|
||||||
|
}
|
||||||
|
|
||||||
|
client, err := mongo.NewClient(options.Client().ApplyURI(mongodb_uri))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return MongoDBStorage{}, err
|
return MongoDBStorage{}, err
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue