Scala使用POI生成Excel的Servlet

lass PatentAnalysisExportExcelResult(requestContext: RequestContext,
  patentAnalysisResult: PatentAnalysisResult) extends ActionResult(requestContext) with Disposable {
  val fieldMap = Map[String, String](("ad_year", "申请日"),
    ("pd_year", "公开公告日"),
    ("paid", "申请人"),
    ("inid", "发明人"),
    ("ic_subclass", "IPC小类"),
    ("ic_supergroup", "IPC大组"),
    ("ic", "IPC小组"),
    ("co", "专利所属国家地区"),
    ("pac", "申请人所在国家地区"),
    ("pcid_1", "一级技术分类"),
    ("pcid_2", "二级技术分类"))
  val idSet = Set[String]("ad_year", "pd_year", "ic_subclass", "ic_supergroup", "ic")

  def execute = {
    response.setContentType("application/excel")
    response.setCharacterEncoding("utf-8")

    patentAnalysisResult.patentAnalysisResultParam.analysisType match {
      case "s" => {
        response.addHeader("Content-Disposition", "attachment; filename=" + "patent-analysis-tableexport-" + patentAnalysisResult.patentAnalysisResultParam.analysisFieldOne + "-" + patentAnalysisResult.patentAnalysisResultParam.analysisFieldTwo + "-" + new SimpleDateFormat("yyyyMMddHHmmss").format(new Date()) + ".xls")
        writeDoubleFieldFacet(patentAnalysisResult.DoubleFieldCounts, patentAnalysisResult.patentAnalysisResultParam.analysisFieldOne, patentAnalysisResult.patentAnalysisResultParam.analysisFieldTwo)
      }
      case _ => {
        response.addHeader("Content-Disposition", "attachment; filename=" + "patent-analysis-tableexport-" + patentAnalysisResult.patentAnalysisResultParam.analysisField + "-" + new SimpleDateFormat("yyyyMMddHHmmss").format(new Date()) + ".xls")

        patentAnalysisResult.patentAnalysisResultParam.analysisField match {
          case "ad_year" => writeFieldFacet(patentAnalysisResult.applicationDateCounts, "申请日分析表格", "申请年", "id")
          case "pd_year" => writeFieldFacet(patentAnalysisResult.publicationDateCounts, "公开公告日分析表格", "公开公告年", "id")
          case "paid" => writeFieldFacet(patentAnalysisResult.applicantCounts, "申请人分析表格", "申请人", "name")
          case "inid" => writeFieldFacet(patentAnalysisResult.inventorCounts, "发明人分析表格", "发明人", "name")
          case "ic_subclass" => writeFieldFacet(patentAnalysisResult.ipcSubClassCounts, "IPC小类分析表格", "IPC小类", "id")
          case "ic_supergroup" => writeFieldFacet(patentAnalysisResult.ipcSuperGroupCounts, "IPC大组分析表格", "IPC大组", "id")
          case "ic" => writeFieldFacet(patentAnalysisResult.ipcCounts, "IPC小类分析表格", "IPC小类", "id")
          case "cb" => writeFieldFacet(patentAnalysisResult.citationCounts, "引用专利分析表格", "公开公告号", "id")
          case "cf" => writeFieldFacet(patentAnalysisResult.cocitationCounts, "被引专利分析表格", "公开公告号", "id")
          case "fm" => writeFieldFacet(patentAnalysisResult.familyStatsCounts, "专利族分析表格", "公开公告号", "id")
          case "co" => writeFieldFacet(patentAnalysisResult.countryCounts, "申请人所在国家地区分析表格", "国家地区", "name")
          case "pac" => writeFieldFacet(patentAnalysisResult.applicantCountryCounts, "申请人所在国家地区分析表格", "国家地区", "name")
          case "pcid_1" => writeFieldFacet(patentAnalysisResult.categoryOneCounts, "一级技术分类分析表格", "分类名称", "name")
          case "pcid_2" => writeFieldFacet(patentAnalysisResult.categoryTwoCounts, "二级技术分类分析表格", "分类名称", "name")
        }
      }
    }
  }

  private def writeFieldFacet(list: Array[FacetCount[IdNameEntity]], title: String, columnName: String, t: String) = {
    val hwb = new HSSFWorkbook()
    val sheet = hwb.createSheet(title)
    val firstrow = sheet.createRow(0)
    val names = Array[String]("排名", columnName, "专利数量")

    for (i <- 0 until names.length) {
      val cell = firstrow.createCell(i)
      cell.setCellValue(new HSSFRichTextString(names(i)))
    }

    for (i <- 0 until list.length) {
      val row = sheet.createRow(i + 1)
      val xlsDto = list(i)

      for (j <- 0 until 3) {
        val sortCell = row.createCell(0)

        sortCell.setCellValue(i + 1)

        val nameCell = row.createCell(1)

        if (t == "id") nameCell.setCellValue(xlsDto.t.id.toString)
        else nameCell.setCellValue(xlsDto.t.name)

        val countCell = row.createCell(2)

        countCell.setCellValue(xlsDto.count)
      }
    }

    using(responseOutputStream) { out =>
      {
        hwb.write(out)
      }
    }
  }

  private def writeDoubleFieldFacet(list: Array[DoubleFacetFiled], fieldOne: String, fieldTwo: String) = {
    val hwb = new HSSFWorkbook()
    val sheet = hwb.createSheet(fieldMap.getOrElse(fieldOne, "") + "-" + fieldMap.getOrElse(fieldTwo, "") + "表格")
    val firstrow = sheet.createRow(0)
    val names = Array[String]("") ++ { if (idSet.contains(fieldOne)) list.map(_.field.id.toString) else list.map(_.field.name) }

    for (i <- 0 until names.length) {
      val cell = firstrow.createCell(i)

      cell.setCellValue(new HSSFRichTextString(names(i)))
    }

    val len = list(0).fields.length

    for (i <- 0 until len) {
      val row = sheet.createRow(i + 1)

      for (j <- 0 to list.length) {
        if (j == 0) {
          val cell = row.createCell(0)
          val field = list(j).fields(i)

          cell.setCellValue(if (idSet.contains(fieldTwo)) field.t.id.toString else field.t.name)
        } else {
          val cell = row.createCell(j)
          val field = list(j-1).fields(i)

          cell.setCellValue(field.count)
        }
      }
    }

    using(responseOutputStream) { out =>
      {
        hwb.write(out)
      }
    }
  }
}