lsm/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2025-08-01 17:46:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

filer mongodb add option poll size and fixed order by list meta

Browse Source
This commit is contained in:
bukton 2020-04-19 17:51:32 +07:00
parent 082f079ead
commit d6d245dc89
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
weed/filer2/mongodb/mongodb_store.go
View File

@ -36,12 +36,19 @@ func (store *MongodbStore) GetName() string {
func (store *MongodbStore) Initialize(configuration util.Configuration, prefix string) (err error) { func (store *MongodbStore) Initialize(configuration util.Configuration, prefix string) (err error) {
store.database = configuration.GetString(prefix + "database") store.database = configuration.GetString(prefix + "database")
store.collectionName = "filemeta" store.collectionName = "filemeta"
return store.connection(configuration.GetString(prefix + "uri")) poolSize := configuration.GetInt(prefix + "option_pool_size")
return store.connection(configuration.GetString(prefix + "uri"), uint64(poolSize))
} }
func (store *MongodbStore) connection(uri string) (err error) { func (store *MongodbStore) connection(uri string, poolSize uint64) (err error) {
ctx, _ := context.WithTimeout(context.Background(), 10*time.Second) ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri)) opts := options.Client().ApplyURI(uri)
if poolSize > 0 {
opts.SetMaxPoolSize(poolSize)
}
client, err := mongo.Connect(ctx, opts)
store.connect = client store.connect = client
return err return err
} }
@ -141,7 +148,8 @@ func (store *MongodbStore) ListDirectoryEntries(ctx context.Context, fullpath ut
} }
} }
optLimit := int64(limit) optLimit := int64(limit)
cur, err := store.connect.Database(store.database).Collection(store.collectionName).Find(ctx, where, &options.FindOptions{Limit: &optLimit}) opts := &options.FindOptions{Limit: &optLimit, Sort: bson.M{ "name": 1 }}
cur, err := store.connect.Database(store.database).Collection(store.collectionName).Find(ctx, where, opts)
for cur.Next(ctx) { for cur.Next(ctx) {
var data Model var data Model
err := cur.Decode(&data) err := cur.Decode(&data)