329 lines
9.8 KiB
Java
329 lines
9.8 KiB
Java
package org.gcube.portlets.user.td.gwtservice.server.storage;
|
|
|
|
import java.io.BufferedInputStream;
|
|
import java.io.BufferedOutputStream;
|
|
import java.io.File;
|
|
import java.io.FileOutputStream;
|
|
import java.io.InputStream;
|
|
import java.net.URL;
|
|
import java.net.URLConnection;
|
|
|
|
import org.gcube.common.homelibrary.home.HomeLibrary;
|
|
import org.gcube.common.homelibrary.home.workspace.Workspace;
|
|
import org.gcube.common.homelibrary.home.workspace.WorkspaceFolder;
|
|
import org.gcube.common.homelibrary.home.workspace.WorkspaceItem;
|
|
import org.gcube.common.homelibrary.home.workspace.WorkspaceSharedFolder;
|
|
import org.gcube.common.homelibrary.util.WorkspaceUtil;
|
|
import org.gcube.common.storagehub.client.StreamDescriptor;
|
|
import org.gcube.common.storagehub.client.dsl.FileContainer;
|
|
import org.gcube.common.storagehub.client.dsl.FolderContainer;
|
|
import org.gcube.common.storagehub.client.dsl.StorageHubClient;
|
|
import org.gcube.common.storagehub.model.items.AbstractFileItem;
|
|
import org.gcube.contentmanagement.blobstorage.service.IClient;
|
|
import org.gcube.contentmanager.storageclient.wrapper.AccessType;
|
|
import org.gcube.contentmanager.storageclient.wrapper.MemoryType;
|
|
import org.gcube.contentmanager.storageclient.wrapper.StorageClient;
|
|
import org.gcube.portlets.user.td.gwtservice.shared.exception.TDGWTServiceException;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
/**
|
|
*
|
|
* @author Giancarlo Panichi
|
|
*
|
|
*
|
|
*/
|
|
|
|
public class FilesStorage {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(FilesStorage.class);
|
|
|
|
/**
|
|
*
|
|
* @param user
|
|
* User
|
|
* @param file
|
|
* File
|
|
* @return File id
|
|
*/
|
|
public String storageCSVTempFile(String user, File file) {
|
|
try {
|
|
IClient client = new StorageClient(user, AccessType.PUBLIC, MemoryType.VOLATILE).getClient();
|
|
String remotePath = "/CSVimport/" + file.getName();
|
|
logger.debug("remotePath: " + remotePath);
|
|
String id = client.put(true).LFile(file.getAbsolutePath()).RFile(remotePath);
|
|
return id;
|
|
|
|
} catch (Throwable e) {
|
|
logger.error("Error no csv file loaded on storage" + e.getLocalizedMessage());
|
|
throw e;
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param user
|
|
* User
|
|
* @param file
|
|
* File
|
|
* @return File id
|
|
*/
|
|
public String storageCodelistMappingTempFile(String user, File file) {
|
|
try {
|
|
IClient client = new StorageClient(user, AccessType.PUBLIC, MemoryType.VOLATILE).getClient();
|
|
String remotePath = "/CodelistMappingImport/" + file.getName();
|
|
logger.debug("remotePath: " + remotePath);
|
|
String id = client.put(true).LFile(file.getAbsolutePath()).RFile(remotePath);
|
|
return id;
|
|
} catch (Throwable e) {
|
|
logger.error("Error no codelist mapping file loaded on storage" + e.getLocalizedMessage());
|
|
throw e;
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param user
|
|
* User
|
|
* @param url
|
|
* Url
|
|
* @return File id
|
|
* @throws TDGWTServiceException
|
|
* Exception
|
|
*/
|
|
public String storageCodelistMappingTempFile(String user, String url) throws TDGWTServiceException {
|
|
try {
|
|
String id = null;
|
|
|
|
URL address = new URL(url);
|
|
try (InputStream is = address.openStream()) {
|
|
|
|
IClient client = new StorageClient(user, AccessType.PUBLIC, MemoryType.VOLATILE).getClient();
|
|
String remotePath = "/CodelistMappingImport/" + address.getFile();
|
|
logger.debug("remotePath: " + remotePath);
|
|
id = client.put(true).LFile(is).RFile(remotePath);
|
|
}
|
|
|
|
return id;
|
|
} catch (Throwable e) {
|
|
logger.error("Error no codelist mapping file loaded on storage" + e.getLocalizedMessage());
|
|
throw new TDGWTServiceException(
|
|
"Error no codelist mapping file loaded on storage" + e.getLocalizedMessage());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* user
|
|
*
|
|
* @param user
|
|
* User
|
|
* @param wi
|
|
* Workspace item
|
|
* @param file
|
|
* File
|
|
* @throws TDGWTServiceException
|
|
* Exception
|
|
*/
|
|
public void retrieveFile(String user, WorkspaceItem wi, File file) throws TDGWTServiceException {
|
|
try {
|
|
logger.debug("retrieveFile: [user=" + user + ", item=" + wi + "]");
|
|
StorageHubClient shc = new StorageHubClient();
|
|
FileContainer fileContainer = shc.open(wi.getId()).asFile();
|
|
StreamDescriptor streamDescriptor = fileContainer.download();
|
|
try (BufferedInputStream bis = new BufferedInputStream(streamDescriptor.getStream())) {
|
|
try (BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(file))) {
|
|
|
|
byte[] buffer = new byte[1024];
|
|
int readCount;
|
|
while ((readCount = bis.read(buffer)) > 0) {
|
|
bos.write(buffer, 0, readCount);
|
|
}
|
|
} catch (Throwable e) {
|
|
logger.error("Error trasferring file from storage: " + e.getLocalizedMessage(), e);
|
|
throw new TDGWTServiceException("Error trasferring file from storage: " + e.getLocalizedMessage(),
|
|
e);
|
|
}
|
|
}
|
|
|
|
} catch (Throwable e) {
|
|
logger.error("Error retrieving file from StorageHub", e);
|
|
throw new TDGWTServiceException("Error retrieving file from StorageHub: " + e.getLocalizedMessage(), e);
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param user
|
|
* User
|
|
*
|
|
* @param wi
|
|
* Workspace item
|
|
* @return Input stream
|
|
* @throws TDGWTServiceException
|
|
* Exception
|
|
*/
|
|
public InputStream retrieveInputStream(String user, WorkspaceItem wi) throws TDGWTServiceException {
|
|
InputStream is = null;
|
|
try {
|
|
logger.debug("retrieveFile: [user=" + user + ", item=" + wi + "]");
|
|
StorageHubClient shc = new StorageHubClient();
|
|
FileContainer fileContainer = shc.open(wi.getId()).asFile();
|
|
StreamDescriptor streamDescriptor = fileContainer.download();
|
|
is = streamDescriptor.getStream();
|
|
|
|
} catch (Throwable e) {
|
|
logger.error("Error retrieving file from StorageHub", e);
|
|
throw new TDGWTServiceException("Error retrieving file from StorageHub: " + e.getLocalizedMessage(), e);
|
|
}
|
|
|
|
return is;
|
|
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param uri
|
|
* Usi
|
|
* @return Input stream
|
|
* @throws TDGWTServiceException
|
|
* Exception
|
|
*/
|
|
public InputStream retrieveInputStream(String uri) throws TDGWTServiceException {
|
|
InputStream is = null;
|
|
try {
|
|
|
|
// SMPUrl smsHome = new SMPUrl(uri);
|
|
URL url = new URL(uri);
|
|
|
|
logger.debug("smsHome: [host:" + url.getHost() + " path:" + url.getPath() + " ref:" + url.getRef()
|
|
+ " userinfo:" + url.getUserInfo() + " ]");
|
|
URLConnection uc = null;
|
|
uc = (URLConnection) url.openConnection();
|
|
is = uc.getInputStream();
|
|
|
|
} catch (Throwable e) {
|
|
logger.error("Error retrieving file from storage", e);
|
|
throw new TDGWTServiceException("Error retrieving file from storage: " + e.getLocalizedMessage(), e);
|
|
}
|
|
|
|
return is;
|
|
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param uri
|
|
* Uri
|
|
* @param user
|
|
* User
|
|
* @param itemName
|
|
* Item name
|
|
* @param itemDescription
|
|
* Item description
|
|
* @param itemMimetype
|
|
* Item mime type
|
|
* @param folderId
|
|
* Destination folder
|
|
* @throws TDGWTServiceException
|
|
* Exception
|
|
*/
|
|
public void createItemOnWorkspace(String uri, String user, String itemName, String itemDescription,
|
|
String itemMimetype, String folderId) throws TDGWTServiceException {
|
|
|
|
try {
|
|
|
|
URL url = new URL(uri);
|
|
|
|
logger.debug("smsHome: [host:" + url.getHost() + " path:" + url.getPath() + " ref:" + url.getRef()
|
|
+ " userinfo:" + url.getUserInfo() + " ]");
|
|
|
|
URLConnection uc = null;
|
|
uc = (URLConnection) url.openConnection();
|
|
try (InputStream is = uc.getInputStream()) {
|
|
|
|
StorageHubClient shc = new StorageHubClient();
|
|
FolderContainer folderContainer = shc.open(folderId).asFolder();
|
|
FileContainer fileContainer = folderContainer.uploadFile(is, itemName, itemDescription);
|
|
AbstractFileItem afi = fileContainer.get();
|
|
logger.debug("Created file on workspace: " + afi.getId());
|
|
}
|
|
|
|
} catch (Throwable e) {
|
|
logger.error("Error creating item on workspace", e);
|
|
throw new TDGWTServiceException("Error creating item on workspace: " + e.getLocalizedMessage(), e);
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param storageId
|
|
* Storage id
|
|
* @param user
|
|
* User
|
|
* @param itemName
|
|
* Item name
|
|
* @param itemDescription
|
|
* Item description
|
|
* @param itemMimetype
|
|
* Item mime type
|
|
* @param itemFolder
|
|
* Destination Folder
|
|
* @throws TDGWTServiceException
|
|
* Exception
|
|
*/
|
|
public void createItemOnWorkspaceByStorageId(String storageId, String user, String itemName,
|
|
String itemDescription, String itemMimetype, String itemFolder) throws TDGWTServiceException {
|
|
|
|
try {
|
|
|
|
Workspace ws = HomeLibrary.getUserWorkspace(user);
|
|
|
|
WorkspaceFolder folder = (WorkspaceFolder) ws.getItem(itemFolder);
|
|
String uniqueName = WorkspaceUtil.getUniqueName(itemName, folder);
|
|
|
|
logger.debug("ws.createExternalFile [folder: " + folder + ", uniqueName: " + uniqueName + ", description: "
|
|
+ itemDescription + ", mimetype: " + itemMimetype + ", StorageId: " + storageId + "]");
|
|
WorkspaceUtil.createExternalFile(folder, uniqueName, itemDescription, itemMimetype, storageId);
|
|
|
|
} catch (Throwable e) {
|
|
logger.error("Error creating item on workspace: " + e.getLocalizedMessage(), e);
|
|
throw new TDGWTServiceException("Error creating item on workspace: " + e.getLocalizedMessage(), e);
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param user
|
|
* User
|
|
* @param scope
|
|
* Scope
|
|
* @return Folder id
|
|
* @throws TDGWTServiceException
|
|
* Exception
|
|
*/
|
|
public String getVREFolderIdByScope(String user, String scope) throws TDGWTServiceException {
|
|
|
|
try {
|
|
|
|
Workspace ws = HomeLibrary.getUserWorkspace(user);
|
|
WorkspaceSharedFolder folder = ws.getVREFolderByScope(scope);
|
|
if (folder != null) {
|
|
return folder.getId();
|
|
} else {
|
|
return null;
|
|
}
|
|
|
|
} catch (Throwable e) {
|
|
logger.error("Error retrieving VRE folder by scope! " + e.getLocalizedMessage(), e);
|
|
throw new TDGWTServiceException("Error retrieving VRE folder by scope: " + e.getLocalizedMessage(), e);
|
|
}
|
|
|
|
}
|
|
|
|
}
|