登录
首页 >  Golang >  Go问答

如何在 Go 二进制文件中捆绑 SQLite 数据库?

来源:Golang技术栈

时间:2023-04-15 11:53:28 441浏览 收藏

学习Golang要努力,但是不要急!今天的这篇文章《如何在 Go 二进制文件中捆绑 SQLite 数据库?》将会介绍到golang等等知识点,如果你想深入学习Golang,可以关注我!我会持续更新相关文章的,希望对大家都能有所帮助!

问题内容

我尝试使用 go-bindata 和 packr,但这些包没有显示如何将 SQLite 数据库文件打包到二进制文件中。

我不需要以任何方式更新数据库,我只想在启动时从中读取数据。

如何在 Go 二进制文件中嵌入 SQLite 数据库文件?

正确答案

SQLite 驱动程序不能从内存中读取数据库文件(例如从字节片)。但是您可以将数据写入临时文件,然后打开它:

//go:generate go run gen.go

package main

import (
    "database/sql"
    "fmt"
    "io/ioutil"
    "log"
    "os"

    _ "github.com/mattn/go-sqlite3"
)

func main() {
    // Create temporary file for database.
    tmpDB, err := ioutil.TempFile("", "db*.sqlite3")
    if err != nil {
        log.Fatal(err)
    }
    // Remove this file after on exit.
    defer func() {
        err := os.Remove(tmpDB.Name())
        if err != nil {
            log.Print(err)
        }
    }()

    // Write database to file.
    _, err = tmpDB.Write(sqlDB)
    if err != nil {
        log.Print(err)
    }
    err = tmpDB.Close()
    if err != nil {
        log.Print(err)
    }

    // Open DB.
    db, err := sql.Open("sqlite3", tmpDB.Name()+"?mode=ro")
    if err != nil {
        log.Fatal(err)
    }

    // Make sure it's loaded correct.
    rows, err := db.Query("select * from test")
    if err != nil {
        log.Fatal(err)
    }

    for rows.Next() {
        var c string
        err := rows.Scan(&c)
        if err != nil {
            log.Fatal(err)
        }
        fmt.Println(c)
    }
}

您可以使用以下内容将数据库写入db.go

// +build generate

package main

import (
    "fmt"
    "io/ioutil"
    "log"
    "os"
    "strings"
)

func main() {
    // Read source database file.
    d, err := ioutil.ReadFile("source.sqlite3")
    if err != nil {
        log.Fatal(err)
    }

    fp, err := os.Create("db.go")
    if err != nil {
        log.Fatal(err)
    }

    _, err = fmt.Fprintf(fp, "// Code generated by gen.go; DO NOT EDIT.\n\n"+
        "package main\n\n"+
        "var sqlDB = %s\n", asbyte(d))
    if err != nil {
        log.Fatal(err)
    }
}

// Write any data as byte array.
func asbyte(s []byte) string {
    var b strings.Builder
    for i, c := range s {
        if i%19 == 0 {
            b.WriteString("\n\t\t")
        }
        b.WriteString(fmt.Sprintf("%#x, ", c))
    }
    return "[]byte{" + b.String() + "}"
}

如果您愿意,您也可以使用 go-bindata 或 packr,但我并没有真正看到优势。


另一种方法是使用内存数据库,这 可能 会更快,具体取决于您想要做什么。

  1. 将你想要的 SQL 模式和行作为字符串嵌入到你的 Go 二进制文件中。
  2. 当你的程序启动时打开一个新的内存数据库(sql.Open("sqlite3",:memory:`)并创建模式并插入行。

此方法没有磁盘访问,因此查询它可能会更快一些,但启动时间较慢(肯定是基准!)

今天关于《如何在 Go 二进制文件中捆绑 SQLite 数据库?》的内容介绍就到此结束,如果有什么疑问或者建议,可以在golang学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!

声明:本文转载于:Golang技术栈 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>
评论列表