From 2c670b3b21d5132f0c804dec0ed87cbe7b6d7389 Mon Sep 17 00:00:00 2001
From: wenxu12345 <44203734@qq.com>
Date: Tue, 8 Feb 2022 20:25:39 +0800
Subject: [PATCH] Refactor code

---
 pkg/common/db/mysql.go | 69 +++++++++++++++++++++---------------------
 1 file changed, 34 insertions(+), 35 deletions(-)

diff --git a/pkg/common/db/mysql.go b/pkg/common/db/mysql.go
index 56ca3648c..efcfc0f63 100644
--- a/pkg/common/db/mysql.go
+++ b/pkg/common/db/mysql.go
@@ -61,41 +61,40 @@ func initMysqlDB() {
 		&Black{}, &ChatLog{})
 	db.Set("gorm:table_options", "CHARSET=utf8")
 
-	//
-	//if !db.HasTable(&Friend{}) {
-	//	log.NewInfo("CreateTable Friend")
-	//	db.CreateTable(&Friend{})
-	//}
-	//
-	//if !db.HasTable(&FriendRequest{}) {
-	//	log.NewInfo("CreateTable FriendRequest")
-	//	db.CreateTable(&FriendRequest{})
-	//}
-	//
-	//if !db.HasTable(&Group{}) {
-	//	log.NewInfo("CreateTable Group")
-	//	db.CreateTable(&Group{})
-	//}
-	//
-	//if !db.HasTable(&GroupMember{}) {
-	//	log.NewInfo("CreateTable GroupMember")
-	//	db.CreateTable(&GroupMember{})
-	//}
-	//
-	//if !db.HasTable(&GroupRequest{}) {
-	//	log.NewInfo("CreateTable GroupRequest")
-	//	db.CreateTable(&GroupRequest{})
-	//}
-	//
-	//if !db.HasTable(&User{}) {
-	//	log.NewInfo("CreateTable User")
-	//	db.CreateTable(&User{})
-	//}
-	//
-	//if !db.HasTable(&Black{}) {
-	//	log.NewInfo("CreateTable Black")
-	//	db.CreateTable(&Black{})
-	//}
+	if !db.HasTable(&Friend{}) {
+		log.NewInfo("CreateTable Friend")
+		db.CreateTable(&Friend{})
+	}
+
+	if !db.HasTable(&FriendRequest{}) {
+		log.NewInfo("CreateTable FriendRequest")
+		db.CreateTable(&FriendRequest{})
+	}
+
+	if !db.HasTable(&Group{}) {
+		log.NewInfo("CreateTable Group")
+		db.CreateTable(&Group{})
+	}
+
+	if !db.HasTable(&GroupMember{}) {
+		log.NewInfo("CreateTable GroupMember")
+		db.CreateTable(&GroupMember{})
+	}
+
+	if !db.HasTable(&GroupRequest{}) {
+		log.NewInfo("CreateTable GroupRequest")
+		db.CreateTable(&GroupRequest{})
+	}
+
+	if !db.HasTable(&User{}) {
+		log.NewInfo("CreateTable User")
+		db.CreateTable(&User{})
+	}
+
+	if !db.HasTable(&Black{}) {
+		log.NewInfo("CreateTable Black")
+		db.CreateTable(&Black{})
+	}
 
 	return