EF Core Azure Cosmos DB プロバイダーを使用したクエリ実行

クエリ実行の基本

EF Core LINQ クエリは、他のデータベース プロバイダーと同じ方法で Azure Cosmos DB に対して実行できます。 次に例を示します。

public class Session
{
    public Guid Id { get; set; }
    public string Category { get; set; }

    public string TenantId { get; set; } = null!;
    public Guid UserId { get; set; }
    public int SessionId { get; set; }
}

var stringResults = await context.Sessions
    .Where(
        e => e.Category.Length > 4
            && e.Category.Trim().ToLower() != "disabled"
            && e.Category.TrimStart().Substring(2, 2).Equals("xy", StringComparison.OrdinalIgnoreCase))
    .ToListAsync();

パーティション キー

パーティション分割の利点は、関連するデータが見つかったパーティションに対してのみクエリが実行されるため、コストを節約でき、結果の速度が速くなることです。 パーティション キーを指定しないクエリはすべてのパーティションで実行されるため、コストがかなり高くなる可能性があります。

EF 9.0 以降では、EF は LINQ クエリの Where 演算子でパーティション キーの比較を自動的に検出して抽出します。 階層パーティション キーを使用して構成された Session エンティティ型に対して次のクエリを実行するとします。

protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    modelBuilder.Entity<Session>()
        .HasPartitionKey(b => new { b.TenantId, b.UserId, b.SessionId })
}

var tenantId = "Microsoft";
var userId = new Guid("00aa00aa-bb11-cc22-dd33-44ee44ee44ee");
var username = "scott";

var sessions = await context.Sessions
    .Where(
        e => e.TenantId == tenantId
             && e.UserId == userId
             && e.SessionId > 0
             && e.Username == username)
    .ToListAsync();

EF によって生成されたログを調べると、このクエリが以下のように実行されます。

Executed ReadNext (166.6985 ms, 2.8 RU) ActivityId='312da0d2-095c-4e73-afab-27072b5ad33c', Container='test', Partition='["Microsoft","00aa00aa-bb11-cc22-dd33-44ee44ee44ee"]', Parameters=[]
SELECT VALUE c
FROM root c
WHERE ((c["SessionId"] > 0) AND CONTAINS(c["Username"], "a"))

これらのログでは、以下のことがわかります。

• 最初の 2 つの比較 (TenantId と UserId) は削除され、ReadNext 句ではなく WHERE "パーティション" に表示されます。つまり、クエリはそれらの値のサブパーティションでのみ実行されます。
• SessionId は階層パーティション キーの一部でもありますが、等値比較の代わりに、より大きい演算子 (>) が使用されるため、リフト アウトできません。これは、通常のプロパティと同様、WHERE 句の一部です。
• Username は通常のプロパティであり、パーティション キーの一部ではないため、WHERE 句にも残ります。

一部のパーティション キー値が指定されていない場合でも、階層パーティション キーでは、最初の 2 つのプロパティに対応するサブパーティションのみを対象にすることができます。 これは、単一のパーティション (3 つのプロパティすべてで識別される) をターゲットにするほど効率的ではありませんが、すべてのパーティションをターゲットにするよりもはるかに効率的です。

Where 演算子でパーティション キー プロパティを参照するのではなく、WithPartitionKey 演算子を使用することにより明示的に指定できます。

var sessions = await context.Sessions
    .WithPartitionKey(tenantId, userId)
    .Where(e => e.SessionId > 0 && e.Username.Contains("a"))
    .ToListAsync();

これは上記のクエリと同じ方法で実行され、クエリでパーティション キーをより明示的にする場合に適しています。 9.0 より前のバージョンの EF では、WithPartitionKey の使用が必要になることがあります。クエリでパーティション キーが期待どおりに使用されていることを確認するため、ログに注意を払ってください。

ポイント読み取り

Azure Cosmos DB では SQL を使用した強力なクエリが可能ですが、このようなクエリにはコストがかなりかかる場合があります。 Azure Cosmos DB は ポイント読み取りもサポートしており、これは id プロパティとパーティション キー全体の両方がわかっている場合に単一のドキュメントを取得するときに使用する必要があります。 ポイント読み取りは、特定のパーティション内の特定のドキュメントを直接識別し、クエリを使用して同じドキュメントを取得する場合と比較して、非常に効率的で低コストで実行します。 可能な限り頻繁にポイント読み取りを活用するようにシステムを設計することをお勧めします。 詳しくは、 Azure Cosmos DBのドキュメントをご覧ください。

前のセクションでは、EF でパーティション キーの比較を識別し、Where 句から抽出することで、クエリをより効率的に実行し、関連するパーティションのみに処理を制限しました。 さらに手順を進め、クエリに id プロパティを指定することも可能です。 以下のクエリを調べてみましょう。

var session = await context.Sessions.SingleAsync(
    e => e.Id == someId
         && e.TenantId == tenantId
         && e.UserId == userId
         && e.SessionId == sessionId);

このクエリでは、 Id プロパティの値 (Azure Cosmos DB id プロパティにマップされます) と、すべてのパーティション キー プロパティの値が提供されます。 さらに、クエリに対する追加のコンポーネントはありません。 これらの条件がすべて満たされると、EF はポイント読み取りとしてクエリを実行することができます。

Executed ReadItem (46 ms, 1 RU) ActivityId='d7391311-2266-4811-ae2d-535904c42c43', Container='test', Id='9', Partition='["Microsoft","00aa00aa-bb11-cc22-dd33-44ee44ee44ee",10.0]'

