-
Notifications
You must be signed in to change notification settings - Fork 15
/
IndexableConfigurationCallback.java
95 lines (80 loc) · 3.94 KB
/
IndexableConfigurationCallback.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
/*
* JBoss, Home of Professional Open Source.
* Copyright 2012, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.capedwarf.services;
import org.hibernate.search.Environment;
import org.hibernate.search.backend.impl.jgroups.JGroupsChannelProvider;
import org.hibernate.search.cfg.EntityMapping;
import org.hibernate.search.cfg.SearchMapping;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.configuration.cache.IndexingConfigurationBuilder;
import org.infinispan.configuration.cache.StoreAsBinaryConfigurationBuilder;
import org.jboss.msc.value.InjectedValue;
import org.jgroups.JChannel;
/**
* Indexable configuration callback.
*
* @author <a href="mailto:ales.justin@jboss.org">Ales Justin</a>
* @author <a href="mailto:mluksa@redhat.com">Marko Luksa</a>
*/
public abstract class IndexableConfigurationCallback extends AbstractConfigurationCallback {
private static final int INDEXING_CACHES = 6;
protected final CacheConfig config;
protected final String appId;
protected final ClassLoader classLoader;
private final InjectedValue<JChannel> channel = new InjectedValue<>();
private final InjectedValue<MuxIdGenerator> generator = new InjectedValue<>();
protected IndexableConfigurationCallback(CacheConfig config, String appId, ClassLoader classLoader) {
this.config = config;
this.appId = appId;
this.classLoader = classLoader;
}
protected SearchMapping applyIndexing(ConfigurationBuilder builder) {
final CacheIndexing ci = config.getIndexing();
if (ci == null)
throw new IllegalArgumentException("Missing cache indexing info: " + config);
final IndexingConfigurationBuilder indexing = builder.indexing();
indexing.addProperty("hibernate.search.default.indexBase", "./indexes_" + appId);
final SearchMapping mapping = new SearchMapping();
for (Class<?> clazz : ci.getClasses(classLoader)) {
final EntityMapping entity = mapping.entity(clazz);
entity.indexed().indexName(getIndexName(clazz.getName()));
}
indexing.setProperty(Environment.MODEL_MAPPING, mapping);
indexing.setProperty(JGroupsChannelProvider.CHANNEL_INJECT, channel.getValue());
indexing.setProperty(JGroupsChannelProvider.CLASSLOADER, classLoader);
short muxId = (short) ((INDEXING_CACHES / 2) * generator.getValue().getMuxId(appId) * ci.getPrefix() + ci.getOffset());
indexing.setProperty(JGroupsChannelProvider.MUX_ID, muxId);
// do we store as binary - e.g. Modules
final StoreAsBinaryConfigurationBuilder storeAsBinaryConfigurationBuilder = builder.storeAsBinary();
storeAsBinaryConfigurationBuilder.enabled(config.storeAsBinary());
return mapping;
}
protected String getIndexName(String className) {
return config.getName() + "_" + appId + "__" + className;
}
public InjectedValue<JChannel> getChannel() {
return channel;
}
public InjectedValue<MuxIdGenerator> getGenerator() {
return generator;
}
}