golang自动导入postgresql脚本

直接代码

package main

import (
    "fmt"
    "golang-objective-go/dataFoundation/dataConvert"
    "io/ioutil"
    "os/exec"
    "strings"
)

func main() {
    dir_list, e := ioutil.ReadDir("./")
    if e != nil {
        fmt.Println("read dir error")
        return
    }

    for _, v := range dir_list {
        fileName := v.Name()

        if strings.Contains(fileName, ".sql") {

            c := exec.Command("psql", "-d", "xxxx", "-f", fileName)
            d, err := c.CombinedOutput()
            info := dataConvert.BytesToString(d)
            if nil != err {
                fmt.Println("失败" + err.Error())
            } else {
                fmt.Println("成功" + info)
            }
        }
    }
}

红色的要注意,不能采用fmt.SPrintf组合字符串,不然会出现死活找不到文件的问题,我在这里浪费了3个小时的时间。

原文地址:https://www.cnblogs.com/ziyouchutuwenwu/p/4740216.html