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();
}
}