GO多类型数据库存储设计

qq_17280559

于 2024-08-29 18:39:58 发布

阅读量428

点赞数 4

文章标签： go postgresql sqlite

本文链接：https://blog.csdn.net/qq_17280559/article/details/141685794

版权

在设计数据库关联时，可以考虑将模型设计得足够灵活，以便能够有效地存储到不同的数据库系统中，如 SQLite 或 PostgreSQL（Pg）。下面我将展示一种通用的设计思路，该设计可以适应 SQLite 和 PostgreSQL，并且能够有效处理不同的数据类型和关联。

数据库设计

假设我们要设计一个简单的内容管理系统（CMS），包括 users、posts 和 comments 表。这些表之间的关联如下：

users 表：存储用户信息。
posts 表：存储文章信息，并关联到 users。
comments 表：存储评论信息，关联到 posts 和 users。

我们希望这个设计能够适应 SQLite 和 PostgreSQL。以下是设计示例及其对应的 SQL 语句：

1. 数据表设计

用户（users）

CREATE TABLE users (
    id SERIAL PRIMARY KEY,          -- 使用 SERIAL (PostgreSQL) 或 INTEGER PRIMARY KEY AUTOINCREMENT (SQLite) 自动递增 ID
    name VARCHAR(255) NOT NULL,    -- 存储用户名称
    email VARCHAR(255) UNIQUE NOT NULL  -- 存储用户邮箱，要求唯一
);

帖子（posts）

CREATE TABLE posts (
    id SERIAL PRIMARY KEY,          -- 使用 SERIAL (PostgreSQL) 或 INTEGER PRIMARY KEY AUTOINCREMENT (SQLite) 自动递增 ID
    title VARCHAR(255) NOT NULL,   -- 存储帖子标题
    content TEXT,                  -- 存储帖子内容
    user_id INTEGER,               -- 存储发帖用户的 ID
    FOREIGN KEY (user_id) REFERENCES users(id) -- 外键关联到 users 表
);

评论（comments）

CREATE TABLE comments (
    id SERIAL PRIMARY KEY,          -- 使用 SERIAL (PostgreSQL) 或 INTEGER PRIMARY KEY AUTOINCREMENT (SQLite) 自动递增 ID
    content TEXT NOT NULL,         -- 存储评论内容
    post_id INTEGER,               -- 存储评论关联的帖子 ID
    user_id INTEGER,               -- 存储评论者的 ID
    FOREIGN KEY (post_id) REFERENCES posts(id),  -- 外键关联到 posts 表
    FOREIGN KEY (user_id) REFERENCES users(id)    -- 外键关联到 users 表
);

使用 GORM 进行数据库操作

GORM 是 Go 语言中的一个 ORM 库，它支持多种数据库，包括 SQLite 和 PostgreSQL。以下是如何使用 GORM 来定义模型并进行操作。

1. 安装 GORM

go get -u gorm.io/gorm
go get -u gorm.io/driver/sqlite  # 对于 SQLite
go get -u gorm.io/driver/postgres  # 对于 PostgreSQL

2. 定义数据模型

package main

import (
    "gorm.io/driver/sqlite"
    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

type User struct {
    ID    uint   `gorm:"primaryKey"`
    Name  string `gorm:"size:255"`
    Email string `gorm:"size:255;unique"`
    Posts []Post
    Comments []Comment
}

type Post struct {
    ID     uint   `gorm:"primaryKey"`
    Title  string `gorm:"size:255"`
    Content string
    UserID uint
    User   User `gorm:"foreignKey:UserID"`
    Comments []Comment
}

type Comment struct {
    ID     uint   `gorm:"primaryKey"`
    Content string
    PostID  uint
    UserID  uint
    Post    Post `gorm:"foreignKey:PostID"`
    User    User `gorm:"foreignKey:UserID"`
}

func main() {
    // PostgreSQL DSN
    // dsn := "user=username password=password dbname=mydb port=5432 sslmode=disable"
    
    // SQLite DSN
    dsn := "file:mydb.db?cache=shared&mode=rwc"

    // Choose database driver
    // db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
    db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
    if err != nil {
        panic("failed to connect database")
    }

    // Auto migrate schema
    db.AutoMigrate(&User{}, &Post{}, &Comment{})
}

数据关联操作

1. 创建和查询数据

func createUser(db *gorm.DB, name, email string) error {
    user := User{Name: name, Email: email}
    return db.Create(&user).Error
}

func createPost(db *gorm.DB, title, content string, userID uint) error {
    post := Post{Title: title, Content: content, UserID: userID}
    return db.Create(&post).Error
}

func createComment(db *gorm.DB, content string, postID, userID uint) error {
    comment := Comment{Content: content, PostID: postID, UserID: userID}
    return db.Create(&comment).Error
}

func getUserWithPosts(db *gorm.DB, userID uint) (User, error) {
    var user User
    err := db.Preload("Posts").First(&user, userID).Error
    return user, err
}

func getPostWithComments(db *gorm.DB, postID uint) (Post, error) {
    var post Post
    err := db.Preload("Comments").First(&post, postID).Error
    return post, err
}