Jon Brisbin
15 years ago
committed by
J. Brisbin
8 changed files with 310 additions and 138 deletions
@ -0,0 +1,201 @@
@@ -0,0 +1,201 @@
|
||||
/* |
||||
* Copyright (c) 2011 by the original author(s). |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.springframework.data.document.mongodb.mapping.event; |
||||
|
||||
import java.lang.reflect.Field; |
||||
import java.util.Collections; |
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
import java.util.Set; |
||||
import java.util.concurrent.ConcurrentHashMap; |
||||
import java.util.concurrent.ExecutorService; |
||||
import java.util.concurrent.Executors; |
||||
import java.util.concurrent.LinkedBlockingQueue; |
||||
|
||||
import com.mongodb.BasicDBObject; |
||||
import com.mongodb.DBCollection; |
||||
import com.mongodb.DBObject; |
||||
import com.mongodb.MongoException; |
||||
import com.mongodb.util.JSON; |
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
import org.springframework.beans.BeansException; |
||||
import org.springframework.context.ApplicationContext; |
||||
import org.springframework.context.ApplicationContextAware; |
||||
import org.springframework.context.ApplicationListener; |
||||
import org.springframework.dao.DataAccessException; |
||||
import org.springframework.data.document.mongodb.CollectionCallback; |
||||
import org.springframework.data.document.mongodb.MongoTemplate; |
||||
import org.springframework.data.document.mongodb.index.CompoundIndex; |
||||
import org.springframework.data.document.mongodb.index.CompoundIndexes; |
||||
import org.springframework.data.document.mongodb.index.IndexDirection; |
||||
import org.springframework.data.document.mongodb.index.Indexed; |
||||
import org.springframework.data.document.mongodb.mapping.Document; |
||||
import org.springframework.data.document.mongodb.mapping.MongoPersistentEntity; |
||||
import org.springframework.data.mapping.PropertyHandler; |
||||
import org.springframework.data.mapping.event.MappingContextEvent; |
||||
import org.springframework.data.mapping.model.PersistentProperty; |
||||
|
||||
/** |
||||
* @author Jon Brisbin <jbrisbin@vmware.com> |
||||
*/ |
||||
public class IndexCreationListener implements ApplicationListener<MappingContextEvent>, ApplicationContextAware { |
||||
|
||||
private static final Logger log = LoggerFactory.getLogger(IndexCreationListener.class); |
||||
|
||||
private Map<String, CompoundIndex> compoundIndexes = new HashMap<String, CompoundIndex>(); |
||||
private Map<String, Indexed> fieldIndexes = new HashMap<String, Indexed>(); |
||||
private ApplicationContext applicationContext; |
||||
private MongoTemplate mongoTemplate; |
||||
private ExecutorService worker = Executors.newFixedThreadPool(1); |
||||
private LinkedBlockingQueue<MappingContextEvent> mappingEvents = new LinkedBlockingQueue<MappingContextEvent>(); |
||||
private Set<Class<?>> classesSeen = Collections.newSetFromMap(new ConcurrentHashMap<Class<?>, Boolean>()); |
||||
|
||||
public IndexCreationListener() { |
||||
worker.submit(new IndexCreationWorker()); |
||||
} |
||||
|
||||
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { |
||||
this.applicationContext = applicationContext; |
||||
} |
||||
|
||||
public MongoTemplate getMongoTemplate() { |
||||
return mongoTemplate; |
||||
} |
||||
|
||||
public void setMongoTemplate(MongoTemplate mongoTemplate) { |
||||
this.mongoTemplate = mongoTemplate; |
||||
} |
||||
|
||||
public void onApplicationEvent(MappingContextEvent event) { |
||||
mappingEvents.add(event); |
||||
} |
||||
|
||||
public void cleanUp() throws InterruptedException { |
||||
while (mappingEvents.size() > 0) { |
||||
Thread.yield(); |
||||
} |
||||
} |
||||
|
||||
private class IndexCreationWorker implements Runnable { |
||||
public void run() { |
||||
while (true) { |
||||
MappingContextEvent event = null; |
||||
try { |
||||
event = mappingEvents.take(); |
||||
if (null == applicationContext) { |
||||
Thread.sleep(500); |
||||
mappingEvents.add(event); |
||||
} |
||||
} catch (InterruptedException ignored) { |
||||
if (log.isDebugEnabled()) { |
||||
log.debug(ignored.getMessage(), ignored); |
||||
} |
||||
break; |
||||
} |
||||
if (event.getPersistentEntity() instanceof MongoPersistentEntity<?>) { |
||||
MongoPersistentEntity<?> entity = (MongoPersistentEntity<?>) event.getPersistentEntity(); |
||||
Class<?> type = event.getTypeInformation().getType(); |
||||
if (!classesSeen.contains(type)) { |
||||
if (log.isDebugEnabled()) { |
||||
log.debug("Analyzing class " + type + " for index information."); |
||||
} |
||||
// Check for special collection setting
|
||||
if (type.isAnnotationPresent(Document.class)) { |
||||
Document doc = type.getAnnotation(Document.class); |
||||
String collection = doc.collection(); |
||||
if ("".equals(collection)) { |
||||
collection = type.getSimpleName().toLowerCase(); |
||||
} |
||||
entity.setCollection(collection); |
||||
} |
||||
|
||||
// Make sure indexes get created
|
||||
if (type.isAnnotationPresent(CompoundIndexes.class)) { |
||||
CompoundIndexes indexes = type.getAnnotation(CompoundIndexes.class); |
||||
for (CompoundIndex index : indexes.value()) { |
||||
String indexColl = index.collection(); |
||||
if ("".equals(indexColl)) { |
||||
indexColl = type.getSimpleName().toLowerCase(); |
||||
} |
||||
if (!compoundIndexes.containsKey(indexColl)) { |
||||
ensureIndex(indexColl, index.name(), index.def(), index.direction(), index.unique(), index.dropDups(), index.sparse()); |
||||
if (log.isDebugEnabled()) { |
||||
log.debug("Created compound index " + index); |
||||
} |
||||
compoundIndexes.put(indexColl, index); |
||||
} |
||||
} |
||||
} |
||||
|
||||
entity.doWithProperties(new PropertyHandler() { |
||||
public void doWithPersistentProperty(PersistentProperty persistentProperty) { |
||||
Field field = persistentProperty.getField(); |
||||
if (field.isAnnotationPresent(Indexed.class)) { |
||||
Indexed index = field.getAnnotation(Indexed.class); |
||||
String collection = index.collection(); |
||||
if ("".equals(collection)) { |
||||
collection = field.getName(); |
||||
} |
||||
if (!fieldIndexes.containsKey(collection)) { |
||||
ensureIndex(collection, index.name(), null, index.direction(), index.unique(), index.dropDups(), index.sparse()); |
||||
if (log.isDebugEnabled()) { |
||||
log.debug("Created property index " + index); |
||||
} |
||||
fieldIndexes.put(collection, index); |
||||
} |
||||
} |
||||
} |
||||
}); |
||||
|
||||
classesSeen.add(type); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
protected void ensureIndex(String collection, |
||||
final String name, |
||||
final String def, |
||||
final IndexDirection direction, |
||||
final boolean unique, |
||||
final boolean dropDups, |
||||
final boolean sparse) { |
||||
mongoTemplate.execute(collection, new CollectionCallback<Object>() { |
||||
public Object doInCollection(DBCollection collection) throws MongoException, DataAccessException { |
||||
DBObject defObj; |
||||
if (null != def) { |
||||
defObj = (DBObject) JSON.parse(def); |
||||
} else { |
||||
defObj = new BasicDBObject(); |
||||
defObj.put(name, (direction == IndexDirection.ASCENDING ? 1 : -1)); |
||||
} |
||||
DBObject opts = new BasicDBObject(); |
||||
if (!"".equals(name)) { |
||||
opts.put("name", name); |
||||
} |
||||
opts.put("dropDups", dropDups); |
||||
opts.put("sparse", sparse); |
||||
opts.put("unique", unique); |
||||
collection.ensureIndex(defObj, opts); |
||||
return null; |
||||
} |
||||
}); |
||||
} |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue