Получить имя таблицы из запроса SQL

Question

Я использую golang SQL parser для извлечения информации, связанной с запросом, из фактической строки запроса SQL. Я могу найти тип запроса, используя следующий код:

queryType := sqlparser.StmtType(sqlparser.Preview(sql))
fmt.Println(queryType)

но я не уверен, как получить реальные имена таблиц из SQL-запроса. Документация также не ясна. Только информация, которую я получаю из функции разбора, - это утверждение

Может кто-нибудь подсказать мне, как получить эту информацию с помощью golang sqlparser?

Answer 1

Я взял фрагмент из комментариев @ rob74, который был фантастическим, и изменил его, чтобы он возвращал только имена таблиц. Оригинальный фрагмент также возвращает псевдонимы таблиц. Например.

select * from my_table as mt join other_table using(my_key)
original snippet returns: [my_table, mt, other_table]
new snippet returns:      [my_table, other_table]

оригинальный фрагмент от rob74: play.golang.org/p/B31wr2w1AL8

package main

import (
    "fmt"
    "github.com/xwb1989/sqlparser"
    "reflect"
)

func main() {
    stmt, _ := sqlparser.Parse("select * from my_table as mt join other_table using(my_key)")
    var tables []string
    tables = getTableNames(reflect.Indirect(reflect.ValueOf(stmt)), tables, 0, false)
    fmt.Printf("%s", tables)
}

func getTableNames(v reflect.Value, tables []string, level int, isTable bool) []string {
    switch v.Kind() {
    case reflect.Struct:
        if v.Type().Name() == "TableIdent" {
            // if this is a TableIdent struct, extract the table name
            tableName := v.FieldByName("v").String()
            if tableName != "" && isTable{
                tables = append(tables, tableName)
            }
        } else {
            // otherwise enumerate all fields of the struct and process further
            for i := 0; i < v.NumField(); i++ {
                tables = getTableNames(reflect.Indirect(v.Field(i)), tables, level+1, isTable)
            }
        }
    case reflect.Array, reflect.Slice:
        for i := 0; i < v.Len(); i++ {
            // enumerate all elements of an array/slice and process further
            tables = getTableNames(reflect.Indirect(v.Index(i)), tables, level+1, isTable)
        }
    case reflect.Interface:
        if v.Type().Name() == "SimpleTableExpr" {
            isTable = true
        }
        // get the actual object that satisfies an interface and process further
        tables = getTableNames(reflect.Indirect(reflect.ValueOf(v.Interface())), tables, level+1, isTable)
    }

    return tables
}

Answer 2

Я написал библиотеку для работы со строками для SQL-запросов, чтобы получить имена таблиц:

queryString := sqlstr.NewQueryString(`SELECT column_name(s)
FROM table1
LEFT JOIN table2
ON table1.column_name = table2.column_name;`)

tableNames := queryString.TableNames()

fmt.Println(tableNames)

// Output:
// [table1 table2]

Answer 3

Чтобы получить все имена таблиц, вам нужно вытащить их из Statement, возвращаемого Parse, возможно, используя отражение. Если вы запустите следующий код:

stmt, _ := sqlparser.Parse("insert into my_table set my_column=1")
fmt.Printf("%#v\n", stmt)

вы получите вывод (с отступом для читабельности):

&sqlparser.Insert{
    Action:"insert", 
    Comments:sqlparser.Comments(nil), 
    Ignore:"", 
    Table:sqlparser.TableName{
        Name:sqlparser.TableIdent{v:"my_table"}, 
        Qualifier:sqlparser.TableIdent{v:""}
    }, 
    Partitions:sqlparser.Partitions(nil), 
    Columns:sqlparser.Columns{sqlparser.ColIdent{_:[0]struct { _ []uint8 }{}, val:"my_column", lowered:""}}, 
    Rows:sqlparser.Values{sqlparser.ValTuple{(*sqlparser.SQLVal)(0xc00000a0c0)}}, 
    OnDup:sqlparser.OnDup(nil)
}

Как вы можете видеть, оно содержит (под) поле типа TableIdent, которое содержит запрошенную таблицу из оператора.