IRIS中实现zip文件的下载、解压及读取

0 前言

项目上做对账需求时，需要通过http方式下载第三方的对账的文件，是一个压缩文件，里边包含一个csv文件。

1 准备工作

1.1 安装解压缩软件

需要根据cache/iris服务器运行环境安装不同的解压缩软件。 本文是在windows环境下实现的，安装了WinRAR解压缩软件，安装目录为：”C:\Program Files\WinRAR“。

1.2 部署测试http服务器

部署一个web服务器，放一个zip文件。 本文在macOS系统中使用MAMP部署了一个简单的服务器，提供一个对外服务：http://192.168.1.107:8888/sss/Archive.zip

2 下载zip文件

主要思路： 通过http下载zip文件，然后另外到服务器上指定的文件夹下，并重命名。

代码

/// d ##class(web.test.DownloadFile).DownloadFile()
ClassMethod DownloadFile()
{
	set filePath="E:\testdata"
	set fileName="BCMData"_$zd(+$h,8)_".zip"
	
	Set httprequest=##class(%Net.HttpRequest).%New()
	set httprequest.Port=8888
	Set httprequest.Server="192.168.1.107"
	set httprequest.ContentCharset="UTF-8"                                    ;编码字符集
	
	set status= httprequest.Get("/sss/Archive.zip")
	//w !,status
	set dataStream = ##class(%Stream.GlobalCharacter).%New()
    Set dataStream=httprequest.HttpResponse.Data
   
	Set stream=##class(%FileBinaryStream).%New()
	Set stream.Filename=filePath_"\"_fileName
	Set status=stream.CopyFromAndSave(dataStream)
    
    //解压文件
    d ##class(web.test.DownloadFile).UnZip(filePath,fileName)
    
}

2 解压zip文件

主要思路： 使用$zf(-1)命令解压文件到指定的目录。

代码

/// d ##class(web.test.DownloadFile).UnZip("E:\testdata","myfile20220909.zip")
ClassMethod UnZip(zipFilePath As %String, zipFileName As %String)
{
	
	set zipFile=zipFilePath_"\"_zipFileName
	set targetDir=zipFilePath_"\"_$p(zipFileName,".",1)
	
	set file=##class(%File).%New()
	if file.DirectoryExists(targetDir) do
	.;如果目录存在先删除
	.//set status=file.RemoveDirectory(targetDir)
	.set status=file.RemoveDirectoryTree(targetDir)
	
	;创建目录
	set status=file.CreateDirectory(targetDir)
	
	//set execCode="""C:\Program Files\WinRAR\WinRAR.exe"" x -t -o-p "_zipFile_" *.txt "_targetDir  //只解压txt文件
	set execCode="""C:\Program Files\WinRAR\WinRAR.exe"" x -t -o-p "_zipFile_" "_targetDir  //所有文件
	
   	//SET execCode=##class(%Library.File).NormalizeFilenameWithSpaces(execCode)
 
   	//w execCode,!
   	WRITE $ZF(-1,execCode)   //解压文件
    
    do file.Close()
    
    //读取csv文件
    do ..ReadFile(targetDir)
}

3 读取zip文件

主要思路： 只读取文件夹中的csv文件，读取时注意文件编码格式。

代码

/// d ##class(web.test.DownloadFile).ReadFile("E:\testdata\myfile20220909")
ClassMethod ReadFile(fileDir As %String)
{
	set file=##class(%File).%New()
	if file.DirectoryExists(fileDir) {
		set num=file.GetDirectoryLength(fileDir)
		set rset=##class(%ResultSet).%New("%Library.File:FileSet")
		do rset.Execute(fileDir)
		while(rset.Next()) {
			set fileName=rset.Data("Name")        //文件名
			set filepx=$p(fileName,".",2)
			if (filepx="csv"){
				Set tmpfile=##class(%File).%New(fileName)
				Do tmpfile.Open("R:/IOTABLE=""UTF8""")   //指定文件编码
				while ('tmpfile.AtEnd){
					set tmpline=tmpfile.ReadLine()	
					;
					//过滤注释行
					continue:$e(tmpline,1,1)="#"
					//过明细列表头
					continue:$e(tmpline,1,1)=""
					//w !,tmpline
					set tradetime=$p($p(tmpline,",",1),"`",2)
					set tradetfinishime=$p($p(tmpline,",",2),"`",2)
					set hisTradeNo=$p($p(tmpline,",",7),"`",2)
					set tradeNo=$p($p(tmpline,",",8),"`",2)
					w !,tradetime_" "_tradetfinishime_" "_hisTradeNo_" "_tradeNo	
				}
				do tmpfile.Close()
			}	
		}
		
	}else{
		w !,"目录不存在"
	}
	do file.Close()
}

4 其他

此方法也可以在cache中使用。 另外，IRIS提供HS.Util.Zip.Adapter等相关类，具体没有使用过，大家可以考虑使用。