146 lines
3.1 KiB
Go
146 lines
3.1 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"net/url"
|
|
|
|
ora "github.com/sijms/go-ora/v2"
|
|
)
|
|
|
|
func openOracle(c ConnReq) (*sql.DB, error) {
|
|
host := c.Host
|
|
if host == "" {
|
|
host = "127.0.0.1"
|
|
}
|
|
port := c.Port
|
|
if port == 0 {
|
|
port = 1521
|
|
}
|
|
user := c.User
|
|
if user == "" {
|
|
user = "system"
|
|
}
|
|
service := c.Service
|
|
if service == "" {
|
|
if c.Database != "" {
|
|
service = c.Database
|
|
} else {
|
|
service = "XEPDB1"
|
|
}
|
|
}
|
|
dsn := fmt.Sprintf("oracle://%s:%s@%s:%d/%s", url.PathEscape(user), url.PathEscape(c.Password), host, port, service)
|
|
mode := EffectiveSSLMode(c)
|
|
if mode != "disable" {
|
|
q := url.Values{}
|
|
q.Set("SSL", "true")
|
|
if mode == "skip-verify" {
|
|
q.Set("SSL VERIFY", "false")
|
|
} else {
|
|
q.Set("SSL VERIFY", "true")
|
|
}
|
|
dsn = dsn + "?" + q.Encode()
|
|
}
|
|
|
|
tlsCfg, err := tlsConfigFrom(c)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if tlsCfg != nil {
|
|
connector := ora.NewConnector(dsn)
|
|
oc, ok := connector.(*ora.OracleConnector)
|
|
if !ok {
|
|
return nil, fmt.Errorf("oracle: connector 类型异常")
|
|
}
|
|
oc.WithTLSConfig(tlsCfg)
|
|
return sql.OpenDB(oc), nil
|
|
}
|
|
return sql.Open("oracle", dsn)
|
|
}
|
|
|
|
func listDatabasesOracle(db *sql.DB) ([]string, error) {
|
|
// 返回当前用户可见 schema
|
|
rows, err := db.Query(`SELECT username FROM all_users ORDER BY username`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
var out []string
|
|
for rows.Next() {
|
|
var name string
|
|
if err := rows.Scan(&name); err != nil {
|
|
return nil, err
|
|
}
|
|
out = append(out, name)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
func listTablesOracle(db *sql.DB, schema string) ([]string, error) {
|
|
var (
|
|
rows *sql.Rows
|
|
err error
|
|
)
|
|
if schema == "" {
|
|
rows, err = db.Query(`SELECT table_name FROM user_tables ORDER BY table_name`)
|
|
} else {
|
|
rows, err = db.Query(`SELECT table_name FROM all_tables WHERE owner = UPPER(:1) ORDER BY table_name`, schema)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
var out []string
|
|
for rows.Next() {
|
|
var name string
|
|
if err := rows.Scan(&name); err != nil {
|
|
return nil, err
|
|
}
|
|
out = append(out, name)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
func listColumnsOracle(dbh *sql.DB, schema, table string) ([]ColumnInfo, error) {
|
|
owner, t := splitSchemaTable(table)
|
|
if owner == "" {
|
|
owner = schema
|
|
}
|
|
var (
|
|
rows *sql.Rows
|
|
err error
|
|
)
|
|
if owner == "" {
|
|
rows, err = dbh.Query(`
|
|
SELECT column_name, data_type, nullable, data_default
|
|
FROM user_tab_columns WHERE table_name = UPPER(:1)
|
|
ORDER BY column_id`, t)
|
|
} else {
|
|
rows, err = dbh.Query(`
|
|
SELECT column_name, data_type, nullable, data_default
|
|
FROM all_tab_columns WHERE owner = UPPER(:1) AND table_name = UPPER(:2)
|
|
ORDER BY column_id`, owner, t)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
var out []ColumnInfo
|
|
for rows.Next() {
|
|
var name, typ, nullable string
|
|
var def sql.NullString
|
|
if err := rows.Scan(&name, &typ, &nullable, &def); err != nil {
|
|
return nil, err
|
|
}
|
|
col := ColumnInfo{Name: name, Type: typ, Nullable: nullable == "Y"}
|
|
if def.Valid {
|
|
col.Default = def.String
|
|
}
|
|
out = append(out, col)
|
|
}
|
|
if out == nil {
|
|
out = []ColumnInfo{}
|
|
}
|
|
return out, rows.Err()
|
|
}
|