客户端从部署服务器或者hdfs的下载功能

最近遇到一个需求,通过系统配置的路径来完成一个下载功能,这个系统配置路径可能是hdfs路径也可能是部署服务器的路径,所以写了一个比较通用的方法,特此跟大家分享一下,也便于自己的记忆,主要是忘了还可以看一看,^_^直接上代码:

 /**
    *@Author huangzx
    *@Description: 文件下载到客户端
    *@Date 5:04 PM 2/23/19
    *@Param [path, response]
    *@return int
    */
    public static int downloadFile(String path, HttpServletResponse response) {
        int result = 1;
        InputStream is = null;
        OutputStream os = null;
        response.reset();
        response.setCharacterEncoding("utf-8");
        response.setContentType("multipart/form-data");
        try {
            if (path.startsWith("hdfs://")) {
                Configuration conf = new Configuration();
                FileSystem fs = FileSystem.get(URI.create(path), conf);
                is = fs.open(new Path(path));
            } else {
                File file = new File(path);
                is = new FileInputStream(file);
            }
            response.setHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode(path.substring(path.lastIndexOf("/")+1), "utf-8"));
            os = response.getOutputStream();
            byte[] b = new byte[2048];
            int length;
            while ((length = is.read(b)) > 0) {
                os.write(b, 0, length);
            }
            os.flush();
        } catch (IOException e) {
            e.printStackTrace();
            result = 0;
        }finally {
            closeInputStream(is);
            closeOutputStream(os);
        }
        return result;
    }
//下面两个是关流的方法

    private static void closeInputStream(InputStream is) {
        try {
            if (is != null) {
                is.close();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private static void closeOutputStream(OutputStream os) {
        try {
            if (os != null) {
                os.close();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值