golang - 下载大文件,实时返回前端下载进度,实现下载进度条

news2024/12/26 21:00:07

示例:

package main

import (
	"fmt"
	"io"
	"net/http"
	"os"
	"path"

	//"github.com/kataras/iris"
	"github.com/kataras/iris/v12"
	"time"
)

func doSomething() {
	time.Sleep(time.Second * 10)
	fmt.Println("sleep end ------")
}

// DownloadFile 下载指定 URL 的文件并保存到指定路径
//func DownloadFile(url string, filePath string) error {
//	// 发起 GET 请求获取文件
//	resp, err := http.Get(url)
//	if err != nil {
//		return err
//	}
//	defer resp.Body.Close()
//
//	// 创建本地文件
//	out, err := os.Create(filePath)
//	if err != nil {
//		return err
//	}
//	defer out.Close()
//
//	// 将响应的内容写入本地文件
//	_, err = io.Copy(out, resp.Body)
//	if err != nil {
//		return err
//	}
//
//	return nil
//}

func DownloadFile(ctx iris.Context, url string, filePath string) error {

	// 获取 http.ResponseWriter
	w := ctx.ResponseWriter()

	// 发起 GET 请求获取文件
	resp, err := http.Get(url)
	if err != nil {
		return err
	}
	defer resp.Body.Close()

	// 创建本地文件
	out, err := os.Create(filePath)
	if err != nil {
		return err
	}
	defer out.Close()

	 将响应的内容写入本地文件
	//_, err = io.Copy(out, resp.Body)
	//if err != nil {
	//	return err
	//}

	writer := &ProgressWriter{
		Total:    resp.ContentLength,
		Response: w,
	}
	_, err = io.Copy(out, io.TeeReader(resp.Body, writer))
	if err != nil {
		fmt.Println("File write failed:", err)
		http.Error(w, "File write failed", http.StatusInternalServerError)
	}

	w.Write([]byte("File download completed"))

	return nil
}

type ProgressWriter struct {
	Total       int64
	Downloaded  int64
	CurrentStep int
	Response    http.ResponseWriter
}

func (pw *ProgressWriter) Write(p []byte) (int, error) {
	n := len(p)
	pw.Downloaded += int64(n)

	// 返回百分比
	//progress := float64(pw.Downloaded) / float64(pw.Total) * 100.0
	//
	//if int(progress) > pw.CurrentStep {
	//	pw.CurrentStep = int(progress)
	//	pw.Response.Write([]byte(fmt.Sprintf("Downloading: %.2f%%\n", progress)))
	//	pw.Response.(http.Flusher).Flush()
	//}

	progress := pw.Downloaded

	if int(progress) > pw.CurrentStep {
		pw.CurrentStep = int(progress)
		pw.Response.Write([]byte(fmt.Sprintf("%d/%d\n", progress, pw.Total)))
		pw.Response.(http.Flusher).Flush()
	}

	return n, nil
}

func main() {
	app := iris.New()

	// 定义路由和处理函数
	app.Get("/", func(ctx iris.Context) {
		go doSomething()
		ctx.HTML("<h1>Welcome to Iris</h1>")
	})

	app.Get("/hello", func(ctx iris.Context) {
		start := time.Now()
		currPath, _ := os.Getwd()
		name := ctx.URLParam("name")
		UpgradePKGPath := path.Join(currPath, "file.bot")
		fmt.Println("UpgradePKGPath: ", UpgradePKGPath)

		DownloadUrl := "https://tests%3D"

		err := DownloadFile(ctx, DownloadUrl, UpgradePKGPath)
		if err != nil {
			fmt.Println("Error:", err)
		}
		duration := time.Since(start)
		fmt.Println("执行时间:", duration)
		ctx.Writef("Hello, %s!", name)
	})

	// 启动应用程序
	app.Run(iris.Addr(":8080"))
}

效果:

 

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/783110.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

大数据学习04-Hbase分布式集群部署

