数据联邦&软件开发

JestClient · 第三方库 · 访问Elasticsea

2021-08-10  本文已影响0人  sinwaj

一、JestClient管理

package org.xwiki.activeinstalls.internal;

import javax.inject.Inject;

import javax.inject.Singleton;

import org.apache.commons.lang3.StringUtils;

import org.slf4j.Logger;

import org.xwiki.activeinstalls.ActiveInstallsConfiguration;

import org.xwiki.component.annotation.Component;

import org.xwiki.component.phase.Disposable;

import org.xwiki.component.phase.Initializable;

import io.searchbox.client.JestClient;

import io.searchbox.client.JestClientFactory;

import io.searchbox.client.config.HttpClientConfig;

public class DefaultJestClientManagerimplements JestClientManager, Initializable, Disposable

{

@Inject

    private Loggerlogger;

@Inject

    private ActiveInstallsConfigurationconfiguration;

/**

* The Jest Client singleton instance to use to connect to the remote instance.

*/

    private JestClientclient;

@Override

    public void initialize()

{

String pingURL =this.configuration.getPingInstanceURL();

if (!StringUtils.isEmpty(pingURL)) {

HttpClientConfig clientConfig =new HttpClientConfig.Builder(pingURL).multiThreaded(true).build();

JestClientFactory factory =new XWikiJestClientFactory(this.configuration);

factory.setHttpClientConfig(clientConfig);

this.client = factory.getObject();

}

}

@Override

    public void dispose()

{

if (this.client !=null) {

this.client.shutdownClient();

}

}

@Override

    public JestClient getClient()

{

return this.client;

}

}

二、应用场景

探测服务器是否工作正常,模拟一个ping动作

package org.xwiki.activeinstalls.internal.client;

import java.util.Collections;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.inject.Inject;

import javax.inject.Provider;

import javax.inject.Singleton;

import org.xwiki.activeinstalls.internal.JestClientManager;

import org.xwiki.component.annotation.Component;

import io.searchbox.client.JestClient;

import io.searchbox.client.JestResult;

import io.searchbox.core.Index;

import io.searchbox.indices.CreateIndex;

import io.searchbox.indices.mapping.PutMapping;

import net.sf.json.JSONObject;

public class DefaultPingSenderimplements PingSender

{

@Inject

    private JestClientManagerjestClientManager;

@Inject

    private Provider>pingDataProviderProvider;

@Override

    public void sendPing()throws Exception

{

// Only send a ping if an ES client is available. Note that an empty ping URL disables the feature.

        JestClient client =this.jestClientManager.getClient();

if (client !=null) {

// Step 1: Create index (if already exists then it'll just be ignored)

            client.execute(new CreateIndex.Builder(JestClientManager.INDEX).build());

// Step 2: Create a mapping so that we can search distribution versions containing hyphens (otherwise they

// are removed by the default tokenizer/analyzer). If mapping already exists then it'll just be ignored.

            PutMapping putMapping =

new PutMapping.Builder(JestClientManager.INDEX, JestClientManager.TYPE, constructJSONMapping()).build();

client.execute(putMapping);

// Step 3: Index the data

            Index index =new Index.Builder(constructIndexJSON())

.index(JestClientManager.INDEX)

.type(JestClientManager.TYPE)

.build();

JestResult result = client.execute(index);

if (!result.isSucceeded()) {

throw new Exception(result.getErrorMessage());

}

}

}

private String constructJSONMapping()

{

Map jsonMap =new HashMap<>();

Map timestampMap =new HashMap<>();

timestampMap.put("enabled",true);

timestampMap.put("store",true);

Map propertiesMap =new HashMap<>();

for (PingDataProvider pingDataProvider :this.pingDataProviderProvider.get()) {

propertiesMap.putAll(pingDataProvider.provideMapping());

}

jsonMap.put("_timestamp", timestampMap);

jsonMap.put("properties", propertiesMap);

return JSONObject.fromObject(Collections.singletonMap(JestClientManager.TYPE, jsonMap)).toString();

}

private String constructIndexJSON()

{

Map jsonMap =new HashMap<>();

for (PingDataProvider pingDataProvider :this.pingDataProviderProvider.get()) {

jsonMap.putAll(pingDataProvider.provideData());

}

return JSONObject.fromObject(jsonMap).toString();

}

}

上一篇下一篇

猜你喜欢

热点阅读