Golang – embed.FS传递参数

huangapple
go评论107阅读模式
英文:

Golang - embed.FS pass argument

问题

我需要修改代码片段，需要使用参数而不是在go:embed中使用硬编码的路径。

我的代码：

  1. package main
  3. import (
  4.     "embed"
  5.     "log"
  6.     
  7.     "github.com/golang-migrate/migrate/v4/source/iofs"
  8. )
  10. // 我需要将`migrations/*.sql`的路径作为参数传递给`migrate`函数，而不是使用硬编码的`migrations/*.sql`
  11. //go:embed migrations/*.sql
  12. var fs embed.FS
  14. func main() {
  15.     const path = "migrations/*.sql"
  16.     if err := migrate(path); err != nil {
  17.         log.Fatal(err)
  18.     }
  19.     log.Println("done")
  20. }
  22. func migrate(path string) error {
  23.     d, err := iofs.New(fs, "migrations")
  24. 	if err != nil {
  25. 		return err
  26. 	}
  27.     
  28.     // 其余的代码...
  29.     return nil
  30. }

我已经尝试使用os.DirFS，但是这个函数的结果只包含dir参数的数据，而不包含migrations/*.sql文件夹中的实际文件。

英文:

I need to modify fragment of code which needs to take argument instead of use hardcoded path in go:embed

My code:

  1. package main
  3. import (
  4.     "embed"
  5.     "log"
  6.     
  7.     "github.com/golang-migrate/migrate/v4/source/iofs"
  8. )
  10. // Instead of use hardcoded `migrations/*.sql` I need to pass 
  11. // `path` to `migrations/*.sql` as a argument to `migrate` function
  12. //go:embed migrations/*.sql
  13. var fs embed.FS
  15. func main() {
  16.     const path = "migrations/*.sql"
  17.     if err := migrate(path); err != nil {
  18.         log.Fatal(err)
  19.     }
  20.     log.Println("done")
  21. }
  23. func migrate(path string) error {
  24.     d, err := iofs.New(fs, "migrations")
  25. 	if err != nil {
  26. 		return err
  27. 	}
  28.     
  29.     // rest of the code...
  30.     return nil
  31. }

I already tried to use os.DirFS however result of this function only contains data with dir parameter without actual files from folder migrations/*.sql

答案1

得分: 1

我不完全确定我理解了问题。您创建了一个包含多个文件的嵌入式文件系统（包含migrations目录中匹配*.sql的所有文件）。您可以通过名称从这个嵌入式文件系统中加载文件。

假设您的migrations目录如下所示：

  1. migrations/
  2. ├── bar.sql
  3. └── foo.sql

其中foo.sql包含this is foo.sql，bar.sql类似。

如果我们编写以下代码：

  1. func main() {
  2.   const path = "migrations/foo.sql"
  3.   if err := migrate(path); err != nil {
  4.     log.Fatal(err)
  5.   }
  6.   log.Println("done")
  7. }
  9. func migrate(path string) error {
  10.   data, err := fs.ReadFile(path)
  11.   if err != nil {
  12.     panic(err)
  13.   }
  15.   fmt.Println(string(data))
  17.   // 其余的代码...
  18.   return nil
  19. }

您将得到输出：

  1. this is foo.sql

如果您想遍历嵌入式文件系统中的文件，可以使用ReadDir方法：

  1. func main() {
  2.   const path = "migrations"
  3.   if err := migrate(path); err != nil {
  4.     log.Fatal(err)
  5.   }
  6.   log.Println("done")
  7. }
  9. func migrate(path string) error {
  10.   items, err := fs.ReadDir(path)
  11.   if err != nil {
  12.     panic(err)
  13.   }
  15.   for _, ent := range items {
  16.     fmt.Println(ent.Name())
  17.   }
  19.   // 其余的代码...
  20.   return nil
  21. }

这将打印：

  1. bar.sql
  2. foo.sql

这对您有帮助吗？

英文:

I'm not entirely sure I understand the question. You've created an embedded filesystem that contains multiple files (everything matching *.sql in the migrations directory). You can load files from this embedded filesystem by name.

Let's assume that your migrations directory looks like:

  1. migrations/
  2. ├── bar.sql
  3. └── foo.sql

Where foo.sql contains this is foo.sql, and similarly for bar.sql.

If we write code like this:

  1. func main() {
  2.   const path = "migrations/foo.sql"
  3.   if err := migrate(path); err != nil {
  4.     log.Fatal(err)
  5.   }
  6.   log.Println("done")
  7. }
  9. func migrate(path string) error {
  10.   data, err := fs.ReadFile(path)
  11.   if err != nil {
  12.     panic(err)
  13.   }
  15.   fmt.Println(string(data))
  17.   // rest of the code...
  18.   return nil
  19. }

You'd get the output:

  1. this is foo.sql

If you want to iterate over files in your embedded filesystem, you can use the ReadDir method:

  1. func main() {
  2.   const path = "migrations"
  3.   if err := migrate(path); err != nil {
  4.     log.Fatal(err)
  5.   }
  6.   log.Println("done")
  7. }
  9. func migrate(path string) error {
  10.   items, err := fs.ReadDir(path)
  11.   if err != nil {
  12.     panic(err)
  13.   }
  15.   for _, ent := range items {
  16.     fmt.Println(ent.Name())
  17.   }
  19.   // rest of the code...
  20.   return nil
  21. }

This would print:

  1. bar.sql
  2. foo.sql

Does that help?

huangapple
  • 本文由 发表于 2023年1月8日 21:02:35
  • 转载请务必保留本文链接：https://go.coder-hub.com/75047973.html
  • go
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定