最新公告
  • 欢迎您光临码农资源网,本站秉承服务宗旨 履行“站长”责任,销售只是起点 服务永无止境!加入我们
  • 使用 jackc/pgx 将空字符串或 null 作为 null 插入 postgres

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

    问题内容

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

    情况1:datedeccurr都是空字符串。

    {
        "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 有潜力,但我不确定应该如何做。

    如有任何建议,我们将不胜感激。谢谢!

    解决方案

    写入数据库时​​,可以通过多种方式表示 nullsql.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的技术知识我们会一点点深入介绍,欢迎大家关注编程网公众号,一起学习编程~

    想要了解更多内容,请持续关注码农资源网,一起探索发现编程世界的无限可能!
    本站部分资源来源于网络,仅限用于学习和研究目的,请勿用于其他用途。
    如有侵权请发送邮件至1943759704@qq.com删除

    码农资源网 » 使用 jackc/pgx 将空字符串或 null 作为 null 插入 postgres
    • 5会员总数(位)
    • 22375资源总数(个)
    • 770本周发布(个)
    • 77 今日发布(个)
    • 176稳定运行(天)

    提供最优质的资源集合

    立即查看 了解详情