mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Rodrigo ZhouandGitHub
5d64c0b65f
## What changed?
Unify Visibility query converters.
The new query converter is the struct `query.QueryConverter[ExprT]`. The
generic parameter specify the output type of the converter (in ES is
`elastic.Query`, in SQL is `sqlparser.Expr`).
The query converter has a `query.StoreQueryConverter[ExprT]`. This is
the Visibility store specific implementation for building the output
query.
For example, the store needs to implement `BuildAndExpr(exprs)` to tell
the query converter how to build the `AND` expression. In SQL, it's
`*sqlparser.AndExpr{exprs}`. In Elasticsearch, it's
`elastic.NewBoolQuery().Filter(exprs)`.
The store query converter for SQL needs a plugin query converter called
`sqlplugin.VisibilityQueryConverter` which is implemented by each plugin
(MySQL, PostgreSQL, SQLite). This interface is needed to build DB
specific syntax for `KeywordList` and `Text` searches as well as to
build the final `SELECT` statements.
Added dynamic config `system.VisibilityEnableUnifiedQueryConverter` that
acts as switch between the legacy and the unified query converters.
## Why?
There are two query converters: one for Elasticsearch, and another for
SQL.
They are somewhat similar, and have quite a few code duplications for
validation, modifying the query, etc.
Unifying the query converter ensures that we provide the exact same
behavior across different Visibility stores.
## How did you test it?
- [x] built
- [x] run locally and tested manually
- [x] covered by existing tests
- [x] added new unit test(s)
- [ ] added new functional test(s)
I've added a lot of unit tests for the new code with coverage close to
100%.
Note that there has been basically no changes to functional tests, which
suggests that the change works well.
## Potential risks
There might be some edge cases that either query converters that this
unified query converter might deal differently.
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package query
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
enumspb "go.temporal.io/api/enums/v1"
|
|
"go.temporal.io/api/serviceerror"
|
|
)
|
|
|
|
type (
|
|
ConverterError struct {
|
|
message string
|
|
}
|
|
)
|
|
|
|
var (
|
|
MalformedSqlQueryErrMessage = "malformed SQL query"
|
|
NotSupportedErrMessage = "operation is not supported"
|
|
InvalidExpressionErrMessage = "invalid expression"
|
|
)
|
|
|
|
func NewConverterError(format string, a ...interface{}) error {
|
|
message := fmt.Sprintf(format, a...)
|
|
return &ConverterError{message: message}
|
|
}
|
|
|
|
func (c *ConverterError) Error() string {
|
|
return c.message
|
|
}
|
|
|
|
func (c *ConverterError) ToInvalidArgument() error {
|
|
return serviceerror.NewInvalidArgumentf("invalid query: %v", c)
|
|
}
|
|
|
|
func NewOperatorNotSupportedError(
|
|
saName string,
|
|
saType enumspb.IndexedValueType,
|
|
operator string,
|
|
) error {
|
|
return NewConverterError(
|
|
"%s: operator '%s' not supported for %s type search attribute '%s'",
|
|
NotSupportedErrMessage,
|
|
strings.ToUpper(operator),
|
|
saType.String(),
|
|
saName,
|
|
)
|
|
}
|
|
|
|
func wrapConverterError(message string, err error) error {
|
|
var converterErr *ConverterError
|
|
if errors.As(err, &converterErr) {
|
|
return NewConverterError("%s: %v", message, converterErr)
|
|
}
|
|
return err
|
|
}
|