Jest는 팩토리 패턴을 사용하는 클래스이다. 그래서 Spring에서 Jest 사용시 FactoryBean으로 생성하여 JestClient를 생성할 수 있다. FactoryBean으로 코딩을 했는데, 최근 유행이 @Configuration을 쓴다고 해서 써봤다.



FactoryBean 을 이용하여 JestClientFactoryBean을 생성한 예이다.



public class JestClientFactoryBean implements FactoryBean<JestClient>, InitializingBean, DisposableBean {

private String endPoint;

private JestClient jestClient;

private int maxTotalConnection = 10;

private int connectionTimeout = 1000;

private int maxTotalConnectionPerRoute = 1000;

private int readTimeout = 3000;


public void setMaxTotalConnection(int maxTotalConnection) {

this.maxTotalConnection = maxTotalConnection;

}


public void setConnectionTimeout(int connectionTimeout) {

this.connectionTimeout = connectionTimeout;

}


public void setSocketTimeout(int socketTimeout) {

this.readTimeout = socketTimeout;

}


public void setEndPoint(String endPoint) {

this.endPoint = endPoint;

}


public void setMaxTotalConnectionPerRoute(int maxTotalConnectionPerRoute) {

this.maxTotalConnectionPerRoute = maxTotalConnectionPerRoute;

}


@Override

public JestClient getObject() throws Exception {

return jestClient;

}


@Override

public Class<?> getObjectType() {

return JestClient.class;

}


@Override

public boolean isSingleton() {

return true;

}


@Override

public void afterPropertiesSet() throws Exception {

Assert.notNull(endPoint, "end point should be configured");

HttpClientConfig clientConfig = new HttpClientConfig.Builder(endPoint)

.multiThreaded(true)

.maxTotalConnection(maxTotalConnection)

.connTimeout(connectionTimeout)

.readTimeout(readTimeout)

.defaultMaxTotalConnectionPerRoute(maxTotalConnectionPerRoute)

.build();


JestClientFactory factory = new JestClientFactory();

factory.setHttpClientConfig(clientConfig);

jestClient = factory.getObject();

}


@Override

public void destroy() throws Exception {

if (jestClient != null) {

jestClient.shutdownClient();

}

}

}


 <bean id="searchJestClient" class="com.google.elasticsearch.JestClientFactoryBean">

    <property name="endPoint" value="http://es-search.google.com:9200"/>

    <property name="maxTotalConnection" value="1000" />

   <property name="maxTotalConnectionPerRoute" value="1000"/>

   <property name="connectionTimeout" value="3000" />

   <property name="socketTimeout" value="3000" />

</bean>




JestClientConfiguration 클래스이다. JestClient Bean을 생성한다. 참고로 DisposableBean의 destroy와 같은 역할은 @Bean(destroyMethod = "shutdownClient") 이 할 수 있다.  코드가 많이 간결해진 느낌이다.

@Configuration은 그 자체로 빈으로 역할을 하지 못해서, @Bean 메소드만 빈 생성을 위해서 사용할 수 있다.



import io.searchbox.client.JestClient;

import io.searchbox.client.JestClientFactory;
import io.searchbox.client.config.HttpClientConfig;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class JestClientConfiguration {

// @Value("${elasticsearch.search.endpoint}")
private String endPoint = "http://es-search.google.com:9200/";

// @Value("${elasticsearch.search.max_connection}")
private int maxTotalConnection = 10;

// @Value("${elasticsearch.search.conn_timeout
private int connTimeout = 1000;

// @Value("${elasticsearch.search.read_timeout
private int readTimeout = 3000;

@Bean(destroyMethod = "shutdownClient")

public JestClient jestClient(){ Assert.notNull(endPoint, "end point should be configured");
HttpClientConfig clientConfig = new HttpClientConfig.Builder(endPoint)
.multiThreaded(true)
.maxTotalConnection(maxTotalConnection)
.connTimeout(connTimeout)
.readTimeout(readTimeout)
.build();

JestClientFactory factory = new JestClientFactory();
factory.setHttpClientConfig(clientConfig);
JestClient client = factory.getObject();
return client;
}

// 다른 property를 이용하여 또 다른 Bean을 생성할 수 있다.

@Bean(destroyMethod = "shutdownClient")
 public JestClient anotherJestClient(){

Assert.notNull(endPoint, "end point should be configured");

HttpClientConfig clientConfig = new HttpClientConfig.Builder(endPoint)
.multiThreaded(true)
.maxTotalConnection(maxTotalConnection)
.connTimeout(connTimeout)
.readTimeout(readTimeout)
.build();

JestClientFactory factory = new JestClientFactory();
factory.setHttpClientConfig(clientConfig);
JestClient client = factory.getObject();
return client;
}
}


