在flatMapToPair中访问HashMap

import org.apache.spark.api.java.JavaPairRDD;
import org.apache.spark.api.java.JavaRDD;
import org.apache.spark.api.java.JavaSparkContext;
import scala.Tuple2;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

public class HashMapAccessExample {
    public static void main(String[] args) {
        // 创建SparkContext
        JavaSparkContext sparkContext = new JavaSparkContext("local", "HashMapAccessExample");

        // 创建一个包含数据的RDD
        JavaRDD<String> dataRDD = sparkContext.parallelize(Arrays.asList("apple", "banana", "orange", "apple"));

        // 创建一个HashMap并添加键值对
        Map<String, Integer> countMap = new HashMap<>();
        countMap.put("apple", 0);
        countMap.put("banana", 0);
        countMap.put("orange", 0);

        // 在flatMapToPair函数中访问HashMap并更新计数
        JavaPairRDD<String, Integer> resultRDD = dataRDD.flatMapToPair(data -> {
            // 获取HashMap的迭代器
            Iterator<Map.Entry<String, Integer>> iterator = countMap.entrySet().iterator();

            // 更新计数
            while (iterator.hasNext()) {
                Map.Entry<String, Integer> entry = iterator.next();
                if (entry.getKey().equals(data)) {
                    entry.setValue(entry.getValue() + 1);
                }
            }

            // 返回键值对
            List<Tuple2<String, Integer>> list = new ArrayList<>();
            for (Map.Entry<String, Integer> entry : countMap.entrySet()) {
                list.add(new Tuple2<>(entry.getKey(), entry.getValue()));
            }
            return list.iterator();
        });

        // 打印结果
        resultRDD.foreach(pair -> System.out.println(pair._1() + ": " + pair._2()));

        // 关闭SparkContext
        sparkContext.close();
    }
}