Java TaskContextImpl类代码示例

import org.infinispan.persistence.TaskContextImpl; //导入依赖的package包/类
@Override
public void process(final KeyFilter<? super K> filter, final CacheLoaderTask<K, V> task, Executor executor, boolean fetchValue, boolean fetchMetadata) {
   ExecutorAllCompletionService eacs = new ExecutorAllCompletionService(executor);
   final TaskContextImpl taskContext = new TaskContextImpl();

   //A while loop since we have to hit the db again for paging.
   boolean shouldContinue = true;
   byte[] id = null;
   while (shouldContinue) {
      final List<MongoDBEntry<K, V>> entries = cache.getPagedEntries(id);
      shouldContinue = !entries.isEmpty();
      if (taskContext.isStopped()) {
         break;
      }
      if (shouldContinue) {
         eacs.submit(() -> {
            for (final MongoDBEntry<K, V> entry : entries) {
               if (taskContext.isStopped()) {
                  break;
               }
               final K marshalledKey = (K) toObject(entry.getKeyBytes());
               if (filter == null || filter.accept(marshalledKey)) {
                  final MarshalledEntry<K, V> marshalledEntry = getMarshalledEntry(entry);
                  if (marshalledEntry != null) {
                     task.processEntry(marshalledEntry, taskContext);
                  }
               }
            }
            return null;
         });
         //get last key so we can get more entries.
         id = entries.get(entries.size() - 1).getKeyBytes();
      }
   }
   eacs.waitUntilAllCompleted();
   if (eacs.isExceptionThrown()) {
      throw new PersistenceException("Execution exception!", eacs.getFirstException());
   }

}
 

import org.infinispan.persistence.TaskContextImpl; //导入依赖的package包/类
@Override
public void process(KeyFilter filter, CacheLoaderTask task, Executor executor, boolean fetchValue,
      boolean fetchMetadata) {
   int batchSize = 100;
   ExecutorCompletionService ecs = new ExecutorCompletionService(executor);
   int tasks = 0;
   final TaskContext taskContext = new TaskContextImpl();

   List<Map.Entry<Object, byte[]>> entries = new ArrayList<Map.Entry<Object, byte[]>>(batchSize);
   
   try {
      for (Map.Entry<Object, byte[]> entry : store.entrySet()) {
         entries.add(entry);
         if (entries.size() == batchSize) {
            final List<Map.Entry<Object, byte[]>> batch = entries;
            entries = new ArrayList<Map.Entry<Object, byte[]>>(batchSize);
            submitProcessTask(task, filter, ecs, taskContext, batch);
            tasks++;
         }
      }
      if (!entries.isEmpty()) {
         submitProcessTask(task, filter,ecs, taskContext, entries);
         tasks++;
      }

      PersistenceUtil.waitForAllTasksToComplete(ecs, tasks);
   } catch (Exception e) {
      throw new CacheLoaderException(e);
   }
}
 

import org.infinispan.persistence.TaskContextImpl; //导入依赖的package包/类
@Override
public void process(KeyFilter<? super K> keyFilter, final CacheLoaderTask<K, V> cacheLoaderTask, Executor executor,
      boolean loadValue, boolean loadMetadata) {
   String nextMarker = null;
   PageSet<? extends StorageMetadata> pageSet;

   do {
      ListContainerOptions listOptions = nextMarker == null ? ListContainerOptions.NONE
            : ListContainerOptions.Builder.afterMarker(nextMarker);

      pageSet = blobStore.list(containerName, listOptions);

      ExecutorAllCompletionService eacs = new ExecutorAllCompletionService(executor);
      final TaskContext taskContext = new TaskContextImpl();

      Set<Object> entries = new HashSet<Object>(BATCH_SIZE);

      Iterator<? extends StorageMetadata> storageMetadataIterator = pageSet.iterator();
      while (storageMetadataIterator.hasNext()) {
         StorageMetadata blobMetadata = storageMetadataIterator.next();
         K key = (K) key2StringMapper.getKeyMapping(blobMetadata.getName());
         if (keyFilter == null || keyFilter.accept(key)) {
            entries.add(key);
         }
         if (entries.size() == BATCH_SIZE) {
            final Set<Object> batch = entries;
            entries = new HashSet<Object>(BATCH_SIZE);
            submitProcessTask(cacheLoaderTask, eacs, taskContext, batch, loadValue, loadMetadata);
         }
      }

      if (!entries.isEmpty()) {
         submitProcessTask(cacheLoaderTask, eacs, taskContext, entries, loadValue, loadMetadata);
      }
      eacs.waitUntilAllCompleted();
      if (eacs.isExceptionThrown()) {
         throw new PersistenceException("Process execution exception!", eacs.getFirstException());
      }

      nextMarker = pageSet.getNextMarker();
   } while (nextMarker != null);
}