系统环境&#xff1a;centos7 软件版本&#xff1a;jdk1.8、zookeeper3.4.8、hadoop2.8.5 一、下载 HBASE官网 cd /home/toolswget https://archive.apache.org/dist/hbase/2.2.4/hbase-2.2.4-bin.tar.gz二、解压 tar -zxvf hbase-2.2.4-bin.tar.gz -C /home/local/移动目…

【弹力设计篇】聊聊降级设计

我们知道在分布式系统中&#xff0c;故障是不可避免的&#xff0c;所以我们需要设计一个高可用的系统&#xff0c;对于接口层面除了幂等&重试机制&#xff0c;还需要保证接口高可用&#xff0c;因此 限流&排队&降级&熔断也需要考虑。本篇主要介绍下接口故障下降…

Qt 之 自定义json配置文件类,QJsonDocument应用

目录 一、前言 二、头文件代码 三、源文件代码 四、使用示例 五、使用效果 一、前言 Qt的配置类QSettings主要是键值结构的配置&#xff0c;若需要的配置项为树形结构&#xff0c;例如配置学校\学院\班级\学生这样&#xff0c;使用键值结构已经不满足我们的需求了&#xf…

【计算机视觉 | 图像分割】arxiv 计算机视觉关于图像分割的学术速递(7 月 21 日论文合集)

文章目录 一、分割|语义相关(14篇)1.1 CNOS: A Strong Baseline for CAD-based Novel Object Segmentation1.2 Spinal nerve segmentation method and dataset construction in endoscopic surgical scenarios1.3 WeakPolyp: You Only Look Bounding Box for Polyp Segmentatio…

【unity】模型裁剪shader(建筑生长动画)

【unity】模型裁剪shader&#xff08;建筑生长动画&#xff09; 思路 使用的核心方法是clip,当传入正值时渲染&#xff0c;传入负值时不渲染。定义一个裁剪向量&#xff0c;使用裁剪向量和模型点点乘&#xff0c;如果模型点和裁剪向量是同一个方向&#xff0c;点乘为正&#…

代码随想录算法训练营第58天|739 496

739 用stack来写 stack里面发index 不要放数值 重点在于 1.填写result数组不需要按顺序填写 根据index就可以 2.遍历的值比top小的话就放入stack 这样stack里面是一个递减数组 遍历的值只需和top比 如果比他大就pop 一直到把stack里面比新加入的值小的都pop完为止 这样stack里…

vue项目的vue.config.js在打包过程中,并不会处理api请求。

主要处理打包选项和静态资源文件 请求是axios处理的

nonebot2聊天机器人插件12:stable_diffusion_webui_api

nonebot2聊天机器人插件12&#xff1a;stable_diffusion_webui_api 1. 插件用途2. 代码实现3. 实际效果 该插件涉及知识点&#xff1a;定时器&#xff0c;调用bot的api发送消息 插件合集&#xff1a;nonebot2聊天机器人插件 该系列为用于QQ群聊天机器人的nonebot2相关插件&…

IPO向上,大模型向下:中国企服寻找新「出口」

2023年&#xff0c;资本市场给企服行业带来的动荡&#xff0c;无疑是一次洗牌机会。只有当SaaS企业深耕产业侧&#xff0c;才能找到实现标准化的解法&#xff0c;才能在一波又一波的浪潮下抓住机遇。 作者|思杭 编辑|皮爷 出品|产业家 2023上半年&#xff0c;企服行业在…

MySQL存储过程——系统变量

1.存储过程中的变量 1.1 查看系统变量 查看所有的系统变量 show variables;查看会话级别的系统变量 show session variables&#xff1b;查看会话和auto相关的变量 show session variables like auto%;查看全局的和auto相关变量 show global variables like auto%;查看某一…

js的几种排序

冒泡排序&#xff1a; function bubbleSort(arr) {var len arr.length;for (var i 0; i < len; i) {for (var j 0; j < len - 1 - i; j) {if (arr[j] > arr[j1]) { //相邻元素两两对比var temp arr[j1]; //元素交换arr[j1] arr[j];arr[j] temp;}}…

