mirror of
https://github.com/grafana/grafana.git
synced 2025-07-30 02:42:39 +08:00
99 lines
2.7 KiB
Go
99 lines
2.7 KiB
Go
//go:build !arm
|
|
|
|
package sql
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
sqle "github.com/dolthub/go-mysql-server"
|
|
mysql "github.com/dolthub/go-mysql-server/sql"
|
|
|
|
"github.com/dolthub/go-mysql-server/sql/analyzer"
|
|
"github.com/grafana/grafana-plugin-sdk-go/data"
|
|
)
|
|
|
|
// DB is a database that can execute SQL queries against a set of Frames.
|
|
type DB struct{}
|
|
|
|
// GoMySQLServerError represents an error from the underlying Go MySQL Server
|
|
type GoMySQLServerError struct {
|
|
Err error
|
|
}
|
|
|
|
// Error implements the error interface
|
|
func (e *GoMySQLServerError) Error() string {
|
|
return fmt.Sprintf("error in go-mysql-server: %v", e.Err)
|
|
}
|
|
|
|
// Unwrap provides the original error for errors.Is/As
|
|
func (e *GoMySQLServerError) Unwrap() error {
|
|
return e.Err
|
|
}
|
|
|
|
// WrapGoMySQLServerError wraps errors from Go MySQL Server with additional context
|
|
func WrapGoMySQLServerError(err error) error {
|
|
// Don't wrap nil errors
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
|
|
// Check if it's a function not found error or other specific GMS errors
|
|
if isFunctionNotFoundError(err) {
|
|
return &GoMySQLServerError{Err: err}
|
|
}
|
|
|
|
// Return original error if it's not one we want to wrap
|
|
return err
|
|
}
|
|
|
|
// isFunctionNotFoundError checks if the error is related to a function not being found
|
|
func isFunctionNotFoundError(err error) bool {
|
|
return mysql.ErrFunctionNotFound.Is(err)
|
|
}
|
|
|
|
// QueryFrames runs the sql query query against a database created from frames, and returns the frame.
|
|
// The RefID of each frame becomes a table in the database.
|
|
// It is expected that there is only one frame per RefID.
|
|
// The name becomes the name and RefID of the returned frame.
|
|
func (db *DB) QueryFrames(ctx context.Context, name string, query string, frames []*data.Frame) (*data.Frame, error) {
|
|
// We are parsing twice due to TablesList, but don't care fow now. We can save the parsed query and reuse it later if we want.
|
|
if allow, err := AllowQuery(query); err != nil || !allow {
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
pro := NewFramesDBProvider(frames)
|
|
session := mysql.NewBaseSession()
|
|
mCtx := mysql.NewContext(ctx, mysql.WithSession(session))
|
|
|
|
// Select the database in the context
|
|
mCtx.SetCurrentDatabase(dbName)
|
|
|
|
// Empty dir does not disable secure_file_priv
|
|
//ctx.SetSessionVariable(ctx, "secure_file_priv", "")
|
|
|
|
// TODO: Check if it's wise to reuse the existing provider, rather than creating a new one
|
|
a := analyzer.NewDefault(pro)
|
|
|
|
engine := sqle.New(a, &sqle.Config{
|
|
IsReadOnly: true,
|
|
})
|
|
|
|
schema, iter, _, err := engine.Query(mCtx, query)
|
|
if err != nil {
|
|
return nil, WrapGoMySQLServerError(err)
|
|
}
|
|
|
|
f, err := convertToDataFrame(mCtx, iter, schema)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
f.Name = name
|
|
f.RefID = name
|
|
|
|
return f, nil
|
|
}
|