From 2047ae9b11d66b33010c76bc2bd1db5a5458b921 Mon Sep 17 00:00:00 2001
From: chenshijun <csj_sky@126.com>
Date: 星期三, 04 九月 2019 19:44:21 +0800
Subject: [PATCH] Merge branch 'master' of ssh://192.168.1.14:29418/syncdb
---
dbself.go | 110 +++++++++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 107 insertions(+), 3 deletions(-)
diff --git a/dbself.go b/dbself.go
index 62f2293..9755dfa 100644
--- a/dbself.go
+++ b/dbself.go
@@ -8,20 +8,20 @@
"path/filepath"
"strings"
"sync"
+ "github.com/jinzhu/gorm"
)
-
const (
PersonSqliteDBPath = "~/workspace/gitblit/dbserver/config/testdb.db"
)
-var syncMut sync.Mutex
+var syncMut sync.Mutex
var SerfDbConn *Conn
// get Conn of db for do execute.
func InitDbConn(dbPath string) error {
- if dbPath == "" {
+ if dbPath == "" {
dbPath = PersonSqliteDBPath
}
@@ -84,6 +84,11 @@
return allResults, nil
}
+var localDb *gorm.DB
+func InitLocalDb(db *gorm.DB) {
+ localDb = db
+}
+
// do exet when get querystring.
func ExecuteQuerySql(sqlString []string) ([]*Rows, error) {
syncMut.Lock()
@@ -96,6 +101,105 @@
return rows, nil
}
+func ExecuteSqlByGorm(sqls []string) (bool,error) {
+ if localDb != nil {
+ localDb.LogMode(false)
+ defer localDb.LogMode(true)
+ var err error
+ tx := localDb.Begin()
+ defer func() {
+ if err !=nil && tx !=nil {
+ tx.Rollback()
+ }
+ }()
+ for _,sql :=range sqls {
+ result := tx.Exec(sql)
+ err = result.Error
+ if err !=nil {
+ return false,result.Error
+ }
+ if result.RowsAffected == 0{
+ fmt.Println("ExecuteSqlByGorm fail")
+ return false,errors.New("ExecuteSqlByGorm fail")
+ }
+ }
+ tx.Commit()
+ return true,nil
+ }
+ return false,errors.New("localDb is nil")
+}
+
+type TableDesc struct {
+ Cid int `json:"cid"`
+ Name string `json:"name"`
+ Type string `json:"type"`
+ Notnull bool `json:"notnull"`
+ DFltValue interface{} `json:"dflt_value"`
+ Pk int `json:"pk"`
+}
+
+type DumpSql struct {
+ Sql string `json:"sql"`
+}
+
+func ExecuteQueryByGorm(tableNames []string) ([]string, error) {
+ localDb.LogMode(false)
+ defer localDb.LogMode(true)
+ if tableNames !=nil {
+ var arr []string
+ var dumpSql []DumpSql
+ for _,table :=range tableNames {
+ dumpSql = make([]DumpSql, 0)
+ var tDescArr []TableDesc
+ tSql := fmt.Sprintf(`PRAGMA table_info("%s")`, table)
+ err := localDb.Raw(tSql).Scan(&tDescArr).Error
+ if err !=nil {
+ return nil,errors.New("tableDesc err")
+ }
+ fmt.Println(table,"'Columns is:",tDescArr)
+ if tDescArr == nil || len(tDescArr) == 0 {
+ return nil,errors.New(table+" has no column")
+ }
+ var columnNames []string
+ for _,col :=range tDescArr {
+ columnNames = append(columnNames, fmt.Sprintf(`'||quote("%s")||'`, col.Name))
+ }
+ if table == "dbTables" {
+ tSql = fmt.Sprintf(`SELECT 'INSERT INTO "%s" VALUES(%s)' as sql FROM "%s" where (analyServerId='' or analyServerId is NULL);`,
+ table,
+ strings.Join(columnNames, ","),
+ table)
+ } else if table == "dbtablepersons" {
+ tSql = fmt.Sprintf(`SELECT 'INSERT INTO "%s" VALUES(%s)' as sql FROM "%s" where tableId in (select id from dbTables where (analyServerId='' or analyServerId is NULL));`,
+ table,
+ strings.Join(columnNames, ","),
+ table)
+ } else {
+ tSql = fmt.Sprintf(`SELECT 'INSERT INTO "%s" VALUES(%s)' as sql FROM "%s";`,
+ table,
+ strings.Join(columnNames, ","),
+ table)
+ }
+
+ fmt.Println("tSql:",tSql)
+
+ err = localDb.Raw(tSql).Scan(&dumpSql).Error
+ if err !=nil {
+ return nil,errors.New("dump err")
+ continue
+ }
+ if len(dumpSql)>0 {
+ for _,d :=range dumpSql {
+ arr = append(arr, d.Sql)
+ }
+ }
+
+ }
+ return arr,nil
+ }
+ return nil,errors.New("tableNames is nil")
+}
+
func Dumpdb() {
var b strings.Builder
--
Gitblit v1.8.0