limit句、offset句追加

titech-cpp · Aug 1, 2020 · 8f75cd3 · 8f75cd3
1 parent f59b7c7
commit 8f75cd3
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 2 deletions.
diff --git a/sample/main.go b/sample/main.go
@@ -35,7 +35,7 @@ func main() {
 
 	fmt.Printf("%+v\n", user)
 
-	users, err := db.Users().Select()
+	users, err := db.Users().Limit(2).Offset(1).Select()
 	if err != nil {
 		panic(err)
 	}

diff --git a/statik/statik.go b/statik/statik.go
diff --git a/template/_select.tpl b/template/_select.tpl
@@ -1,4 +1,16 @@
 {{define "select"}}
+func (q *{{.Name.UpperCamel}}Query) Limit(limit int) *{{.Name.UpperCamel}}Query {
+    q.limit = limit
+
+    return q
+}
+
+func (q *{{.Name.UpperCamel}}Query) Offset(offset int) *{{.Name.UpperCamel}}Query {
+    q.offset = offset
+
+    return q
+}
+
 func (q *{{.Name.UpperCamel}}Query) Find() (*{{.Name.UpperCamel}}Table,error) {
     whereMap := q.createWhereMap()
     whereQuery, whereArgs := q.whereStruct.Where(whereMap)
@@ -44,8 +56,19 @@ func (q *{{.Name.UpperCamel}}Query) Select() ([]*{{.Name.UpperCamel}}Table,error
     if len(whereArgs) != 0 {
         args = append(args, whereArgs...)
     }
+
     query := fmt.Sprintf("SELECT * FROM {{.Name.Snake}} %s", whereQuery)
 
+    if q.limit != 0 {
+        query += " LIMIT ?"
+        args = append(args, q.limit)
+    }
+
+    if q.offset != 0 {
+        query += " OFFSET ?"
+        args = append(args, q.offset)
+    }
+
     rows, err := q.db.Query(query, args...)
     if err != nil {
         return nil, fmt.Errorf("Execute Query Error: %w", err)

diff --git a/template/queries.go.tpl b/template/queries.go.tpl
@@ -32,6 +32,8 @@ type {{.Name.UpperCamel}}Table struct {
 type {{.Name.UpperCamel}}Query struct {
     db *sql.DB
     whereStruct *query.Where
+    limit int
+    offset int
     table {{.Name.UpperCamel}}
 }