Move analysis completed processing to an image gallery task

This commit is contained in:
apriestman 2021-04-27 14:27:08 -04:00
parent 5796ef4c6c
commit 457a9ce00c
2 changed files with 65 additions and 13 deletions

View File

@ -0,0 +1,64 @@
/*
* Autopsy Forensic Browser
*
* Copyright 2021 Basis Technology Corp.
* Contact: carrier <at> sleuthkit <dot> org
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.sleuthkit.autopsy.imagegallery;
import java.sql.SQLException;
import java.util.logging.Level;
import org.sleuthkit.autopsy.coreutils.Logger;
import org.sleuthkit.autopsy.imagegallery.datamodel.DrawableDB;
import org.sleuthkit.datamodel.TskCoreException;
/**
* Task to run when data source analysis is complete.
*/
class HandleDataSourceAnalysisCompleteTask extends DrawableDbTask {
private final ImageGalleryController controller;
private final long dataSourceObjId;
private static final Logger logger = Logger.getLogger(HandleDataSourceAnalysisCompleteTask.class.getName());
HandleDataSourceAnalysisCompleteTask(long dataSourceObjId, ImageGalleryController controller) {
this.controller = controller;
this.dataSourceObjId = dataSourceObjId;
}
@Override
public void run() {
controller.getGroupManager().resetCurrentPathGroup();
try {
DrawableDB drawableDB = controller.getDrawablesDatabase();
if (drawableDB.getDataSourceDbBuildStatus(dataSourceObjId) == DrawableDB.DrawableDbBuildStatusEnum.IN_PROGRESS) {
// If at least one file in CaseDB has mime type, then set to COMPLETE
// Otherwise, back to UNKNOWN since we assume file type module was not run
DrawableDB.DrawableDbBuildStatusEnum datasourceDrawableDBStatus
= controller.hasFilesWithMimeType(dataSourceObjId)
? DrawableDB.DrawableDbBuildStatusEnum.COMPLETE
: DrawableDB.DrawableDbBuildStatusEnum.UNKNOWN;
drawableDB.insertOrUpdateDataSource(dataSourceObjId, datasourceDrawableDBStatus);
}
drawableDB.freeFileMetaDataCache();
} catch (TskCoreException | SQLException ex) {
logger.log(Level.WARNING, "Error handling data source analysis completed event", ex);
}
}
}

View File

@ -966,19 +966,7 @@ public final class ImageGalleryController {
* of the local drawables database. * of the local drawables database.
*/ */
if (isListeningEnabled()) { if (isListeningEnabled()) {
groupManager.resetCurrentPathGroup(); queueDBTask(new HandleDataSourceAnalysisCompleteTask(dataSourceObjId, this));
if (drawableDB.getDataSourceDbBuildStatus(dataSourceObjId) == DrawableDB.DrawableDbBuildStatusEnum.IN_PROGRESS) {
// If at least one file in CaseDB has mime type, then set to COMPLETE
// Otherwise, back to UNKNOWN since we assume file type module was not run
DrawableDB.DrawableDbBuildStatusEnum datasourceDrawableDBStatus
= hasFilesWithMimeType(dataSourceObjId)
? DrawableDB.DrawableDbBuildStatusEnum.COMPLETE
: DrawableDB.DrawableDbBuildStatusEnum.UNKNOWN;
drawableDB.insertOrUpdateDataSource(dataSource.getId(), datasourceDrawableDBStatus);
}
drawableDB.freeFileMetaDataCache();
} }
} else if (((AutopsyEvent) event).getSourceType() == AutopsyEvent.SourceType.REMOTE) { } else if (((AutopsyEvent) event).getSourceType() == AutopsyEvent.SourceType.REMOTE) {
/* /*