翻译:Hadoop权威指南之Spark-2

本文原始地址

A Scala Standalone Application

在Spark shell中运行了一个小程序之后,你可能想要把它打包成自包含应用,这样就可以多次运行了。

示例19-1. 使用Spark找出最高气温的Scala应用

import org.apache.spark.SparkContext._
import org.apache.spark.{SparkConf, SparkContext}

object MaxTemperature {
    def main(args: Array[String]) {
        val conf = new SparkConf().setAppName("Max Temperature")
        val sc = new SparkContext(conf)
        sc.textFile(args(0))
          .map(_.split("\t"))
          .filter(rec => (rec(1) != "9999" && rec(2).matches("[01459]")))
          .map(rec => (rec(0).toInt, rec(1).toInt))
          .reduceByKey((a, b) => Math.max(a, b))
          .saveAsTextFile(args(1))
    }
}

运行独立程序时,没有shell为我们提供SparkContext,我们需要自己创建。我们用一个SparkConf来创建这个实例。SparkConf可以用来向应用中传递多个Spark属性,这里我们仅仅设置应用的名字。

还有一些别的微小变化。首先是我们使用命令行参数来指定输入和输出路径。另外还使用了方法链来避免为每一个RDD创建中间变量,这样程序更紧凑,如果需要的话,我们仍然可以在Scala IDE中查看每次转变(transformation)的类型信息。

并非所有的Spark定义的transformation都可用于RDD类本身。在本例中,reduceByKey()(仅仅在键值对的RDD上起作用)实际上定义在PairRDDFunctions类中,但我们能用下面的import来让Scala隐含地把RDD[(Int, Int)]转为PairRDDFunctions:

import org.apache.spark.SparkContext._

这个import不同于Spark使用的隐式转型函数,因此理所当然地值得包含在程序中。

这一次我们使用spark-submit来运行这个程序,把包含编译后的Scala程序的JAR包作为参数传入,接着传入命令行参数(输入输出路径):

% spark-submit --class MaxTemperature --master local \
spark-examples.jar input/ncdc/micro-tab/sample.txt output
% cat output/part-*
(1950,22)
(1949,111)

我们还指定了两个选项:--class 告诉Spark应用类的名字,--master 指定job的运行方式,local值告诉Spark在本地机器的单个JVM中运行,在“Executors and Cluster Managers”一节我们将会学到在集群中运行的选项。接下来,我们看看怎样用Java语言来使用Spark。

A Java Example

Spark是使用Scala实现的,Scala是基于JVM的语言,可以和Java完美集成。同样的例子用Java来表达,很直接,也很啰嗦(使用Java 8的lambda表达式可以使这个版本更紧凑)。

示例19-2. 使用Spark找出最高气温的Java应用

public class MaxTemperatureSpark {
    public static void main(String[] args) throws Exception {
        if (args.length != 2) {
            System.err.println("Usage: MaxTemperatureSpark  ");
            System.exit(-1);
        }
        SparkConf conf = new SparkConf();
        JavaSparkContext sc = new JavaSparkContext("local", "MaxTemperatureSpark", conf);
        JavaRDD lines = sc.textFile(args[0]);
        JavaRDD records = lines.map(new Function() {
            @Override public String[] call(String s) {
                return s.split("\t");
            }
        });
        JavaRDD filtered = records.filter(new Function() {
            @Override public Boolean call(String[] rec) {
                return rec[1] != "9999" && rec[2].matches("[01459]");
            }
        });
        JavaPairRDD tuples = filtered.mapToPair(
            new PairFunction() {
                @Override public Tuple2 call(String[] rec) {
                    return new Tuple2(
                        Integer.parseInt(rec[0]), Integer.parseInt(rec[1]));
                }
        });
        JavaPairRDD maxTemps = tuples.reduceByKey(
            new Function2() {
                @Override public Integer call(Integer i1, Integer i2) {
                    return Math.max(i1, i2);
                }
        });
        maxTemps.saveAsTextFile(args[1]);
    }
}

在Spark的Java API中,一个RDD由JavaRDD的实例表示,在键值对RDD的特殊情况下是JavaPairRDD 。这两个类都实现了JavaRDDLike接口,该接口中可以找到操作RDD的大多数方法。

运行这个程序和运行Scala版本一样,除了类名字是MaxTemperatureSpark 。

A Python Example

Spark也支持Python语言,API叫做PySpark。由于Python语言有lambda表达式,例子程序非常接近Scala的版本。

示例19-3. 使用Spark找出最高气温的Python应用

form pyspark import SparkContext
import re, sys

sc = SparkContext("local", "Max Temperature")
sc.textFile(sys.argv[1]) \
  .map(lambda s: s.split("\t")) \
  .filter(lambda rec: (rec[1] != "9999" and re.match("[01459]", rec[2]))) \
  .map(lambda rec: (int(rec[0]), int(rec[1]))) \
  .reduceByKey(max) \
  .saveAsTextFile(sys.argv[2])

注意到在reduceByKey()的转变中,我们可以使用Python语言内建的max函数。

需要留意的重点是,这个程序是用CPython写的,Spark会创建一个Python子进程来执行用户的Python代码(在启动程序launcher和在集群上运行用户任务的executor上)。两个进程间使用socket通讯来传递RDD分区数据。

要运行这个程序,只需指定Python文件即可:

% spark-submit --master local \
  ch19-spark/src/main/python/MaxTemperature.py \
  input/ncdc/micro-tab/sample.txt output

还可以使用pyspark命令,以交互模式运行Spark和Python。

你可能感兴趣的:(翻译:Hadoop权威指南之Spark-2)