bingyi/gf
1
0
Fork 0
mirror of https://gitee.com/johng/gf synced 2026-06-06 02:25:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add tables method of MSSQL, Oracle and PgSQL

Browse Source
This commit is contained in:
wenzi1
2020-03-02 23:41:33 +08:00
parent 270af8accb
commit 8971ad8445
3 changed files with 45 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
database/gdb/gdb_mssql.go
View File

@ -163,8 +163,22 @@ func (db *dbMssql) parseSql(sql string) string {
}
// Tables retrieves and returns the tables of current schema.
// TODO
func (db *dbMssql) Tables(schema ...string) (tables []string, err error) {
var result Result
link, err := db.getSlave(schema...)
if err != nil {
return nil, err
}
result, err = db.doGetAll(link, `SELECT NAME FROM SYSOBJECTS WHERE XTYPE='U' AND STATUS >= 0 ORDER BY NAME`)
if err != nil {
return
}
for _, m := range result {
for _, v := range m {
tables = append(tables, strings.ToLower(v.String()))
}
}
return
}

12
database/gdb/gdb_oracle.go
View File

@ -123,8 +123,18 @@ func (db *dbOracle) parseSql(sql string) string {
}
// Tables retrieves and returns the tables of current schema.
// TODO
func (db *dbOracle) Tables(schema ...string) (tables []string, err error) {
var result Result
result, err = db.doGetAll(nil, "SELECT TABLE_NAME FROM USER_TABLES ORDER BY TABLE_NAME")
if err != nil {
return
}
for _, m := range result {
for _, v := range m {
tables = append(tables, strings.ToLower(v.String()))
}
}
return
}

20
database/gdb/gdb_pgsql.go
View File

@ -62,8 +62,26 @@ func (db *dbPgsql) handleSqlBeforeExec(sql string) string {
}
// Tables retrieves and returns the tables of current schema.
// TODO
func (db *dbPgsql) Tables(schema ...string) (tables []string, err error) {
var result Result
link, err := db.getSlave(schema...)
if err != nil {
return nil, err
}
query := "SELECT TABLENAME FROM PG_TABLES WHERE SCHEMANAME = 'public' ORDER BY TABLENAME"
if len(schema) > 0 && schema[0] != "" {
query = fmt.Sprintf("SELECT TABLENAME FROM PG_TABLES WHERE SCHEMANAME = '%s' ORDER BY TABLENAME", schema[0])
}
result, err = db.doGetAll(link, query)
if err != nil {
return
}
for _, m := range result {
for _, v := range m {
tables = append(tables, v.String())
}
}
return
}