我在 Go 中创建了一个非关系型的对象映射，它非常简单。

我有几个看起来像这样的结构:

type Message struct {
    Id       int64
    Message  string
    ReplyTo  sql.NullInt64 `db:"reply_to"`
    FromId   int64         `db:"from_id"`
    ToId     int64         `db:"to_id"`
    IsActive bool          `db:"is_active"`
    SentTime int64         `db:"sent_time"`
    IsViewed bool          `db:"is_viewed"`

    Method   string `db:"-"`
    AppendTo int64  `db:"-"`
}

要创建一条新消息，我只需运行此函数:

func New() *Message {
    return &Message{
        IsActive: true,
        SentTime: time.Now().Unix(),
        Method:   "new",
    }
}

然后我有这个结构的 message_crud.go 文件，如下所示:

要按唯一列(例如按 ID)查找消息，我运行此函数:

func ByUnique(column string, value interface{}) (*Message, error) {

    query := fmt.Sprintf(`
        SELECT *
        FROM message
        WHERE %s = ?
        LIMIT 1;
    `, column)

    message := &Message{}
    err := sql.DB.QueryRowx(query, value).StructScan(message)
    if err != nil {
        return nil, err
    }
    return message, nil
}

为了保存消息(在数据库中插入或更新)，我运行了这个方法:

func (this *Message) save() error {
    s := ""
    if this.Id == 0 {
        s = "INSERT INTO message SET %s;"
    } else {
        s = "UPDATE message SET %s WHERE id=:id;"
    }
    query := fmt.Sprintf(s, sql.PlaceholderPairs(this))

    nstmt, err := sql.DB.PrepareNamed(query)
    if err != nil {
        return err
    }

    res, err := nstmt.Exec(*this)
    if err != nil {
        return err
    }

    if this.Id == 0 {
        lastId, err := res.LastInsertId()
        if err != nil {
            return err
        }
        this.Id = lastId
    }

    return nil
}

sql.PlaceholderPairs() 函数如下所示:

func PlaceholderPairs(i interface{}) string {

    s := ""
    val := reflect.ValueOf(i).Elem()
    count := val.NumField()

    for i := 0; i < count; i++ {
        typeField := val.Type().Field(i)
        tag := typeField.Tag

        fname := strings.ToLower(typeField.Name)

        if fname == "id" {
            continue
        }

        if t := tag.Get("db"); t == "-" {
            continue
        } else if t != "" {
            s += t + "=:" + t
        } else {
            s += fname + "=:" + fname
        }
        s += ", "
    }
    s = s[:len(s)-2]
    return s
}

但是每次我创建一个新的结构，例如一个用户结构，我必须复制粘贴上面的“crud 部分”并创建一个 user_crud.go 文件，并将“消息”替换为“用户”，以及“消息”与“用户”。我重复了很多代码，它不是很枯燥。有什么我可以做的，而不是为我会重用的东西重复这段代码吗？我总是有一个 save() 方法，并且总是有一个 ByUnique() 函数，我可以在其中返回一个结构并按唯一列进行搜索。

在 PHP 中这很容易，因为 PHP 不是静态类型的。

这在 Go 中可以做到吗？

最佳答案

您的 ByUnique 几乎已经是通用的了。只需拉出变化的部分(表格和目的地):

func ByUnique(table string, column string, value interface{}, dest interface{}) error {
    query := fmt.Sprintf(`
            SELECT *
            FROM %s
            WHERE %s = ?
            LIMIT 1;
        `, table, column)

    return sql.DB.QueryRowx(query, value).StructScan(dest)
}

func ByUniqueMessage(column string, value interface{}) (*Message, error) {
    message := &Message{}
    if err := ByUnique("message", column, value, &message); err != nil {
        return nil, err
    }
    return message, error
}

您的save 非常相似。您只需要按照以下行制作一个通用的保存功能:

func Save(table string, identifier int64, source interface{}) { ... }

然后在 (*Message)save 内部，您只需调用一般的 Save() 函数。看起来很简单。

旁注:不要使用 this 作为方法内对象的名称。有关更多信息，请参阅@OneOfOne 的链接。并且不要沉迷于 DRY。它本身不是目标。 Go 专注于代码简单、清晰和可靠。不要仅仅为了避免输入一行简单的错误处理而创建一些复杂而脆弱的东西。这并不意味着您不应该提取重复的代码。这只是意味着在 Go 中，通常最好稍微重复一些简单的代码，而不是创建复杂的代码来避免它。

编辑:如果您想使用接口(interface)实现Save，那没问题。只需创建一个 Identifier 接口(interface)即可。

type Ider interface {
    Id() int64
    SetId(newId int64)
}

func (msg *Message) Id() int64 {
    return msg.Id
}

func (msg *Message) SetId(newId int64) {
    msg.Id = newId
}

func Save(table string, source Ider) error {
    s := ""
    if source.Id() == 0 {
        s = fmt.Sprintf("INSERT INTO %s SET %%s;", table)
    } else {
        s = fmt.Sprintf("UPDATE %s SET %%s WHERE id=:id;", table)
    }
    query := fmt.Sprintf(s, sql.PlaceholderPairs(source))

    nstmt, err := sql.DB.PrepareNamed(query)
    if err != nil {
        return err
    }

    res, err := nstmt.Exec(source)
    if err != nil {
        return err
    }

    if source.Id() == 0 {
        lastId, err := res.LastInsertId()
        if err != nil {
            return err
        }
        source.SetId(lastId)
    }

    return nil
}

func (msg *Message) save() error {
    return Save("message", msg)
}

可能与此有关的部分是对 Exec 的调用。我不知道您使用的是什么包，如果您向它传递接口(interface)而不是实际结构，Exec 可能无法正常工作，但它可能会工作。也就是说，我可能只是传递标识符而不是添加此开销。

关于go - 我怎样才能使这个对象映射在 Go 中更干燥和可重用？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/35260122/