我开源了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。

相关推荐
「QT(C++)开发工程师」6 分钟前
UML | 最好的类图设计工具结合CSDN天启呈现-领路架构师
数据库·uml·类视图
Damon小智9 分钟前
玩转ClaudeCode:用Database-MCP实现自然语言操作数据库
数据库·ai编程·claude·vibe coding·claude code
相与还10 分钟前
godot+c#使用godot-sqlite连接数据库
数据库·c#·godot
ZHOU_World17 分钟前
数据库主从同步
数据库·mysql
茉莉玫瑰花茶18 分钟前
MySQL 用户管理
数据库·mysql
FL162386312921 分钟前
C#winform流程图工具箱源码支持画矩形箭头圆形菱形保存为图片
数据库·oracle
博一波27 分钟前
MySQL 备份与复制:类似“手机数据管理”
数据库·mysql
semantist@语校28 分钟前
第二十篇|SAMU教育学院的教育数据剖析:制度阈值、能力矩阵与升学网络
大数据·数据库·人工智能·百度·语言模型·矩阵·prompt
xqlily35 分钟前
SQL 数据库简介
数据库·sql
正在走向自律1 小时前
Java连接电科金仓数据库(KingbaseES)实战指南
java·数据库·oracle·国产数据库·kingbase