大数据-88 Spark Super Word Count 全流程实现（Scala + MySQL）

武子康2025-09-05 9:52

点一下关注吧！！！非常感谢！！持续更新！！！

🚀 AI篇持续更新中！（长期更新）

AI炼丹日志-31- 千呼万唤始出来 GPT-5 发布！"快的模型 + 深度思考模型 + 实时路由"，持续打造实用AI工具指南！📐🤖

💻 Java篇正式开启！（300篇）

目前2025年09月01日更新到： Java-113 深入浅出 MySQL 扩容全攻略：触发条件、迁移方案与性能优化 MyBatis 已完结，Spring 已完结，Nginx已完结，Tomcat已完结，分布式服务正在更新！深入浅出助你打牢基础！

📊 大数据板块已完成多项干货更新（300篇）：

包括 Hadoop、Hive、Kafka、Flink、ClickHouse、Elasticsearch 等二十余项核心组件，覆盖离线+实时数仓全栈！ 大数据-278 Spark MLib - 基础介绍 机器学习算法 梯度提升树 GBDT案例 详解

章节内容

上节完成的内容如下：

  • Spark案例编写 Scala
  • 计算圆周率
  • 找共同的好友

Super Word Count

需求背景详细说明

文本预处理任务分解

  1. 大小写转换：将输入文本中的所有字母统一转换为小写形式，确保后续处理不受大小写影响。例如："Hello World" → "hello world"

  2. 标点符号处理：

    • 默认去除常见标点符号：! " # $ % & ' ( ) * + , - . / : ; < = > ? @ [ \ ] ^ _ ` { | } ~
    • 支持自定义标点符号集合，用户可传入特定符号列表进行过滤
    • 示例：输入"Hello, world!" → 处理后"hello world"

  3. 停用词过滤：

    • 默认使用英文常见停用词列表（如NLTK的停用词集）
    • 支持用户自定义停用词列表，可灵活添加业务相关停用词
    • 示例：输入"the quick brown fox" → 过滤后"quick brown fox"（假设"the"在停用词表中）

  4. 词频统计：

    • 对处理后的单词进行计数
    • 按词频降序排列结果
    • 示例：输入"apple banana apple" → 输出[('apple',2), ('banana',1)]

  5. 数据存储：

    • 结果保存到MySQL数据库
    • 需要设计包含以下字段的表：
      • id（主键）
      • word（单词）
      • count（词频）
      • create_time（记录创建时间）
    • 支持批量插入和更新操作

技术实现要点

  • 提供配置接口允许用户自定义：
    • punctuation_set: 标点符号集合
    • stop_words: 停用词列表
  • 实现高效的正则表达式处理标点符号
  • 使用字典结构进行词频统计优化性能
  • MySQL连接使用连接池管理
  • 添加异常处理机制确保数据完整性

典型应用场景

  1. 舆情分析系统中的关键词提取
  2. 搜索引擎的索引构建
  3. 文本分类任务的前期处理
  4. 用户评论的词云生成
  5. 日志分析中的高频词统计

扩展功能考虑

  • 支持处理多语言文本（如中文分词）
  • 添加词干提取（stemming）或词形还原（lemmatization）选项
  • 实现定时任务自动处理新增文本
  • 增加可视化统计结果导出功能

编写代码

先实现到MySQL保存前的内容，我们需要先编写测试一下我们的代码是否正确

scala 复制代码 
object SuperWordCount1 {

  private val stopWords = "in on to from by a an the is are were was i we you your he his".split("\\s+")

  private val punctuation = "[\\)\\.,:;'!\\?]"

  def main(args: Array[String]): Unit = {
    val conf = new SparkConf()
      .setAppName("ScalaSuperWordCount1")
      .setMaster("local[*]")
    val sc = new SparkContext(conf)
    sc.setLogLevel("WARN")

    val lines: RDD[String] = sc.textFile(args(0))
    lines
      .flatMap(_.split("\\s+"))
      .map(_.toLowerCase)
      .map(_.replaceAll(punctuation, ""))
      .filter(word => !stopWords.contains(word) && word.trim.nonEmpty)
      .map((_, 1))
      .reduceByKey(_ + _)
      .sortBy(_._2, false)
      .collect()
      .foreach(println)
    
    sc.stop()
  }

}

