448 lines
14 KiB
Lua
448 lines
14 KiB
Lua
-- ------------------------------------------------------------------------------ --
|
|
-- TradeSkillMaster --
|
|
-- https://tradeskillmaster.com --
|
|
-- All Rights Reserved - Detailed license information included with addon. --
|
|
-- ------------------------------------------------------------------------------ --
|
|
|
|
local _, TSM = ...
|
|
local QueryClause = TSM.Init("Util.DatabaseClasses.QueryClause")
|
|
local Constants = TSM.Include("Util.DatabaseClasses.Constants")
|
|
local Util = TSM.Include("Util.DatabaseClasses.Util")
|
|
local ObjectPool = TSM.Include("Util.ObjectPool")
|
|
local LibTSMClass = TSM.Include("LibTSMClass")
|
|
local DatabaseQueryClause = LibTSMClass.DefineClass("DatabaseQueryClause")
|
|
local private = {
|
|
objectPool = nil,
|
|
}
|
|
|
|
|
|
|
|
-- ============================================================================
|
|
-- Module Loading
|
|
-- ============================================================================
|
|
|
|
QueryClause:OnModuleLoad(function()
|
|
private.objectPool = ObjectPool.New("DATABASE_QUERY_CLAUSES", DatabaseQueryClause, 1)
|
|
end)
|
|
|
|
|
|
|
|
-- ============================================================================
|
|
-- Module Functions
|
|
-- ============================================================================
|
|
|
|
function QueryClause.Get(query, parent)
|
|
local clause = private.objectPool:Get()
|
|
clause:_Acquire(query, parent)
|
|
return clause
|
|
end
|
|
|
|
|
|
|
|
-- ============================================================================
|
|
-- Class Method Methods
|
|
-- ============================================================================
|
|
|
|
function DatabaseQueryClause.__init(self)
|
|
self._query = nil
|
|
self._operation = nil
|
|
self._parent = nil
|
|
-- comparison
|
|
self._field = nil
|
|
self._value = nil
|
|
self._boundValue = nil
|
|
self._otherField = nil
|
|
-- or / and
|
|
self._subClauses = {}
|
|
end
|
|
|
|
function DatabaseQueryClause._Acquire(self, query, parent)
|
|
self._query = query
|
|
self._parent = parent
|
|
end
|
|
|
|
function DatabaseQueryClause._Release(self)
|
|
self._query = nil
|
|
self._operation = nil
|
|
self._parent = nil
|
|
self._field = nil
|
|
self._value = nil
|
|
self._boundValue = nil
|
|
self._otherField = nil
|
|
for _, clause in ipairs(self._subClauses) do
|
|
clause:_Release()
|
|
end
|
|
wipe(self._subClauses)
|
|
private.objectPool:Recycle(self)
|
|
end
|
|
|
|
|
|
|
|
-- ============================================================================
|
|
-- Public Class Method
|
|
-- ============================================================================
|
|
|
|
function DatabaseQueryClause.Equal(self, field, value, otherField)
|
|
return self:_SetComparisonOperation("EQUAL", field, value, otherField)
|
|
end
|
|
|
|
function DatabaseQueryClause.NotEqual(self, field, value, otherField)
|
|
return self:_SetComparisonOperation("NOT_EQUAL", field, value, otherField)
|
|
end
|
|
|
|
function DatabaseQueryClause.LessThan(self, field, value, otherField)
|
|
return self:_SetComparisonOperation("LESS", field, value, otherField)
|
|
end
|
|
|
|
function DatabaseQueryClause.LessThanOrEqual(self, field, value, otherField)
|
|
return self:_SetComparisonOperation("LESS_OR_EQUAL", field, value, otherField)
|
|
end
|
|
|
|
function DatabaseQueryClause.GreaterThan(self, field, value, otherField)
|
|
return self:_SetComparisonOperation("GREATER", field, value, otherField)
|
|
end
|
|
|
|
function DatabaseQueryClause.GreaterThanOrEqual(self, field, value, otherField)
|
|
return self:_SetComparisonOperation("GREATER_OR_EQUAL", field, value, otherField)
|
|
end
|
|
|
|
function DatabaseQueryClause.Matches(self, field, value)
|
|
return self:_SetComparisonOperation("MATCHES", field, value)
|
|
end
|
|
|
|
function DatabaseQueryClause.Contains(self, field, value)
|
|
return self:_SetComparisonOperation("CONTAINS", field, value)
|
|
end
|
|
|
|
function DatabaseQueryClause.StartsWith(self, field, value)
|
|
return self:_SetComparisonOperation("STARTS_WITH", field, value)
|
|
end
|
|
|
|
function DatabaseQueryClause.IsNil(self, field)
|
|
return self:_SetComparisonOperation("IS_NIL", field)
|
|
end
|
|
|
|
function DatabaseQueryClause.IsNotNil(self, field)
|
|
return self:_SetComparisonOperation("IS_NOT_NIL", field)
|
|
end
|
|
|
|
function DatabaseQueryClause.Custom(self, func, arg)
|
|
return self:_SetComparisonOperation("CUSTOM", func, arg)
|
|
end
|
|
|
|
function DatabaseQueryClause.HashEqual(self, fields, value)
|
|
return self:_SetComparisonOperation("HASH_EQUAL", fields, value)
|
|
end
|
|
|
|
function DatabaseQueryClause.InTable(self, field, value)
|
|
return self:_SetComparisonOperation("IN_TABLE", field, value)
|
|
end
|
|
|
|
function DatabaseQueryClause.NotInTable(self, field, value)
|
|
return self:_SetComparisonOperation("NOT_IN_TABLE", field, value)
|
|
end
|
|
|
|
function DatabaseQueryClause.Or(self)
|
|
return self:_SetSubClauseOperation("OR")
|
|
end
|
|
|
|
function DatabaseQueryClause.And(self)
|
|
return self:_SetSubClauseOperation("AND")
|
|
end
|
|
|
|
|
|
|
|
-- ============================================================================
|
|
-- Private Class Method
|
|
-- ============================================================================
|
|
|
|
function DatabaseQueryClause._GetParent(self)
|
|
return self._parent
|
|
end
|
|
|
|
function DatabaseQueryClause._IsTrue(self, row)
|
|
local value = self._value
|
|
if value == Constants.BOUND_QUERY_PARAM then
|
|
value = self._boundValue
|
|
elseif value == Constants.OTHER_FIELD_QUERY_PARAM then
|
|
value = row:GetField(self._otherField)
|
|
end
|
|
local operation = self._operation
|
|
if operation == "EQUAL" then
|
|
return row[self._field] == value
|
|
elseif operation == "NOT_EQUAL" then
|
|
return row[self._field] ~= value
|
|
elseif operation == "LESS" then
|
|
return row[self._field] < value
|
|
elseif operation == "LESS_OR_EQUAL" then
|
|
return row[self._field] <= value
|
|
elseif operation == "GREATER" then
|
|
return row[self._field] > value
|
|
elseif operation == "GREATER_OR_EQUAL" then
|
|
return row[self._field] >= value
|
|
elseif operation == "MATCHES" then
|
|
return strfind(strlower(row[self._field]), value) and true or false
|
|
elseif operation == "CONTAINS" then
|
|
return strfind(strlower(row[self._field]), value, 1, true) and true or false
|
|
elseif operation == "STARTS_WITH" then
|
|
return strsub(strlower(row[self._field]), 1, #value) == value
|
|
elseif operation == "IS_NIL" then
|
|
return row[self._field] == nil
|
|
elseif operation == "IS_NOT_NIL" then
|
|
return row[self._field] ~= nil
|
|
elseif operation == "CUSTOM" then
|
|
return self._field(row, value) and true or false
|
|
elseif operation == "HASH_EQUAL" then
|
|
return row:CalculateHash(self._field) == value
|
|
elseif operation == "IN_TABLE" then
|
|
return value[row[self._field]] ~= nil
|
|
elseif operation == "NOT_IN_TABLE" then
|
|
return value[row[self._field]] == nil
|
|
elseif operation == "OR" then
|
|
for i = 1, #self._subClauses do
|
|
if self._subClauses[i]:_IsTrue(row) then
|
|
return true
|
|
end
|
|
end
|
|
return false
|
|
elseif operation == "AND" then
|
|
for i = 1, #self._subClauses do
|
|
if not self._subClauses[i]:_IsTrue(row) then
|
|
return false
|
|
end
|
|
end
|
|
return true
|
|
else
|
|
error("Invalid operation: " .. tostring(operation))
|
|
end
|
|
end
|
|
|
|
function DatabaseQueryClause._GetIndexValue(self, indexField)
|
|
if self._operation == "EQUAL" then
|
|
if self._field ~= indexField then
|
|
return
|
|
end
|
|
if self._value == Constants.OTHER_FIELD_QUERY_PARAM then
|
|
return
|
|
elseif self._value == Constants.BOUND_QUERY_PARAM then
|
|
local result = Util.ToIndexValue(self._boundValue)
|
|
return result, result
|
|
else
|
|
local result = Util.ToIndexValue(self._value)
|
|
return result, result
|
|
end
|
|
elseif self._operation == "LESS_OR_EQUAL" then
|
|
if self._field ~= indexField then
|
|
return
|
|
end
|
|
if self._value == Constants.OTHER_FIELD_QUERY_PARAM then
|
|
return
|
|
elseif self._value == Constants.BOUND_QUERY_PARAM then
|
|
return nil, Util.ToIndexValue(self._boundValue)
|
|
else
|
|
return nil, Util.ToIndexValue(self._value)
|
|
end
|
|
elseif self._operation == "GREATER_OR_EQUAL" then
|
|
if self._field ~= indexField then
|
|
return
|
|
end
|
|
if self._value == Constants.OTHER_FIELD_QUERY_PARAM then
|
|
return
|
|
elseif self._value == Constants.BOUND_QUERY_PARAM then
|
|
return Util.ToIndexValue(self._boundValue), nil
|
|
else
|
|
return Util.ToIndexValue(self._value), nil
|
|
end
|
|
elseif self._operation == "STARTS_WITH" then
|
|
if self._field ~= indexField then
|
|
return
|
|
end
|
|
local minValue = nil
|
|
if self._value == Constants.OTHER_FIELD_QUERY_PARAM then
|
|
return
|
|
elseif self._value == Constants.BOUND_QUERY_PARAM then
|
|
minValue = Util.ToIndexValue(self._boundValue)
|
|
else
|
|
minValue = Util.ToIndexValue(self._value)
|
|
end
|
|
-- calculate the max value
|
|
assert(gsub(minValue, "\255", "") ~= "")
|
|
local maxValue = nil
|
|
for i = #minValue, 1, -1 do
|
|
if strsub(minValue, i, i) ~= "\255" then
|
|
maxValue = strsub(minValue, 1, i - 1)..strrep("\255", #minValue - i + 1)
|
|
break
|
|
end
|
|
end
|
|
return minValue, maxValue
|
|
elseif self._operation == "OR" then
|
|
local numSubClauses = #self._subClauses
|
|
if numSubClauses == 0 then
|
|
return
|
|
end
|
|
-- all of the subclauses need to support the same index
|
|
local valueMin, valueMax = self._subClauses[1]:_GetIndexValue(indexField)
|
|
for i = 2, numSubClauses do
|
|
local subClauseValueMin, subClauseValueMax = self._subClauses[i]:_GetIndexValue(indexField)
|
|
if subClauseValueMin ~= valueMin or subClauseValueMax ~= valueMax then
|
|
return
|
|
end
|
|
end
|
|
return valueMin, valueMax
|
|
elseif self._operation == "AND" then
|
|
-- get the most constrained range of index values from the subclauses
|
|
local valueMin, valueMax = nil, nil
|
|
for _, subClause in ipairs(self._subClauses) do
|
|
local subClauseValueMin, subClauseValueMax = subClause:_GetIndexValue(indexField)
|
|
if subClauseValueMin ~= nil and (valueMin == nil or subClauseValueMin > valueMin) then
|
|
valueMin = subClauseValueMin
|
|
end
|
|
if subClauseValueMax ~= nil and (valueMax == nil or subClauseValueMax < valueMax) then
|
|
valueMax = subClauseValueMax
|
|
end
|
|
end
|
|
return valueMin, valueMax
|
|
end
|
|
end
|
|
|
|
function DatabaseQueryClause._GetTrigramIndexValue(self, indexField)
|
|
if self._operation == "EQUAL" then
|
|
if self._field ~= indexField then
|
|
return
|
|
end
|
|
if self._value == Constants.OTHER_FIELD_QUERY_PARAM then
|
|
return
|
|
elseif self._value == Constants.BOUND_QUERY_PARAM then
|
|
return self._boundValue
|
|
else
|
|
return self._value
|
|
end
|
|
elseif self._operation == "CONTAINS" then
|
|
if self._field ~= indexField then
|
|
return
|
|
end
|
|
if self._value == Constants.OTHER_FIELD_QUERY_PARAM then
|
|
return
|
|
elseif self._value == Constants.BOUND_QUERY_PARAM then
|
|
return self._boundValue
|
|
else
|
|
return self._value
|
|
end
|
|
elseif self._operation == "OR" then
|
|
-- all of the subclauses need to support the same trigram value
|
|
local value = nil
|
|
for i = 1, #self._subClauses do
|
|
local subClause = self._subClauses[i]
|
|
local subClauseValue = subClause:_GetTrigramIndexValue(indexField)
|
|
if not subClauseValue then
|
|
return
|
|
end
|
|
if i == 1 then
|
|
value = subClauseValue
|
|
elseif subClauseValue ~= value then
|
|
return
|
|
end
|
|
end
|
|
return value
|
|
elseif self._operation == "AND" then
|
|
-- at least one of the subclauses need to support the trigram
|
|
for _, subClause in ipairs(self._subClauses) do
|
|
local value = subClause:_GetTrigramIndexValue(indexField)
|
|
if value then
|
|
return value
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
function DatabaseQueryClause._IsStrictIndex(self, indexField, indexValueMin, indexValueMax)
|
|
if self._value == Constants.OTHER_FIELD_QUERY_PARAM then
|
|
return false
|
|
end
|
|
if self._operation == "EQUAL" and self._field == indexField and indexValueMin == indexValueMax then
|
|
if self._value == Constants.BOUND_QUERY_PARAM then
|
|
return Util.ToIndexValue(self._boundValue) == indexValueMin
|
|
else
|
|
return Util.ToIndexValue(self._value) == indexValueMin
|
|
end
|
|
elseif self._operation == "GREATER_OR_EQUAL" and self._field == indexField then
|
|
if self._value == Constants.BOUND_QUERY_PARAM then
|
|
return Util.ToIndexValue(self._boundValue) == indexValueMin
|
|
else
|
|
return Util.ToIndexValue(self._value) == indexValueMin
|
|
end
|
|
elseif self._operation == "LESS_OR_EQUAL" and self._field == indexField then
|
|
if self._value == Constants.BOUND_QUERY_PARAM then
|
|
return Util.ToIndexValue(self._boundValue) == indexValueMax
|
|
else
|
|
return Util.ToIndexValue(self._value) == indexValueMax
|
|
end
|
|
elseif self._operation == "OR" and #self._subClauses == 1 then
|
|
return self._subClauses[1]:_IsStrictIndex(indexField, indexValueMin, indexValueMax)
|
|
elseif self._operation == "AND" then
|
|
-- must be strict for all subclauses
|
|
for _, subClause in ipairs(self._subClauses) do
|
|
if not subClause:_IsStrictIndex(indexField, indexValueMin, indexValueMax) then
|
|
return false
|
|
end
|
|
end
|
|
return true
|
|
else
|
|
return false
|
|
end
|
|
end
|
|
|
|
function DatabaseQueryClause._UsesField(self, field)
|
|
if field == self._field or self._operation == "CUSTOM" then
|
|
return true
|
|
end
|
|
if self._operation == "OR" or self._operation == "AND" then
|
|
for i = 1, #self._subClauses do
|
|
if self._subClauses[i]:_UsesField(field) then
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
function DatabaseQueryClause._InsertSubClause(self, subClause)
|
|
assert(self._operation == "OR" or self._operation == "AND")
|
|
tinsert(self._subClauses, subClause)
|
|
self._query:_MarkResultStale()
|
|
return self
|
|
end
|
|
|
|
function DatabaseQueryClause._SetComparisonOperation(self, operation, field, value, otherField)
|
|
assert(not self._operation)
|
|
assert(value == Constants.OTHER_FIELD_QUERY_PARAM or not otherField)
|
|
self._operation = operation
|
|
self._field = field
|
|
self._value = value
|
|
self._otherField = otherField
|
|
self._query:_MarkResultStale()
|
|
return self
|
|
end
|
|
|
|
function DatabaseQueryClause._SetSubClauseOperation(self, operation)
|
|
assert(not self._operation)
|
|
self._operation = operation
|
|
assert(#self._subClauses == 0)
|
|
self._query:_MarkResultStale()
|
|
return self
|
|
end
|
|
|
|
function DatabaseQueryClause._BindParams(self, ...)
|
|
if self._value == Constants.BOUND_QUERY_PARAM then
|
|
self._boundValue = ...
|
|
self._query:_MarkResultStale()
|
|
return 1
|
|
end
|
|
local valuesUsed = 0
|
|
for _, clause in ipairs(self._subClauses) do
|
|
valuesUsed = valuesUsed + clause:_BindParams(select(valuesUsed + 1, ...))
|
|
end
|
|
self._query:_MarkResultStale()
|
|
return valuesUsed
|
|
end
|