Merge pull request #99 from kaiso/F#72

[F#72] Enable multiple mongoTemplate support

Author: kaiso

.gitignore

@@ -5,3 +5,5 @@ pom.xml.versionsBackup
 .settings/
 .sass-cache
 docs/robots.txt
+docs/src/_site
+docs/src/.ruby-version

docs/src/Gemfile

@@ -3,6 +3,7 @@
 source 'https://rubygems.org'
 #gems
 gem "jekyll-theme-architect"
+gem "kramdown-parser-gfm"
 group :jekyll_plugins do
  gem 'jekyll-sitemap'
 end

docs/src/Gemfile.lock

@@ -4,17 +4,17 @@ GEM
  addressable (2.7.0)
  public_suffix (>= 2.0.2, < 5.0)
  colorator (1.1.0)
- concurrent-ruby (1.1.7)
+ concurrent-ruby (1.1.8)
  em-websocket (0.5.2)
  eventmachine (>= 0.12.9)
  http_parser.rb (~> 0.6.0)
  eventmachine (1.2.7)
- ffi (1.13.1)
+ ffi (1.15.0)
  forwardable-extended (2.6.0)
  http_parser.rb (0.6.0)
  i18n (0.9.5)
  concurrent-ruby (~> 1.0)
- jekyll (3.9.0)
+ jekyll (3.9.1)
  addressable (~> 2.4)
  colorator (~> 1.0)
  em-websocket (~> 0.5)
@@ -38,10 +38,12 @@ GEM
  jekyll-seo-tag (~> 2.0)
  jekyll-watch (2.2.1)
  listen (~> 3.0)
- kramdown (2.3.0)
+ kramdown (2.3.1)
  rexml
+ kramdown-parser-gfm (1.1.0)
+ kramdown (~> 2.0)
  liquid (4.0.3)
- listen (3.2.1)
+ listen (3.5.1)
  rb-fsevent (~> 0.10, >= 0.10.3)
  rb-inotify (~> 0.9, >= 0.9.10)
  mercenary (0.3.6)
@@ -51,8 +53,8 @@ GEM
  rb-fsevent (0.10.4)
  rb-inotify (0.10.1)
  ffi (~> 1.0)
- rexml (3.2.4)
- rouge (3.24.0)
+ rexml (3.2.5)
+ rouge (3.26.0)
  safe_yaml (1.0.5)
  sass (3.7.4)
  sass-listen (~> 4.0.0)
@@ -66,6 +68,7 @@ PLATFORMS
 DEPENDENCIES
  jekyll-sitemap
  jekyll-theme-architect
+ kramdown-parser-gfm
 
 BUNDLED WITH
- 2.1.1
+ 2.2.16

pom.xml

@@ -5,7 +5,7 @@
 
 <groupId>io.github.kaiso.relmongo</groupId>
 <artifactId>relmongo</artifactId>
-<version>3.4.2</version>
+<version>3.5.0-RC4</version>
 <packaging>jar</packaging>
 
 <name>relmongo</name>

src/main/java/io/github/kaiso/relmongo/config/EnableRelMongo.java

@@ -20,22 +20,33 @@
 import io.github.kaiso.relmongo.annotation.OneToOne;
 
 import org.springframework.context.annotation.Import;
+import org.springframework.data.mongodb.core.MongoTemplate;
 
 import java.lang.annotation.ElementType;
 import java.lang.annotation.Retention;
 import java.lang.annotation.RetentionPolicy;
 import java.lang.annotation.Target;
+
 /**
  * Enables the RelMongo framework processing.<br>
- * RelMongo allows to implement relations and associations between MongoDB collections <br>
+ * RelMongo allows to implement relations and associations between MongoDB
+ * collections <br>
+ * 
  * @see OneToMany
  * @see OneToOne
  * @author Kais OMRI
  *
  */
 @Retention(RetentionPolicy.RUNTIME)
 @Target(value = { ElementType.TYPE })
-@Import(value = RelMongoConfiguration.class)
+@Import(value = RelMongoProcessorRegistrar.class)
 public @interface EnableRelMongo {
+ 
+ /**
+ * Configures the name of the {@link MongoTemplate} bean to be used with RelMongo.
+ *
+ * @return
+ */
+ String mongoTemplateRef() default "mongoTemplate";
 
 }

src/main/java/io/github/kaiso/relmongo/config/EnableRelationalMongo.java

@@ -0,0 +1,77 @@
+package io.github.kaiso.relmongo.config;
+
+import io.github.kaiso.relmongo.events.processor.RelMongoProcessor;
+
+import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.BeanInitializationException;
+import org.springframework.beans.factory.config.BeanPostProcessor;
+import org.springframework.context.ApplicationEventPublisher;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.mapping.event.MongoMappingEvent;
+
+import java.lang.reflect.Field;
+
+public class RelMongoBeanPostProcessor implements BeanPostProcessor {
+
+ private final String mongoTemplateRef;
+
+ public RelMongoBeanPostProcessor(String mongoTemplateRef) {
+ super();
+ this.mongoTemplateRef = mongoTemplateRef;
+ }
+
+ @Override
+ public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
+
+ if (bean instanceof MongoTemplate && beanName.equals(mongoTemplateRef)) {
+
+ /*
+ * Enhancer enhancer = new Enhancer();
+ * enhancer.setSuperclass(bean.getClass());
+ * enhancer.setCallbacks(new MethodInterceptor[] { new
+ * RelMongoTemplateInvocationHandler() });
+ * 
+ * Object proxy = enhancer.create(new Class<?>[] { MongoDbFactory.class,
+ * MongoConverter.class },
+ * new Object[] { ((MongoTemplate) bean).getMongoDbFactory(), ((MongoTemplate)
+ * bean).getConverter() });
+ * 
+ * ((MongoTemplate) proxy).setApplicationContext(applicationContext);
+ * 
+ * return proxy;
+ */
+
+ try {
+ Field ep = MongoTemplate.class.getDeclaredField("eventPublisher");
+ ep.setAccessible(true);
+ ep.set(bean, new RelMongoEventPublisher((MongoTemplate) bean, (ApplicationEventPublisher) ep.get(bean)));
+ } catch (NoSuchFieldException | SecurityException | IllegalArgumentException | IllegalAccessException e) {
+ throw new BeanInitializationException("Fatal: failed to init the RelMongo Engine", e);
+ }
+ }
+
+ return bean;
+ }
+
+ private static final class RelMongoEventPublisher implements ApplicationEventPublisher {
+
+ private final RelMongoProcessor relMongoProcessor;
+ private final MongoTemplate mongoTemplate;
+ private final ApplicationEventPublisher eventPublisher;
+
+ public RelMongoEventPublisher(MongoTemplate mongoTemplate, ApplicationEventPublisher eventPublisher) {
+ super();
+ this.relMongoProcessor = new RelMongoProcessor();
+ this.mongoTemplate = mongoTemplate;
+ this.eventPublisher = eventPublisher;
+ }
+
+ @Override
+ public void publishEvent(Object event) {
+ relMongoProcessor.onApplicationEvent((MongoMappingEvent<?>) event, mongoTemplate);
+ eventPublisher.publishEvent(event);
+ }
+
+ }
+
+}