golang官方嵌入文件到可执行程序的示例详解-巨人网络通讯

主页 > 知识库 > golang官方嵌入文件到可执行程序的示例详解

golang官方嵌入文件到可执行程序的示例详解

前言

在go官方出嵌入文件的方法前我在网上找过，并且自己还研究过，虽然没有问题，但是既然官方支持还是用起来吧。
看了下go源码embed/embed.go很简单，并且看embed/internal/embedtest/embed_test.go就知道如何使用。

示例程序

嵌入文件直接访问

原理是使用go:embed标签来完成。下面是直接读取文件内容，需要注意下面几点。
文件不是utf8编码时，输出内容为中文会乱码。
测试过嵌入文件只能为源码文件同级目录和子目录下的文件，试过其他目录的绝对路径或相对路径会报错。
我测试过几个能想到的场景，有些会报错，所以在使用时需要注意。

package main
 
import (
  _ "embed"
)
 
//go:embed test.txt
var testString string // 当前目录，解析为string类型
 
//go:embed test.txt
var testByte []byte // 当前目录，解析为[]byte类型
 
//go:embed test/test.txt
var testAbsolutePath string // 子目录，解析为string类型
 
//go:embed notExistsFile
var testErr0 string // 文件不存在，编译报错：pattern notExistsFile: no matching files found
 
//go:embed dir
var testErr1 string // dir是目录，编译报错：pattern dir: cannot embed directory dir: contains no embeddable files
 
//go:embed ../test.txt
var testErr2 string // 相对路径，不是当前目录或子目录，编译报错：pattern ../test.txt: invalid pattern syntax
 
//go:embed D:\test.txt
var testErr3 string // 绝对路径，编译报错：pattern D:\test.txt: no matching files found
 
func main() {
  println(testString)
  println(string(testByte))
  println(testAbsolutePath)
}

嵌入文件列表

package main
 
import (
  "embed"
  "io"
  "os"
)
 
//go:embed test0.txt test1.txt test1*.txt
//go:embed test/test0.txt test/test1.txt
//go:embed test0
var fileList embed.FS
/*
使用上述方法可以将多个文件或目录添加到fileList中。
1. 添加多个文件，且支持"*"号通配文件。
2. 支持子目录文件。
3. 支持嵌入一个目录。
*/
 
func main() {
  testDir, err := fileList.ReadDir("test0")
  if err != nil {
    panic(err)
  }
  for _, v := range testDir {
    println(v.Name()) // 打印嵌入的目录内容
  }
 
  // 使用fileList.Open可以生成一个对象，可以通过文件流那样读出来
  testFile, err := fileList.Open("test0.txt")
  if err != nil {
    panic(err)
  }
  io.Copy(os.Stdout, testFile)
 
  testFile, err = fileList.Open("test112.txt")
  if err != nil {
    panic(err)
  }
  io.Copy(os.Stdout, testFile)
 
  testFile, err = fileList.Open("test/test1.txt")
  if err != nil {
    panic(err)
  }
  io.Copy(os.Stdout, testFile)
 
  // 直接将文件内容读出来
  data, err := fileList.ReadFile("test111.txt")
  if err != nil {
    panic(err)
  }
  println(string(data))
}