ReadItem に注目してください。これは、クエリが効率的なポイント読み取りとして実行されたことを示しています。SQL クエリは関係していません。

パーティション キーの抽出と同様、EF 9.0 ではこのメカニズムが大幅に改善されています。それより古いバージョンでは、ポイント読み取りを確実に検出して使用することはできません。

改ページ位置の自動修正

ページネーションとは、結果を一度にすべて取得するのではなく、ページ単位で取得することを指します。これは通常、大規模な結果セットに対して行われ、ユーザー インターフェイスが表示され、ユーザーが結果のページ間を移動できるようになります。

データベースで改ページ位置の自動修正を実装する一般的な方法は、Skip および Take LINQ 演算子 (SQL では OFFSET および LIMIT) を使用する方法です。 10 件の結果のページ サイズの場合、次のようにして 3 ページ目を EF Core でフェッチできます。

var position = 20;
var nextPage = await context.Session
    .OrderBy(s => s.Id)
    .Skip(position)
    .Take(10)
    .ToListAsync();

残念ながら、この手法はかなり効率が悪いため、クエリのコストを大幅に増加させる可能性があります。 Azure Cosmos DB には、 後続トークンを使うことにより、クエリの結果を通じてページ分割するための特別なメカニズムが用意されています。

CosmosPage firstPage = await context.Sessions
    .OrderBy(s => s.Id)
    .ToPageAsync(pageSize: 10, continuationToken: null);

string continuationToken = firstPage.ContinuationToken;
foreach (var session in firstPage.Values)
{
    // Display/send the sessions to the user
}

LINQ クエリを ToListAsync などで終了するのではなく、ToPageAsync メソッドを使用して、ページごとに最大 10 個の項目を取得するよう指示します (データベース内の項目数はこれより少なくなる場合がある点に注意してください)。 これは最初のクエリなので、最初から結果を取得し、後続トークンとして null を渡したいと思います。 ToPageAsync は、後続トークンとページ内の値 (最大 10 項目) を公開する CosmosPage<T> を返します。 プログラムは通常、後続トークンと共にこれらの値をクライアントに送信します。これにより、後でクエリを再開し、より多くの結果をフェッチできるようになります。

ユーザーが UI の [次へ] ボタンをクリックし、次の 10 項目を要求したとします。 その後、次のようにクエリを実行できます。

CosmosPage nextPage = await context.Sessions.OrderBy(s => s.Id).ToPageAsync(10, continuationToken);
string continuationToken = nextPage.ContinuationToken;
foreach (var session in nextPage.Values)
{
    // Display/send the sessions to the user
}

同じクエリを実行しますが、今回は最初の実行で受け取った後続トークンを渡します。これにより、Cosmos DB は中断したところからクエリエンジンを続行し、次の 10 個の項目をフェッチするよう指示します。 最後のページを取得し、結果がなくなると、後続トークンは null になり、[次へ] ボタンがグレー表示になります。このページネーション方法は、Skip や Take を使用する場合に比べて、非常に効率がよく、コスト効率に優れています。

Azure Cosmos DB におけるページネーションについて詳しくは、 このページをご覧ください。

FindAsync

FindAsync は、エンティティを主キーにより取得し、エンティティが既に読み込まれて、コンテキストによって追跡されている場合にデータベース ラウンドトリップを回避するのに役立つ API です。

リレーショナル データベースに慣れている開発者は、エンティティ型の主キー (Id プロパティなど) に使用されます。 EF Azure Cosmos DB プロバイダーを使用するとき、主キーには、JSON id プロパティにマップされたプロパティに加えて、パーティション キー プロパティが含まれています。これは、Azure Cosmos DB では異なるパーティションに同じ JSON id プロパティを持つドキュメントを含めることができるためであり、そのため、 id とパーティション キーの組み合わせのみがコンテナー内の単一のドキュメントを一意に識別します。

public class Session
{
    public Guid Id { get; set; }
    public string PartitionKey { get; set; }
    ...
}

var mySession = await context.FindAsync(id, pkey);

階層パーティション キーがある場合、すべてのパーティション キー値を、構成された順序で FindAsync に渡す必要があります。

SQL クエリ

クエリは、SQL で直接記述することもできます。 次に例を示します。

var rating = 3;
_ = await context.Blogs
    .FromSql($"SELECT VALUE c FROM root c WHERE c.Rating > {rating}")
    .ToListAsync();

このクエリの結果、次のクエリが実行されます。

SELECT VALUE s
FROM (
    SELECT VALUE c FROM root c WHERE c.Angle1 <= @p0
) s

FromSql は EF 9.0 で導入された点に注意してください。 以前のバージョンでは、代わりに FromSqlRaw を使用できますが、メソッドは SQL インジェクション攻撃に対して脆弱である点に注意してください。

SQL クエリについて詳しくは、 SQL クエリに関する関連ドキュメントをご覧ください。その内容のほとんどは、Azure Cosmos DB プロバイダーにも関連しています。

関数のマッピング

このセクションでは、Azure Cosmos DB プロバイダーでクエリを実行するとき、どの .NET メソッドとメンバーがどの SQL 関数に変換されるかを示します。

日時の関数

¹ 他のコンポーネント メンバーも変換されます (Month, Day...)。

Numeric 関数

String 関数

ベクター検索とフルテキスト検索

ベクター検索の詳細については、 ドキュメントを参照してください。 フルテキスト検索の詳細については、 ドキュメントを参照してください。

その他の関数

¹CoalesceUndefined は undefined ではなく null と合体することに注意してください。 null を合体するには、標準 C# ?? 演算子を使用します。