scala小工具：自动生成跑批文件

最新推荐文章于 2024-07-10 18:31:14 发布

Carltmf

最新推荐文章于 2024-07-10 18:31:14 发布

阅读量626

点赞数

CC 4.0 BY-SA版权

分类专栏： scala

本文链接：https://blog.youkuaiyun.com/qq1084502122/article/details/104455961

scala 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种使用Scala编写的工具，该工具可以快速生成用于批量处理SQL脚本的批处理文件，适用于需要补历史数据的数仓项目，通过简单的输入即可自动生成一系列SQL执行命令。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

背景：

由于数仓项目中经常需要手工跑批补历史数据，通过excel手工拉取每次都要写函数不太方便，通过小工具生成，省时省力。

语言：

scala

脚本：

import scala.io.StdIn
import java.time.format.DateTimeFormatter
import java.time.LocalDate._
import java.io.PrintWriter
import java.time.LocalDate
object TestBasic {
  def main(args: Array[String]): Unit = {
    //输入作业名(不含后缀)
    val jobName = args(0)
    //指定日期格式
    val pattern = DateTimeFormatter.ofPattern("yyyyMMdd")
    //获取开始日期
    var startDt = LocalDate.parse(StdIn.readLine("开始日期(YYYYMMDD)："), pattern)
    //获取结束日期
    val endDt = LocalDate.parse(StdIn.readLine("结束日期(YYYYMMDD)："), pattern)
    //打开文件(写入)
    val writer = new PrintWriter(jobName+".bat")
    while (startDt.isBefore(endDt)) {
      //写入内容
      writer.write("dsql -c logon.env -f " + jobName + ".dsql TX_DATE=" + startDt.format(pattern) + ">" + jobName + "_" + startDt.format(pattern) + ".sql\n")
      startDt = startDt.plusDays(1)
      writer.write("\n")
    }
    //关闭文件
    writer.close()
    //结束
    println("work is done!")
  }
}