mstoeck3/autopsy-flatpak
1
0
Fork 0
mirror of https://github.com/overcuriousity/autopsy-flatpak.git synced 2025-07-17 10:17:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

More work on embedded server

Browse Source
This commit is contained in:
Eugene Livis 2019-12-16 16:53:02 -05:00
parent d215c3a46e
commit 58b3055cd9
1 changed files with 42 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
KeywordSearch/src/org/sleuthkit/autopsy/keywordsearch/Server.java
View File

@ -54,6 +54,7 @@ import org.apache.solr.client.solrj.impl.XMLResponseParser;
import org.apache.solr.client.solrj.request.CollectionAdminRequest;
import org.apache.solr.client.solrj.response.CollectionAdminResponse;
import org.apache.solr.client.solrj.request.CoreAdminRequest;
import org.apache.solr.client.solrj.response.CoreAdminResponse;
import java.util.Properties;
import org.apache.solr.client.solrj.impl.BaseHttpSolrClient.RemoteSolrException;
import org.apache.solr.client.solrj.response.QueryResponse;
@ -919,6 +920,25 @@ public class Server {
if (!collectionExists(collectionName)) {
throw new KeywordSearchModuleException(NbBundle.getMessage(this.getClass(), "Server.openCore.exception.noIndexDir.msg"));
}
} else {
if (theCase.getCaseType() == CaseType.SINGLE_USER_CASE) {
// for single user cases, we unload the core when we close the case. So we have to load the core again.
// core name is (collectionName + "_shard1_replica_n1")
String coreName = collectionName + "_shard1_replica_n1";
CoreAdminRequest.Create createCoreRequest = new CoreAdminRequest.Create();
createCoreRequest.setDataDir(dataDir.getAbsolutePath());
createCoreRequest.setCoreName(coreName);
createCoreRequest.setConfigSet("AutopsyConfig"); //NON-NLS
createCoreRequest.setIsLoadOnStartup(false);
createCoreRequest.setIsTransient(true);
currentSolrServer.request(createCoreRequest);
if (!coreIndexFolderExists(coreName)) {
throw new KeywordSearchModuleException(NbBundle.getMessage(this.getClass(), "Server.openCore.exception.noIndexDir.msg"));
}
}
}
// ELTODO do we need to verify that the index directory exists?
@ -930,6 +950,28 @@ public class Server {
}
}
/**
* Determines whether or not the index files folder for a Solr core
* exists.
*
* @param coreName the name of the core.
*
* @return true or false
*
* @throws SolrServerException
* @throws IOException
*/
private boolean coreIndexFolderExists(String coreName) throws SolrServerException, IOException {
CoreAdminResponse response = CoreAdminRequest.getStatus(coreName, currentSolrServer);
Object dataDirPath = response.getCoreStatus(coreName).get("dataDir"); //NON-NLS
if (null != dataDirPath) {
File indexDir = Paths.get((String) dataDirPath, "index").toFile(); //NON-NLS
return indexDir.exists();
} else {
return false;
}
}
/**
* Get the host and port for a multiuser case. If the file solrserver.txt
* exists, then use the values from that file. Otherwise use the settings