golang实现文件上传并转存数据库功能详解

共计 1632 个字符，预计需要花费 5 分钟才能阅读完成。

本文实例为大家分享了 golang 实现文件上传并转存数据库的具体代码，供大家参考，具体内容如下

上传图片，且可选择将图片保存到数据中。

获取文件，并返回 base64string 流

代码如下（示例）：

func setIconPost(c *gin.Context)  {
    // 获取文件，icon 实现对上传文件的访问，header 是对上传文件信息的标记
    icon,header,err :=c.Request.FormFile("file")
    dangerous(err)
    defer icon.Close()
    //path.Ext 是取后缀，Tolower 小写
    ext := strings.ToLower(path.Ext(header.Filename))
    if header.Size>1024*1024*2{fmt.Println("文件过大")
    }
    buf := bytes.NewBuffer(nil)
    // 读取 icon 的数据存入 buf 中
    if _,err := io.Copy(buf,icon);err != nil{return}
    // 将 base64 返回前端
    renderData(c, gin.H{"base64":base64.StdEncoding.EncodeToString(buf.Bytes()),
        "icon-ext":ext,
    },nil)
}

代码如下（示例）：

func setEntPost(c *gin.Context)  {
    var f Identical
    bind(c,&f)
    models.EtpSave(f.Copyright,"copyright")
    models.EtpSave(f.Introduction,"introduction")
    models.EtpSave(f.Icon,"icon")
    models.EtpSave(f.Logo,"logo")
    models.EtpSave(f.Version,"version")
    models.EtpSave(f.Belong,"belong")
    renderMessage(c,nil)
}

type Etp struct {
    Id    int         `json:"id"`
    Ckey  string     `json:"ckey"`
    Cval  string    `json:"cval"`
    Kind  int         `json:"kind"`
}
func EtpSave(cval, ckey string) error {
    var obj Etp
    // 数据库是否存在
    has,err := DB["rdb"].Table("configs").Where("ckey=?",ckey).Get(&obj)
    if err != nil{return err}
    // 不存在
    if !has {_, err = DB["rdb"].Table("configs").Where("ckey=?",ckey).Insert(Etp{
            Ckey: ckey,
            Cval: cval,
            Kind: 1,
        })
    }else{
        obj.Cval = cval
        DB["rdb"].Table("configs").Where("ckey=?",ckey).Cols("cval").Update(&obj)
    }
    return err
}

若上传图片稍大，在转存 MySQL 时会报错。Data too long for column ‘……’ at row 1

解决方法

将数据库字段格式设置为 longtext

本需求主要难点在于对于 golang 核心库方法的掌握，包括上传文件，[]bytes 和 string 之间的转换。

整体框架：

一、接收文件 / 图片接口

二、修改 / 保存图片的接口

三、保存数据的方法

阿里云 2 核 2G 服务器 3M 带宽 61 元 1 年，有高配

腾讯云新客低至 82 元 / 年，老客户 99 元 / 年

代金券：在阿里云专用满减优惠券