fix: do not try to reset nil objects

planetscale · Sep 14, 2023 · 9c89c01 · 9c89c01
1 parent cb44ee8
commit 9c89c01
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/features/pool/pool.go b/features/pool/pool.go
@@ -50,6 +50,7 @@ func (p *pool) message(message *protogen.Message) {
 	p.P(`}`)
 
 	p.P(`func (m *`, ccTypeName, `) ResetVT() {`)
+	p.P(`if m != nil {`)
 	var saved []*protogen.Field
 	for _, field := range message.Fields {
 		fieldName := field.GoName
@@ -89,6 +90,7 @@ func (p *pool) message(message *protogen.Message) {
 		p.P(`m.`, field.GoName, ` = `, fmt.Sprintf("f%d", i))
 	}
 	p.P(`}`)
+	p.P(`}`)
 
 	p.P(`func (m *`, ccTypeName, `) ReturnToVTPool() {`)
 	p.P(`if m != nil {`)