From d8ac88cfb72e3aac3a89c3cfe77774be3024a24c Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 11 八月 2023 17:54:43 +0800
Subject: [PATCH] update

---
 model/index.go |  217 +++++++++++++++++++++++++++---------------------------
 1 files changed, 109 insertions(+), 108 deletions(-)

diff --git a/model/index.go b/model/index.go
index 04856fa..0da8fa8 100644
--- a/model/index.go
+++ b/model/index.go
@@ -1,108 +1,109 @@
-package model
-
-import (
-	"aps_crm/conf"
-	"aps_crm/pkg/logx"
-	"aps_crm/pkg/mysqlx"
-	"aps_crm/pkg/safe"
-	"gorm.io/gorm/schema"
-)
-
-func Init() error {
-	if err := mysqlx.Init(&conf.Conf.Mysql, logx.GetLogger()); err != nil {
-		return err
-	}
-
-	if err := RegisterTables(); err != nil {
-		return err
-	}
-	safe.Go(func() {
-		InsertDefaultData()
-	})
-	return nil
-}
-
-func RegisterTables() error {
-	db := mysqlx.GetDB()
-	err := db.AutoMigrate(
-		User{},
-		Menu{},
-		JwtBlacklist{},
-		Country{},
-		Province{},
-		City{},
-		Client{},
-		ClientLevel{},
-		ClientOrigin{},
-		ClientStatus{},
-		ClientType{},
-		Contact{},
-		EnterpriseNature{},
-		EnterpriseScale{},
-		FollowRecord{},
-		Industry{},
-		RegisteredCapital{},
-		SaleChance{},
-		SalesLeads{},
-		SalesSources{},
-		SaleStage{},
-		SaleType{},
-		RegularCustomers{},
-		Possibility{},
-		Status{},
-		Quotation{},
-		MasterOrder{},
-		SubOrder{},
-		Product{},
-		SalesDetails{},
-		SalesReturn{},
-		SalesRefund{},
-		Contract{},
-		Plan{},
-		ServiceContract{},
-		OrderManage{},
-		ServiceFollowup{},
-		CustomerServiceSheet{},
-		ServiceFeeManage{},
-		Authority{},
-		Api{},
-		Department{},
-		Satisfaction{},
-		TimelyRate{},
-		SolveRate{},
-		IsVisit{},
-		IsVisit{},
-		ReportSource{},
-		OrderType{},
-		ServiceContractStatus{},
-		ServiceContractType{},
-		RefundMethod{},
-		IsInvoice{},
-		AccountId{},
-		SalesReturnStatus{},
-		Repository{},
-		QuotationStatus{},
-		Currency{},
-	)
-	return err
-}
-
-type InitDefaultData interface {
-	InitDefaultData() error
-}
-
-func InsertDefaultData() {
-	models := []interface{}{
-		NewServiceTypeSearch(),
-		NewPriorityLevelSearch(),
-		NewSeveritySearch(),
-		NewTimeSpentSearch(),
-	}
-	for _, model := range models {
-		if id, ok := model.(InitDefaultData); ok {
-			if err := id.InitDefaultData(); err != nil {
-				logx.Errorf("InitDefaultData for table: %v, err :%v", model.(schema.Tabler).TableName(), err.Error())
-			}
-		}
-	}
-}
+package model
+
+import (
+	"aps_crm/conf"
+	"aps_crm/pkg/logx"
+	"aps_crm/pkg/mysqlx"
+	"aps_crm/pkg/safe"
+	"gorm.io/gorm/schema"
+)
+
+func Init() error {
+	if err := mysqlx.Init(&conf.Conf.Mysql, logx.GetLogger()); err != nil {
+		return err
+	}
+
+	if err := RegisterTables(); err != nil {
+		return err
+	}
+	safe.Go(func() {
+		InsertDefaultData()
+	})
+	return nil
+}
+
+func RegisterTables() error {
+	db := mysqlx.GetDB()
+	err := db.AutoMigrate(
+		User{},
+		Menu{},
+		JwtBlacklist{},
+		Country{},
+		Province{},
+		City{},
+		Client{},
+		ClientLevel{},
+		ClientOrigin{},
+		ClientStatus{},
+		ClientType{},
+		Contact{},
+		EnterpriseNature{},
+		EnterpriseScale{},
+		FollowRecord{},
+		Industry{},
+		RegisteredCapital{},
+		SaleChance{},
+		SalesLeads{},
+		SalesSources{},
+		SaleStage{},
+		SaleType{},
+		RegularCustomers{},
+		Possibility{},
+		Status{},
+		Quotation{},
+		MasterOrder{},
+		SubOrder{},
+		Product{},
+		SalesDetails{},
+		SalesReturn{},
+		SalesRefund{},
+		Contract{},
+		Plan{},
+		ServiceContract{},
+		OrderManage{},
+		ServiceFollowup{},
+		CustomerServiceSheet{},
+		ServiceFeeManage{},
+		Authority{},
+		Api{},
+		Department{},
+		Satisfaction{},
+		TimelyRate{},
+		SolveRate{},
+		IsVisit{},
+		IsVisit{},
+		ReportSource{},
+		OrderType{},
+		ServiceContractStatus{},
+		ServiceContractType{},
+		RefundMethod{},
+		IsInvoice{},
+		AccountId{},
+		SalesReturnStatus{},
+		Repository{},
+		QuotationStatus{},
+		Currency{},
+		CollectionProjection{},
+	)
+	return err
+}
+
+type InitDefaultData interface {
+	InitDefaultData() error
+}
+
+func InsertDefaultData() {
+	models := []interface{}{
+		NewServiceTypeSearch(),
+		NewPriorityLevelSearch(),
+		NewSeveritySearch(),
+		NewTimeSpentSearch(),
+	}
+	for _, model := range models {
+		if id, ok := model.(InitDefaultData); ok {
+			if err := id.InitDefaultData(); err != nil {
+				logx.Errorf("InitDefaultData for table: %v, err :%v", model.(schema.Tabler).TableName(), err.Error())
+			}
+		}
+	}
+}

--
Gitblit v1.8.0