org.springframework.data.repository.config.RepositoryConfigurationSource Java Examples

@Override
protected AbstractBeanDefinition getDefaultKeyValueTemplateBeanDefinition(
		RepositoryConfigurationSource configurationSource) {

	RootBeanDefinition keyValueTemplateDefinition = new RootBeanDefinition(VaultKeyValueTemplate.class);

	ConstructorArgumentValues constructorArgumentValuesForKeyValueTemplate = new ConstructorArgumentValues();
	constructorArgumentValuesForKeyValueTemplate.addIndexedArgumentValue(0,
			new RuntimeBeanReference(VAULT_ADAPTER_BEAN_NAME));

	constructorArgumentValuesForKeyValueTemplate.addIndexedArgumentValue(1,
			new RuntimeBeanReference(VAULT_MAPPING_CONTEXT_BEAN_NAME));

	keyValueTemplateDefinition.setConstructorArgumentValues(constructorArgumentValuesForKeyValueTemplate);

	return keyValueTemplateDefinition;
}
 

private static void assertHashRepo(Class<?> repositoryInterface,
                                   Collection<RepositoryConfiguration<RepositoryConfigurationSource>> configs) {
    for (final RepositoryConfiguration<?> config : configs) {
        if (config.getRepositoryInterface().equals(repositoryInterface.getName())) {
            return;
        }
    }

    fail("expected to find config for repository interface "
            + repositoryInterface.getName() + ", but got: " + configs.toString());
}
 

@Override
protected AbstractBeanDefinition getDefaultKeyValueTemplateBeanDefinition(
		RepositoryConfigurationSource configurationSource) {

	BeanDefinitionBuilder adapterBuilder = BeanDefinitionBuilder.rootBeanDefinition(MapKeyValueAdapter.class);
	adapterBuilder.addConstructorArgValue(getMapTypeToUse(configurationSource));

	BeanDefinitionBuilder builder = BeanDefinitionBuilder.rootBeanDefinition(KeyValueTemplate.class);
	builder
			.addConstructorArgValue(ParsingUtils.getSourceBeanDefinition(adapterBuilder, configurationSource.getSource()));
	builder.setRole(BeanDefinition.ROLE_SUPPORT);

	return ParsingUtils.getSourceBeanDefinition(builder, configurationSource.getSource());
}
 

@Override
protected AbstractBeanDefinition getDefaultKeyValueTemplateBeanDefinition(RepositoryConfigurationSource configurationSource) {
    RootBeanDefinition keyValueTemplateDefinition = new RootBeanDefinition(KeyValueTemplate.class);
    ConstructorArgumentValues constructorArgumentValuesForKeyValueTemplate = new ConstructorArgumentValues();
    constructorArgumentValuesForKeyValueTemplate
            .addIndexedArgumentValue(0, new RuntimeBeanReference(HAZELCAST_ADAPTER_BEAN_NAME));
    constructorArgumentValuesForKeyValueTemplate
            .addIndexedArgumentValue(1, new RuntimeBeanReference(MAPPING_CONTEXT_BEAN_NAME));

    keyValueTemplateDefinition.setConstructorArgumentValues(constructorArgumentValuesForKeyValueTemplate);

    return keyValueTemplateDefinition;
}
 

@Override
public void registerBeansForRoot(BeanDefinitionRegistry registry, RepositoryConfigurationSource configurationSource) {
    // register HazelcastKeyValueAdapter
    String hazelcastInstanceRef = configurationSource.getAttribute("hazelcastInstanceRef").get();

    RootBeanDefinition hazelcastKeyValueAdapterDefinition = new RootBeanDefinition(HazelcastKeyValueAdapter.class);
    ConstructorArgumentValues constructorArgumentValuesForHazelcastKeyValueAdapter = new ConstructorArgumentValues();
    constructorArgumentValuesForHazelcastKeyValueAdapter
            .addIndexedArgumentValue(0, new RuntimeBeanReference(hazelcastInstanceRef));
    hazelcastKeyValueAdapterDefinition.setConstructorArgumentValues(constructorArgumentValuesForHazelcastKeyValueAdapter);
    registerIfNotAlreadyRegistered(() -> hazelcastKeyValueAdapterDefinition, registry, HAZELCAST_ADAPTER_BEAN_NAME,
            configurationSource);

    super.registerBeansForRoot(registry, configurationSource);
}
 

@Override
public void postProcess(BeanDefinitionBuilder builder,
		RepositoryConfigurationSource source) {

	Optional<String> transactionManagerRef = source
			.getAttribute("transactionManagerRef");
	builder.addPropertyValue("transactionManager",
			transactionManagerRef.orElse(DEFAULT_TRANSACTION_MANAGER_BEAN_NAME));
	Optional<String> sqlSessionTemplateRef = source
			.getAttribute("sqlSessionTemplateRef");
	builder.addPropertyReference("sqlSessionTemplate",
			sqlSessionTemplateRef.orElse(DEFAULT_SQL_SESSION_TEMPLATE_BEAN_NAME));
	builder.addPropertyReference("mappingContext", MYBATIS_MAPPING_CONTEXT_BEAN_NAME);

}
 

