public ValueTask <SerializableClassifiedSpans> GetSemanticClassificationsAsync(
PinnedSolutionInfo solutionInfo,
DocumentId documentId,
TextSpan span,
ClassificationOptions options,
StorageDatabase database,
bool isFullyLoaded,
CancellationToken cancellationToken)
{
return(RunServiceAsync(async cancellationToken =>
{
var solution = await GetSolutionAsync(solutionInfo, cancellationToken).ConfigureAwait(false);
var document = solution.GetDocument(documentId) ?? await solution.GetSourceGeneratedDocumentAsync(documentId, cancellationToken).ConfigureAwait(false);
Contract.ThrowIfNull(document);
using var _ = ArrayBuilder <ClassifiedSpan> .GetInstance(out var temp);
await AbstractClassificationService.AddSemanticClassificationsInCurrentProcessAsync(
document, span, options, temp, cancellationToken).ConfigureAwait(false);
if (isFullyLoaded)
{
// Once fully loaded, there's no need for us to keep around any of the data we cached in-memory
// during the time the solution was loading.
lock (_cachedData)
_cachedData.Clear();
// Enqueue this document into our work queue to fully classify and cache.
_workQueue.AddWork((document, options, database));
}
return SerializableClassifiedSpans.Dehydrate(temp.ToImmutable());
}, cancellationToken));
}
public Task <SerializableClassifiedSpans> GetSemanticClassificationsAsync(
PinnedSolutionInfo solutionInfo, DocumentId documentId,
TextSpan span, CancellationToken cancellationToken)
{
return(RunServiceAsync(async() =>
{
using (UserOperationBooster.Boost())
{
var solution = await GetSolutionAsync(solutionInfo, cancellationToken).ConfigureAwait(false);
var document = solution.GetDocument(documentId);
using var _ = ArrayBuilder <ClassifiedSpan> .GetInstance(out var temp);
await AbstractClassificationService.AddSemanticClassificationsInCurrentProcessAsync(
document, span, temp, cancellationToken).ConfigureAwait(false);
return SerializableClassifiedSpans.Dehydrate(temp.ToImmutable());
}
}, cancellationToken));
}
public ValueTask <SerializableClassifiedSpans> GetClassificationsAsync(
Checksum solutionChecksum,
DocumentId documentId,
TextSpan span,
ClassificationType type,
ClassificationOptions options,
bool isFullyLoaded,
CancellationToken cancellationToken)
{
return(RunServiceAsync(solutionChecksum, async solution =>
{
var document = solution.GetDocument(documentId) ?? await solution.GetSourceGeneratedDocumentAsync(documentId, cancellationToken).ConfigureAwait(false);
Contract.ThrowIfNull(document);
if (options.ForceFrozenPartialSemanticsForCrossProcessOperations)
{
// Frozen partial semantics is not automatically passed to OOP, so enable it explicitly when desired
document = document.WithFrozenPartialSemantics(cancellationToken);
}
using var _ = ArrayBuilder <ClassifiedSpan> .GetInstance(out var temp);
await AbstractClassificationService.AddClassificationsInCurrentProcessAsync(
document, span, type, options, temp, cancellationToken).ConfigureAwait(false);
if (isFullyLoaded)
{
// Once fully loaded, there's no need for us to keep around any of the data we cached in-memory
// during the time the solution was loading.
lock (_cachedData)
_cachedData.Clear();
// Enqueue this document into our work queue to fully classify and cache.
_workQueue.AddWork((document, type, options));
}
return SerializableClassifiedSpans.Dehydrate(temp.ToImmutable());
}, cancellationToken));
}
