package sparkcore.java;
import java.util.Arrays;
import java.util.Iterator;
import org.apache.spark.SparkConf;
import org.apache.spark.api.java.JavaPairRDD;
import org.apache.spark.api.java.JavaRDD;
import org.apache.spark.api.java.JavaSparkContext;
import org.apache.spark.api.java.function.FlatMapFunction;
import org.apache.spark.api.java.function.Function2;
import org.apache.spark.api.java.function.PairFunction;
import org.apache.spark.api.java.function.VoidFunction;
import scala.Tuple2;
/**
* 使用java开发本地测试的wordcount程序
*/
public class WordCountLocal {
public static void main(String[] args) {
// 编写Spark应用程序
// 本地执行,是能够执行在eclipse中的main方法中,执行的
// 第一步:建立SparkConf对象,设置Spark应用的配置信息
// 使用setMaster()能够设置Spark应用程序要链接的Spark集群的master节点的url
// 可是若是设置为local则表明,在本地运行
SparkConf conf = new SparkConf().setAppName("WordCountLocal").setMaster("local");
// 以spark-submit提交运行时,须要设置,如直接以Java应用程序启动,则能够去掉
conf.set("spark.testing.memory", "536870912");// 512M
// 第二步:建立JavaSparkContext对象
// 在Spark中,SparkContext是Spark全部功能的一个入口,你不管是用java、scala,甚至是python编写
// 都必需要有一个SparkContext,它的主要做用,包括初始化Spark应用程序所需的一些核心组件,包括
// 调度器(DAGSchedule、TaskScheduler),还会去到Spark Master节点上进行注册,等等
// 一句话,SparkContext,是Spark应用中,能够说是最最重要的一个对象
// 可是呢,在Spark中,编写不一样类型的Spark应用程序,使用的SparkContext是不一样的,若是使用scala,
// 使用的就是原生的SparkContext对象
// 可是若是使用Java,那么就是JavaSparkContext对象
// 若是是开发Spark SQL程序,那么就是SQLContext、HiveContext
// 若是是开发Spark Streaming程序,那么就是它独有的SparkContext
// 以此类推
JavaSparkContext sc = new JavaSparkContext(conf);
// 第三步:要针对输入源(hdfs文件、本地文件,等等),建立一个初始的RDD
// 输入源中的数据会打散,分配到RDD的每一个partition中,从而造成一个初始的分布式的数据集
// 咱们这里呢,由于是本地测试,因此呢,就是针对本地文件
// SparkContext中,用于根据文件类型的输入源建立RDD的方法,叫作textFile()方法
// 在Java中,建立的普通RDD,都叫作JavaRDD
// 在这里呢,RDD中,有元素这种概念,若是是hdfs或者本地文件呢,建立的RDD,每个元素就至关于
是文件里的一行
// JavaRDD<String> lines = sc.textFile("file:///D:/eclipse-jee-neon-3/workspace/sparkcore-java/test.txt");
// 若是本地文件为Linux
JavaRDD<String> lines = sc.textFile("file:////root/spark/core/test.txt");
// 若是文件放在HDFS上时,须要先启动Hadoop
// JavaRDD<String> lines = sc.textFile("hdfs://node1:8020/test.txt");
// 第四步:对初始RDD进行transformation操做,也就是一些计算操做
// 一般操做会经过建立function,并配合RDD的map、flatMap等算子来执行
// function,一般,若是比较简单,则建立指定Function的匿名内部类
// 可是若是function比较复杂,则会单首创建一个类,做为实现这个function接口的类
// 先将每一行拆分红单个的单词
// FlatMapFunction,有两个泛型参数,分别表明了输入和输出类型
// 咱们这里呢,输入确定是String,由于是一行一行的文本,输出,其实也是String,只不过是多个放在集合中
// 这里先简要介绍flatMap算子的做用,其实就是,将RDD的一个元素,给拆分红一个或多个元素
JavaRDD<String> words = lines.flatMap(new FlatMapFunction<String, String>() {
private static final long serialVersionUID = 1L;
public Iterator<String> call(String line) throws Exception {
return Arrays.asList(line.split(" ")).iterator();
}
});
// 接着,须要将每个单词,映射为(单词, 1)的这种格式
// 由于只有这样,后面才能根据单词做为key,来进行每一个单词的出现次数的累加
// mapToPair,其实就是将每一个元素,映射为一个(v1,v2)这样的Tuple2类型的元素
// 若是你们还记得scala里面讲的tuple,那么没错,这里的tuple2就是scala类型,包含了两个值
// mapToPair这个算子,要求的是与PairFunction配合使用,第一个泛型参数表明了输入类型
// 第二个和第三个泛型参数,表明的输出的Tuple2的第一个值和第二个值的类型
// JavaPairRDD的两个泛型参数,分别表明了tuple元素的第一个值和第二个值的类型
JavaPairRDD<String, Integer> pairs = words.mapToPair(new PairFunction<String, String, Integer>() {
private static final long serialVersionUID = 1L;
public Tuple2<String, Integer> call(String word) throws Exception {
return new Tuple2<String, Integer>(word, 1);
}
});
// 接着,须要以单词做为key,统计每一个单词出现的次数
// 这里要使用reduceByKey这个算子,对每一个key对应的value,都进行reduce操做
// 好比JavaPairRDD中有几个元素,分别为(hello, 1) (hello, 1) (hello, 1) (world, 1)
// reduce操做,至关因而把第一个值和第二个值进行计算,而后再将结果与第三个值进行计算
// 好比这里的hello,那么就至关因而,首先是1 + 1 = 2,而后再将2 + 1 = 3
// 最后返回的JavaPairRDD中的元素,也是tuple,可是第一个值就是每一个key,第二个值就是key的value
// reduce以后的结果,至关于就是每一个单词出现的次数
JavaPairRDD<String, Integer> wordCounts = pairs.reduceByKey(
// 第一与第二个参数为输入类型(为两个Tuple2的第二个元素类型),第三个为输出类型
new Function2<Integer, Integer, Integer>() {
private static final long serialVersionUID = 1L;
public Integer call(Integer v1, Integer v2) throws Exception {
return v1 + v2;
}
});
// 到这里为止,咱们经过几个Spark算子操做,已经统计出了单词的次数
// 可是,以前咱们使用的flatMap、mapToPair、reduceByKey这种操做,都叫作transformation操做
// 一个Spark应用中,光是有transformation操做,是不行的,是不会执行的,必需要有一种叫作action
// 接着,最后,可使用一种叫作action操做的,好比说,foreach,来触发程序的执行
wordCounts.foreach(new VoidFunction<Tuple2<String, Integer>>() {
private static final long serialVersionUID = 1L;
public void call(Tuple2<String, Integer> wordCount) throws Exception {
System.out.println(wordCount._1 + " : " + wordCount._2);
}
});
sc.close();
}
}