@Override
public void postProcess(BeanDefinitionBuilder builder, RepositoryConfigurationSource source) {

	builder.addPropertyValue("transactionManager",
		source.getAttribute("transactionManagerRef").orElse(DEFAULT_TRANSACTION_MANAGER_BEAN_NAME));
	builder.addPropertyReference("neo4jOperations",
		source.getAttribute("neo4jTemplateRef").orElse(DEFAULT_NEO4J_TEMPLATE_BEAN_NAME));
	builder.addPropertyReference("neo4jMappingContext",
		source.getAttribute("neo4jMappingContextRef").orElse(DEFAULT_MAPPING_CONTEXT_BEAN_NAME));
}
 

private static void assertHashRepo(Class<?> repositoryInterface,
                                   Collection<RepositoryConfiguration<RepositoryConfigurationSource>> configs) {
    for (final RepositoryConfiguration<?> config : configs) {
        if (config.getRepositoryInterface().equals(repositoryInterface.getName())) {
            return;
        }
    }

    fail("expected to find config for repository interface "
            + repositoryInterface.getName() + ", but got: " + configs.toString());
}
 

@Override
public void registerBeansForRoot(BeanDefinitionRegistry registry, RepositoryConfigurationSource config) {
    super.registerBeansForRoot(registry, config);

    if (!registry.containsBeanDefinition(Constants.COSMOS_MAPPING_CONTEXT)) {
        final RootBeanDefinition definition = new RootBeanDefinition(CosmosMappingContext.class);
        definition.setRole(AbstractBeanDefinition.ROLE_INFRASTRUCTURE);
        definition.setSource(config.getSource());

        registry.registerBeanDefinition(Constants.COSMOS_MAPPING_CONTEXT, definition);
    }
}
 

@Override
public void registerBeansForRoot(BeanDefinitionRegistry registry, RepositoryConfigurationSource config) {
    super.registerBeansForRoot(registry, config);

    if (!registry.containsBeanDefinition(Constants.COSMOS_MAPPING_CONTEXT)) {
        final RootBeanDefinition definition = new RootBeanDefinition(CosmosMappingContext.class);
        definition.setRole(AbstractBeanDefinition.ROLE_INFRASTRUCTURE);
        definition.setSource(config.getSource());

        registry.registerBeanDefinition(Constants.COSMOS_MAPPING_CONTEXT, definition);
    }
}
 

private void registerSolrMappingContextIfNotPresent(BeanDefinitionRegistry registry,
		RepositoryConfigurationSource configurationSource) {

	RootBeanDefinition definition = new RootBeanDefinition(SimpleSolrMappingContext.class);
	definition.setRole(AbstractBeanDefinition.ROLE_INFRASTRUCTURE);
	definition.setSource(configurationSource.getSource());

	registerIfNotAlreadyRegistered(definition, registry, BeanDefinition.SOLR_MAPPTING_CONTEXT.getBeanName(),
			definition);
}
 

@Override
public void registerBeansForRoot(BeanDefinitionRegistry registry,
		RepositoryConfigurationSource configurationSource) {

	super.registerBeansForRoot(registry, configurationSource);

	registerSolrMappingContextIfNotPresent(registry, configurationSource);

	registerIfNotAlreadyRegistered(
			BeanDefinitionBuilder.genericBeanDefinition(SolrExceptionTranslator.class).getBeanDefinition(),
			registry, "solrExceptionTranslator", configurationSource);
}
 

@Override
public void postProcess(BeanDefinitionBuilder builder, RepositoryConfigurationSource source) {

	builder.addPropertyValue("transactionManager",
		source.getAttribute("transactionManagerRef").orElse(DEFAULT_TRANSACTION_MANAGER_BEAN_NAME));
	builder.addPropertyReference("neo4jOperations",
		source.getAttribute("neo4jTemplateRef").orElse(DEFAULT_NEO4J_TEMPLATE_BEAN_NAME));
	builder.addPropertyReference("neo4jMappingContext",
		source.getAttribute("neo4jMappingContextRef").orElse(DEFAULT_MAPPING_CONTEXT_BEAN_NAME));
}
 

@Override
public void postProcess(BeanDefinitionBuilder builder, RepositoryConfigurationSource source) {
    super.postProcess(builder, source);
}
 

@Override
public void postProcess(BeanDefinitionBuilder builder, RepositoryConfigurationSource source) {
    super.postProcess(builder, source);
}
 

