feat: gen.R to gen.RowsAffected

a631807682 · Oct 8, 2021 · 8d2f7dc · 8d2f7dc
1 parent eb1ac5e
commit 8d2f7dc
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/generator.go b/generator.go
@@ -29,8 +29,8 @@ type T interface{}
 // M map[string]interface{}
 type M map[string]interface{}
 
-// R execute affected raws
-type R int64
+// RowsAffected execute affected raws
+type RowsAffected int64
 
 // NewGenerator create a new generator
 func NewGenerator(cfg Config) *Generator {

diff --git a/internal/check/checkinterface.go b/internal/check/checkinterface.go
@@ -186,7 +186,7 @@ func (m *InterfaceMethod) checkResult(result []parser.Param) (err error) {
 			m.ResultData = param
 		case param.IsInterface():
 			return fmt.Errorf("query method can not return interface in [%s.%s]", m.InterfaceName, m.MethodName)
-		case param.IsGenR():
+		case param.IsGenRowsAffected():
 			param.Type = "int64"
 			param.Package = ""
 			param.SetName("rowsAffected")

diff --git a/internal/parser/parser.go b/internal/parser/parser.go
@@ -132,8 +132,8 @@ func (p *Param) IsGenM() bool {
 	return p.Package == "gen" && p.Type == "M"
 }
 
-func (p *Param) IsGenR() bool {
-	return p.Package == "gen" && p.Type == "R"
+func (p *Param) IsGenRowsAffected() bool {
+	return p.Package == "gen" && p.Type == "RowsAffected"
 }
 
 func (p *Param) IsMap() bool {