」工欲善其事,必先利其器。「—孔子《論語.錄靈公》
首頁 > 程式設計 > Gorm:自訂資料類型先睹為快

Gorm:自訂資料類型先睹為快

發佈於2024-11-08
瀏覽:181

欢迎回来,伙计们?!今天,我们讨论在数据库之间来回移动数据时可能遇到的一个特定用例。首先,让我为今天的挑战设定界限。为了坚持现实生活中的例子,让我们借用美国陆军的一些概念?我们的任务是编写一个小软件来保存和读取军官在职业生涯中取得的成绩。

Gorm 的自定义数据类型

我们的软件需要处理军官及其各自的等级。乍一看,这似乎很简单,而且我们可能不需要任何自定义数据类型。但是,为了展示此功能,让我们使用非常规方式来表示数据。因此,我们被要求定义 Go 结构和数据库关系之间的自定义映射。此外,我们必须定义特定的逻辑来解析数据。让我们通过查看该计划的目标来扩展这一点?.

要处理的用例

为了方便起见,我们用一张图来描述代码和 SQL 对象之间的关系:

Gorm: Sneak Peek of Custom Data Types

让我们逐一关注每个容器。

Go 结构?

在这里,我们定义了两个结构体。 Grade 结构体包含军事等级的非详尽列表?️。该结构不会是数据库中的表。相反,Officer 结构体包含 ID、姓名和指向 Grade 结构体的指针,指示该军官到目前为止已取得的成绩。

每当我们向数据库写入官员时,grades_achieved 列必须包含一个字符串数组,其中填充了所达到的成绩(Grade 结构中具有 true 的成绩)。

数据库关系?

关于 SQL 对象,我们只有军官表。 id 和 name 列是不言自明的。然后,我们有 Grades_achieved 列,将官员的成绩保存在字符串集合中。

每当我们从数据库中解码官员时,我们都会解析 Grades_achieved 列并创建 Grade 结构的匹配“实例”。

您可能已经注意到这种行为不是标准行为。我们必须做出一些安排,以期望的方式实现它。

这里,模型的布局故意过于复杂。请尽可能坚持使用更简单的解决方案。

自定义数据类型

Gorm 为我们提供了自定义数据类型。它们为我们定义数据库的检索和保存提供了极大的灵活性。我们必须实现两个接口:Scanner 和 Valuer?。前者指定从数据库获取数据时要应用的自定义行为。后者指示如何将值写入数据库。两者都帮助我们实现我们需要的非常规映射逻辑。

我们必须实现的函数签名是 Scan(value interface{}) error 和 Value() (driver.Value, error)。现在,让我们看一下代码。

守则

此示例的代码位于两个文件中:domain/models.go 和 main.go。让我们从第一个开始,处理模型(在 Go 中翻译为结构体)。

域/models.go 文件

首先,让我展示这个文件的代码:

package models

import (
 "database/sql/driver"
 "slices"
 "strings"
)

type Grade struct {
 Lieutenant bool
 Captain    bool
 Colonel    bool
 General    bool
}

type Officer struct {
 ID             uint64 `gorm:"primaryKey"`
 Name           string
 GradesAchieved *Grade `gorm:"type:varchar[]"`
}

func (g *Grade) Scan(value interface{}) error {
 // we should have utilized the "comma, ok" idiom
 valueRaw := value.(string)
 valueRaw = strings.Replace(strings.Replace(valueRaw, "{", "", -1), "}", "", -1)
 grades := strings.Split(valueRaw, ",")
 if slices.Contains(grades, "lieutenant") {
 g.Lieutenant = true
 }
 if slices.Contains(grades, "captain") {
 g.Captain = true
 }
 if slices.Contains(grades, "colonel") {
 g.Colonel = true
 }
 if slices.Contains(grades, "general") {
 g.General = true
 }
 return nil
}

func (g Grade) Value() (driver.Value, error) {
 grades := make([]string, 0, 4)
 if g.Lieutenant {
 grades = append(grades, "lieutenant")
 }
 if g.Captain {
 grades = append(grades, "captain")
 }
 if g.Colonel {
 grades = append(grades, "colonel")
 }
 if g.General {
 grades = append(grades, "general")
 }
 return grades, nil
}

