package jp.ac.u_ryukyu.ie.cr.jungle.test;

import com.mongodb.client.FindIterable;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import org.bson.Document;


public class findMongoAttributeThread extends DataBaseBenchMarkThread {
    private long findCount;
    boolean loop = true;
    MongoCollection<Document> coll;

    public findMongoAttributeThread( MongoCollection<Document> coll) {
        this.coll = coll;
        this.findCount = 0;
    }

    @Override
    public long getFindCount() {
        System.out.println("thread count = " + findCount);
        return findCount;
    }

    public void set(boolean loop) {
        this.loop = loop;
    }

    @Override
    public void run() {
        while (loop) {
            FindIterable<Document> searchResultList = coll.find(new Document("key","1"));
            MongoCursor<Document> iterator = searchResultList.iterator();
            if (iterator.hasNext()) {
                findCount++;
            }
        }
    }
}