@SuppressWarnings({ "unchecked", "rawtypes" })
private static Class<? extends Map> getMapTypeToUse(RepositoryConfigurationSource source) {

	return (Class<? extends Map>) ((AnnotationMetadata) source.getSource()).getAnnotationAttributes(
			EnableMapRepositories.class.getName()).get("mapType");
}
 

@Override
public void registerBeansForRoot(BeanDefinitionRegistry registry,
		RepositoryConfigurationSource configurationSource) {

	Optional<String> vaultTemplateRef = configurationSource.getAttribute("vaultTemplateRef");

	RootBeanDefinition mappingContextDefinition = createVaultMappingContext(configurationSource);
	mappingContextDefinition.setSource(configurationSource.getSource());

	registerIfNotAlreadyRegistered(() -> mappingContextDefinition, registry, VAULT_MAPPING_CONTEXT_BEAN_NAME,
			configurationSource);

	// register Adapter
	RootBeanDefinition vaultKeyValueAdapterDefinition = new RootBeanDefinition(VaultKeyValueAdapter.class);

	ConstructorArgumentValues constructorArgumentValuesForVaultKeyValueAdapter = new ConstructorArgumentValues();

	constructorArgumentValuesForVaultKeyValueAdapter.addIndexedArgumentValue(0,
			new RuntimeBeanReference(vaultTemplateRef.orElse("vaultTemplate")));

	vaultKeyValueAdapterDefinition.setConstructorArgumentValues(constructorArgumentValuesForVaultKeyValueAdapter);

	registerIfNotAlreadyRegistered(() -> vaultKeyValueAdapterDefinition, registry, VAULT_ADAPTER_BEAN_NAME,
			configurationSource);

	Optional<String> keyValueTemplateName = configurationSource.getAttribute(KEY_VALUE_TEMPLATE_BEAN_REF_ATTRIBUTE);

	// No custom template reference configured and no matching bean definition found
	if (keyValueTemplateName.isPresent() && getDefaultKeyValueTemplateRef().equals(keyValueTemplateName.get())
			&& !registry.containsBeanDefinition(keyValueTemplateName.get())) {

		registerIfNotAlreadyRegistered(() -> getDefaultKeyValueTemplateBeanDefinition(configurationSource),
				registry, keyValueTemplateName.get(), configurationSource.getSource());
	}

	super.registerBeansForRoot(registry, configurationSource);
}
 

@Override
public void registerBeansForRoot(BeanDefinitionRegistry registry, RepositoryConfigurationSource configurationSource) {

	super.registerBeansForRoot(registry, configurationSource);

	registerIfNotAlreadyRegistered(() -> {

		RootBeanDefinition definitionefinition = new RootBeanDefinition(KeyValueMappingContext.class);
		definitionefinition.setSource(configurationSource.getSource());

		return definitionefinition;

	}, registry, getMappingContextBeanRef(), configurationSource);

	Optional<String> keyValueTemplateName = configurationSource.getAttribute(KEY_VALUE_TEMPLATE_BEAN_REF_ATTRIBUTE);

	// No custom template reference configured and no matching bean definition found
	if (keyValueTemplateName.isPresent() && getDefaultKeyValueTemplateRef().equals(keyValueTemplateName.get())
			&& !registry.containsBeanDefinition(keyValueTemplateName.get())) {

		AbstractBeanDefinition beanDefinition = getDefaultKeyValueTemplateBeanDefinition(configurationSource);

		if (beanDefinition != null) {
			registerIfNotAlreadyRegistered(() -> beanDefinition, registry, keyValueTemplateName.get(),
					configurationSource.getSource());
		}
	}
}
 

@Override
public void registerBeansForRoot(BeanDefinitionRegistry registry,
		RepositoryConfigurationSource config) {

	super.registerBeansForRoot(registry, config);

	Object source = config.getSource();

	registerLazyIfNotAlreadyRegistered(
			() -> new RootBeanDefinition(MybatisMappingContextFactoryBean.class),
			registry, MYBATIS_MAPPING_CONTEXT_BEAN_NAME, source);

}
 

private RootBeanDefinition createVaultMappingContext(RepositoryConfigurationSource configurationSource) {

		ConstructorArgumentValues mappingContextArgs = new ConstructorArgumentValues();

		RootBeanDefinition mappingContextBeanDef = new RootBeanDefinition(VaultMappingContext.class);
		mappingContextBeanDef.setConstructorArgumentValues(mappingContextArgs);
		mappingContextBeanDef.setSource(configurationSource.getSource());

		return mappingContextBeanDef;
	}
 

/**
 * Get the default {@link RootBeanDefinition} for {@link org.springframework.data.keyvalue.core.KeyValueTemplate}.
 *
 * @return {@literal null} to explicitly not register a template.
 * @see #getDefaultKeyValueTemplateRef()
 */
@Nullable
protected AbstractBeanDefinition getDefaultKeyValueTemplateBeanDefinition(
		RepositoryConfigurationSource configurationSource) {
	return null;
}