e2e/mongo: add e2e test for yaml multiple object (#271)

cmd/gen.go

@@ -17,6 +17,7 @@ package cmd
 import (
 	"bytes"
 	"fmt"
+	"io"
 	"io/fs"
 	"io/ioutil"
 	"os"
@@ -68,7 +69,7 @@ var genCmd = &cobra.Command{
 				if err != nil {
 					return err
 				}
-				if err := yaml.NewDecoder(bytes.NewReader(data)).Decode(&objs); err != nil {
+				if err := decode(data, &objs); err != nil {
 					return err
 				}
 				return nil
@@ -80,7 +81,7 @@ var genCmd = &cobra.Command{
 			if err != nil {
 				return err
 			}
-			if err := yaml.NewDecoder(bytes.NewReader(data)).Decode(&objs); err != nil {
+			if err := decode(data, &objs); err != nil {
 				return err
 			}
 		}
@@ -90,7 +91,6 @@ var genCmd = &cobra.Command{
 		if genGoPackageName == "" {
 			genGoPackageName = genPackageName
 		}
-
 		if plugin != "" {
 			generators = append(generators, generator.NewPluginGenerator(plugin))
 		}
@@ -115,6 +115,19 @@ var genCmd = &cobra.Command{
 	},
 }
 
+func decode(data []byte, meta *generator.Metadata) error {
+	dc := yaml.NewDecoder(bytes.NewReader(data))
+	for {
+		if err := dc.Decode(meta); err != nil {
+			if err == io.EOF {
+				return nil
+			} else {
+				return err
+			}
+		}
+	}
+}
+
 var (
 	input            string
 	output           string

e2e/mongo/nested/nested.yaml

@@ -1,3 +1,4 @@
+---
 User:
   db: mongo
   fields:
@@ -12,7 +13,7 @@ User:
     - Blogs: list<Blog>
     - RegisterDate: time.Time
       attrs: {bsonTag: "registerDate", jsonTag: "registerDate"}
-
+---
 Blog:
   db: mongo
   # To describe that the Blog should only generate the structure , exclude the orm methods

e2e/mongo/user/user.yaml

@@ -1,3 +1,4 @@
+---
 User:
   db: mongo
   fields:
@@ -15,7 +16,7 @@ User:
   table: test_user
   dbname: test
   comment: "all registered user use our systems"
-
+---
 UserBlog:
   db: mongo
   fields:

e2e/mysql/blog.yaml

@@ -1,3 +1,4 @@
+---
 Blog:
   db: mysql
   dbname: test

e2e/mysql/user.yaml

@@ -1,3 +1,4 @@
+---
 User:
   db: mysql
   table: test_user

e2e/mysqlr/blog.yaml

@@ -1,4 +1,4 @@
-
+---
 Blog:
   db: mysqlr
   dbname: test

e2e/mysqlr/user.yaml

@@ -1,3 +1,4 @@
+---
 User:
   db: mysqlr
   dbname: test