scala小工具:读取EXCEL自动生成初始化语句-阿里云开发者社区

开发者社区> 开发与运维> 正文
登录阅读全文

scala小工具:读取EXCEL自动生成初始化语句

简介: 数仓项目中经常遇到业务手工发数据,然后要求补录数据的情况。每次通过excel拉取费时费力,使用小工具提高效率

maven依赖:

<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>4.1.0</version>

脚本:

package Excel

import java.io.{File, FileInputStream, InputStream, PrintWriter}

import org.apache.poi.hssf.usermodel.HSSFWorkbook
import org.apache.poi.ss.usermodel.{Cell, CellType}

object TestExcel {
def main(args: Array[String]): Unit = {

//读取excel
val file = new File(args(0))
val stream = new FileInputStream(file)
//输出文件
var output=new PrintWriter("script.sql")
//工作簿
val workBook = new HSSFWorkbook(stream)
//获取sheets集合
val sheets = workBook.sheetIterator()
//sheet递归
while (sheets.hasNext) {
  //获取当前sheet
  val sheet = sheets.next()
  //获取rows集合
  val rows = sheet.rowIterator
  //行递归
  while (rows.hasNext) {
    //获取当前row
    val row = rows.next()
    //获取cell集合
    val cells = row.cellIterator()
    //sql语句
    var sql = "insert into " + sheet.getSheetName + " values("
    //内容
    var content: String = ""
    //cell遍历
    while (cells.hasNext) {
      var cell = cells.next()
      content += (cell.getCellType match {
        case CellType.NUMERIC => "," + cell.getNumericCellValue + ""
        case _ => ",'" + cell.getStringCellValue.toString.replace("'","\\'") + "'"
      }
        )
    }
    //生成sql
    sql+=content.replaceFirst(",","")+");"
    //写入文件
    output.write(sql)
    output.write("\n")
  }
}
output.close()

}
}

版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。

分享:
开发与运维
使用钉钉扫一扫加入圈子
+ 订阅

集结各类场景实战经验,助你开发运维畅行无忧

其他文章
最新文章
相关文章