bingyi/gf
1
0
Fork 0
mirror of https://gitee.com/johng/gf synced 2026-07-08 06:35:16 +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

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
}