SearchService 이다. Jest를 통해 간단한 질의를 하고  POJO로 바인딩한다. JestClientConfiguration이 두 개의 Bean이 있어서 @Inject 대신 @Autowired @Qualifier를 동시에 사용한다. 

@Service
public class SearchService {
// JestClientConfiguration이 하나의 Bean만 가진다면, JestClient Bean을 바로 생성할 수 있다. // @Inject
// private JestClient jestClient;
@Autowired
@Qualifier("jestClient")
private JestClient jestClient; // SearchObject는 POJO이며 Jest의 Json을 객체로 Marshalling해준다.public List<SearchObject> search(String queryString) {

if (StringUtils.isEmpty(queryString)) {
return Collections.emptyList();
}

String query = "{\n"
+ " \"query\": {\n"
+ " \"filtered\" : {\n"
+ " \"query\" : {\n"
+ " \"query_string\" : {\n"
+ " \"query\" : \"" + queryString + "\"\n"
+ " }\n"
+ " }\n"
+ " }\n"
+ " },\n"
+ " \"size\" : 10"
+ "}";

Search.Builder searchBuilder = new Search.Builder(query).addIndex(INDEX_NAME).addType(TYPE_NAME);
Search search = searchBuilder.build();
List<SearchObject> searchObjects = Lists.newArrayList();
try {
SearchResult result = execute(search);

if (result == null) {
return Collections.emptyList();
}

List<SearchResult.Hit<SearchObject, Void>> hits = result.getHits(SearchObject.class);

for (SearchResult.Hit< SearchObject, Void> hit: hits) {
SearchObject objectSource = hit.source;
searchObjects.add(objectSource);
}

} catch (Exception e) {
logger.error(e.getMessage(), e);
}

return searchObjects;
}

private SearchResult execute(Search action) {
try {
SearchResult result = jestClient.execute(action);
if (!result.isSucceeded()) {
logger.warn("Failed to search elasticSearch action: " + result.getErrorMessage()
+ " " + result.getJsonString());
}
return result;
} catch (Exception e) {
logger.error(e.getMessage(), e);
}
return null;
}

@Autowired, @Qualifier 사용하는 대신 아래 코드처럼 @Autowired 만 사용할 수 있다. 


또한, JestClient를 Java8의 Optional 객체로 감쌀 수 있다. maven 빌드는 문제 없으나 Intellij IDEA14.1.x에서는 에러를 표시하지만 잘 돌아간다.

(Could not autowired..라는 에러가 보이지만, 이는 IDEA에서 Java8 연동을 완벽하지 않은 것으로 보인다.)


Optional을 사용하면 객체가 바로 null이 아닌 null을 포함한 객체가 되므로 NPE를 최대한 방지 할 수 있다. Optional 클래스를 이용하면 isPresent()를 통해 null 체크를 할 수 있다. 


@Autowired
private Optional<JestClient> jestClient;
...
private SearchResult execute(Search action) {
if (!jestClient.isPresent()) {
return null;
}
try {
SearchResult result = jestClient.get().execute(action);
logger.error(result.getJsonString());
if (!result.isSucceeded()) {
logger.warn("Failed to search ElasticSearch action: " + result.getErrorMessage()
+ " " + result.getJsonString());
}
return result;
} catch (Exception e) {
logger.error(e.getMessage(), e);
}
return null;
}




그리고, 빈 생성이 되는지 테스트는 이렇게 할 수 있다. 


public class JestClientConfigurationTest {

@Test
public void test() {
AnnotationConfigApplicationContext appContext = new AnnotationConfigApplicationContext(JestClientConfiguration.class);
JestClient jestClient = (JestClient) appContext.getBean("jestClient");
if (jestClient != null) System.out.println("OK");
}
}




Posted by '김용환'

댓글을 달아 주세요