generator.go 602 字节
Newer Older
aaronchen2k2k's avatar
init  
aaronchen2k2k 已提交
1 2
package action

aaronchen2k2k's avatar
aaronchen2k2k 已提交
3 4
import (
	"github.com/easysoft/zendata/src/model"
aaronchen2k2k's avatar
aaronchen2k2k 已提交
5
	"github.com/easysoft/zendata/src/gen"
aaronchen2k2k's avatar
aaronchen2k2k 已提交
6 7 8 9
	logUtils "github.com/easysoft/zendata/src/utils/log"
	"gopkg.in/yaml.v2"
	"io/ioutil"
)
aaronchen2k2k's avatar
init  
aaronchen2k2k 已提交
10

aaronchen2k2k's avatar
init  
aaronchen2k2k 已提交
11
func Generate(file string, count int, fields string, out string, table string) {
aaronchen2k2k's avatar
aaronchen2k2k 已提交
12 13 14
	definition := model.Definition{}

	yamlContent, err := ioutil.ReadFile(file)
aaronchen2k2k's avatar
aaronchen2k2k 已提交
15 16 17 18
	if err != nil {
		logUtils.Screen("fail to read " + file)
		return
	}
aaronchen2k2k's avatar
aaronchen2k2k 已提交
19

aaronchen2k2k's avatar
aaronchen2k2k 已提交
20 21 22 23
	err = yaml.Unmarshal(yamlContent, &definition)
	if err != nil {
		logUtils.Screen("fail to parse " + file)
		return
aaronchen2k2k's avatar
aaronchen2k2k 已提交
24
	}
aaronchen2k2k's avatar
aaronchen2k2k 已提交
25 26

	gen.Generate(definition, count, fields, out, table)
aaronchen2k2k's avatar
init  
aaronchen2k2k 已提交
27
}