diff --git a/.gitignore b/.gitignore index 1744184..1f97881 100644 --- a/.gitignore +++ b/.gitignore @@ -23,6 +23,8 @@ go.work.sum # env file .env +# IDEs +.idea/ # mac .DS_Store diff --git a/internal/pkg/object/command/trino/client.go b/internal/pkg/object/command/trino/client.go index de52133..e8e1f4c 100644 --- a/internal/pkg/object/command/trino/client.go +++ b/internal/pkg/object/command/trino/client.go @@ -7,6 +7,7 @@ import ( "io" "net/http" "os" + "strings" "github.com/patterninc/heimdall/pkg/object/cluster" "github.com/patterninc/heimdall/pkg/object/job" @@ -61,6 +62,7 @@ func newRequest(r *plugin.Runtime, j *job.Job, c *cluster.Cluster) (*request, er return nil, err } } + jobCtx.Query = normalizeTrinoQuery(jobCtx.Query) // form context for trino request req := &request{ @@ -177,3 +179,8 @@ func (r *request) api(req *http.Request) error { return nil } + +func normalizeTrinoQuery(query string) string { + // Trino does not support semicolon at the end of the query, so we remove it if present + return strings.TrimSuffix(query, ";") +}