详细解释

object SuperWordCount1 { ... }

  • SuperWordCount1 是一个 Scala 对象，定义了一个单例对象用于运行单词计数程序。

private val stopWords = "in on to from by a an the is are were was i we you your he his".split("\s+")

  • 这里定义了一个 stopWords 列表，包含了常见的停用词，这些词在统计单词频率时会被过滤掉。
  • split("\s+") 方法将这些停用词用空白字符分割成数组，便于后续的查找和过滤。

private val punctuation = "[\)\.,:;'!\?]"

  • 定义了一个正则表达式 punctuation，用于匹配常见的标点符号。这些标点符号在统计单词频率时会被去除。

def main(args: Array[String]): Unit = { ... }

  • main 方法是程序的入口点，args 是命令行参数，其中 args(0) 通常表示输入文件的路径。

val conf = new SparkConf().setAppName("ScalaSuperWordCount1").setMaster("local[*]")

  • SparkConf() 用于配置 Spark 应用程序。setAppName("ScalaSuperWordCount1") 设置应用程序的名称。
  • setMaster("local[*]") 指定应用程序以本地模式运行，使用所有可用的 CPU 核心。

val sc = new SparkContext(conf)

  • SparkContext 是 Spark 应用程序的核心，用于与 Spark 集群进行交互。

sc.setLogLevel("WARN")

  • 设置日志级别为 "WARN"，减少日志输出，方便查看重要信息。

val lines: RDD[String] = sc.textFile(args(0))

  • sc.textFile(args(0)) 从指定的文本文件路径加载数据，创建一个 RDD[String]，其中每一行文本都作为一个字符串元素。
  • lines 是包含输入文本数据的 RDD。

flatMap(_.split("\s+"))

  • flatMap 方法将每一行字符串按空白字符拆分成单词，并将其展开成单个单词的 RDD。

map(_.toLowerCase)

  • 将每个单词转换为小写，以确保统计时不区分大小写。

map(_.replaceAll(punctuation, ""))

  • 使用正则表达式 punctuation 去除单词中的标点符号，使得统计结果更加准确。

filter(word => !stopWords.contains(word) && word.trim.nonEmpty)

  • filter 方法过滤掉停用词和空白单词：
  • !stopWords.contains(word) 确保当前单词不在停用词列表中。
  • word.trim.nonEmpty 确保单词在去除前后空白字符后不是空字符串。

map((_, 1))

  • 将每个单词映射为 (word, 1) 的键值对，表示每个单词出现一次。

reduceByKey(_ + _)

  • reduceByKey 方法根据键（单词）对值（计数）进行累加，统计每个单词的总出现次数。

sortBy(_._2, false)

  • 将统计结果按值（单词出现的次数）从大到小排序。

collect().foreach(println)

  • collect() 方法将 RDD 中的数据收集到驱动程序中（即本地），然后使用 foreach(println) 输出每个单词及其出现的次数。
  • 由于 collect 会将数据从分布式环境中拉到本地，需要注意数据量大的情况下可能导致内存不足的问题。

sc.stop()

  • 在计算完成后，调用 sc.stop() 方法停止 SparkContext，释放资源。

添加依赖

shell 复制代码 
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>8.0.28</version>
</dependency>

同时我们需要在build的部分，也要加入对应的内容，让驱动可以加载进来：

