feat: FindByPage return all result when limit<=0

BigeYoung · Apr 7, 2022 · 61f3ee2 · 61f3ee2
1 parent bd3f480
commit 61f3ee2
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/internal/template/method.go b/internal/template/method.go
@@ -204,17 +204,12 @@ func ({{.S}} {{.NewStructName}}Do) FirstOrCreate() (*{{.StructInfo.Package}}.{{.
 }
 
 func ({{.S}} {{.NewStructName}}Do) FindByPage(offset int, limit int) (result []*{{.StructInfo.Package}}.{{.StructInfo.Type}}, count int64, err error) {
-	if limit <= 0 {
-		count, err = {{.S}}.Count()
-		return
-	}
-
 	result, err = {{.S}}.Offset(offset).Limit(limit).Find()
 	if err != nil{
 		return
 	}
 
-	if size := len(result); 0 < size && size < limit {
+	if size := len(result); 0 < limit && 0 < size && size < limit {
 		count = int64(size+offset)
 		return
 	}