在 Go 中获取文件的内容类型时遇到问题

目前golang学习网上已经有很多关于Golang的文章了，自己在初次阅读这些文章中，也见识到了很多学习思路；那么本文《在 Go 中获取文件的内容类型时遇到问题》，也希望能帮助到大家，如果阅读完后真的对你学习Golang有帮助，欢迎动动手指，评论留言并分享~

我有一个函数，可以接收 base64 字符串并获取其内容（pdf 或 jpeg）。 我读入 base64 内容，将其转换为字节并将其解码为文件本身。 然后，我创建一个文件，在其中输出解码的文件（jpeg 或 pdf）。 然后我将字节写入其中。 然后我对其调用 getfilecontenttype，它返回一个空字符串。 如果我单独运行这些函数，就像我首先使用第一个函数来创建解码文件，然后结束它。然后调用第二个函数来获取内容类型，它会工作并将其作为 jpeg 或 pdf 返回。 我在这里做错了什么？ 有没有更好的方法来做到这一点？

func ConvertToJPEGBase64(
    src string,
    dst string,
) error {

    b, err := ioutil.ReadFile(src)
    if err != nil {
        return err
    }
    str := string(b)

    byteArray, err := base64.StdEncoding.DecodeString(str)
    if err != nil {
        return err
    }

    f, err := os.Create(dst)
    if err != nil {
        return err
    }

    if _, err := f.Write(byteArray); err != nil {
        return err
    }

    f.Sync()
    filetype, err := client.GetFileContentType(f)
    if err != nil {
        return err
    }
    if strings.Contains(filetype, "jpeg") {
             // do something
    } else {
            // do something else
    }

    return nil
}

// GetFileContentType tells us the type of file
func GetFileContentType(out *os.File) (string, error) {

    // Only the first 512 bytes are used to sniff the content type.
    buffer := make([]byte, 512)
    _, err := out.Read(buffer)
    if err != nil {
        return "", err
    }
    contentType := http.DetectContentType(buffer)

    return contentType, nil
}

解决方案

问题在于 getfilecontenttype 从文件末尾读取。修复此问题，在调用 getfilecontenttype 之前返回文件开头：

if _, err := f.seek(io.seekstart, 0); err != nil {
     return err
 }

更好的解决方法是使用内存中已有的文件数据。这将代码简化到不需要 getfilecontenttype 函数的程度。

func converttojpegbase64(
    src string,
    dst string,
) error {

    b, err := ioutil.readfile(src)
    if err != nil {
        return err
    }
    str := string(b)

    bytearray, err := base64.stdencoding.decodestring(str)
    if err != nil {
        return err
    }

    f, err := os.create(dst)
    if err != nil {
        return err
    }
    defer f.close() // <-- close the file on return.

    if _, err := f.write(bytearray); err != nil {
        return err
    }

    filetype := http.detectcontenttype(bytearray) // <-- use data in memory
    if strings.contains(filetype, "jpeg") {
             // do something
    } else {
            // do something else
    }

    return nil
}

使用 ioutil.writefile 可以消除更多代码：

func ConvertToJPEGBase64(src, dst string) error {
    b, err := ioutil.ReadFile(src)
    if err != nil {
        return err
    }

    byteArray, err := base64.StdEncoding.DecodeString(string(b))
    if err != nil {
        return err
    }

    if err := ioutil.WriteFile(dst, byteArray, 0666); err != nil {
        return err
    }

    fileType := http.DetectContentType(byteArray)
    if strings.Contains(fileType, "jpeg") {
        // do something
    } else {
        // do something else
    }

    return nil
}

理论要掌握，实操不能落！以上关于《在 Go 中获取文件的内容类型时遇到问题》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！