go-gorm / gorm

The fantastic ORM library for Golang, aims to be developer friendly
https://gorm.io
MIT License
36.48k stars 3.9k forks source link

How to chanage value in Callback.Row register function? #7117

Closed LaoBan-ywcm closed 6 days ago

LaoBan-ywcm commented 1 month ago

How to chanage value in Callback.Row register function?

Some fields in my database are encrypted. When performing a query, I need to join multiple tables and use the scan method. I want to decrypt in plugins.

I register the callback like this: db.Callback().Row().After("gorm:row").Register("*", decryptRowValue).

In decryptRowValue, I want to modify the value of wallet_name, but modifying it this way does not work. How should I set it up?

func decryptRowValue(db *gorm.DB) {
    rows, err := db.Statement.ConnPool.QueryContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)
    if err != nil {
        db.AddError(err)
        return
    }
    defer rows.Close()

    columns, err := rows.Columns()
    if err != nil {
        db.AddError(err)
        return
    }

    values := make([]interface{}, len(columns))
    valuePtrs := make([]interface{}, len(columns))
    for i := range values {
        valuePtrs[i] = &values[i]
    }

    if rows.Next() {
        if err = rows.Scan(valuePtrs...); err != nil {
            db.AddError(err)
            return
        }

        for i, col := range columns {
            if col == "wallet_name" {
                values[i] = "test"
            }

            fmt.Println(col, values[i])
        }
    }
}
github-actions[bot] commented 1 month ago

The issue has been automatically marked as stale as it missing playground pull request link, which is important to help others understand your issue effectively and make sure the issue hasn't been fixed on latest master, checkout https://github.com/go-gorm/playground for details. it will be closed in 30 days if no further activity occurs. if you are asking question, please use the Question template, most likely your question already answered https://github.com/go-gorm/gorm/issues or described in the document https://gorm.ioSearch Before Asking