mirror of
https://github.com/sosedoff/pgweb.git
synced 2024-12-15 11:52:12 +03:00
400 lines
8.2 KiB
Go
400 lines
8.2 KiB
Go
package client
|
|
|
|
import (
|
|
"fmt"
|
|
neturl "net/url"
|
|
"reflect"
|
|
"strings"
|
|
"time"
|
|
|
|
_ "github.com/lib/pq"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/sosedoff/pgweb/pkg/command"
|
|
"github.com/sosedoff/pgweb/pkg/connection"
|
|
"github.com/sosedoff/pgweb/pkg/history"
|
|
"github.com/sosedoff/pgweb/pkg/shared"
|
|
"github.com/sosedoff/pgweb/pkg/statements"
|
|
)
|
|
|
|
type Client struct {
|
|
db *sqlx.DB
|
|
tunnel *Tunnel
|
|
serverVersion string
|
|
lastQueryTime time.Time
|
|
External bool
|
|
History []history.Record `json:"history"`
|
|
ConnectionString string `json:"connection_string"`
|
|
}
|
|
|
|
// Struct to hold table rows browsing options
|
|
type RowsOptions struct {
|
|
Where string // Custom filter
|
|
Offset int // Number of rows to skip
|
|
Limit int // Number of rows to fetch
|
|
SortColumn string // Column to sort by
|
|
SortOrder string // Sort direction (ASC, DESC)
|
|
}
|
|
|
|
func getSchemaAndTable(str string) (string, string) {
|
|
chunks := strings.Split(str, ".")
|
|
if len(chunks) == 1 {
|
|
return "public", chunks[0]
|
|
}
|
|
return chunks[0], chunks[1]
|
|
}
|
|
|
|
func New() (*Client, error) {
|
|
str, err := connection.BuildString(command.Opts)
|
|
|
|
if command.Opts.Debug && str != "" {
|
|
fmt.Println("Creating a new client for:", str)
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
db, err := sqlx.Open("postgres", str)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
client := Client{
|
|
db: db,
|
|
ConnectionString: str,
|
|
History: history.New(),
|
|
}
|
|
|
|
client.setServerVersion()
|
|
return &client, nil
|
|
}
|
|
|
|
func NewFromUrl(url string, sshInfo *shared.SSHInfo) (*Client, error) {
|
|
var tunnel *Tunnel
|
|
|
|
if sshInfo != nil {
|
|
if command.Opts.Debug {
|
|
fmt.Println("Opening SSH tunnel for:", sshInfo)
|
|
}
|
|
|
|
tunnel, err := NewTunnel(sshInfo, url)
|
|
if err != nil {
|
|
tunnel.Close()
|
|
return nil, err
|
|
}
|
|
|
|
err = tunnel.Configure()
|
|
if err != nil {
|
|
tunnel.Close()
|
|
return nil, err
|
|
}
|
|
|
|
go tunnel.Start()
|
|
|
|
uri, err := neturl.Parse(url)
|
|
if err != nil {
|
|
tunnel.Close()
|
|
return nil, err
|
|
}
|
|
|
|
// Override remote postgres port with local proxy port
|
|
url = strings.Replace(url, uri.Host, fmt.Sprintf("127.0.0.1:%v", tunnel.Port), 1)
|
|
}
|
|
|
|
if command.Opts.Debug {
|
|
fmt.Println("Creating a new client for:", url)
|
|
}
|
|
|
|
db, err := sqlx.Open("postgres", url)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
client := Client{
|
|
db: db,
|
|
tunnel: tunnel,
|
|
ConnectionString: url,
|
|
History: history.New(),
|
|
}
|
|
|
|
client.setServerVersion()
|
|
return &client, nil
|
|
}
|
|
|
|
func (client *Client) setServerVersion() {
|
|
res, err := client.query("SELECT version()")
|
|
if err != nil || len(res.Rows) < 1 {
|
|
return
|
|
}
|
|
|
|
version := res.Rows[0][0].(string)
|
|
client.serverVersion = strings.Split(version, " ")[1]
|
|
}
|
|
|
|
func (client *Client) Test() error {
|
|
return client.db.Ping()
|
|
}
|
|
|
|
func (client *Client) Info() (*Result, error) {
|
|
return client.query(statements.Info)
|
|
}
|
|
|
|
func (client *Client) Databases() ([]string, error) {
|
|
return client.fetchRows(statements.Databases)
|
|
}
|
|
|
|
func (client *Client) Schemas() ([]string, error) {
|
|
return client.fetchRows(statements.Schemas)
|
|
}
|
|
|
|
func (client *Client) Objects() (*Result, error) {
|
|
return client.query(statements.Objects)
|
|
}
|
|
|
|
func (client *Client) Table(table string) (*Result, error) {
|
|
schema, table := getSchemaAndTable(table)
|
|
return client.query(statements.TableSchema, schema, table)
|
|
}
|
|
|
|
func (client *Client) MaterializedView(name string) (*Result, error) {
|
|
return client.query(statements.MaterializedView, name)
|
|
}
|
|
|
|
func (client *Client) TableRows(table string, opts RowsOptions) (*Result, error) {
|
|
schema, table := getSchemaAndTable(table)
|
|
sql := fmt.Sprintf(`SELECT * FROM "%s"."%s"`, schema, table)
|
|
|
|
if opts.Where != "" {
|
|
sql += fmt.Sprintf(" WHERE %s", opts.Where)
|
|
}
|
|
|
|
if opts.SortColumn != "" {
|
|
if opts.SortOrder == "" {
|
|
opts.SortOrder = "ASC"
|
|
}
|
|
|
|
sql += fmt.Sprintf(` ORDER BY "%s" %s`, opts.SortColumn, opts.SortOrder)
|
|
}
|
|
|
|
if opts.Limit > 0 {
|
|
sql += fmt.Sprintf(" LIMIT %d", opts.Limit)
|
|
}
|
|
|
|
if opts.Offset > 0 {
|
|
sql += fmt.Sprintf(" OFFSET %d", opts.Offset)
|
|
}
|
|
|
|
return client.query(sql)
|
|
}
|
|
|
|
func (client *Client) TableRowsCount(table string, opts RowsOptions) (*Result, error) {
|
|
schema, table := getSchemaAndTable(table)
|
|
sql := fmt.Sprintf(`SELECT COUNT(1) FROM "%s"."%s"`, schema, table)
|
|
|
|
if opts.Where != "" {
|
|
sql += fmt.Sprintf(" WHERE %s", opts.Where)
|
|
}
|
|
|
|
return client.query(sql)
|
|
}
|
|
|
|
func (client *Client) TableInfo(table string) (*Result, error) {
|
|
return client.query(statements.TableInfo, table)
|
|
}
|
|
|
|
func (client *Client) TableIndexes(table string) (*Result, error) {
|
|
schema, table := getSchemaAndTable(table)
|
|
res, err := client.query(statements.TableIndexes, schema, table)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return res, err
|
|
}
|
|
|
|
func (client *Client) TableConstraints(table string) (*Result, error) {
|
|
schema, table := getSchemaAndTable(table)
|
|
res, err := client.query(statements.TableConstraints, schema, table)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return res, err
|
|
}
|
|
|
|
// Returns all active queriers on the server
|
|
func (client *Client) Activity() (*Result, error) {
|
|
chunks := strings.Split(client.serverVersion, ".")
|
|
version := strings.Join(chunks[0:2], ".")
|
|
|
|
query := statements.Activity[version]
|
|
if query == "" {
|
|
query = statements.Activity["default"]
|
|
}
|
|
|
|
return client.query(query)
|
|
}
|
|
|
|
func (client *Client) Query(query string) (*Result, error) {
|
|
res, err := client.query(query)
|
|
|
|
// Save history records only if query did not fail
|
|
if err == nil && !client.hasHistoryRecord(query) {
|
|
client.History = append(client.History, history.NewRecord(query))
|
|
}
|
|
|
|
return res, err
|
|
}
|
|
|
|
func (client *Client) SetReadOnlyMode() error {
|
|
var value string
|
|
if err := client.db.Get(&value, "SHOW default_transaction_read_only;"); err != nil {
|
|
return err
|
|
}
|
|
|
|
if value == "off" {
|
|
_, err := client.db.Exec("SET default_transaction_read_only=on;")
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (client *Client) ServerVersion() string {
|
|
return client.serverVersion
|
|
}
|
|
|
|
func (client *Client) query(query string, args ...interface{}) (*Result, error) {
|
|
// Update the last usage time
|
|
defer func() {
|
|
client.lastQueryTime = time.Now().UTC()
|
|
}()
|
|
|
|
// We're going to force-set transaction mode on every query.
|
|
// This is needed so that default mode could not be changed by user.
|
|
if command.Opts.ReadOnly {
|
|
if err := client.SetReadOnlyMode(); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
action := strings.ToLower(strings.Split(query, " ")[0])
|
|
if action == "update" || action == "delete" {
|
|
res, err := client.db.Exec(query, args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
affected, err := res.RowsAffected()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := Result{
|
|
Columns: []string{"Rows Affected"},
|
|
Rows: []Row{
|
|
Row{affected},
|
|
},
|
|
}
|
|
|
|
return &result, nil
|
|
}
|
|
|
|
rows, err := client.db.Queryx(query, args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer rows.Close()
|
|
|
|
cols, err := rows.Columns()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Make sure to never return null colums
|
|
if cols == nil {
|
|
cols = []string{}
|
|
}
|
|
|
|
result := Result{
|
|
Columns: cols,
|
|
Rows: []Row{},
|
|
}
|
|
|
|
for rows.Next() {
|
|
obj, err := rows.SliceScan()
|
|
|
|
for i, item := range obj {
|
|
if item == nil {
|
|
obj[i] = nil
|
|
} else {
|
|
t := reflect.TypeOf(item).Kind().String()
|
|
|
|
if t == "slice" {
|
|
obj[i] = string(item.([]byte))
|
|
}
|
|
}
|
|
}
|
|
|
|
if err == nil {
|
|
result.Rows = append(result.Rows, obj)
|
|
}
|
|
}
|
|
|
|
result.PrepareBigints()
|
|
|
|
return &result, nil
|
|
}
|
|
|
|
// Close database connection
|
|
func (client *Client) Close() error {
|
|
if client.tunnel != nil {
|
|
client.tunnel.Close()
|
|
}
|
|
|
|
if client.db != nil {
|
|
return client.db.Close()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (client *Client) IsIdle() bool {
|
|
return time.Since(client.lastQueryTime).Hours() > 1
|
|
}
|
|
|
|
// Fetch all rows as strings for a single column
|
|
func (client *Client) fetchRows(q string) ([]string, error) {
|
|
res, err := client.query(q)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Init empty slice so json.Marshal will encode it to "[]" instead of "null"
|
|
results := make([]string, 0)
|
|
|
|
for _, row := range res.Rows {
|
|
results = append(results, row[0].(string))
|
|
}
|
|
|
|
return results, nil
|
|
}
|
|
|
|
func (client *Client) hasHistoryRecord(query string) bool {
|
|
result := false
|
|
|
|
for _, record := range client.History {
|
|
if record.Query == query {
|
|
result = true
|
|
break
|
|
}
|
|
}
|
|
|
|
return result
|
|
}
|