fix(generate): fix GetFieldByName panic

go-gorm · Dec 15, 2021 · 7ee5fc3 · 7ee5fc3
1 parent 14d5b60
commit 7ee5fc3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/internal/template/struct.go b/internal/template/struct.go
@@ -87,7 +87,10 @@ func ({{.S}} {{.NewStructName}}) clone(db *gorm.DB) {{.NewStructName}} {
 	getFieldMethod = `
 func ({{.S}} *{{.NewStructName}}) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
 	_f, ok := {{.S}}.fieldMap[fieldName]
-	return _f.(field.OrderExpr), ok
+	if !ok || _f == nil {
+		return nil, false
+	}
+	return _f.(field.OrderExpr), true
 }
 `
 	relationship = `{{range .Members}}{{if .IsRelation}}` +