ArrayListBackedIterator.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:2k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.mapred.join;
  19. import java.io.IOException;
  20. import java.util.ArrayList;
  21. import java.util.Iterator;
  22. import org.apache.hadoop.io.Writable;
  23. import org.apache.hadoop.io.WritableUtils;
  24. /**
  25.  * This class provides an implementation of ResetableIterator. The
  26.  * implementation uses an {@link java.util.ArrayList} to store elements
  27.  * added to it, replaying them as requested.
  28.  * Prefer {@link StreamBackedIterator}.
  29.  */
  30. public class ArrayListBackedIterator<X extends Writable>
  31.     implements ResetableIterator<X> {
  32.   private Iterator<X> iter;
  33.   private ArrayList<X> data;
  34.   private X hold = null;
  35.   public ArrayListBackedIterator() {
  36.     this(new ArrayList<X>());
  37.   }
  38.   public ArrayListBackedIterator(ArrayList<X> data) {
  39.     this.data = data;
  40.     this.iter = this.data.iterator();
  41.   }
  42.   public boolean hasNext() {
  43.     return iter.hasNext();
  44.   }
  45.   public boolean next(X val) throws IOException {
  46.     if (iter.hasNext()) {
  47.       WritableUtils.cloneInto(val, iter.next());
  48.       if (null == hold) {
  49.         hold = WritableUtils.clone(val, null);
  50.       } else {
  51.         WritableUtils.cloneInto(hold, val);
  52.       }
  53.       return true;
  54.     }
  55.     return false;
  56.   }
  57.   public boolean replay(X val) throws IOException {
  58.     WritableUtils.cloneInto(val, hold);
  59.     return true;
  60.   }
  61.   public void reset() {
  62.     iter = data.iterator();
  63.   }
  64.   public void add(X item) throws IOException {
  65.     data.add(WritableUtils.clone(item, null));
  66.   }
  67.   public void close() throws IOException {
  68.     iter = null;
  69.     data = null;
  70.   }
  71.   public void clear() {
  72.     data.clear();
  73.     reset();
  74.   }
  75. }