代码之家 › 专栏 › 技术社区 › Bussiere

如何正确调用gorm别名?

go-gorm go

Bussiere · 技术社区 · 7 年前

package main

import (
    "fmt"
    "time"

    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/sqlite"
)

type ClientCustomer struct {
    Id       int `json:"Id"`
    Name     string
    Created  time.Time
    key      string
    UserId   int `gorm:"user_id"`
    Modified time.Time
}

func (ClientCustomer) TableName() string {
    return "Client_customer"
}

type ClientCustomerInvitation struct {
    Id               int
    CustomerId       int `gorm:"customer_id"`
    CodeInvitationId int `gorm:"codeinvitation_id"`
}

func (ClientCustomerInvitation) TableName() string {
    return "Client_customer_invitation"
}

func main() {
    db, err := gorm.Open("sqlite3", "db.sqlite3?cache=shared&mode=rwc")
    if err != nil {
        panic("failed to connect database")
    }
    defer db.Close()
    var clientCustomer ClientCustomer
    rows, err := db.Model(&ClientCustomer{}).Rows()
    defer rows.Close()
    if err != nil {
        panic(err)
    }
    var clientCustomerInvitation ClientCustomerInvitation
    for rows.Next() {
        db.ScanRows(rows, &clientCustomer)
        db.First(&clientCustomerInvitation, "customer_id = ?", clientCustomer.Id)
        fmt.Println(clientCustomer)
        fmt.Println(clientCustomerInvitation)

    }

}

但我不喜欢这句话:

db.First(&clientCustomerInvitation, "customer_id = ?", clientCustomer.Id)

理想情况下,我想做如下事情:

 db.First(&clientCustomerInvitation, ClientCustomerInvitation.CustomerId.gormAlias+" = ?", clientCustomer.Id)

1 回复 | 直到 6 年前

novalagung Alexander Kim 7 年前

从特定结构字段获取标记值的唯一方法是使用 reflect .

func getGormAlias(obj interface{}, fieldName string) string {
    if field, ok := reflect.TypeOf(obj).FieldByName(fieldName); ok {
        return field.Tag.Get("gorm")
    }

    return ""
}

然后使用它获取标记值。

gormAliasCustomerId := getGormAlias(ClientCustomerInvitation{}, "CustomerId")
db.First(&clientCustomerInvitation, gormAliasCustomerId + " = ?", clientCustomer.Id)

基本上是什么 getGormAlias() 函数可以:

使用 reflect.Type obj 为了得到 反射类型 价值。
然后打电话 .FieldByName() reflect.Value 来自选定字段名的对象。
.Tag 财产。使用它来获取 gorm .

推荐文章

ginad · 如何在go-app组件中执行javascript代码

9 月前

Jason Fingar · 方法在另一个方法的上下文中不可访问

10 月前

Flissi Hamed · 从抓取aliexpress到使用chromedp的无头浏览器

10 月前

Adam Presley · GORM错误:“运算符不存在:bigint=text”

10 月前

Jess The Witch · GCP云功能中处理延迟任务的模式

10 月前

Moonlit · 在Golang中将`func()Foo`转换为`func)any`?

10 月前

pmoubed · 这是使用计时器的goroutine泄漏吗?

11 月前

Kurt Peek · 在Go模板中,如何仅在字符串值非空的情况下添加一行?

11 月前

Harimbola Santatra · 如何获取JSON中包含unicode代码点的键的值?

11 月前

techStud · 为什么Go在变量超出作用域后立即取消引用悬挂指针时不会抛出错误?

11 月前