xml 复制代码 
<build>
    <plugins>
        <plugin>
            <groupId>net.alchim31.maven</groupId>
            <artifactId>scala-maven-plugin</artifactId>
            <version>4.4.0</version>
            <executions>
                <execution>
                    <goals>
                        <goal>compile</goal>
                        <goal>testCompile</goal>
                    </goals>
                </execution>
            </executions>
        </plugin>
        <plugin>
            <groupId>org.apache.maven.plugins</groupId>
            <artifactId>maven-assembly-plugin</artifactId>
            <version>3.3.0</version>
            <configuration>
                <archive>
                    <manifest>
                        <mainClass>cn.lagou.sparkcore.WordCount</mainClass>
                    </manifest>
                </archive>
                <descriptorRefs>
                    <descriptorRef>jar-with-dependencies</descriptorRef>
                </descriptorRefs>
            </configuration>
            <executions>
                <execution>
                    <phase>package</phase>
                    <goals>
                        <goal>single</goal>
                    </goals>
                </execution>
            </executions>
        </plugin>
    </plugins>
</build>

创建库表

我们新建一个数据库，也要新建一个数据表

shell 复制代码 
CREATE TABLE `wordcount` (
  `word` varchar(255) DEFAULT NULL,
  `count` int(11) DEFAULT NULL
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;

写入SQL-未优化

我们在 foreach 中保存了数据，此时需要创建大量的MySQL连接，效率是比较低的。

shell 复制代码 
object SuperWordCount2 {

  private val stopWords = "in on to from by a an the is are were was i we you your he his".split("\\s+")

  private val punctuation = "[\\)\\.,:;'!\\?]"

  def main(args: Array[String]): Unit = {
    val conf = new SparkConf()
      .setAppName("ScalaSuperWordCount2")
      .setMaster("local[*]")
    val sc = new SparkContext(conf)
    sc.setLogLevel("WARN")

    val lines: RDD[String] = sc.textFile(args(0))

    val words: RDD[String] = lines
      .flatMap(_.split("\\s+"))
      .map(_.trim.toLowerCase())

    val clearWords: RDD[String] = words
      .filter(!stopWords.contains(_))
      .map(_.replaceAll(punctuation, ""))

    val result: RDD[(String, Int)] = clearWords
      .map((_, 1))
      .reduceByKey(_ + _)
      .sortBy(_._2, false)
    result.foreach(println)

    // 输出到 MySQL
    val username = "hive"
    val password = "hive@wzk.icu"
    val url = "jdbc:mysql://h122.wzk.icu:3306/spark-test?useUnicode=true&characterEncoding=utf-8&useSSL=false"

    var conn: Connection = null
    var stmt: PreparedStatement = null
    var sql = "insert into wordcount values(?, ?)"

    result.foreach{
      case (word, count) => try {
        conn = DriverManager.getConnection(url, username, password)
        stmt = conn.prepareStatement(sql)
        stmt.setString(1, word)
        stmt.setInt(2, count)
      } catch {
        case e: Exception => e.printStackTrace()
      } finally {
        if (stmt != null) {
          stmt.close()
        }
        if (conn != null) {
          conn.close()
        }
      }
    }

    sc.stop()
  }

}

写入SQL-优化版

优化后使用 foreachPartition 保存数据，一个分区创建一个链接：cache RDD 注意：

  • SparkSQL 有方便的读写MySQL的方法，给参数直接调用即可
  • 但掌握这个方法很重要，因为SparkSQL不是支持所有类型的数据库
scala 复制代码 
object SuperWordCount3 {

  private val stopWords = "in on to from by a an the is are were was i we you your he his".split("\\s+")
  private val punctuation = "[\\)\\.,:;'!\\?]"
  private val username = "hive"
  private val password = "hive@wzk.icu"
  private val url = "jdbc:mysql://h122.wzk.icu:3306/spark-test?useUnicode=true&characterEncoding=utf-8&useSSL=false"

  def main(args: Array[String]): Unit = {
    val conf = new SparkConf()
      .setAppName("ScalaSuperWordCount2")
      .setMaster("local[*]")
    val sc = new SparkContext(conf)
    sc.setLogLevel("WARN")

    val lines: RDD[String] = sc.textFile(args(0))

    val words: RDD[String] = lines
      .flatMap(_.split("\\s+"))
      .map(_.trim.toLowerCase())

    val clearWords: RDD[String] = words
      .filter(!stopWords.contains(_))
      .map(_.replaceAll(punctuation, ""))

    val result: RDD[(String, Int)] = clearWords
      .map((_, 1))
      .reduceByKey(_ + _)
      .sortBy(_._2, false)
    result.foreach(println)

    result.foreachPartition(saveAsMySQL)

    sc.stop()
  }

  def saveAsMySQL(iter: Iterator[(String, Int)]): Unit = {
    var conn: Connection = null
    var stmt: PreparedStatement = null
    var sql = "insert into wordcount values(?, ?)"

    try {
      conn = DriverManager.getConnection(url, username, password)
      stmt = conn.prepareStatement(sql)
      iter.foreach{
        case (word, count) =>
          stmt.setString(1, word)
          stmt.setInt(2, count)
      }
    } catch {
      case e: Exception => e.printStackTrace()
    } finally {
      if (stmt != null) {
        stmt.close()
      }
      if (conn != null) {
        conn.close()
      }
    }
  }

}

打包上传

shell 复制代码 
mvn clean package

打包并上传到项目：

运行项目

不写入SQL版

shell 复制代码 
spark-submit --master local[*] --class icu.wzk.SuperWordCount1 spark-wordcount-1.0-SNAPSHOT.jar /opt/wzk/goodtbl.java

运行结果如下图：

写入SQL-未优化版

shell 复制代码 
spark-submit --master local[*] --class icu.wzk.SuperWordCount2 spark-wordcount-1.0-SNAPSHOT.jar /opt/wzk/goodtbl.java

写入SQL-优化版

shell 复制代码 
spark-submit --master local[*] --class icu.wzk.SuperWordCount3 spark-wordcount-1.0-SNAPSHOT.jar /opt/wzk/goodtbl.java

运行结果如下图：

查看数据

查看数据库，内容如下：

相关推荐
老邓计算机毕设3 小时前
Springboot乐家流浪猫管理系统16lxw（程序+源码+数据库+调试部署+开发环境）带论文文档1万字以上，文末可获取，系统界面在最后面。
数据库·spring boot·后端
知其然亦知其所以然3 小时前
别再只会背八股了！一文带你彻底搞懂UNION与UNION ALL的区别
后端·mysql·面试
羑悻3 小时前
再续传输层协议UDP ：从低可靠到极速传输的协议重生之路，揭秘无连接通信的二次进化密码！
后端
就是帅我不改3 小时前
99%的Java程序员都踩过的高并发大坑
后端·面试
BingoGo3 小时前
PHP Swoole/WebMan/Laravel Octane 等长驻进程框架内存泄露诊断与解决方案
后端·php
秦清3 小时前
组态可视化软件【导入属性】
前端·javascript·后端
用户4099322502123 小时前
为什么你的单元测试需要Mock数据库才能飞起来？
后端·ai编程·trae
hwjfqr3 小时前
VSCode终端中文乱码问题解决
前端·后端
费益洲3 小时前
用 Shields.io 定制 README 个性徽章
后端
热门推荐
012025 年高教社杯全国大学生数学建模竞赛C 题 NIPT 的时点选择与胎儿的异常判定 完整成品思路模型代码分享，全网首发高质量！！！02UV安装并设置国内源03奈飞工厂官网，国内Netflix影视在线看|中文网页电脑版入口04Xget：下一代开源资源获取加速引擎，让你的文件下载、储存库克隆和镜像拉取快如闪电05A股预测还能更准？开源大模型Kronos带你跑通预测+回测全流程06Qwen3-Coder 快速上手教程 | Qwen Code + Claude Code07不再让Windows更新！&Edge游戏助手卸载及关闭自动更新08KGG转MP3工具|非KGM文件|解密音频09Linux下V2Ray安装配置指南10jdk21下载、安装（Windows、Linux、macOS）