现在,让我们突出显示它的相关部分?:

  1. Grade 结构仅列出我们在软件中预测的成绩
  2. Officer 结构定义了实体的特征。该实体是数据库中的关系。我们应用了两种 Gorm 符号:
    1. gorm:ID 字段上的“primaryKey”将其定义为我们关系的主键
    2. gorm:"type:varchar[]" 将字段 GradesAchieved 映射为数据库中的 varchar 数组。否则,它会转换为单独的数据库表或官员表中的附加列
  3. Grade 结构体实现 Scan 函数。在这里,我们获取原始值,调整它,在 g 变量上设置一些字段,然后返回
  4. Grade 结构体还将 Value 函数实现为值接收器类型(这次我们不需要更改接收器,我们不使用 * 引用)。我们返回要写入军官表的 Grades_achieved 列中的值

借助这两种方法,我们可以控制在数据库交互期间如何发送和检索类型 Grade。现在,让我们看一下 main.go 文件。

main.go 文件?

在这里,我们准备数据库连接,将对象迁移到关系(ORM代表Object Relation Mapping),然后插入和获取记录下来测试逻辑。代码如下:

package main

import (
 "encoding/json"
 "fmt"
 "os"

 "gormcustomdatatype/models"

 "gorm.io/driver/postgres"
 "gorm.io/gorm"
)

func seedDB(db *gorm.DB, file string) error {
 data, err := os.ReadFile(file)
 if err != nil {
  return err
 }
 if err := db.Exec(string(data)).Error; err != nil {
  return err
 }
 return nil
}

// docker run -d -p 54322:5432 -e POSTGRES_PASSWORD=postgres postgres
func main() {
 dsn := "host=localhost port=54322 user=postgres password=postgres dbname=postgres sslmode=disable"
 db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
 if err != nil {
 fmt.Fprintf(os.Stderr, "could not connect to DB: %v", err)
  return
 }
 db.AutoMigrate(&models.Officer{})
 defer func() {
 db.Migrator().DropTable(&models.Officer{})
 }()
 if err := seedDB(db, "data.sql"); err != nil {
 fmt.Fprintf(os.Stderr, "failed to seed DB: %v", err)
  return
 }
 // print all the officers
 var officers []models.Officer
 if err := db.Find(&officers).Error; err != nil {
 fmt.Fprintf(os.Stderr, "could not get the officers from the DB: %v", err)
  return
 }
 data, _ := json.MarshalIndent(officers, "", "\t")
 fmt.Fprintln(os.Stdout, string(data))

 // add a new officer
 db.Create(&models.Officer{
 Name: "Monkey D. Garp",
 GradesAchieved: &models.Grade{
 Lieutenant: true,
 Captain:    true,
 Colonel:    true,
 General:    true,
  },
 })
 var garpTheHero models.Officer
 if err := db.First(&garpTheHero, 4).Error; err != nil {
 fmt.Fprintf(os.Stderr, "failed to get officer from the DB: %v", err)
  return
 }
 data, _ = json.MarshalIndent(&garpTheHero, "", "\t")
 fmt.Fprintln(os.Stdout, string(data))
}

现在,让我们看看这个文件的相关部分。首先,我们定义seedDB函数来在数据库中添加虚拟数据。数据位于 data.sql 文件中,内容如下:

INSERT INTO public.officers
(id, "name", grades_achieved)
VALUES(nextval('officers_id_seq'::regclass), 'john doe', '{captain,lieutenant}'),
(nextval('officers_id_seq'::regclass), 'gerard butler', '{general}'),
(nextval('officers_id_seq'::regclass), 'chuck norris', '{lieutenant,captain,colonel}');

main() 函数首先设置数据库连接。在本演示中,我们使用了 PostgreSQL。然后,我们确保军官表存在于数据库中,并且与最新版本的 models.Officer 结构保持同步。由于该程序是一个示例,因此我们做了两件事:

  • 在main()函数末尾删除表(当程序终止时,我们也想删除表)
  • 一些虚拟数据的播种

最后,为了确保一切按预期进行,我们做了几件事:

  1. 获取数据库中的所有记录
  2. 添加(并取回)新官员

这个文件就是这样。现在,让我们测试一下我们的工作?.

真相时刻

在运行代码之前,请确保您的计算机上正在运行 PostgreSQL 实例。使用 Docker ?,你可以运行这个命令:

docker run -d -p 54322:5432 -e POSTGRES_PASSWORD=postgres postgres

现在,我们可以通过发出命令来安全地运行我们的应用程序: go run 。 ?

输出为:

[
        {
                "ID": 1,
                "Name": "john doe",
                "GradesAchieved": {
                        "Lieutenant": true,
                        "Captain": true,
                        "Colonel": false,
                        "General": false
                }
        },
        {
                "ID": 2,
                "Name": "gerard butler",
                "GradesAchieved": {
                        "Lieutenant": false,
                        "Captain": false,
                        "Colonel": false,
                        "General": true
                }
        },
        {
                "ID": 3,
                "Name": "chuck norris",
                "GradesAchieved": {
                        "Lieutenant": true,
                        "Captain": true,
                        "Colonel": true,
                        "General": false
                }
        }
]
{
        "ID": 4,
        "Name": "Monkey D. Garp",
        "GradesAchieved": {
                "Lieutenant": true,
                "Captain": true,
                "Colonel": true,
                "General": true
        }
}

