Merge pull request #7 from swiftcarrot/insert-all

InsertAll

Author: wangzuo

generator/client/golang/templates/[model]_query.gotmpl

@@ -1,6 +1,6 @@
 // Code generated by queryx, DO NOT EDIT.
 
-package {{.packageName}}
+package {{ $.packageName }}
 
 import (
 	"database/sql"
@@ -126,6 +126,40 @@ func (q *{{ $.model.Name }}Query) Create(change *queryx.{{ $.model.Name }}Change
 	return record, nil
 }
 
+func (q *{{.model.Name}}Query) InsertAll(changes []*queryx.{{.model.Name}}Change) (int64, error){
+	if q.err != nil {
+		return 0, q.err
+	}
+
+	if len(changes) == 0 {
+		return 0, ErrInsertAllEmptyChanges
+	}
+
+	{{- if eq $.model.Timestamps true }}
+	now := queryx.Now("2006-01-02 15:04:05.000")
+	for _, change := range changes {
+		if !change.CreatedAt.Set {
+			change.SetCreatedAt(now)
+		}
+		if !change.UpdatedAt.Set {
+			change.SetUpdatedAt(now)
+		}
+	}
+	{{- end }}
+
+	s := queryx.NewInsert().Into("{{ $.model.TableName }}")
+	for i, change := range changes {
+		columns, values := change.Changes()
+		if i == 0 {
+			s.Columns(columns...)
+		}
+		s.Values(values...)
+	}
+
+	query, args := s.ToSQL()
+	return q.adapter.Exec(query, args...)
+}
+
 {{- if $.model.PrimaryKey }}
 
 func (q *{{ $.model.Name }}Query) Delete({{ template "primary_key_params" $.model.PrimaryKey }}) (int64, error) {

generator/client/golang/templates/errors.gotmpl

@@ -0,0 +1,7 @@
+// Code generated by queryx, DO NOT EDIT.
+
+package {{ $.packageName }}
+
+import "errors"
+
+var ErrInsertAllEmptyChanges = errors.New("queryx: insert all with empty changes")

internal/integration/client_test.go

@@ -10,6 +10,7 @@ import (
 
 	"github.com/stretchr/testify/require"
 	"github.com/swiftcarrot/queryx/internal/integration/db"
+	"github.com/swiftcarrot/queryx/internal/integration/db/queryx"
 )
 
 var c *db.QXClient
@@ -65,6 +66,23 @@ func TestCreate(t *testing.T) {
 	require.True(t, user.ID > 0)
 }
 
+func TestInsertAll(t *testing.T) {
+	_, err := c.QueryUserPost().DeleteAll()
+	require.NoError(t, err)
+
+	_, err = c.QueryPost().InsertAll(nil)
+	require.ErrorIs(t, err, db.ErrInsertAllEmptyChanges)
+	_, err = c.QueryPost().InsertAll([]*queryx.PostChange{})
+	require.ErrorIs(t, err, db.ErrInsertAllEmptyChanges)
+
+	inserted, err := c.QueryPost().InsertAll([]*queryx.PostChange{
+		c.ChangePost().SetTitle("title1"),
+		c.ChangePost().SetTitle("title2"),
+	})
+	require.NoError(t, err)
+	require.Equal(t, int64(2), inserted)
+}
+
 func TestCreateEmpty(t *testing.T) {
 	tag, err := c.QueryTag().Create(nil)
 	require.NoError(t, err)