Kratos: http上传、下载和预览图片

在Web开发中,经常需要对图片上传、下载或预览。Kratos一套轻量级Go微服务框架,本文将介绍在Kratos中要如何实现。

接着我们将以Kratos初始化的项目(kratos new helloworld -r https://gitee.com/go-kratos/kratos-layout.git)为基点,进行修改,详细如下:

  1. 通过http请求form-data的方式,上传图片并保存
  2. 通过http请求下载、预览图片

1. 修改Service:上传、下载和预览

1.1 上传:

go 复制代码
package service

import (
    "context"
    transporthttp "github.com/go-kratos/kratos/v2/transport/http"
    v1 "helloworld/api/helloworld/v1"
    "helloworld/internal/biz"
    "io"
    "os"
) 
// 上传
func (s *GreeterService) Upload(ctx transporthttp.Context) (err error) {
    file, header, err := ctx.Request().FormFile("file")
    if err != nil {
       return
    }
    defer file.Close()

    // 修改文件名并创建保存图片
    imageFile, err := os.Create("./band_" + header.Filename) 
    if err != nil {
       return
    }
    defer imageFile.Close()

    // 将文件内容复制到保存的文件中
    _, err = io.Copy(imageFile, file)
    if err != nil {
       return
    }
    return nil
}

上例中,我们通过ctx.Request().FormFile 获取文件,通过os.Create将文件的名称修改为band_ + w文件名,最后io.Copy将文件内容复制到保存的文件中。

1.2 下载:

go 复制代码
func (s *GreeterService) Download(ctx transporthttp.Context) error {
    imageFile, err := os.Open("./go.png")
    if err != nil {
       return err
    }
    defer imageFile.Close()
    payload, err := io.ReadAll(imageFile)
    if err != nil {
       return err
    } 
    // 设置HTTP响应头
    // 打开为预览
    //ctx.Response().Header().Set("Content-Type", "image/png")
    // 打开为下载
    ctx.Response().Header().Set("Content-Type", "application/octet-stream")
    ctx.Response().Header().Set("Content-Disposition", "attachment; filename=go2.png") 
    // 将结果写入
    _, err = ctx.Response().Write(payload)
    return err
}

上例中,我们通过设置了Header的不同Content-Type来分别实现是预览或者下载。 其中

  • application/octet-stream,通知浏览器以二进制流的形式下载文件。

  • Content-Disposition,其中attachment表示将文件作为附件下载,filename=download.png指定了下载文件的名称。

2. 注册路由

go 复制代码
package server

import (
    v1 "helloworld/api/helloworld/v1"
    "helloworld/internal/conf"
    "helloworld/internal/service"

    "github.com/go-kratos/kratos/v2/log"
    "github.com/go-kratos/kratos/v2/middleware/recovery"
    "github.com/go-kratos/kratos/v2/transport/http"
)

// NewHTTPServer new a HTTP server.
func NewHTTPServer(c *conf.Server, greeter *service.GreeterService, logger log.Logger) *http.Server {
    var opts = []http.ServerOption{
       http.Middleware(
          recovery.Recovery(),
       ),
    }
    if c.Http.Network != "" {
       opts = append(opts, http.Network(c.Http.Network))
    }
    if c.Http.Addr != "" {
       opts = append(opts, http.Address(c.Http.Addr))
    }
    if c.Http.Timeout != nil {
       opts = append(opts, http.Timeout(c.Http.Timeout.AsDuration()))
    }
    srv := http.NewServer(opts...)
    v1.RegisterGreeterHTTPServer(srv, greeter)

    //路由注册
    route := srv.Route("/")
    //图片上传
    route.POST("/upload", greeter.Upload)
    //图片下载
    route.GET("/download", greeter.Download)

    return srv
} 

测试1. 上传

  1. 启动Kratos HelloWorld Http 服务
js 复制代码
> msg=[HTTP] server listening on: [::]:8001
  1. 我们通过http服务localhost:8001, 路由/upload,上传本地图片heibao.jpg,curl如下
cmd 复制代码
curl --location 'http://localhost:8001/upload?file=null' \
--form 'file=@"/D:/heibao.jpg"'

在工作目录下,多出了一个band_heibao.jpg的文件,说明上传并图片成功。尝试打开图片

正常显示。

测试2: 下载、预览

在GeeteeingService.Download的代码,通过修改http.Header设置来实现预览

go 复制代码
ctx.Response().Header().Set("Content-Type", "image/png") 

打开浏览器,输入http://localhost:8001/download,显示如下:

如果修改为

go 复制代码
//ctx.Response().Header().Set("Content-Type", "image/png")
ctx.Response().Header().Set("Content-Type", "application/octet-stream")
ctx.Response().Header().Set("Content-Disposition", "attachment; filename=go2.png")

则为浏览器下载

参考

相关推荐
煎鱼eddycjy11 小时前
新提案:由迭代器启发的 Go 错误函数处理
go
煎鱼eddycjy11 小时前
Go 语言十五周年!权力交接、回顾与展望
go
不爱说话郭德纲1 天前
聚焦 Go 语言框架,探索创新实践过程
go·编程语言
ZHOU_WUYI2 天前
React 中使用 Axios 进行 HTTP 请求
javascript·react.js·http
0x派大星2 天前
【Golang】——Gin 框架中的 API 请求处理与 JSON 数据绑定
开发语言·后端·golang·go·json·gin
可涵不会debug2 天前
【Linux|计算机网络】HTTPS工作原理与安全机制详解
linux·网络协议·http·网络安全·https
lu云之东2 天前
Harbor2.11.1生成自签证和配置HTTPS访问
网络协议·http·docker·https·harbor
helloWorldZMY2 天前
超文本传输协议(HTTP)与超文本传输安全协议(HTTPS)
网络协议·计算机网络·http·https
IT书架2 天前
golang高频面试真题
面试·go