使用 jackc/pgx 将空字符串或 null 作为 null 插入 postgres

欢迎各位小伙伴来到编程网，相聚于此都是缘哈哈哈！今天我给大家带来《使用 jackc/pgx 将空字符串或 null 作为 null 插入 postgres》，这篇文章主要讲到等等知识，如果你对golang相关的知识非常感兴趣或者正在自学，都可以关注我，我会持续更新相关文章！当然，有什么建议也欢迎在评论留言提出！一起学习！

我使用的外部 JSON api 处理缺失值的方式不一致。有时 json 值显示为空字符串，有时显示为 null。例如...

情况1：datedec和curr都是空字符串。

{
    "symbol": "xyz",
    "dateex": "2020-09-01",
    "datedec": "",
    "amount": "1.25",
    "curr": "",
    "freq": "annual"
}

情况2：datedec 为空。 curr 已填充。

{
    "symbol": "xyz",
    "dateex": "2020-09-01",
    "datedec": null,
    "amount": "1.25",
    "curr": "usd",
    "freq": "annual"
}

这是我用来表示股息的结构：

type dividend struct {
    symbol   string `json:"symbol"`
    dateex   string `json:"dateex"`
    datedec  string `json:"datedec"`
    amount   string `json:"amount"`
    curr     string `json:"curr"`
    freq     string `json:"freq"`
}

我遇到的问题是如何将空字符串或 null 作为 null 插入数据库。我知道我可以使用 omitempty json 标签，但是如何编写一个函数来处理我不知道会丢失的值？例如，这是我当前使用 jackc/pgx 包将红利插入到 postgresql 的函数：

func InsertDividend(d Dividend) error {
    sql := `INSERT INTO dividends 
    (symbol, dateex, datedec, amount, curr, freq)
    VALUES ($1, $2, $3, $4, $5, $6)`
    conn, err := pgx.Connect(ctx, "DATABASE_URL")
    // handle error 
    defer conn.Close(ctx)
    tx, err := conn.Begin()
    // handle error
    defer tx.Rollback(ctx)
    _, err = tx.Exec(ctx, sql, d.symbol, d.dateex, d.datedec, d.amount, d.curr, d.freq)
    // handle error
    }
    err = tx.Commit(ctx)
    // handle error
    return nil
}

如果某个值（例如 datedec 或 curr）丢失，则此函数将出错。从这篇文章Golang insert null into sql代替空字符串我看到了如何解决case1。但是有没有更通用的方法来处理这两种情况（空或空字符串）？

我一直在浏览数据库/sql 和 jackc/pgx 文档，但我还没有找到任何东西。我认为 sql.nullstring 有潜力，但我不确定应该如何做。

如有任何建议，我们将不胜感激。谢谢！

解决方案

写入数据库时​​，可以通过多种方式表示 null。 sql.nullstring 是一个选项，与使用指针一样 (nil = null)；选择实际上取决于您认为更容易理解的内容。罗斯·考克斯 commented：

没有任何有效的区别。我们认为人们可能想要使用 nullstring，因为它很常见，并且可能比 *string 更清楚地表达意图。但两者都可以。

我怀疑在您的情况下使用指针将是最简单的方法。例如，以下内容可能会满足您的需求：

type dividend struct {
    symbol  string  `json:"symbol"`
    dateex  string  `json:"dateex"`
    datedec *string `json:"datedec"`
    amount  string  `json:"amount"`
    curr    *string `json:"curr"`
    freq    string  `json:"freq"`
}

func unmarshal(in[]byte, div *dividend) {
    err := json.unmarshal(in, div)
    if err != nil {
        panic(err)
    }
    // the below is not necessary unless if you want to ensure that blanks
    // and missing values are both written to the database as null...
    if div.datedec != nil && len(*div.datedec) == 0 {
        div.datedec = nil
    }
    if div.curr != nil && len(*div.curr) == 0 {
        div.curr = nil
    }
}

试试in the playground。

您可以像现在写入数据库时​​一样使用 dividend 结构体； sql 驱动程序会将 nil 写入为 null。

您还可以使用 pgtypes 并使用 value() 函数从任何 pgtype 获取 sql 驱动程序值：

https://GitHub.com/jackc/pgtype

Https://github.com/jackc/pgtype/blob/master/text.go

type Dividend struct {
    symbol   pgtype.Text `json:"symbol"`
    dateex   pgtype.Text `json:"dateex"`
    datedec  pgtype.Text `json:"datedec"`
    amount   pgtype.Text `json:"amount"`
    curr     pgtype.Text `json:"curr"`
    freq     pgtype.Text `json:"freq"`
}

func InsertDividend(d Dividend) error {
    // --> get SQL values from d
    var err error
    symbol, err := d.symbol.Value() // see https://github.com/jackc/pgtype/blob/4db2a33562c6d2d38da9dbe9b8e29f2d4487cc5b/text.go#L174
    if err != nil {
        return err
    }
    dateex, err := d.dateex.Value()
    if err != nil {
        return err
    }
    // ...

    sql := `INSERT INTO dividends 
    (symbol, dateex, datedec, amount, curr, freq)
    VALUES ($1, $2, $3, $4, $5, $6)`
    conn, err := pgx.Connect(ctx, "DATABASE_URL")
    defer conn.Close(ctx)
    tx, err := conn.Begin()
    defer tx.Rollback(ctx)
    // --> exec your query using the SQL values your get earlier
    _, err = tx.Exec(ctx, sql, symbol, dateex, datedec, amount, curr, freq)
    // handle error
    }
    err = tx.Commit(ctx)
    // handle error
    return nil
}

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注编程网公众号，一起学习编程~