use of com.b2international.snowowl.core.events.Notifications in project snow-owl by b2ihealthcare.
the class RepositoryPlugin method init.
@Override
public void init(SnowOwlConfiguration configuration, Environment env) throws Exception {
final IManagedContainer container = env.container();
final boolean gzip = configuration.isGzip();
final RpcConfiguration rpcConfig = configuration.getModuleConfig(RpcConfiguration.class);
LOG.debug("Preparing RPC communication (config={},gzip={})", rpcConfig, gzip);
RpcUtil.prepareContainer(container, rpcConfig, gzip);
LOG.debug("Preparing EventBus communication (gzip={})", gzip);
RepositoryConfiguration repositoryConfiguration = configuration.getModuleConfig(RepositoryConfiguration.class);
env.services().registerService(RepositoryConfiguration.class, repositoryConfiguration);
int maxThreads = repositoryConfiguration.getMaxThreads();
EventBusNet4jUtil.prepareContainer(container, gzip, maxThreads);
env.services().registerService(IEventBus.class, EventBusNet4jUtil.getBus(container, maxThreads));
LOG.debug("Preparing JSON support");
final ObjectMapper mapper = JsonSupport.getDefaultObjectMapper();
mapper.registerModule(new PrimitiveCollectionModule());
env.services().registerService(ObjectMapper.class, mapper);
// initialize Notification support
env.services().registerService(Notifications.class, new Notifications(env.service(IEventBus.class), env.plugins().getCompositeClassLoader()));
env.services().registerService(RepositoryCommitNotificationSender.class, new RepositoryCommitNotificationSender());
// initialize Index Settings
final IndexSettings indexSettings = new IndexSettings();
indexSettings.putAll(initIndexSettings(env));
env.services().registerService(IndexSettings.class, indexSettings);
}
use of com.b2international.snowowl.core.events.Notifications in project snow-owl by b2ihealthcare.
the class ClassifyOperation method run.
/**
* Allocates a reasoner instance, performs the requested operation, then releases the borrowed instance back to the pool.
* @param monitor an {@link IProgressMonitor} to monitor operation progress
* @return the value returned by {@link #processResults(IProgressMonitor, long)}
* @throws OperationCanceledException
*/
public T run(final IProgressMonitor monitor) throws OperationCanceledException {
monitor.beginTask("Classification in progress...", IProgressMonitor.UNKNOWN);
try {
final String classificationId = UUID.randomUUID().toString();
final String jobId = IDs.sha1(classificationId);
final Notifications notifications = getServiceForClass(Notifications.class);
final BlockingQueue<RemoteJobEntry> jobQueue = Queues.newArrayBlockingQueue(1);
final Observable<RemoteJobEntry> jobObservable = notifications.ofType(RemoteJobNotification.class).filter(RemoteJobNotification::isChanged).filter(notification -> notification.getJobIds().contains(jobId)).concatMap(notification -> JobRequests.prepareSearch().one().filterById(jobId).buildAsync().execute(getEventBus())).map(RemoteJobs::first).map(Optional<RemoteJobEntry>::get).filter(RemoteJobEntry::isDone);
// "One-shot" subscription; it should self-destruct after the first notification
jobObservable.subscribe(new DisposableObserver<RemoteJobEntry>() {
@Override
public void onComplete() {
dispose();
}
@Override
public void onError(final Throwable t) {
dispose();
}
@Override
public void onNext(final RemoteJobEntry job) {
try {
jobQueue.put(job);
} catch (InterruptedException e) {
throw new SnowowlRuntimeException("Interrupted while trying to add a remote job entry to the queue.", e);
} finally {
dispose();
}
}
});
ClassificationRequests.prepareCreateClassification().setClassificationId(classificationId).setReasonerId(reasonerId).setUserId(userId).addAllConcepts(additionalConcepts).setParentLockContext(parentLockContext).build(branch).get(ApplicationContext.getServiceForClass(Environment.class));
while (true) {
if (monitor.isCanceled()) {
throw new OperationCanceledException();
}
try {
final RemoteJobEntry jobEntry = jobQueue.poll(CHECK_JOB_INTERVAL_SECONDS, TimeUnit.SECONDS);
if (jobEntry == null) {
continue;
}
switch(jobEntry.getState()) {
// $FALL-THROUGH$
case SCHEDULED:
case RUNNING:
case CANCEL_REQUESTED:
break;
case FINISHED:
try {
return processResults(classificationId);
} finally {
deleteEntry(jobId);
}
case CANCELED:
deleteEntry(jobId);
throw new OperationCanceledException();
case FAILED:
deleteEntry(jobId);
throw new SnowowlRuntimeException("Failed to retrieve the results of the classification.");
default:
throw new IllegalStateException("Unexpected state '" + jobEntry.getState() + "'.");
}
} catch (final InterruptedException e) {
// Nothing to do
}
}
} finally {
monitor.done();
}
}
Aggregations