进程(process)与线程(thread)以及线程的三种实现方法

一、线程和进程区别 说起进程&#xff0c;就不得不说下程序。程序是指令和数据的集合&#xff0c;其本身没有任何运行的含义&#xff0c;是一个静态的概念。 而进程则是执行程序的一次执行过程&#xff0c;它是一个动态的概念。是系统资源分配的单位。 通常在一个进程中可以…

BGP对SR-MPLS的支持

目录 BGP的SID类型 BGP Prefix-SID BGP Anycast-SID BGP Peer-SID BGP SID的通告 通过Prefix-SID属性 通告Prefix-SID 通过BGP EPE 通告Peer-SID 为什么要使用BGP作为SR-MPLS的控制平面 IGP for SR-MPSL只可以在自治系统AS内分配SID&#xff0c;规划出AS域内的最优路径 …

第12章 STM32+BH1750光照传感器+OLED模块显示环境光照强度

今天给大家介绍一块嵌入式毕设中也经常用到的一款传感器——BH1750光照传感器&#xff0c;如下图。&#xff08;该传感器的购买链接和代码我已放在资料里&#xff0c;想要资料的同学&#xff0c;评论区留下邮箱即可&#xff09;相比光敏传感器&#xff0c;它可以直接输出环境光…

【Unity2D】设置一物体默认在其他物体之上不被遮挡

比如我想让机器人显示在箱子的前面。 点击箱子&#xff0c;将其层级设置在机器人的后面。 即修改箱子的Order in Layer 在机器人之后 物体默认的Order in Layer 都是0 &#xff0c;将箱子的Order in Layer修改为-1即可 这样将确保先绘制机器人&#xff0c;然后绘制箱子。这样…

ConstraintLayout(约束布局)替代LinearLayout权重,解决多View一行省略问题

1.看上面的设计图中圈红的地方&#xff1a;左边设计图是一张直播间消息流&#xff0c;其中标红的消息流意思是&#xff1a; 用户的等级标签&#xff08;一张图片&#xff09; 用户名字写死的文案send,要求这三个View写一行&#xff0c;但是当用户名字过长时会让用户名出现.....…

数据结构双向循环链表,增删改查基本操作

一、双向循环链表的描述 和单链表的循环类似&#xff0c;双向链表也可以有循环表&#xff0c;循环表的引进是为了弥补双向链表不能向前遍历的弊端。 在双向循环链表中&#xff0c;头结点的直接前驱为尾结点&#xff0c;而尾结点的直接后继为头结点。 二、双向循环链表的存储结…

黑马B站视频JAVA部分的知识与学习-【思维导图知识范围】

JAVA本系列黑马的JAVA学习路线–详解JAVA部分的学习语言视频选择收录专辑链接C张雪峰推荐选择了计算机专业之后-在大学期间卷起来-【大学生活篇】JAVA黑马B站视频JAVA部分的知识范围、学习步骤详解JAVAWEB黑马B站视频JAVAWEB部分的知识范围、学习步骤详解SpringBootSpringBoot知…

如何在 Outlook 中将导航栏从左侧移动到底部?

对于最新版本的 Outlook 的用户&#xff0c;这是 Microsoft 365 MSO&#xff08;版本 2211 内部版本 16.0.15831.20098&#xff09;&#xff0c;您可以更快地完成工作。 Office 刚刚在最新版本的 Outlook 365 中添加了一个选项。由于这次更新&#xff0c;您只需单击几下即可将导…

RT-Thread快速入门-定时器管理

1时钟节拍 任何操作系统都需要提供一个时钟节拍&#xff0c;以供系统处理所有和时间有关的事件&#xff0c;如延时、线程的时间片轮转调度以及定时器超时等。时钟节拍&#xff08;OS Tick&#xff09;是操作系统中最小的时间单位。 时钟节拍是特定的周期性中断&#xff0c;这…