瞧!一切都按预期进行。我们可以多次重新运行代码并始终得到相同的输出。

这是一个包裹

我希望您喜欢这篇关于 Gorm自定义数据类型 的博文。我始终建议您坚持使用最直接的方法。仅当您最终需要时才选择此选项。这种方法增加了灵活性,但代价是使代码变得更复杂和更不健壮(结构定义的微小变化可能会导致错误和需要额外的工作)。

记住这一点。如果您坚持约定,则整个代码库中的内容就会变得不那么冗长。

这是结束这篇博文的一个很好的引用。
如果您意识到需要自定义数据类型,这篇博文应该是一个很好的起点,可以为您提供一个可行的解决方案。

请告诉我您的感受和想法。任何反馈总是值得赞赏!如果您对某个特定主题感兴趣,请联系我,我会将其列入候选名单。下次见,注意安全,再见!

版本聲明 本文轉載於:https://dev.to/ossan/gorm-sneak-peek-of-custom-data-types-97n?1如有侵犯,請聯絡[email protected]刪除
最新教學 更多>
  • Bootstrap 4 Beta 中的列偏移發生了什麼事?
    Bootstrap 4 Beta 中的列偏移發生了什麼事?
    Bootstrap 4 Beta:列偏移的刪除和恢復Bootstrap 4 在其Beta 1 版本中引入了重大更改柱子偏移了。然而,隨著 Beta 2 的後續發布,這些變化已經逆轉。 從 offset-md-* 到 ml-auto在 Bootstrap 4 Beta 1 中, offset-md-*...
    程式設計 發佈於2024-12-23
  • 在 Go 中使用 WebSocket 進行即時通信
    在 Go 中使用 WebSocket 進行即時通信
    构建需要实时更新的应用程序(例如聊天应用程序、实时通知或协作工具)需要一种比传统 HTTP 更快、更具交互性的通信方法。这就是 WebSockets 发挥作用的地方!今天,我们将探讨如何在 Go 中使用 WebSocket,以便您可以向应用程序添加实时功能。 在这篇文章中,我们将介绍: WebSoc...
    程式設計 發佈於2024-12-23
  • 大批
    大批
    方法是可以在物件上呼叫的 fns 數組是對象,因此它們在 JS 中也有方法。 slice(begin):將陣列的一部分提取到新數組中,而不改變原始數組。 let arr = ['a','b','c','d','e']; // Usecase: Extract till index ...
    程式設計 發佈於2024-12-23
  • 儘管程式碼有效,為什麼 POST 請求無法擷取 PHP 中的輸入?
    儘管程式碼有效,為什麼 POST 請求無法擷取 PHP 中的輸入?
    解決PHP 中的POST 請求故障在提供的程式碼片段中:action=''而非:action="<?php echo $_SERVER['PHP_SELF'];?>";?>"檢查$_POST陣列:表單提交後使用 var_dump 檢查 $_POST 陣列的內...
    程式設計 發佈於2024-12-23
  • 如何在 PHP 中組合兩個關聯數組,同時保留唯一 ID 並處理重複名稱?
    如何在 PHP 中組合兩個關聯數組,同時保留唯一 ID 並處理重複名稱?
    在 PHP 中組合關聯數組在 PHP 中,將兩個關聯數組組合成一個數組是常見任務。考慮以下請求:問題描述:提供的代碼定義了兩個關聯數組,$array1 和 $array2。目標是建立一個新陣列 $array3,它合併兩個陣列中的所有鍵值對。 此外,提供的陣列具有唯一的 ID,而名稱可能重疊。要求是建...
    程式設計 發佈於2024-12-23
  • 插入資料時如何修復「常規錯誤:2006 MySQL 伺服器已消失」?
    插入資料時如何修復「常規錯誤:2006 MySQL 伺服器已消失」?
    插入記錄時如何解決「一般錯誤:2006 MySQL 伺服器已消失」介紹:將資料插入MySQL 資料庫有時會導致錯誤「一般錯誤:2006 MySQL 伺服器已消失」。當與伺服器的連線遺失時會出現此錯誤,通常是由於 MySQL 配置中的兩個變數之一所致。 解決方案:解決此錯誤的關鍵是調整wait_tim...
    程式設計 發佈於2024-12-23
  • 為什麼我的 Angular HTTP POST 值在 PHP 中未定義,如何修復它?
    為什麼我的 Angular HTTP POST 值在 PHP 中未定義,如何修復它?
    Angular HTTP POST 到PHP:處理未定義的POST 值在AngularJS 中,對PHP 端點執行HTTP POST 請求有時會導致未定義的值伺服器端的POST 值。當預期資料格式與 Angular 應用程式傳送的實際資料不符時,就會發生這種情況。 要解決此問題,確保正確設定 Con...
    程式設計 發佈於2024-12-23
  • Go可以存取初始標準輸入流嗎?
    Go可以存取初始標準輸入流嗎?
    在 Go 中,您可以存取初始標準輸入嗎? 在 Go 中,使用 os.Stdin 從原始標準輸入讀取應該會產生所需的結果,如圖所示通過這個代碼片段:package main import "os" import "log" import "io&quo...
    程式設計 發佈於2024-12-23
  • 極簡密碼管理器桌面應用程式:進軍 Golang 的 Wails 框架(第 2 部分)
    極簡密碼管理器桌面應用程式:進軍 Golang 的 Wails 框架(第 2 部分)
    Hi again, coders! In the first part of this short series we saw the creation and operation of a desktop application to store and encrypt our passwords...
    程式設計 發佈於2024-12-23
  • ES6 React 元件:何時使用基於類別與函數式?
    ES6 React 元件:何時使用基於類別與函數式?
    在ES6 基於類別和函數式ES6 React 元件之間做出選擇使用React 時,開發人員面臨著使用ES6 基於類別的選擇組件或功能ES6 組件。了解每種類型的適當用例對於最佳應用程式開發至關重要。 函數式 ES6 元件:無狀態且簡單函數式元件是無狀態的,這表示它們不維護任何內部狀態。他們只是接收道...
    程式設計 發佈於2024-12-23
  • 如何在 PHP 中找到兩個平面數組之間的唯一值?
    如何在 PHP 中找到兩個平面數組之間的唯一值?
    在平面數組之間查找唯一值給定兩個數組,任務是確定僅存在於其中一個數組中的值。此操作通常稱為尋找兩個集合之間的差異。 在 PHP 中,您可以利用 array_merge、array_diff 和 array_diff 函數來實現此操作。詳細解法如下:$array1 = [64, 98, 112, 92...
    程式設計 發佈於2024-12-23
  • CSS 可以在內聯區塊元素中本機插入換行符號嗎?
    CSS 可以在內聯區塊元素中本機插入換行符號嗎?
    CSS 在行內區塊元素中插入換行符:理論探索在不斷發展的Web 開發領域,這種能力操縱內容流仍然是最重要的。經常出現的一個特殊挑戰涉及在內聯區塊元素中插入換行符。 考慮以下 HTML 結構:<h3 id="features">Features</h3> &...
    程式設計 發佈於2024-12-23
  • 如何在 PHP 中輕鬆轉換時區之間的時間和日期?
    如何在 PHP 中輕鬆轉換時區之間的時間和日期?
    在PHP 中轉換時區之間的時間和日期使用PHP,您可以輕鬆地在不同時區之間轉換時間和日期。此功能在處理全球資料的應用程式或與來自不同位置的使用者一起工作時特別有用。 取得時區偏移量要取得與 GMT 的時間偏移量,您可以使用 DateTimeZone 類別。它提供了時區及其各自偏移量的完整清單。 $t...
    程式設計 發佈於2024-12-23
  • HTML 格式標籤
    HTML 格式標籤
    HTML 格式化元素 **HTML Formatting is a process of formatting text for better look and feel. HTML provides us ability to format text without us...
    程式設計 發佈於2024-12-23
  • 如何在Windows上安裝並使用Pip進行Python套件管理?
    如何在Windows上安裝並使用Pip進行Python套件管理?
    Pip:在Windows 上安裝Python 套件的輕鬆方式在Windows 上安裝Python 套件可能是一項艱鉅的任務,特別是如果您在使用EasyInstall 時遇到困難。幸運的是,EasyInstall 的後繼者 Pip 提供了更簡化和簡化的解決方案。 在Windows 上逐步安裝Pip若要...
    程式設計 發佈於2024-12-23

免責聲明: 提供的所有資源部分來自互聯網,如果有侵犯您的版權或其他權益,請說明詳細緣由並提供版權或權益證明然後發到郵箱:[email protected] 我們會在第一時間內為您處理。

Copyright© 2022 湘ICP备2022001581号-3