com.bq.oss.lib.mongo.config.MongoCommonRepositoryFactoryBean Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2014 StarTIC
 */
package com.bq.oss.lib.mongo.config;

import java.io.Serializable;

import com.bq.oss.lib.mongo.repository.impl.ExtendedRepository;
import org.springframework.data.mongodb.core.MongoOperations;
import org.springframework.data.mongodb.repository.MongoRepository;
import org.springframework.data.mongodb.repository.support.MongoRepositoryFactory;
import org.springframework.data.mongodb.repository.support.MongoRepositoryFactoryBean;
import org.springframework.data.repository.core.RepositoryMetadata;
import org.springframework.data.repository.core.support.RepositoryFactorySupport;

/**
 * @author Alexander De Leon
 * 
 */
public class MongoCommonRepositoryFactoryBean, T, I extends Serializable> extends
		MongoRepositoryFactoryBean {

	@Override
	protected RepositoryFactorySupport getFactoryInstance(MongoOperations operations) {
		return new MongoCommonRepositoryFactory(operations);
	}

	private class MongoCommonRepositoryFactory extends MongoRepositoryFactory {

		private final MongoOperations mongoOperations;

		public MongoCommonRepositoryFactory(MongoOperations mongoOperations) {
			super(mongoOperations);
			this.mongoOperations = mongoOperations;
		}

		@Override
		protected Class getRepositoryBaseClass(RepositoryMetadata metadata) {
			return ExtendedRepository.class;
		}

		@SuppressWarnings({ "rawtypes", "unchecked" })
		@Override
		protected Object getTargetRepository(RepositoryMetadata metadata) {
			return new ExtendedRepository(getEntityInformation(metadata.getDomainType()), mongoOperations);
		}
	}

}