遍历嵌套的映射

Question

我整理了以下方法，以帮助迭代嵌套地图（针对另一个SO问题）。可以清楚地看到，前两个方法实际上几乎完全相同，除了它们的泛型和其中一个调用和另一个调用。有没有办法将它们折叠成一个方法，或者至少将机制的明显重复放在一个地方？

如果正确执行，应该可以迭代任意深度的嵌套地图。

// Iterating across Maps of Maps of Maps.
static <K1, K2, K3, V> Iterator<Iterator<Iterator<V>>> iiiV(Map<K1, Map<K2, Map<K3, V>>> mmm) {
  final Iterator<Map<K2, Map<K3, V>>> mmi = iV(mmm);
  return new Iterator<Iterator<Iterator<V>>>() {
    @Override
    public boolean hasNext() {
      return mmi.hasNext();
    }

    @Override
    public Iterator<Iterator<V>> next() {
      return iiV(mmi.next());
    }

    @Override
    public void remove() {
      mmi.remove();
    }
  };
}

// Iterating across Maps of Maps.
static <K1, K2, V> Iterator<Iterator<V>> iiV(Map<K1, Map<K2, V>> mm) {
  final Iterator<Map<K2, V>> mi = iV(mm);
  return new Iterator<Iterator<V>>() {
    @Override
    public boolean hasNext() {
      return mi.hasNext();
    }

    @Override
    public Iterator<V> next() {
      return iV(mi.next());
    }

    @Override
    public void remove() {
      mi.remove();
    }
  };
}

// Iterating across Map values.
static <K, V> Iterator<V> iV(final Map<K, V> map) {
  return iV(map.entrySet().iterator());
}

// Iterating across Map.Entries.
static <K, V> Iterator<V> iV(final Iterator<Map.Entry<K, V>> mei) {
  return new Iterator<V>() {
    @Override
    public boolean hasNext() {
      return mei.hasNext();
    }

    @Override
    public V next() {
      return mei.next().getValue();
    }

    @Override
    public void remove() {
      mei.remove();
    }
  };
}

- OldCurmudgeon

---

如果要在n层深度的地图上进行迭代，方法签名和返回类型应该是什么？ - khachik

@khachik - 请查看链接中的完整答案。您可以使用NestedIterator(NestedIterator(Iterator<Iterator<Iterator<V>>> i))返回Iterator<V>。这些方法只是帮助构建嵌套的迭代器，用于NestedIterator构造函数中的嵌套Map。 - OldCurmudgeon

Answer 1

为什么不这样做：

static <K, V> Iterator<V> iV(final Iterator<V> mei) {
   return new Iterator<V>() {
     @Override
     public boolean hasNext() {
       return mei.hasNext();
     }

     @Override
     public V next() {
       if (Iterator.class.isAssignableFrom(V.class)) {
         Iterator it = iV((Iterator)V);
         return it.hasNext() ? it.next() : ((Iterator)mei.next()).next();
       }
       else
         return mei.next().getValue();
     }

     @Override
     public void remove() {
       mei.remove();
    }
  };
}

（未测试）

- Cedric Reichenbach

---

不确定由于类型擦除是否可以执行 V.class，但我喜欢这个想法。也许可以考虑使用 mei.getClass()。 - OldCurmudgeon

哦，我认为你是对的。但是mei.getClass()只会返回Iterator.class。你应该取一个元素并使用element.getClass()检查它，但要小心考虑空迭代器等情况。我已经遇到过类似的情况，我认为没有更干净的解决方案... :( - Cedric Reichenbach