- README.md:项目结构、技术栈、三条业务需求、MySQL 表、构建运行步骤、已知问题 - CODE_REVIEW.md:分三级(严重/主要/次要)列出 11 个发现,含密码硬编码、空类、Any 分发等 本次仅提交两份文档;工作树里 App.scala 的 data path 切换与 ratings_all.csv 由用户后续单独提交
7.9 KiB
代码审查
审查范围:
src/main/scala、src/main/resources、pom.xml、data/(目录约定) 审查日期:2026-09-21
总体评价
项目用 Scala 2.11 + Spark 2.4.3 + MySQL(HikariCP + DBUtils)实现了 MovieLens 风格数据集上的三条聚合分析需求,整体结构清晰、能跑通,但存在若干 正确性 / 安全 / 工程规范 级别的问题,需要修。
下面按严重程度分级。
🔴 严重(必须修)
1. MySQL 密码硬编码在源码里
src/main/scala/com/example/utils/JDBCUtil.scala:14
private val password = "root123456"
private val url = "jdbc:mysql://192.168.48.10:3306/test"
- 风险:密码、IP、库名都明文进 Git;任何克隆者都能拿到内网 MySQL 凭证。
- 建议:抽到
application.conf(Typesafe Config)或环境变量;至少把密码挪出源码。dev和prod用不同 profile。
2. create_table.sql 字段注释 copy-paste 错误
src/main/resources/create_table.sql:7, 20
`avg_rating` decimal(10, 2) not null comment '电影名称',
...
`avg_rating` DECIMAL(10, 2) NOT NULL COMMENT '电影类别平均评分',
第 7 行 top_ten_avg_ratings.avg_rating 的注释是“电影名称”,明显复制粘贴残留。虽然不影响运行,但生产里这种注释会误导接手人。
3. System.exit(-1) 出现在库代码里
JDBCUtil.scala:51
case None =>
println("Mysql 连接失败")
System.exit(-1)
- 问题:JDBCUtil 应该是个工具对象,不应该决定进程生死。
App.main想在 MySQL 不可用时降级(比如只打印到控制台)就没办法。 - 建议:抛
RuntimeException,让main顶层try/catch或让 Spark Driver 自行退出。
🟠 主要(建议修)
4. Entitys.scala 里的空类 class Entitys
src/main/scala/com/example/spark/Entitys.scala:3
class Entitys {
}
完全没人用,是死代码。文件命名也误导(复数 Entitys,里面只有一个空类 + 三个 case class + 一个无关 case class)。建议:
- 要么删空类,把文件名改成
Models.scala或Schemas.scala; - 要么拆成
Movie.scala/Rating.scala/Results.scala三个文件。
5. 结果 case class 字段类型与 Spark SQL 不一致
Entitys.scala:11, 13
case class TopTenAvgRatings(movieId: String, title: String, avgRating: String)
case class TopGenresByAverageRating(genres: String, avgRating: String)
Movie.movieId: Int、TopGenresByAverageRating 里 avgRating 在 SQL 是 decimal / double,但这里全是 String。原因应该是为了避免 Int.box/Long.box,但代价是:
- 类型擦掉了,编译期查不出 bug;
.as[TopTenAvgRatings]实际是反射转 String,靠 Spark 的内部转换,调试时容易踩坑;- 写库时 DBUtils 走
setObject(String, ...),MySQL 驱动能转,但语义不清晰。
建议:直接用 Int/Double,让 Driver 自己装箱;性能差异在这点数据量上完全可忽略。
6. insert2Mysql(res: Any) 用 Any 做运行时分发
JDBCUtil.scala:44, 56
def insert2Mysql(res: Any): Unit = ...
private def upsert(res: Any, queryRunner: QueryRunner): Unit = {
res match {
case data: TopTenAvgRatings => ...
case data: TopGenresByAverageRating => ...
case data: MovieRatingCount => ...
case _ => println(s"不支持的数据类型:${res.getClass.getName}")
}
}
- 没有编译期保证,新增 case class 时静默落入
case _。 case _只println一下就吞掉,数据丢了不抛错。
建议:用 trait + 各自实现,例如:
sealed trait Writable { def sql: String; def params: Seq[Any] }
object TopTenAvgRatings { def toRow(r: TopTenAvgRatings): Writable = ... }
或者最简单:拆成 insertTopTen(Seq[TopTenAvgRatings])、insertGenres(...)、insertMovieCount(...),foreach 在外部调用。
7. App.scala 里的 SQL 注释被去掉执行了
App.scala:44, 60-62
// - 需求1:查找电影评分个数超过5000,并且平均分较高的前十部电影名称及其对应的平均评分
val sql1 = ...
spark.sql(sql1).show()
val result1 = spark.sql(sql1).as[TopTenAvgRatings]
result1.foreach(insert2Mysql(_))
sql1算了两次(一次show()、一次as[]),每次都是从头跑 Job。show()算调试输出,跟"写库"是同一份逻辑,没必要算两遍。
建议:拿一个 result1 用 .show() + .foreach(...),或者干脆用 cache() 复用。
8. SQL 3 里 max(m.movieId) / max(m.title) 是误导写法
App.scala:82-91
select
max(m.movieId) movieId
, max(m.title) title
, count(1) cn
from movies m
left join ratings r
on m.movieId = r.movieId
group by m.movieId
group by m.movieId 时 m.movieId 已经是 GROUP BY key,max(m.movieId) 等于 m.movieId 本身;m.title 函数依赖 movieId,max(title) 是合法的但语义上是"随便挑一个"。
建议:
select m.movieId, m.title, count(r.userId) as cn
from movies m left join ratings r on m.movieId = r.movieId
group by m.movieId, m.title
order by cn desc
limit 10
更直白。
9. SparkSession 创建后没有 stop()
App.scala:11
程序退出靠 JVM 终止,自然回收。但 Driver 日志会一直说 SparkSession 没正常关;在 cluster / spark-submit 模式下漏掉 stop() 可能影响下一次提交。
建议:
try { ... } finally { spark.stop() }
10. 没有用到的 import
App.scala:4
import org.apache.spark.sql.{DataFrame, Dataset, SparkSession}
DataFrame 和 Dataset 都没引用,IDE 静态检查能识别。可删。
11. CSV 用 inferSchema=true
App.scala:21-23, 30-32
.option("inferSchema","true")
每次跑都扫描一次 CSV 推 schema。movies.csv 小无所谓,ratings_all.csv 如果上 GB 级别,启动慢且 schema 可能飘。
建议:用 StructType 显式声明 schema。
🟡 次要(可改可不改)
| 位置 | 问题 |
|---|---|
JDBCUtil.scala:103-105 |
Int.box / Long.box 不必要,DBUtils update(sql, args*) 会自动装箱 |
JDBCUtil.scala:29 |
HikariDataSource 是 object 字段,永不 close;CLI 应用无所谓,做 library 时会泄漏 |
JDBCUtil.scala:33-40 |
getQueryRunner: Option 模式过度工程化;dataSource 在类初始化时已建好,根本不会抛 |
App.scala:8-9 |
数据路径硬编码;应该走 args / config |
App.scala:14 |
master("local[2]") 写死;集群提交时得改代码 |
log4j.properties |
只配了 error 级别 + stdout,没文件 appender;线上调试不方便 |
pom.xml:71-75 |
引了 JUnit 3.8.1,但 src/test 整个目录都没有;要么补测试要么删依赖 |
| 缩进风格 | 需求 2/3 里 val result2/3 没对齐 result1,缩进风格不统一 |
sql2 |
没过滤 category is null or category = '(no genres listed)',可能脏数据 |
改进建议(按优先级)
- 把密码 / DB 连接信息挪到 config 文件,加
.gitignore。 - 删
class Entitys死代码;改文件名。 - 修
create_table.sql注释。 - 用 sealed trait / 独立函数替换
Any模式匹配;case _改成抛异常。 sql1去重计算 +spark.stop()+ 显式 CSV schema。- 简化
TopTenAvgRatings等 case class 的字段类型。 - 补 1-2 个最关键的单元测试(CSV 解析 / SQL 结果行数)。
测试覆盖
仓库目前没有任何测试代码。src/test/ 目录是空的,但 pom.xml 引了 JUnit 3.8.1。至少建议补:
Entitys的 case class 序列化(防 schema 漂移)JDBCUtil.upsert至少对一个类型做 round-trip- 三个 SQL 在 fixture 数据上的预期行数 / Top-1 断言
总结
作为学习项目骨架是合格的,但距离"可上线"差上述红/橙清单。最大的两件事:把密码挪出源码、把空类 / 无用 import / Any 分发清理掉。