io.djigger.collector.accessors.stackref.AbstractAccessor.java Source code

Introduction

Here is the source code for io.djigger.collector.accessors.stackref.AbstractAccessor.java

Source

/*******************************************************************************
 * (C) Copyright 2016 Jrme Comte and Dorian Cransac
 *  
 *  This file is part of djigger
 *  
 *  djigger is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Affero General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *  
 *  djigger is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Affero General Public License for more details.
 *  
 *  You should have received a copy of the GNU Affero General Public License
 *  along with djigger.  If not, see <http://www.gnu.org/licenses/>.
 *
 *******************************************************************************/
package io.djigger.collector.accessors.stackref;

import java.util.concurrent.TimeUnit;

import org.bson.Document;

import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.IndexOptions;

public class AbstractAccessor {

    public AbstractAccessor() {
        super();
    }

    protected void createOrUpdateIndex(MongoCollection<Document> collection, String attribute) {
        Document index = getIndex(collection, attribute);
        if (index == null) {
            collection.createIndex(new Document(attribute, 1));
        }
    }

    protected void createOrUpdateTTLIndex(MongoCollection<Document> collection, String attribute, Long ttl) {
        Document ttlIndex = getIndex(collection, attribute);
        if (ttlIndex == null) {
            if (ttl != null && ttl > 0) {
                createTimestampIndexWithTTL(collection, attribute, ttl);
            } else {
                createTimestampIndex(collection, attribute);
            }
        } else {
            if (ttl != null && ttl > 0) {
                if (!ttlIndex.containsKey("expireAfterSeconds")
                        || !ttlIndex.getLong("expireAfterSeconds").equals(ttl)) {
                    dropIndex(collection, ttlIndex);
                    createTimestampIndexWithTTL(collection, attribute, ttl);
                }
            } else {
                if (ttlIndex.containsKey("expireAfterSeconds")) {
                    dropIndex(collection, ttlIndex);
                    createTimestampIndex(collection, attribute);
                }
            }
        }
    }

    private void dropIndex(MongoCollection<Document> collection, Document ttlIndex) {
        collection.dropIndex(ttlIndex.getString("name"));
    }

    private void createTimestampIndexWithTTL(MongoCollection<Document> collection, String attribute, Long ttl) {
        IndexOptions options = new IndexOptions();
        options.expireAfter(ttl, TimeUnit.SECONDS);
        createTimestampIndexWithOptions(collection, attribute, options);
    }

    private void createTimestampIndex(MongoCollection<Document> collection, String attribute) {
        IndexOptions options = new IndexOptions();
        createTimestampIndexWithOptions(collection, attribute, options);
    }

    private void createTimestampIndexWithOptions(MongoCollection<Document> collection, String attribute,
            IndexOptions options) {
        collection.createIndex(new Document(attribute, 1), options);
    }

    private Document getIndex(MongoCollection<Document> collection, String indexName) {
        for (Document index : collection.listIndexes()) {
            Object o = index.get("key");
            if (o instanceof Document) {
                if (((Document) o).containsKey(indexName)) {
                    return (Document) index;
                }
            }
        }
        return null;
    }

}