Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/**
* Copyright 2013-2017 the original author or authors.
*
* 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.mybatis.cdi;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import javax.enterprise.event.Observes;
import javax.enterprise.inject.spi.AfterBeanDiscovery;
import javax.enterprise.inject.spi.AnnotatedMember;
import javax.enterprise.inject.spi.AnnotatedType;
import javax.enterprise.inject.spi.Bean;
import javax.enterprise.inject.spi.BeanManager;
import javax.enterprise.inject.spi.Extension;
import javax.enterprise.inject.spi.InjectionPoint;
import javax.enterprise.inject.spi.InjectionTarget;
import javax.enterprise.inject.spi.ProcessAnnotatedType;
import javax.enterprise.inject.spi.ProcessInjectionTarget;
import javax.enterprise.inject.spi.ProcessProducer;
import javax.inject.Named;
import javax.inject.Qualifier;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.logging.Level;
import java.util.logging.Logger;
/**
* MyBatis CDI extension
*
* @author Frank D. Martinez [mnesarco]
*/
public class MybatisExtension implements Extension {
private static final Logger LOGGER = Logger.getLogger(MybatisExtension.class.getName());
private final Set sessionProducers = new HashSet();
private final Set mapperTypes = new HashSet();
private final Set injectionPoints = new HashSet();
/**
* Collect types of all mappers annotated with Mapper.
*
* @param
* @param pat
*/
@SuppressWarnings("UnusedDeclaration")
protected void processAnnotatedType(@Observes final ProcessAnnotatedType pat) {
final AnnotatedType at = pat.getAnnotatedType();
if (at.isAnnotationPresent(Mapper.class)) {
LOGGER.log(Level.INFO, "MyBatis CDI Module - Found class with @Mapper-annotation: {0}", at.getJavaClass()
.getSimpleName());
this.mapperTypes.add(at.getBaseType());
}
}
/**
* Collect all SqlSessionFactory producers annotated with SessionFactoryProvider.
*
* @param
* @param
* @param pp
*/
@SuppressWarnings("UnusedDeclaration")
protected void processProducer(@Observes final ProcessProducer pp) {
final AnnotatedMember am = pp.getAnnotatedMember();
final boolean isAnnotated = am.isAnnotationPresent(SessionFactoryProvider.class), isSqlSessionFactory = am
.getBaseType().equals(SqlSessionFactory.class);
final Object[] logData = { am.getJavaMember().getDeclaringClass().getSimpleName(), am.getJavaMember().getName() };
if (isAnnotated && isSqlSessionFactory) {
LOGGER.log(Level.INFO, "MyBatis CDI Module - SqlSessionFactory producer {0}.{1}", logData);
this.sessionProducers.add(new BeanKey((Class) (Type) SqlSession.class, am.getAnnotations()));
} else if (isAnnotated && !isSqlSessionFactory) {
LOGGER
.log(Level.SEVERE, "MyBatis CDI Module - Invalid return type (Must be SqlSessionFactory): {0}.{1}", logData);
pp.addDefinitionError(new MybatisCdiConfigurationException(String.format(
"SessionFactoryProvider producers must return SqlSessionFactory (%s.%s)", logData[0], logData[1])));
} else if (!isAnnotated && isSqlSessionFactory) {
LOGGER
.log(
Level.WARNING,
"MyBatis CDI Module - Ignored SqlSessionFactory producer because it is not annotated with @SessionFactoryProvider: {0}.{1}",
logData);
}
}
/**
* Collect all targets to match Mappers and Session providers dependency.
*
* @param
* @param event
*/
protected void processInjectionTarget(@Observes ProcessInjectionTarget event) {
final InjectionTarget it = event.getInjectionTarget();
for (final InjectionPoint ip : it.getInjectionPoints()) {
injectionPoints.add(ip);
}
}
/**
* Register all mybatis injectable beans.
*
* @param abd
* @param bm
*/
protected void afterBeanDiscovery(@Observes AfterBeanDiscovery abd, BeanManager bm) {
LOGGER.log(Level.INFO, "MyBatis CDI Module - Activated");
Set mappers = new HashSet();
Set sessionTargets = new HashSet();
for (InjectionPoint ip : injectionPoints) {
if (this.mapperTypes.contains(ip.getAnnotated().getBaseType())) {
LOGGER.log(Level.INFO, "MyBatis CDI Module - Found a bean, which needs a Mapper {0}", new Object[] { ip
.getAnnotated().getBaseType() });
mappers.add(new BeanKey((Class) ip.getAnnotated().getBaseType(), ip.getAnnotated().getAnnotations()));
} else if (SqlSession.class.equals(ip.getAnnotated().getBaseType())) {
sessionTargets.add(new BeanKey((Class) ip.getAnnotated().getBaseType(), ip.getAnnotated()
.getAnnotations()));
}
}
this.injectionPoints.clear();
// Mappers -----------------------------------------------------------------
for (BeanKey key : mappers) {
LOGGER.log(Level.INFO, "MyBatis CDI Module - Managed Mapper dependency: {0}, {1}", new Object[] { key.getKey(),
key.type.getName() });
abd.addBean(key.createBean(bm));
}
this.mapperTypes.clear();
// SqlSessionFactories -----------------------------------------------------
for (BeanKey key : this.sessionProducers) {
LOGGER.log(Level.INFO, "MyBatis CDI Module - Managed SqlSession: {0}, {1}",
new Object[] { key.getKey(), key.type.getName() });
abd.addBean(key.createBean(bm));
sessionTargets.remove(key);
}
this.sessionProducers.clear();
// Unmanaged SqlSession targets --------------------------------------------
for (BeanKey key : sessionTargets) {
LOGGER.log(Level.WARNING, "MyBatis CDI Module - Unmanaged SqlSession: {0}, {1}", new Object[] { key.getKey(),
key.type.getName() });
}
}
/**
* Unique key for fully qualified Mappers and Sessions.
*/
private static final class BeanKey implements Comparable {
private final String key;
private final List qualifiers;
private final Class type;
private final String sqlSessionManagerName;
public BeanKey(Class type, Set annotations) {
this.type = type;
this.qualifiers = sort(filterQualifiers(annotations));
// Create key = type(.qualifier)*(.name)?
final StringBuilder sb = new StringBuilder();
String name = null;
sb.append(type.getName());
for (Annotation q : this.qualifiers) {
if (q instanceof Named) {
name = ((Named) q).value();
} else {
sb.append(".").append(q.annotationType().getSimpleName());
}
}
if (name != null) {
sb.append("_").append(name);
}
this.key = sb.toString();
this.sqlSessionManagerName = name;
}
private Set filterQualifiers(Set annotations) {
final Set set = new HashSet();
for (Annotation a : annotations) {
if (a.annotationType().isAnnotationPresent(Qualifier.class)) {
set.add(a);
}
}
return set;
}
private List sort(Set annotations) {
final List list = new ArrayList(annotations);
Collections.sort(list, new Comparator() {
@Override
public int compare(Annotation a, Annotation b) {
return a.getClass().getName().compareTo(b.getClass().getName());
}
});
return list;
}
@Override
public int compareTo(BeanKey o) {
return this.key.compareTo(o.key);
}
@Override
public int hashCode() {
int hash = 3;
hash = 43 * hash + (this.key != null ? this.key.hashCode() : 0);
return hash;
}
@Override
public boolean equals(Object obj) {
if (obj == null) {
return false;
}
if (getClass() != obj.getClass()) {
return false;
}
final BeanKey other = (BeanKey) obj;
return !(this.key == null ? other.key != null : !this.key.equals(other.key));
}
public Bean createBean(BeanManager bm) {
return new MyBatisBean(this.key, this.type, new HashSet(this.qualifiers), this.sqlSessionManagerName);
}
public String getKey() {
return this.key;
}
@Override
public String toString() {
return this.key;
}
}
}
