Skip to content

Commit f8acda5

Browse files
Fixed handling of null values
1 parent 032e740 commit f8acda5

File tree

1 file changed

+16
-5
lines changed

1 file changed

+16
-5
lines changed

internal/getters/getters.go

+16-5
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,11 @@ func (r *RandomString) Value() interface{} {
107107
}
108108

109109
func (r *RandomString) String() string {
110-
return fmt.Sprintf("%q", r.Value())
110+
v := r.Value()
111+
if v == nil {
112+
return "NULL"
113+
}
114+
return fmt.Sprintf("%q", v)
111115
}
112116

113117
func NewRandomString(name string, maxSize int64, allowNull bool) Getter {
@@ -231,7 +235,10 @@ func (r *RandomEnum) Value() interface{} {
231235
}
232236

233237
func (r *RandomEnum) String() string {
234-
return fmt.Sprintf("%q", r.Value())
238+
if v := r.Value(); v != nil {
239+
return fmt.Sprintf("%q", v)
240+
}
241+
return "NULL"
235242
}
236243

237244
func NewRandomEnum(allowedValues []string, allowNull bool) Getter {
@@ -253,11 +260,15 @@ func (r *RandomSample) Value() interface{} {
253260
}
254261

255262
func (r *RandomSample) String() string {
256-
switch r.Value().(type) {
263+
v := r.Value()
264+
if v == nil {
265+
return "NULL"
266+
}
267+
switch v.(type) {
257268
case string:
258-
return fmt.Sprintf("%q", r.Value())
269+
return fmt.Sprintf("%q", v)
259270
default:
260-
return fmt.Sprintf("%v", r.Value())
271+
return fmt.Sprintf("%v", v)
261272
}
262273
}
263274

0 commit comments

Comments
 (0)