From 28b35dae2af440fd9586fc3c45246914751b3d96 Mon Sep 17 00:00:00 2001
From: fwhez <1728565484@qq.com>
Date: Mon, 12 Nov 2018 11:12:32 +0800
Subject: [PATCH] Improve code quality

---
 main.go      |  2 +-
 main_test.go | 12 ++++++------
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/main.go b/main.go
index 65ed4fd5..f4aa5573 100644
--- a/main.go
+++ b/main.go
@@ -114,7 +114,7 @@ func (s *DB) CopyDB() (*sql.DB, error) {
 	if s.copyDB != nil {
 		return s.copyDB, nil
 	}
-		return sql.Open(s.Dialect().GetName(), s.DataSource())
+	return sql.Open(s.Dialect().GetName(), s.DataSource())
 }
 
 type closer interface {
diff --git a/main_test.go b/main_test.go
index 62e798b1..03eaf8e7 100644
--- a/main_test.go
+++ b/main_test.go
@@ -1061,8 +1061,8 @@ func TestBlockGlobalUpdate(t *testing.T) {
 
 func TestDB_DataSource(t *testing.T) {
 	source := "user=gorm password=gorm DB.name=gorm port=9920 sslmode=disable"
-	db,er := gorm.Open("postgres", source)
-	if er!=nil {
+	db, er := gorm.Open("postgres", source)
+	if er != nil {
 		panic(fmt.Sprintf("No error should happen when connecting to test database, but got err=%+v", er))
 	}
 	if db.DataSource() != source {
@@ -1072,12 +1072,12 @@ func TestDB_DataSource(t *testing.T) {
 }
 func TestDB_CopyIn(t *testing.T) {
 	source := "user=gorm password=gorm DB.name=gorm port=9920 sslmode=disable"
-	db,er := gorm.Open("postgres", source)
-	if er!=nil {
+	db, er := gorm.Open("postgres", source)
+	if er != nil {
 		panic(fmt.Sprintf("No error should happen when connecting to test database, but got err=%+v", er))
 	}
-	e:=db.Exec("create table if not exists example(name varchar, age integer)").Error
-	defer func(){
+	e := db.Exec("create table if not exists example(name varchar, age integer)").Error
+	defer func() {
 		er := db.Exec("drop table if exists example").Error
 		if er != nil {
 			t.Fatal(e.Error())