import data

cmd/test/comm/dao.go

@@ -65,6 +65,8 @@ func GetDB() *gorm.DB {
 		&model.IdiomSimple{},
 		&model.PoetryAncient{},
 		&model.PoetryCategory{},
+		&model.Medicine{},
+		&model.Joke{},
 	)
 	if err != nil {
 		fmt.Printf(color.RedString("migrate models failed, error: %s.", err.Error()))

cmd/test/import/advert.go

@@ -25,9 +25,7 @@ func main() {
 		col1 := arr[0]
 
 		po := model.Advert{
-			DataComm: model.DataComm{
-				Name: col1,
-			},
+			Name: col1,
 		}
 
 		if po.Name != "" {

cmd/test/import/joke.go

+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/easysoft/zendata/cmd/test/comm"
+	"github.com/easysoft/zendata/cmd/test/model"
+	fileUtils "github.com/easysoft/zendata/pkg/utils/file"
+	"strings"
+)
+
+func main() {
+	filePath := "/Users/aaron/work/zentao/product/zd/行业数据/笑话.json"
+
+	tableName := model.Joke{}.TableName()
+	db := comm.GetDB()
+	err := db.Exec(fmt.Sprintf(comm.TruncateTable, tableName)).Error
+	if err != nil {
+		panic(err)
+	}
+
+	content := fileUtils.ReadFileBuf(filePath)
+
+	arr := make([]string, 0)
+
+	json.Unmarshal(content, &arr)
+
+	for _, item := range arr {
+		item = strings.TrimSpace(item)
+
+		cate := model.Joke{
+			Content: item,
+		}
+
+		if cate.Content != "" {
+			db.Save(&cate)
+		}
+	}
+}

cmd/test/import/medicine.go

+package main
+
+import (
+	"fmt"
+	"github.com/easysoft/zendata/cmd/test/comm"
+	"github.com/easysoft/zendata/cmd/test/model"
+	fileUtils "github.com/easysoft/zendata/pkg/utils/file"
+	"strings"
+)
+
+func main() {
+	filePath := "/Users/aaron/work/zentao/product/zd/行业数据/医学.txt"
+
+	tableName := model.Medicine{}.TableName()
+	db := comm.GetDB()
+	err := db.Exec(fmt.Sprintf(comm.TruncateTable, tableName)).Error
+	if err != nil {
+		panic(err)
+	}
+
+	content := fileUtils.ReadFile(filePath)
+
+	for _, line := range strings.Split(content, "\n") {
+		arr := strings.Split(strings.TrimSpace(line), " ")
+		col1 := arr[0]
+
+		po := model.Medicine{
+			Name: col1,
+		}
+
+		if po.Name != "" {
+			db.Save(&po)
+		}
+	}
+}

cmd/test/model/model.go

@@ -676,3 +676,21 @@ type PoetryCategory struct {
 func (PoetryCategory) TableName() string {
 	return "biz_data_poetry_category"
 }
+
+type Medicine struct {
+	DataComm
+	Name string `json:"name"`
+}
+
+func (Medicine) TableName() string {
+	return "biz_data_medicine"
+}
+
+type Joke struct {
+	DataComm
+	Content string `json:"content"`
+}
+
+func (Joke) TableName() string {
+	return "biz_data_joke"
+}