mongo.groovy

import com.mongodb.MongoClient
import com.mongodb.client.MongoCollection
import com.mongodb.client.MongoDatabase
import groovy.transform.Field
import io.codearte.jfairy.Fairy
import io.codearte.jfairy.producer.text.TextProducer
import org.bson.Document

@Grab('io.codearte.jfairy:jfairy:0.4.3')
@Grab('org.mongodb:mongo-java-driver:3.0.3')

import java.util.logging.Logger

@Field Logger log = Logger.getLogger("")

@Field final int HOW_MANY = 100000
@Field final int HOW_MANY_CACHE = 10
@Field final boolean CLEAN = false
@Field MongoClient mongoClient = new MongoClient('public.mongo.projectmgr2015.tk')
@Field MongoDatabase database = mongoClient.getDatabase('cache')
@Field MongoCollection<Document> apiCollection = database.getCollection('api')
@Field MongoCollection<Document> cacheCollection = database.getCollection('cache')

@Field Fairy fairy = Fairy.create(new Locale('PL'))
@Field TextProducer textProducer = fairy.textProducer()

// MAIN
//cleanCollections()
def apis = createApis()
log.info 'creating cache'
HOW_MANY_CACHE.times {
    println it
    createCacheKeys(apis)
}
createIndexes()
if(CLEAN) {
    cleanCollections()
}

def cleanCollections() {
    println 'cleaning collections'
    apiCollection.deleteMany(new Document())
    cacheCollection.deleteMany(new Document())
}

private List createApis() {
    log.info 'creating api'
    List<Document> apis = (1..HOW_MANY).collect {
        return new Document([
                _id    : UUID.randomUUID().toString(),
        ])
    }
    apiCollection.insertMany(apis)
    return apis.collect {Document doc -> doc.getString('_id')}
}

private List createCacheKeys(List apis) {
    Random rand = new Random()
    List comments = apis.collect {
        def api = it
        new Document(
                api: api,
                key: textProducer.latinWord(1)+'_'+rand.nextLong(),
                value: textProducer.latinWord(4),
        )
    }
//    log.info 'shuffling'
//    Collections.shuffle(comments)
    cacheCollection.insertMany(comments)
}

private void createIndexes() {
    log.info 'creating index'
    cacheCollection.createIndex(new Document('api': 1, 'key': 1))
    cacheCollection.createIndex(new Document('api': 1))
}