From 968035909d5aa12622ddf804fefcd920f78814f7 Mon Sep 17 00:00:00 2001
From: shoopea <shoopea@siteop.biz>
Date: Mon, 8 Jul 2019 11:49:03 +0800
Subject: [PATCH] test

---
 sql.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/sql.go b/sql.go
index 4f273c6..926d44c 100644
--- a/sql.go
+++ b/sql.go
@@ -28,8 +28,13 @@ func initDB() {
 		err = rows.Scan(&name)
 		failOnError(err, "initDB : show tables listing")
 
-		_, err = tx.Exec("drop table " + name)
-		failOnError(err, "initDB : drop table "+name)
+		if ok, _ := regexp.MatchString(`^.*_v$`, name); !ok {
+			_, err = tx.Exec("drop view " + name)
+			failOnError(err, "initDB : drop view "+name)
+		} else {
+			_, err = tx.Exec("drop table " + name)
+			failOnError(err, "initDB : drop table "+name)
+		}
 	}
 	err = rows.Err()
 	failOnError(err, "initDB : show tables listing end")