1
0
mirror of https://github.com/redis/go-redis.git synced 2025-07-29 17:41:15 +03:00

fix: nil pointer dereferencing in writeArg (#3271)

* fixed bug with nil dereferencing in writeArg, added hset struct example, added tests

* removed password from example

* added omitempty

* reverted xxhash versioning

* reverted xxhash versioning

* removed password

* removed password

---------

Co-authored-by: Nedyalko Dyakov <nedyalko.dyakov@gmail.com>
This commit is contained in:
Ali Error
2025-02-20 17:54:11 +03:00
committed by GitHub
parent 747190e231
commit 37accb4b28
7 changed files with 274 additions and 29 deletions

View File

@ -0,0 +1,7 @@
# Example for setting struct fields as hash fields
To run this example:
```shell
go run .
```

View File

@ -0,0 +1,15 @@
module github.com/redis/go-redis/example/scan-struct
go 1.18
replace github.com/redis/go-redis/v9 => ../..
require (
github.com/davecgh/go-spew v1.1.1
github.com/redis/go-redis/v9 v9.6.2
)
require (
github.com/cespare/xxhash/v2 v2.2.0 // indirect
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
)

View File

@ -0,0 +1,10 @@
github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs=
github.com/bsm/gomega v1.27.10 h1:yeMWxP2pV2fG3FgAODIY8EiRE3dy0aeFYt4l7wh6yKA=
github.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=
github.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78=
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc=

129
example/hset-struct/main.go Normal file
View File

@ -0,0 +1,129 @@
package main
import (
"context"
"time"
"github.com/davecgh/go-spew/spew"
"github.com/redis/go-redis/v9"
)
type Model struct {
Str1 string `redis:"str1"`
Str2 string `redis:"str2"`
Str3 *string `redis:"str3"`
Str4 *string `redis:"str4"`
Bytes []byte `redis:"bytes"`
Int int `redis:"int"`
Int2 *int `redis:"int2"`
Int3 *int `redis:"int3"`
Bool bool `redis:"bool"`
Bool2 *bool `redis:"bool2"`
Bool3 *bool `redis:"bool3"`
Bool4 *bool `redis:"bool4,omitempty"`
Time time.Time `redis:"time"`
Time2 *time.Time `redis:"time2"`
Time3 *time.Time `redis:"time3"`
Ignored struct{} `redis:"-"`
}
func main() {
ctx := context.Background()
rdb := redis.NewClient(&redis.Options{
Addr: ":6379",
})
_ = rdb.FlushDB(ctx).Err()
t := time.Date(2025, 02, 8, 0, 0, 0, 0, time.UTC)
data := Model{
Str1: "hello",
Str2: "world",
Str3: ToPtr("hello"),
Str4: nil,
Bytes: []byte("this is bytes !"),
Int: 123,
Int2: ToPtr(0),
Int3: nil,
Bool: true,
Bool2: ToPtr(false),
Bool3: nil,
Time: t,
Time2: ToPtr(t),
Time3: nil,
Ignored: struct{}{},
}
// Set some fields.
if _, err := rdb.Pipelined(ctx, func(rdb redis.Pipeliner) error {
rdb.HMSet(ctx, "key", data)
return nil
}); err != nil {
panic(err)
}
var model1, model2 Model
// Scan all fields into the model.
if err := rdb.HGetAll(ctx, "key").Scan(&model1); err != nil {
panic(err)
}
// Or scan a subset of the fields.
if err := rdb.HMGet(ctx, "key", "str1", "int").Scan(&model2); err != nil {
panic(err)
}
spew.Dump(model1)
// Output:
// (main.Model) {
// Str1: (string) (len=5) "hello",
// Str2: (string) (len=5) "world",
// Str3: (*string)(0xc000016970)((len=5) "hello"),
// Str4: (*string)(0xc000016980)(""),
// Bytes: ([]uint8) (len=15 cap=16) {
// 00000000 74 68 69 73 20 69 73 20 62 79 74 65 73 20 21 |this is bytes !|
// },
// Int: (int) 123,
// Int2: (*int)(0xc000014568)(0),
// Int3: (*int)(0xc000014560)(0),
// Bool: (bool) true,
// Bool2: (*bool)(0xc000014570)(false),
// Bool3: (*bool)(0xc000014548)(false),
// Bool4: (*bool)(<nil>),
// Time: (time.Time) 2025-02-08 00:00:00 +0000 UTC,
// Time2: (*time.Time)(0xc0000122a0)(2025-02-08 00:00:00 +0000 UTC),
// Time3: (*time.Time)(0xc000012288)(0001-01-01 00:00:00 +0000 UTC),
// Ignored: (struct {}) {
// }
// }
spew.Dump(model2)
// Output:
// (main.Model) {
// Str1: (string) (len=5) "hello",
// Str2: (string) "",
// Str3: (*string)(<nil>),
// Str4: (*string)(<nil>),
// Bytes: ([]uint8) <nil>,
// Int: (int) 123,
// Int2: (*int)(<nil>),
// Int3: (*int)(<nil>),
// Bool: (bool) false,
// Bool2: (*bool)(<nil>),
// Bool3: (*bool)(<nil>),
// Bool4: (*bool)(<nil>),
// Time: (time.Time) 0001-01-01 00:00:00 +0000 UTC,
// Time2: (*time.Time)(<nil>),
// Time3: (*time.Time)(<nil>),
// Ignored: (struct {}) {
// }
// }
}
func ToPtr[T any](v T) *T {
return &v
}

View File

@ -11,9 +11,12 @@ import (
type Model struct {
Str1 string `redis:"str1"`
Str2 string `redis:"str2"`
Str3 *string `redis:"str3"`
Bytes []byte `redis:"bytes"`
Int int `redis:"int"`
Int2 *int `redis:"int2"`
Bool bool `redis:"bool"`
Bool2 *bool `redis:"bool2"`
Ignored struct{} `redis:"-"`
}
@ -29,8 +32,11 @@ func main() {
if _, err := rdb.Pipelined(ctx, func(rdb redis.Pipeliner) error {
rdb.HSet(ctx, "key", "str1", "hello")
rdb.HSet(ctx, "key", "str2", "world")
rdb.HSet(ctx, "key", "str3", "")
rdb.HSet(ctx, "key", "int", 123)
rdb.HSet(ctx, "key", "int2", 0)
rdb.HSet(ctx, "key", "bool", 1)
rdb.HSet(ctx, "key", "bool2", 0)
rdb.HSet(ctx, "key", "bytes", []byte("this is bytes !"))
return nil
}); err != nil {