我开源了Go语言连接数据库和一键生成结构体的包【实用】

项目地址https://gitee.com/zht639/my_gopkg

autosql

autosql 是一个简化数据库使用的模块,支持常见的数据库(MySQL、PostgreSQL、SQLite、SQL Server)。该模块不仅提供了数据库连接函数,还能自动生成数据表对应的结构体,极大简化了数据库的初始化操作。

一、使用方法

打开项目根目录,在该目录下打开命令行,然后按如下步骤操作。

(1)初始化mod(有则跳过此步)

bash 复制代码
go mod init 你的模块名

(2)下载包的最新版本(第一次下载可省略@master)

bash 复制代码
go get gitee.com/zht639/my_gopkg@master

(3)导包

go 复制代码
import "gitee.com/zht639/my_gopkg/autosql"

二、功能概述

  • 连接函数:

    • ConnMysql(username, password, host, port, dbname string): 连接 MySQL 数据库。
    • ConnPostgres(username, password, host, port, dbname string): 连接 PostgreSQL 数据库。
    • ConnSQLite(filepath string): 连接 SQLite 数据库。
    • ConnSQLServer(username, password, host, port, dbname string): 连接 SQL Server 数据库。
  • 结构体生成:

    • GenStruct(db *gorm.DB, tableName string): 生成数据库中指定表的结构体。
    • GenAllStructs(db *gorm.DB): 生成数据库中所有表的结构体。

    使用生成结构体功能可以将数据库中的表映射为 Go 语言的结构体,极大地提高了开发效率和代码的可维护性。

三、使用示例

以mysql为例,连接数据库并生成表的结构体:

go 复制代码
package main

import (
	"fmt"

	"gitee.com/zht639/my_gopkg/autosql"
)

func main() {
	db, err := autosql.ConnMysql("root", "123456", "localhost", "3306", "db01")
	if err != nil {
		fmt.Println(err)
		return
	}
	// 生成数据库中所有表的结构体
	autosql.GenAllStructs(db)
	//或者生成指定表对应的结构体
	//autosql.GenStruct(db, "student")
}

该函数将会在当前目录下生成一个model包,包含数据库中所有表的结构体,

目录结构如下:

bash 复制代码
├─model
│    grade.gen.go
│    student.gen.go
│    user.gen.go

打开其中一个文件,它不仅包含了结构体定义,还包含了json绑定字符串和返回表名的方法:

go 复制代码
// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.

package model

import (
	"time"
)

const TableNameStudent = "student"

// Student mapped from table <student>
type Student struct {
	Stuid int32     `gorm:"column:stuid;primaryKey;autoIncrement:true" json:"stuid"`
	Stuno string    `gorm:"column:stuno" json:"stuno"`
	Name  string    `gorm:"column:name" json:"name"`
	Sex   string    `gorm:"column:sex" json:"sex"`
	Age   int32     `gorm:"column:age" json:"age"`
	Birth time.Time `gorm:"column:birth" json:"birth"`
}

// TableName Student's table name
func (*Student) TableName() string {
	return TableNameStudent
}

之后若数据库中的表结构发生变化,只需用GenStruct()GenAllStructs()重新生成结构体即可。

注意

1.必须在mod同级目录调用GenStruct()或GenAllStructs()。

2.该包基于GORM相关的包,但本人的包已包含相关依赖,直接下载本人的包即可使用GORM。

相关推荐
luckys.one2 小时前
第9篇:Freqtrade量化交易之config.json 基础入门与初始化
javascript·数据库·python·mysql·算法·json·区块链
言之。3 小时前
Django中的软删除
数据库·django·sqlite
是誰萆微了承諾5 小时前
【golang学习笔记 gin 】1.2 redis 的使用
笔记·学习·golang
阿里嘎多哈基米5 小时前
SQL 层面行转列
数据库·sql·状态模式·mapper·行转列
抠脚学代码5 小时前
Ubuntu Qt x64平台搭建 arm64 编译套件
数据库·qt·ubuntu
jakeswang5 小时前
全解MySQL之死锁问题分析、事务隔离与锁机制的底层原理剖析
数据库·mysql
Heliotrope_Sun6 小时前
Redis
数据库·redis·缓存
一成码农6 小时前
MySQL问题7
数据库·mysql
吃饭最爱6 小时前
JUnit技术的核心和用法
数据库·oracle·sqlserver
专注API从业者6 小时前
Python/Java 代码示例:手把手教程调用 1688 API 获取商品详情实时数据
java·linux·数据库·python