Java tutorial
/** * License Agreement for OpenSearchServer * * Copyright (C) 2011-2013 Emmanuel Keller / Jaeksoft * * http://www.open-search-server.com * * This file is part of OpenSearchServer. * * OpenSearchServer is free software: you can redistribute it and/or * modify it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * OpenSearchServer 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 General Public License for more details. * * You should have received a copy of the GNU General Public License * along with OpenSearchServer. * If not, see <http://www.gnu.org/licenses/>. **/ package com.jaeksoft.searchlib.webservice.document; import java.io.IOException; import java.io.InputStream; import java.net.URISyntaxException; import java.security.NoSuchAlgorithmException; import java.util.ArrayList; import java.util.Arrays; import java.util.List; import javax.ws.rs.core.Response.Status; import javax.xml.transform.stream.StreamSource; import org.apache.commons.collections.CollectionUtils; import org.apache.commons.lang3.StringUtils; import com.jaeksoft.searchlib.Client; import com.jaeksoft.searchlib.ClientFactory; import com.jaeksoft.searchlib.SearchLibException; import com.jaeksoft.searchlib.index.IndexDocument; import com.jaeksoft.searchlib.request.AbstractSearchRequest; import com.jaeksoft.searchlib.request.SearchPatternRequest; import com.jaeksoft.searchlib.schema.SchemaField; import com.jaeksoft.searchlib.schema.SchemaFieldList; import com.jaeksoft.searchlib.user.Role; import com.jaeksoft.searchlib.webservice.CommonResult; import com.jaeksoft.searchlib.webservice.CommonServices; import com.jaeksoft.searchlib.webservice.query.document.IndexDocumentResult; public class DocumentImpl extends CommonServices implements SoapDocument, RestDocument { private final static String UPDATED_COUNT = "updatedCount"; private final static String DELETED_COUNT = "deletedCount"; @Override public CommonResult deleteByQuery(String index, String login, String key, String query) { try { Client client = getLoggedClient(index, login, key, Role.INDEX_UPDATE); ClientFactory.INSTANCE.properties.checkApi(); AbstractSearchRequest request = new SearchPatternRequest(client); request.setQueryString(query); int count = client.deleteDocuments(request); return new CommonResult(true, count + " document(s) deleted").addDetail(DELETED_COUNT, count); } catch (InterruptedException e) { throw new CommonServiceException(e); } catch (IOException e) { throw new CommonServiceException(e); } catch (SearchLibException e) { throw new CommonServiceException(e); } } @Override public CommonResult deleteByValue(String index, String login, String key, String field, List<String> values) { try { Client client = getLoggedClient(index, login, key, Role.INDEX_UPDATE); ClientFactory.INSTANCE.properties.checkApi(); SchemaFieldList schemaFieldList = client.getSchema().getFieldList(); SchemaField schemaField = field != null ? schemaFieldList.get(field) : schemaFieldList.getUniqueField(); if (schemaField == null) throw new CommonServiceException(Status.NOT_FOUND, "Field not found: " + field); int count = 0; if (!CollectionUtils.isEmpty(values)) count = client.deleteDocuments(schemaField.getName(), values); return new CommonResult(true, count + " document(s) deleted by " + schemaField.getName()) .addDetail(DELETED_COUNT, count); } catch (SearchLibException e) { throw new CommonServiceException(e); } catch (InterruptedException e) { throw new CommonServiceException(e); } catch (IOException e) { throw new CommonServiceException(e); } } @Override public CommonResult deleteByValue(String index, String login, String key, String field, String values) { String[] valueArray = StringUtils.split(values, '/'); return deleteByValue(index, login, key, field, Arrays.asList(valueArray)); } private int updateDocument(Client client, List<DocumentUpdate> documents) throws NoSuchAlgorithmException, IOException, URISyntaxException, SearchLibException, InstantiationException, IllegalAccessException, ClassNotFoundException { if (documents == null || documents.size() == 0) throw new CommonServiceException(Status.NO_CONTENT, "No documents"); List<IndexDocument> indexDocuments = new ArrayList<IndexDocument>(0); for (DocumentUpdate document : documents) { IndexDocument indexDocument = DocumentUpdate.getIndexDocument(document); if (indexDocument != null) indexDocuments.add(indexDocument); } return client.updateDocuments(indexDocuments); } @Override public CommonResult update(String index, String login, String key, List<DocumentUpdate> documents) { try { Client client = getLoggedClient(index, login, key, Role.INDEX_UPDATE); ClientFactory.INSTANCE.properties.checkApi(); int count = updateDocument(client, documents); return new CommonResult(true, count + " document(s) updated").addDetail(UPDATED_COUNT, count); } catch (SearchLibException e) { throw new CommonServiceException(e); } catch (NoSuchAlgorithmException e) { throw new CommonServiceException(e); } catch (IOException e) { throw new CommonServiceException(e); } catch (URISyntaxException e) { throw new CommonServiceException(e); } catch (InstantiationException e) { throw new CommonServiceException(e); } catch (IllegalAccessException e) { throw new CommonServiceException(e); } catch (ClassNotFoundException e) { throw new CommonServiceException(e); } catch (InterruptedException e) { throw new CommonServiceException(e); } } @Override public CommonResult updateRaw(String index, String login, String key, List<IndexDocumentResult> indexDocuments) { try { Client client = getLoggedClient(index, login, key, Role.INDEX_UPDATE); ClientFactory.INSTANCE.properties.checkApi(); if (indexDocuments == null || indexDocuments.size() == 0) throw new CommonServiceException(Status.NO_CONTENT, "No documents"); int count = client.updateIndexDocuments(indexDocuments); return new CommonResult(true, count + " document(s) updated").addDetail(UPDATED_COUNT, count); } catch (IOException e) { throw new CommonServiceException(e); } catch (InterruptedException e) { throw new CommonServiceException(e); } catch (SearchLibException e) { throw new CommonServiceException(e); } } @Override public CommonResult update(String index, String login, String key, String capturePattern, List<String> fields, Integer langPosition, String charset, Integer bufferSize, InputStream inputStream) { try { Client client = getLoggedClient(index, login, key, Role.INDEX_UPDATE); ClientFactory.INSTANCE.properties.checkApi(); CommonResult result = new CommonResult(true, null); StreamSource streamSource = new StreamSource(inputStream); int count = client.updateTextDocuments(streamSource, charset, bufferSize, capturePattern, langPosition, fields, result); return result.addDetail(UPDATED_COUNT, count); } catch (InterruptedException e) { throw new CommonServiceException(e); } catch (IOException e) { throw new CommonServiceException(e); } catch (NoSuchAlgorithmException e) { throw new CommonServiceException(e); } catch (SearchLibException e) { throw new CommonServiceException(e); } catch (URISyntaxException e) { throw new CommonServiceException(e); } catch (InstantiationException e) { throw new CommonServiceException(e); } catch (IllegalAccessException e) { throw new CommonServiceException(e); } catch (ClassNotFoundException e) { throw new CommonServiceException(e); } } }