use of org.bimserver.interfaces.objects.SCheckoutResult in project BIMserver by opensourceBIM.
the class LongDownloadOrCheckoutAction method convertModelToCheckoutResult.
protected SCheckoutResult convertModelToCheckoutResult(Project project, String username, IfcModelInterface model, RenderEnginePlugin renderEnginePlugin, DownloadParameters downloadParameters) throws UserException, NoSerializerFoundException {
SCheckoutResult checkoutResult = new SCheckoutResult();
checkoutResult.setSerializerOid(downloadParameters.getSerializerOid());
if (model.isValid()) {
checkoutResult.setProjectName(project.getName());
checkoutResult.setRevisionNr(model.getModelMetaData().getRevisionId());
try {
Serializer serializer = getBimServer().getSerializerFactory().create(project, username, model, renderEnginePlugin, downloadParameters);
if (serializer == null) {
throw new UserException("Error, no serializer found " + downloadParameters.getSerializerOid());
}
if (getBimServer().getServerSettingsCache().getServerSettings().getCacheOutputFiles() && serializer.allowCaching()) {
if (getBimServer().getDiskCacheManager().contains(downloadParameters)) {
checkoutResult.setFile(new CachingDataHandler(getBimServer().getDiskCacheManager(), downloadParameters));
} else {
checkoutResult.setFile(new DataHandler(new CacheStoringEmfSerializerDataSource(serializer, model.getModelMetaData().getName(), getBimServer().getDiskCacheManager().startCaching(downloadParameters))));
}
} else {
checkoutResult.setFile(new DataHandler(new EmfSerializerDataSource(serializer, model.getModelMetaData().getName())));
}
} catch (SerializerException e) {
LOGGER.error("", e);
}
}
return checkoutResult;
}
use of org.bimserver.interfaces.objects.SCheckoutResult in project BIMserver by opensourceBIM.
the class LongDownloadOrCheckoutAction method executeAction.
protected void executeAction(BimDatabaseAction<? extends IfcModelInterface> action, DownloadParameters downloadParameters, DatabaseSession session, boolean commit) throws BimserverDatabaseException, UserException, NoSerializerFoundException, ServerException {
try {
if (action == null) {
checkoutResult = new SCheckoutResult();
checkoutResult.setFile(new CachingDataHandler(getBimServer().getDiskCacheManager(), downloadParameters));
checkoutResult.setSerializerOid(downloadParameters.getSerializerOid());
} else {
Revision revision = session.get(StorePackage.eINSTANCE.getRevision(), downloadParameters.getRoid(), OldQuery.getDefault());
if (revision == null) {
throw new UserException("Revision with roid " + downloadParameters.getRoid() + " not found");
}
// Little hack to make
revision.getProject().getGeoTag().load();
// sure this is
// lazily loaded,
// because after the
// executeAndCommitAction
// the session won't
// be usable
IfcModelInterface ifcModel = session.executeAndCommitAction(action);
// Session is closed after this
DatabaseSession newSession = getBimServer().getDatabase().createSession();
RenderEnginePlugin renderEnginePlugin = null;
try {
PluginConfiguration serializerPluginConfiguration = newSession.get(StorePackage.eINSTANCE.getPluginConfiguration(), downloadParameters.getSerializerOid(), OldQuery.getDefault());
if (serializerPluginConfiguration != null) {
if (serializerPluginConfiguration instanceof MessagingSerializerPluginConfiguration) {
try {
messagingSerializer = getBimServer().getSerializerFactory().createMessagingSerializer(getUserName(), ifcModel, downloadParameters);
} catch (SerializerException e) {
e.printStackTrace();
}
} else if (serializerPluginConfiguration instanceof SerializerPluginConfiguration) {
RenderEnginePluginConfiguration renderEngine = ((SerializerPluginConfiguration) serializerPluginConfiguration).getRenderEngine();
if (renderEngine != null) {
renderEnginePlugin = getBimServer().getPluginManager().getRenderEnginePlugin(renderEngine.getPluginDescriptor().getPluginClassName(), true);
}
checkoutResult = convertModelToCheckoutResult(revision.getProject(), getUserName(), ifcModel, renderEnginePlugin, downloadParameters);
}
}
} catch (BimserverDatabaseException e) {
LOGGER.error("", e);
} finally {
newSession.close();
}
}
} finally {
done();
}
}
use of org.bimserver.interfaces.objects.SCheckoutResult in project BIMserver by opensourceBIM.
the class ServiceImpl method triggerRevisionService.
@Override
public void triggerRevisionService(Long roid, Long soid) throws ServerException, UserException {
DatabaseSession session = getBimServer().getDatabase().createSession();
try {
Revision revision = (Revision) session.get(StorePackage.eINSTANCE.getRevision(), roid, OldQuery.getDefault());
if (revision == null) {
throw new UserException("No revision found for roid " + roid);
}
NewService newService = session.get(StorePackage.eINSTANCE.getNewService(), soid, OldQuery.getDefault());
if (revision.getServicesLinked().contains(newService)) {
// We don't want no loops
return;
}
String url = newService.getResourceUrl();
SerializerPluginConfiguration serializer = newService.getSerializer();
PackageMetaData pmd = getBimServer().getMetaDataManager().getPackageMetaData(revision.getProject().getSchema());
Query query = DefaultQueries.all(pmd);
Long topicId = download(Collections.singleton(roid), new JsonQueryObjectModelConverter(pmd).toJson(query).toString(), serializer.getOid(), false);
CloseableHttpClient httpclient = HttpClients.createDefault();
HttpPost httpPost = new HttpPost(url);
LongAction<?> longAction = getBimServer().getLongActionManager().getLongAction(topicId);
if (longAction == null) {
throw new UserException("No data found for topicId " + topicId);
}
SCheckoutResult result;
if (longAction instanceof LongStreamingDownloadAction) {
LongStreamingDownloadAction longStreamingDownloadAction = (LongStreamingDownloadAction) longAction;
if (longStreamingDownloadAction.getErrors().isEmpty()) {
try {
result = longStreamingDownloadAction.getCheckoutResult();
} catch (SerializerException e) {
throw new UserException(e);
}
} else {
LOGGER.error(longStreamingDownloadAction.getErrors().get(0));
throw new ServerException(longStreamingDownloadAction.getErrors().get(0));
}
} else {
LongDownloadOrCheckoutAction longDownloadAction = (LongDownloadOrCheckoutAction) longAction;
try {
longDownloadAction.waitForCompletion();
if (longDownloadAction.getErrors().isEmpty()) {
result = longDownloadAction.getCheckoutResult();
} else {
LOGGER.error(longDownloadAction.getErrors().get(0));
throw new ServerException(longDownloadAction.getErrors().get(0));
}
} catch (Exception e) {
LOGGER.error("", e);
throw new ServerException(e);
}
}
ByteArrayOutputStream baos = new ByteArrayOutputStream();
LOGGER.info("Starting serialization");
DataSource datasource = result.getFile().getDataSource();
if (datasource instanceof ExtendedDataSource) {
((ExtendedDataSource) datasource).writeToOutputStream(baos, null);
}
LOGGER.info("Serialization done");
if (newService.getAccessToken() != null) {
httpPost.setHeader("Authorization", "Bearer " + newService.getAccessToken());
}
httpPost.setHeader("Input-Type", newService.getInput());
httpPost.setHeader("Output-Type", newService.getOutput());
httpPost.setEntity(new ByteArrayEntity(baos.toByteArray()));
CloseableHttpResponse response = httpclient.execute(httpPost);
LOGGER.info(response.getStatusLine().toString());
if (response.getStatusLine().getStatusCode() == 401) {
throw new UserException("Remote service responded with a 401 Unauthorized");
} else if (response.getStatusLine().getStatusCode() == 200) {
Header[] headers = response.getHeaders("Content-Disposition");
String filename = "unknown";
if (headers.length > 0) {
String contentDisposition = headers[0].getValue();
if (contentDisposition.contains("filename=")) {
int indexOf = contentDisposition.indexOf("filename=") + 10;
filename = contentDisposition.substring(indexOf, contentDisposition.indexOf("\"", indexOf + 1));
} else {
filename = contentDisposition;
}
}
Header dataTitleHeader = response.getFirstHeader("Data-Title");
String dataTitle = newService.getName() + " Results";
if (dataTitleHeader != null) {
dataTitle = dataTitleHeader.getValue();
}
byte[] responseBytes = ByteStreams.toByteArray(response.getEntity().getContent());
Action action = newService.getAction();
if (action instanceof StoreExtendedData) {
SFile file = new SFile();
file.setData(responseBytes);
file.setFilename(filename);
file.setSize(responseBytes.length);
file.setMime(response.getHeaders("Content-Type")[0].getValue());
Long fileId = uploadFile(file);
SExtendedData extendedData = new SExtendedData();
extendedData.setAdded(new Date());
extendedData.setRevisionId(roid);
extendedData.setTitle(dataTitle);
extendedData.setSize(responseBytes.length);
extendedData.setFileId(fileId);
extendedData.setSchemaId(getExtendedDataSchemaByName(newService.getOutput()).getOid());
addExtendedDataToRevision(roid, extendedData);
} else if (action instanceof CheckinRevision) {
CheckinRevision checkinRevision = (CheckinRevision) action;
Project targetProject = checkinRevision.getProject();
String extension = filename.substring(filename.lastIndexOf(".") + 1);
SDeserializerPluginConfiguration deserializer = getSuggestedDeserializerForExtension(extension, targetProject.getOid());
Long checkingTopicId = initiateCheckin(targetProject.getOid(), deserializer.getOid());
checkinInitiatedInternal(checkingTopicId, targetProject.getOid(), dataTitle, deserializer.getOid(), (long) responseBytes.length, filename, new DataHandler(new ByteArrayDataSource(responseBytes, "ifc")), false, true, newService.getOid());
}
} else {
throw new UserException("Remote service responded with a " + response.getStatusLine());
}
} catch (Exception e) {
handleException(e);
} finally {
session.close();
}
}
Aggregations