douhengdao4499 2018-03-28 22:24
浏览 1993
已采纳

如何在Golang中使用命令执行SQL文件

everyone, I have some troubles with postgresql in golang. I have an sql file (database.sql), and before starting my server I want to exec some command, it look's like that

CREATE TABLE forums (
  id BIGSERIAL primary key,
  slug TEXT NOT NULL UNIQUE,
  title TEXT,
  author TEXT references users(login),
  threads BIGINT DEFAULT 0,
  posts BIGINT DEFAULT 0
);

I know, that I should use db.Exec(request), but i have many requests ("CREATE TABLE user" and others...)

I have no idea, how to do that

Help, and thanks a lot!

  • 写回答

1条回答 默认 最新

  • doushuangdui5419 2018-03-29 07:34
    关注

    You can read the file's contents into a string and pass that to Exec.

    query, err := ioutil.ReadFile("path/to/database.sql")
    if err != nil {
        panic(err)
    }
    if _, err := db.Exec(query); err != nil {
        panic(err)
    }
    

    If your database.sql is formated in such a way, or it contains queries that break db.Exec, then you could use os/exec together with psql (if it's installed on the machine on which the code is being run).

    cmd := exec.Command("psql", "-d", "database_name", "-f", "path/to/database.sql")
    stderr, err := cmd.StderrPipe()
    if err != nil {
        panic(err)
    }
    
    if err := cmd.Start(); err != nil {
        panic(err)
    }
    
    errout, _ := ioutil.ReadAll(stderr)
    if err := cmd.Wait(); err != nil {
        fmt.Println(errout)
        panic(err)
    }
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 修改一下代码,考虑进程到达时间不同的情况
  • ¥15 华为nova10pro ,关闭热动热点用量记录
  • ¥15 JavaFX启动SceneBuilder报错,找不到fxml文件
  • ¥15 树莓派ubuntu使用conda就提示core dumped
  • ¥15 帮我看看这是个啥题,带解题过程和结果,条件如下FCF = 290471.33 g1 = 15% r = 8% g2 = 4% n = 5
  • ¥15 edem模拟颗粒不显示或者生成失败
  • ¥15 Python代码编写
  • ¥15 php 将rtmp协议转hls协议,无法播放
  • ¥20 python代码编写
  • ¥20 使用MPI广播数据遇到阻塞