Java River类代码示例

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void masterOperation(GetRiverStateRequest request, ClusterState state, ActionListener<GetRiverStateResponse> listener) throws ElasticsearchException {
    MetaData metaData = state.metaData();
    RiverStatesMetaData riverStates = metaData.custom(RiverStatesMetaData.TYPE);
    if (request.getRiverName() == null || request.getRiverType() == null) {
        listener.onResponse(new GetRiverStateResponse(request, riverStates.rivers()));
    } else {
        String riverName = request.getRiverName();
        String riverType = request.getRiverType();
        ImmutableList.Builder<RiverState> builder = ImmutableList.builder();
        for (Map.Entry<RiverName, River> entry : RiverHelper.rivers(injector).entrySet()) {
            RiverName name = entry.getKey();
            if (("*".equals(riverName) || name.getName().equals(riverName)) && ("*".equals(riverType) || name.getType().equals(riverType)) && entry.getValue() instanceof StatefulRiver) {
                StatefulRiver river = (StatefulRiver) entry.getValue();
                if (river.getRiverState() != null) {
                    builder.add(river.getRiverState());
                }
            }
        }
        listener.onResponse(new GetRiverStateResponse(request, builder.build()));
    }
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected NodeRiverExecuteResponse nodeOperation(NodeRiverExecuteRequest request) throws ElasticsearchException {
    NodeInfo nodeInfo = nodeService.info(false, true, false, true, false, false, true, false, true);
    String riverType = request.getRiverType();
    String riverName = request.getRiverName();
    for (Map.Entry<RiverName, River> entry : RiverHelper.rivers(injector).entrySet()) {
        RiverName name = entry.getKey();
        if ((riverName == null || name.getName().equals(riverName))
                && (riverType == null || name.getType().equals(riverType))
                && entry.getValue() instanceof RunnableRiver) {
            RunnableRiver river = (RunnableRiver) entry.getValue();
            river.run();
            return new NodeRiverExecuteResponse(nodeInfo.getNode()).setExecuted(true);
        }
    }
    return new NodeRiverExecuteResponse(nodeInfo.getNode()).setExecuted(false);
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
/**
 * Retrieve the registered rivers using reflection. This hack can be removed when RiversService gets a public API.
 *
 * @param injector injector
 * @return map of rivers or null if not possible
 */
@SuppressWarnings({"unchecked"})
public static ImmutableMap<RiverName, River> rivers(Injector injector) {
    RiversService riversService = injector.getInstance(RiversService.class);
    try {
        Field field = RiversService.class.getDeclaredField("rivers");
        if (field != null) {
            field.setAccessible(true);
            return (ImmutableMap<RiverName, River>) field.get(riversService);
        }
    } catch (Throwable e) {
        //
    }
    return ImmutableMap.of();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
    bind(River.class).to(DynamoDbRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
	bind(River.class).to(SnapshotsRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
	bind(River.class).to(BigQueryRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
    bind(River.class).to(ZookeeperRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
	bind(River.class).to(TMDBRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
    bind(River.class).to(GitHubRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
	bind(River.class).to(NvdRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure()
{
	bind(River.class).to(CoherenceRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
    bind(River.class).to(JDBCRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
    bind(River.class).to(KafkaRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
	bind(River.class).to(HBaseRiver.class).asEagerSingleton();
}
 

import org.elasticsearch.river.River; //导入依赖的package包/类
@Override
protected void configure() {
    bind(River.class).to(Neo4jDriver.class).asEagerSingleton();
}