| /******************************************************************************** |
| * Copyright (c) 2015-2019 Contributors to the Eclipse Foundation |
| * |
| * See the NOTICE file(s) distributed with this work for additional |
| * information regarding copyright ownership. |
| * |
| * This program and the accompanying materials are made available under the |
| * terms of the Eclipse Public License v. 2.0 which is available at |
| * http://www.eclipse.org/legal/epl-2.0. |
| * |
| * SPDX-License-Identifier: EPL-2.0 |
| * |
| ********************************************************************************/ |
| |
| package org.eclipse.mdm.filerelease.control; |
| |
| import java.io.BufferedInputStream; |
| import java.io.BufferedOutputStream; |
| import java.io.File; |
| import java.io.FileInputStream; |
| import java.io.FileOutputStream; |
| import java.io.IOException; |
| import java.io.ObjectInputStream; |
| import java.io.ObjectOutputStream; |
| import java.util.HashMap; |
| import java.util.Map; |
| |
| import javax.ejb.Stateless; |
| |
| import org.eclipse.mdm.filerelease.entity.FileRelease; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| /** |
| * |
| * FileReleasePersistance bean implementation. |
| * |
| * @author Sebastian Dirsch, Gigatronik Ingolstadt GmbH |
| * |
| */ |
| @Stateless |
| public class FileReleasePersistance { |
| |
| private static final Logger LOG = LoggerFactory.getLogger(FileReleasePersistance.class); |
| |
| private static final String TARGET_FILE_NAME = "mdm_filerelease_storage.sav"; |
| |
| /** |
| * Persists the given file release map. |
| * |
| * @param map The map to persist. |
| */ |
| public void save(Map<String, FileRelease> map) { |
| |
| String userHomePath = System.getProperty("user.home"); |
| File directory = new File(userHomePath); |
| |
| File targetFile = new File(directory, TARGET_FILE_NAME); |
| writeFile(targetFile, map); |
| } |
| |
| /** |
| * |
| * Loads the {@link FileRelease} into a map. |
| * |
| * @return The map that contains the {@link FileRelease}s |
| */ |
| public Map<String, FileRelease> load() { |
| |
| String userHomePath = System.getProperty("user.home"); |
| File directory = new File(userHomePath); |
| |
| File targetFile = new File(directory, TARGET_FILE_NAME); |
| |
| return loadFile(targetFile); |
| } |
| |
| private void writeFile(File targetFile, Map<String, FileRelease> map) { |
| |
| LOG.debug("Writing FileRelease storage file to '" + targetFile.getAbsolutePath() + "'"); |
| |
| try { |
| if (targetFile.exists()) { |
| deleteFile(targetFile); |
| } |
| |
| try (ObjectOutputStream oos = new ObjectOutputStream( |
| new BufferedOutputStream(new FileOutputStream(targetFile)))) { |
| oos.writeObject(map); |
| } |
| } catch (IOException e) { |
| throw new FileReleaseException(e.getMessage(), e); |
| } |
| } |
| |
| @SuppressWarnings("unchecked") |
| private Map<String, FileRelease> loadFile(File targetFile) { |
| |
| LOG.debug("Loading FileRelease storage file from '" + targetFile.getAbsolutePath() + "'"); |
| |
| try { |
| if (!targetFile.exists()) { |
| LOG.warn("Storage file does not exist at '" + targetFile.getAbsolutePath() |
| + "'. Using an empty FileRelease pool"); |
| return new HashMap<String, FileRelease>(); |
| } |
| |
| try (ObjectInputStream ois = new ObjectInputStream( |
| new BufferedInputStream(new FileInputStream(targetFile)))) { |
| return (Map<String, FileRelease>) ois.readObject(); |
| } |
| } catch (IOException e) { |
| throw new FileReleaseException(e.getMessage(), e); |
| } catch (ClassNotFoundException e) { |
| throw new FileReleaseException(e.getMessage(), e); |
| } |
| } |
| |
| private void deleteFile(File targetFile) { |
| boolean deleted = targetFile.delete(); |
| if (!deleted) { |
| throw new FileReleaseException( |
| "Unable to delete FileRelease storage file at '" + targetFile.getAbsolutePath() + "'"); |
| } |
| } |
| } |