前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >golang连接MySQL并导入执行SQL文件

golang连接MySQL并导入执行SQL文件

作者头像
唯一Chat
发布2022-11-16 16:43:27
2.9K0
发布2022-11-16 16:43:27
举报
文章被收录于专栏:陶士涵的菜地陶士涵的菜地

实现golang连接MySQL数据库导入执行一个sql文件

package tools

import (
    "fmt"
    _ "github.com/go-sql-driver/mysql"
    "github.com/jinzhu/gorm"
    "io/ioutil"
    "log"
    "os"
    "strings"
    "time"
)

type ImportSqlTool struct {
    SqlPath                                    string
    Username, Password, Server, Port, Database string
}

func (this *ImportSqlTool) ImportSql() error {
    _, err := os.Stat(this.SqlPath)
    if os.IsNotExist(err) {
        log.Println("数据库SQL文件不存在:", err)
        return err
    }

    dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", this.Username, this.Password, this.Server, this.Port, this.Database)
    db, err := gorm.Open("mysql", dsn)
    if err != nil {
        log.Println("数据库连接失败:", err)
        //panic("数据库连接失败!")
        return err
    }
    db.SingularTable(true)
    db.LogMode(true)
    db.DB().SetMaxIdleConns(10)
    db.DB().SetMaxOpenConns(100)
    db.DB().SetConnMaxLifetime(59 * time.Second)

    sqls, _ := ioutil.ReadFile(this.SqlPath)
    sqlArr := strings.Split(string(sqls), ";")
    for _, sql := range sqlArr {
        sql = strings.TrimSpace(sql)
        if sql == "" {
            continue
        }
        err := db.Exec(sql).Error
        if err != nil {
            log.Println("数据库导入失败:" + err.Error())
            return err
        } else {
            log.Println(sql, "\t success!")
        }
    }
    return nil
}

使用方法

    tool:=&ImportSqlTool{
        SqlPath:  "../import.sql",
        Username: "go-fly",
        Password: "go-fly",
        Server:   "127.0.0.1",
        Port:     "3306",
        Database: "go-fly",
    }
    tool.ImportSql()
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2022-11-09,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
云数据库 MySQL
腾讯云数据库 MySQL(TencentDB for MySQL)为用户提供安全可靠,性能卓越、易于维护的企业级云数据库服务。其具备6大企业级特性,包括企业级定制内核、企业级高可用、企业级高可靠、企业级安全、企业级扩展以及企业级智能运维。通过使用腾讯云数据库 MySQL,可实现分钟级别的数据库部署、弹性扩展以及全自动化的运维管理,不仅经济实惠,而且稳定可靠,易于运维。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档