public SoodaQuerySource(SoodaTransaction transaction, Sooda.Schema.ClassInfo classInfo, SoqlBooleanExpression where)
{
_transaction = transaction;
_classInfo = classInfo;
_options = SoodaSnapshotOptions.Default;
_where = where;
}
public SoodaQuerySource(SoodaTransaction transaction, Sooda.Schema.ClassInfo classInfo, SoodaSnapshotOptions options)
{
_transaction = transaction;
_classInfo = classInfo;
_options = options;
_where = null;
}
public override IDataReader LoadObjectList(SchemaInfo schemaInfo, ClassInfo classInfo, SoodaWhereClause whereClause, SoodaOrderBy orderBy, int startIdx, int pageCount, SoodaSnapshotOptions options, out TableInfo[] tables)
{
try
{
Queue <_QueueItem> queue = new Queue <_QueueItem>();
List <TableInfo> tablesArrayList = new List <TableInfo>(classInfo.UnifiedTables.Count);
SoqlQueryExpression queryExpression = new SoqlQueryExpression();
queryExpression.StartIdx = startIdx;
queryExpression.PageCount = pageCount;
queryExpression.From.Add(classInfo.Name);
queryExpression.FromAliases.Add("");
foreach (TableInfo ti in classInfo.UnifiedTables)
{
tablesArrayList.Add(ti);
foreach (FieldInfo fi in ti.Fields)
{
SoqlPathExpression pathExpr = new SoqlPathExpression(fi.Name);
queryExpression.SelectExpressions.Add(pathExpr);
queryExpression.SelectAliases.Add("");
if (fi.ReferencedClass != null && fi.PrefetchLevel > 0 && ((options & SoodaSnapshotOptions.PrefetchRelated) != 0))
{
_QueueItem item = new _QueueItem();
item.classInfo = fi.ReferencedClass;
item.level = fi.PrefetchLevel;
item.prefix = pathExpr;
queue.Enqueue(item);
}
}
}
while (queue.Count > 0)
{
_QueueItem it = queue.Dequeue();
foreach (TableInfo ti in it.classInfo.UnifiedTables)
{
tablesArrayList.Add(ti);
foreach (FieldInfo fi in ti.Fields)
{
// TODO - this relies on the fact that path expressions
// are never reconstructed or broken. We simply share previous prefix
// perhaps it's cleaner to Clone() the expression here
SoqlPathExpression extendedExpression = new SoqlPathExpression(it.prefix, fi.Name);
queryExpression.SelectExpressions.Add(extendedExpression);
queryExpression.SelectAliases.Add("");
if (it.level >= 1 && fi.PrefetchLevel > 0 && fi.ReferencedClass != null)
{
_QueueItem newItem = new _QueueItem();
newItem.classInfo = fi.ReferencedClass;
newItem.prefix = extendedExpression;
newItem.level = it.level - 1;
queue.Enqueue(newItem);
}
}
}
}
if (whereClause != null && whereClause.WhereExpression != null)
{
queryExpression.WhereClause = whereClause.WhereExpression;
}
if (orderBy != null)
{
queryExpression.SetOrderBy(orderBy);
}
string query = SoqlToSql(queryExpression, schemaInfo, false);
IDbCommand cmd = Connection.CreateCommand();
try
{
cmd.CommandTimeout = CommandTimeout;
}
catch (NotSupportedException e)
{
logger.Debug("CommandTimeout not supported. {0}", e.Message);
}
if (Transaction != null)
{
cmd.Transaction = this.Transaction;
}
SqlBuilder.BuildCommandWithParameters(cmd, false, query, whereClause.Parameters, false);
tables = tablesArrayList.ToArray();
return(TimedExecuteReader(cmd));
}
catch (Exception ex)
{
logger.Error("Exception in LoadObjectList: {0}", ex);
throw;
}
}
public abstract IDataReader LoadObjectList(SchemaInfo schemaInfo, ClassInfo classInfo, SoodaWhereClause whereClause, SoodaOrderBy orderBy, int startIdx, int pageCount, SoodaSnapshotOptions options, out TableInfo[] tables);
private void LoadList(SoodaTransaction transaction, SoodaWhereClause whereClause, SoodaOrderBy orderBy, int startIdx, int pageCount, SoodaSnapshotOptions options, string[] involvedClassNames, bool useCache)
{
ISoodaObjectFactory factory = transaction.GetFactory(classInfo);
string cacheKey = null;
if (useCache)
{
// cache makes sense only on clean database
if (!transaction.HasBeenPrecommitted(classInfo))
{
cacheKey = SoodaCache.GetCollectionKey(classInfo, whereClause);
}
IEnumerable keysCollection = transaction.LoadCollectionFromCache(cacheKey, logger);
if (keysCollection != null)
{
foreach (object o in keysCollection)
{
SoodaObject obj = factory.GetRef(transaction, o);
// this binds to cache
obj.EnsureFieldsInited();
items.Add(obj);
}
if (orderBy != null)
{
items.Sort(orderBy.GetComparer());
}
count = items.Count;
if (startIdx > 0)
{
if (startIdx < count)
{
items.RemoveRange(0, startIdx);
}
else
{
items.Clear();
}
}
if (pageCount != -1 && pageCount < items.Count)
{
items.RemoveRange(pageCount, items.Count - pageCount);
}
return;
}
}
SoodaDataSource ds = transaction.OpenDataSource(classInfo.GetDataSource());
if ((options & SoodaSnapshotOptions.KeysOnly) != 0)
{
if (pageCount != -1)
{
using (IDataReader reader = ds.LoadMatchingPrimaryKeys(transaction.Schema, classInfo, whereClause, orderBy, 0, -1))
{
count = 0;
while (reader.Read())
{
count++;
}
}
}
using (IDataReader reader = ds.LoadMatchingPrimaryKeys(transaction.Schema, classInfo, whereClause, orderBy, startIdx, pageCount))
{
while (reader.Read())
{
SoodaObject obj = SoodaObject.GetRefFromKeyRecordHelper(transaction, factory, reader);
items.Add(obj);
}
if (pageCount == -1)
{
count = items.Count;
}
}
}
else
{
if (pageCount != -1)
{
using (IDataReader reader = ds.LoadMatchingPrimaryKeys(transaction.Schema, classInfo, whereClause, orderBy, 0, -1))
{
count = 0;
while (reader.Read())
{
count++;
}
}
}
TableInfo[] loadedTables;
using (IDataReader reader = ds.LoadObjectList(transaction.Schema, classInfo, whereClause, orderBy, startIdx, pageCount, options, out loadedTables))
{
while (reader.Read())
{
SoodaObject obj = SoodaObject.GetRefFromRecordHelper(transaction, factory, reader, 0, loadedTables, 0);
if ((options & SoodaSnapshotOptions.VerifyAfterLoad) != 0 && whereClause != null && !whereClause.Matches(obj, false))
{
continue; // don't add the object
}
items.Add(obj);
}
if (pageCount == -1)
{
count = items.Count;
}
}
}
if (cacheKey != null && useCache && startIdx == 0 && pageCount == -1 && involvedClassNames != null)
{
TimeSpan expirationTimeout;
bool slidingExpiration;
if (transaction.CachingPolicy.GetExpirationTimeout(
classInfo, whereClause, orderBy, startIdx, pageCount, items.Count,
out expirationTimeout, out slidingExpiration))
{
transaction.StoreCollectionInCache(cacheKey, classInfo, items, involvedClassNames, (options & SoodaSnapshotOptions.KeysOnly) == 0, expirationTimeout, slidingExpiration);
}
}
}
public SoodaObjectListSnapshot(SoodaTransaction t, SoodaWhereClause whereClause, SoodaOrderBy orderBy, int startIdx, int pageCount, SoodaSnapshotOptions options, ClassInfo ci)
{
this.classInfo = ci;
string[] involvedClasses = null;
bool useCache;
if ((options & SoodaSnapshotOptions.NoCache) != 0)
{
useCache = false;
}
else if ((options & SoodaSnapshotOptions.Cache) != 0)
{
useCache = true;
}
else
{
useCache = t.CachingPolicy.ShouldCacheCollection(ci, whereClause, orderBy, startIdx, pageCount);
}
if (whereClause != null && whereClause.WhereExpression != null)
{
if ((options & SoodaSnapshotOptions.NoWriteObjects) == 0 || useCache)
{
try
{
GetInvolvedClassesVisitor gic = new GetInvolvedClassesVisitor(classInfo);
gic.GetInvolvedClasses(whereClause.WhereExpression);
involvedClasses = gic.ClassNames;
}
catch
{
// logger.Warn("{0}", ex);
// cannot detect involved classes (probably because of RAWQUERY)
// - precommit all objects
// if we get here, involvedClasses remains set to null
}
}
}
else
{
// no where clause
involvedClasses = new string[] { ci.Name };
}
if ((options & SoodaSnapshotOptions.NoWriteObjects) == 0)
{
t.PrecommitClasses(involvedClasses);
}
LoadList(t, whereClause, orderBy, startIdx, pageCount, options, involvedClasses, useCache);
}
private void LoadList(SoodaTransaction transaction, SoodaWhereClause whereClause, SoodaOrderBy orderBy, int startIdx, int pageCount, SoodaSnapshotOptions options, string[] involvedClassNames, bool useCache)
{
ISoodaObjectFactory factory = transaction.GetFactory(classInfo);
string cacheKey = null;
if (useCache)
{
// cache makes sense only on clean database
if (!transaction.HasBeenPrecommitted(classInfo))
{
cacheKey = SoodaCache.GetCollectionKey(classInfo, whereClause);
}
IEnumerable keysCollection = transaction.LoadCollectionFromCache(cacheKey, logger);
if (keysCollection != null)
{
foreach (object o in keysCollection)
{
SoodaObject obj = factory.GetRef(transaction, o);
// this binds to cache
obj.EnsureFieldsInited();
items.Add(obj);
}
if (orderBy != null)
{
items.Sort(orderBy.GetComparer());
}
count = items.Count;
if (startIdx > 0)
{
if (startIdx < count)
items.RemoveRange(0, startIdx);
else
items.Clear();
}
if (pageCount != -1 && pageCount < items.Count)
{
items.RemoveRange(pageCount, items.Count - pageCount);
}
return;
}
}
SoodaDataSource ds = transaction.OpenDataSource(classInfo.GetDataSource());
if ((options & SoodaSnapshotOptions.KeysOnly) != 0)
{
if (pageCount != -1)
{
using (IDataReader reader = ds.LoadMatchingPrimaryKeys(transaction.Schema, classInfo, whereClause, orderBy, 0, -1))
{
count = 0;
while (reader.Read())
count++;
}
}
using (IDataReader reader = ds.LoadMatchingPrimaryKeys(transaction.Schema, classInfo, whereClause, orderBy, startIdx, pageCount))
{
while (reader.Read())
{
SoodaObject obj = SoodaObject.GetRefFromKeyRecordHelper(transaction, factory, reader);
items.Add(obj);
}
if (pageCount == -1)
count = items.Count;
}
}
else
{
if (pageCount != -1)
{
using (IDataReader reader = ds.LoadMatchingPrimaryKeys(transaction.Schema, classInfo, whereClause, orderBy, 0, -1))
{
count = 0;
while (reader.Read())
count++;
}
}
TableInfo[] loadedTables;
using (IDataReader reader = ds.LoadObjectList(transaction.Schema, classInfo, whereClause, orderBy, startIdx, pageCount, options, out loadedTables))
{
while (reader.Read())
{
SoodaObject obj = SoodaObject.GetRefFromRecordHelper(transaction, factory, reader, 0, loadedTables, 0);
if ((options & SoodaSnapshotOptions.VerifyAfterLoad) != 0 && whereClause != null && !whereClause.Matches(obj, false))
continue; // don't add the object
items.Add(obj);
}
if (pageCount == -1)
count = items.Count;
}
}
if (cacheKey != null && useCache && startIdx == 0 && pageCount == -1 && involvedClassNames != null)
{
TimeSpan expirationTimeout;
bool slidingExpiration;
if (transaction.CachingPolicy.GetExpirationTimeout(
classInfo, whereClause, orderBy, startIdx, pageCount, items.Count,
out expirationTimeout, out slidingExpiration))
{
transaction.StoreCollectionInCache(cacheKey, classInfo, items, involvedClassNames, (options & SoodaSnapshotOptions.KeysOnly) == 0, expirationTimeout, slidingExpiration);
}
}
}
public SoodaObjectListSnapshot(SoodaTransaction t, SoodaWhereClause whereClause, SoodaOrderBy orderBy, int startIdx, int pageCount, SoodaSnapshotOptions options, ClassInfo ci)
{
this.classInfo = ci;
string[] involvedClasses = null;
bool useCache;
if ((options & SoodaSnapshotOptions.NoCache) != 0)
useCache = false;
else if ((options & SoodaSnapshotOptions.Cache) != 0)
useCache = true;
else
useCache = t.CachingPolicy.ShouldCacheCollection(ci, whereClause, orderBy, startIdx, pageCount);
if (whereClause != null && whereClause.WhereExpression != null)
{
if ((options & SoodaSnapshotOptions.NoWriteObjects) == 0 || useCache)
{
try
{
GetInvolvedClassesVisitor gic = new GetInvolvedClassesVisitor(classInfo);
gic.GetInvolvedClasses(whereClause.WhereExpression);
involvedClasses = gic.ClassNames;
}
catch
{
// logger.Warn("{0}", ex);
// cannot detect involved classes (probably because of RAWQUERY)
// - precommit all objects
// if we get here, involvedClasses remains set to null
}
}
}
else
{
// no where clause
involvedClasses = new string[] { ci.Name };
}
if ((options & SoodaSnapshotOptions.NoWriteObjects) == 0)
t.PrecommitClasses(involvedClasses);
LoadList(t, whereClause, orderBy, startIdx, pageCount, options, involvedClasses, useCache);
}
public override IDataReader LoadObjectList(SchemaInfo schemaInfo, ClassInfo classInfo, SoodaWhereClause whereClause, SoodaOrderBy orderBy, int startIdx, int pageCount, SoodaSnapshotOptions options, out TableInfo[] tables)
{
try
{
Queue<_QueueItem> queue = new Queue<_QueueItem>();
List<TableInfo> tablesArrayList = new List<TableInfo>(classInfo.UnifiedTables.Count);
SoqlQueryExpression queryExpression = new SoqlQueryExpression();
queryExpression.StartIdx = startIdx;
queryExpression.PageCount = pageCount;
queryExpression.From.Add(classInfo.Name);
queryExpression.FromAliases.Add("");
foreach (TableInfo ti in classInfo.UnifiedTables)
{
tablesArrayList.Add(ti);
foreach (FieldInfo fi in ti.Fields)
{
SoqlPathExpression pathExpr = new SoqlPathExpression(fi.Name);
queryExpression.SelectExpressions.Add(pathExpr);
queryExpression.SelectAliases.Add("");
if (fi.ReferencedClass != null && fi.PrefetchLevel > 0 && ((options & SoodaSnapshotOptions.PrefetchRelated) != 0))
{
_QueueItem item = new _QueueItem();
item.classInfo = fi.ReferencedClass;
item.level = fi.PrefetchLevel;
item.prefix = pathExpr;
queue.Enqueue(item);
}
}
}
while (queue.Count > 0)
{
_QueueItem it = queue.Dequeue();
foreach (TableInfo ti in it.classInfo.UnifiedTables)
{
tablesArrayList.Add(ti);
foreach (FieldInfo fi in ti.Fields)
{
// TODO - this relies on the fact that path expressions
// are never reconstructed or broken. We simply share previous prefix
// perhaps it's cleaner to Clone() the expression here
SoqlPathExpression extendedExpression = new SoqlPathExpression(it.prefix, fi.Name);
queryExpression.SelectExpressions.Add(extendedExpression);
queryExpression.SelectAliases.Add("");
if (it.level >= 1 && fi.PrefetchLevel > 0 && fi.ReferencedClass != null)
{
_QueueItem newItem = new _QueueItem();
newItem.classInfo = fi.ReferencedClass;
newItem.prefix = extendedExpression;
newItem.level = it.level - 1;
queue.Enqueue(newItem);
}
}
}
}
if (whereClause != null && whereClause.WhereExpression != null)
{
queryExpression.WhereClause = whereClause.WhereExpression;
}
if (orderBy != null)
{
queryExpression.SetOrderBy(orderBy);
}
string query = SoqlToSql(queryExpression, schemaInfo, false);
IDbCommand cmd = Connection.CreateCommand();
try
{
cmd.CommandTimeout = CommandTimeout;
}
catch (NotSupportedException e)
{
logger.Debug("CommandTimeout not supported. {0}", e.Message);
}
if (Transaction != null)
cmd.Transaction = this.Transaction;
SqlBuilder.BuildCommandWithParameters(cmd, false, query, whereClause.Parameters, false);
tables = tablesArrayList.ToArray();
return TimedExecuteReader(cmd);
}
catch (Exception ex)
{
logger.Error("Exception in LoadObjectList: {0}", ex);
throw;
}
}
public abstract IDataReader LoadObjectList(SchemaInfo schemaInfo, ClassInfo classInfo, SoodaWhereClause whereClause, SoodaOrderBy orderBy, int startIdx, int pageCount, SoodaSnapshotOptions options, out TableInfo[] tables);