io.zulia.server.config.cluster.MongoIndexService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zulia-server Show documentation
$project.description
There is a newer version: 1.6.4
package io.zulia.server.config.cluster;

import com.google.protobuf.util.JsonFormat;
import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.ReplaceOptions;
import io.zulia.message.ZuliaIndex.IndexMapping;
import io.zulia.message.ZuliaIndex.IndexSettings;
import io.zulia.server.config.IndexService;
import org.bson.Document;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class MongoIndexService implements IndexService {

	private static final String MAPPING = "mapping";
	private static final String INDEX_SETTINGS = "indexSettings";
	private static final String ID = "_id";
	private static final String INDEX_MAPPING = "indexMapping";

	private final MongoCollection settingsCollection;
	private final MongoCollection mappingCollection;

	public MongoIndexService(MongoClient mongoClient, String clusterName) {
		settingsCollection = mongoClient.getDatabase(clusterName).getCollection(SETTINGS);
		mappingCollection = mongoClient.getDatabase(clusterName).getCollection(MAPPING);
	}

	@Override
	public List getIndexes() throws Exception {

		List indexSettings = new ArrayList<>();
		for (Document doc : settingsCollection.find()) {
			Document indexSettingsDoc = (Document) doc.get(INDEX_SETTINGS);
			IndexSettings.Builder builder = IndexSettings.newBuilder();
			JsonFormat.parser().merge(indexSettingsDoc.toJson(), builder);
			indexSettings.add(builder.build());
		}

		if (!indexSettings.isEmpty()) {
			return indexSettings;
		}

		return Collections.emptyList();
	}

	@Override
	public IndexSettings getIndex(String indexName) throws Exception {
		Document doc = settingsCollection.find(new Document(ID, indexName)).first();

		if (doc != null) {
			Document indexSettingsDoc = (Document) doc.get(INDEX_SETTINGS);
			IndexSettings.Builder builder = IndexSettings.newBuilder();
			JsonFormat.parser().merge(indexSettingsDoc.toJson(), builder);
			return builder.build();
		}

		return null;
	}

	@Override
	public void createIndex(IndexSettings indexSettings) throws Exception {

		Document indexSettingsDoc = new Document(ID, indexSettings.getIndexName())
				.append(INDEX_SETTINGS, Document.parse(JsonFormat.printer().print(indexSettings)));

		settingsCollection.replaceOne(new Document(ID, indexSettings.getIndexName()), indexSettingsDoc, new ReplaceOptions().upsert(true));

	}

	@Override
	public void removeIndex(String indexName) throws Exception {
		settingsCollection.deleteOne(new Document(ID, indexName));
	}

	@Override
	public void removeIndexMapping(String indexName) throws Exception {
		mappingCollection.deleteOne(new Document(ID, indexName));
	}

	@Override
	public List getIndexMappings() throws Exception {

		List indexMappings = new ArrayList<>();
		for (Document doc : mappingCollection.find()) {

			Document indexMappingDoc = (Document) doc.get(INDEX_MAPPING);
			IndexMapping.Builder builder = IndexMapping.newBuilder();
			JsonFormat.parser().merge(indexMappingDoc.toJson(), builder);

			indexMappings.add(builder.build());

		}

		if (!indexMappings.isEmpty()) {
			return indexMappings;
		}

		return Collections.emptyList();

	}

	@Override
	public IndexMapping getIndexMapping(String indexName) throws Exception {
		Document doc = mappingCollection.find(new Document(ID, indexName)).first();

		if (doc != null) {
			Document indexingDoc = (Document) doc.get(INDEX_MAPPING);

			IndexMapping.Builder builder = IndexMapping.newBuilder();
			JsonFormat.parser().merge(indexingDoc.toJson(), builder);
			return builder.build();
		}

		return null;

	}

	@Override
	public void storeIndexMapping(IndexMapping indexMapping) throws Exception {

		Document indexMappingDoc = new Document(ID, indexMapping.getIndexName())
				.append(INDEX_MAPPING, Document.parse(JsonFormat.printer().print(indexMapping)));

		mappingCollection.replaceOne(new Document(ID, indexMapping.getIndexName()), indexMappingDoc, new ReplaceOptions().upsert(true));
	}
}