MongoDBで要素が存在しない場合は配列を挿入し、存在する場合は複数条件で更新する方法

MongoDBで要素が存在しない場合は配列を挿入し、存在する場合は複数条件で更新する方法

MongoDBでは、bulkWrite()メソッドを利用することで、「配列要素が存在しない場合は新しく挿入し、すでに存在する場合は複数の条件を指定して更新する」という一連の処理をまとめて実行できます。

本記事では、実際のサンプルコードを使いながら、その手順をわかりやすく解説します。

サンプルコレクションの作成

まず、ドキュメントを含むコレクションを作成します。次のクエリを実行してください。

> db.demo105.insertOne( { _id:'101', Name:'Chris', Details:[{ Marks1:60, Marks2:70, Marks3:70 }, { Marks1:70, Marks2:70, Marks3:90 }] } );
{ "acknowledged" : true, "insertedId" : "101" }

作成したドキュメントの確認

find()メソッドを使って、コレクション内のすべてのドキュメントを表示してみましょう。

> db.demo105.find().pretty();

このクエリを実行すると、次のような出力が得られます。

{
   "_id" : "101",
   "Name" : "Chris",
   "Details" : [
      {
         "Marks1" : 60,
         "Marks2" : 70,
         "Marks3" : 70
      },
      {
         "Marks1" : 70,
         "Marks2" : 70,
         "Marks3" : 90
      }
   ]
}

bulkWrite()による挿入・更新の実装

以下が、要素が存在しない場合には配列を挿入し、存在する場合には複数の条件で更新を行うクエリです。

> db.demo105.bulkWrite([
...    { "updateOne": {
...       "filter": {
...          "_id": "101",
...          "Details": {
...             "$elemMatch": { Marks2: 70, Marks3: 70 }
...          }
...       },
...       "update": {
...          "$set": { "Details.$.Marks3": 96 }
...       }
...    }},
...    { "updateOne": {
...       "filter": {
...          "_id": "101",
...          "Details": {
...             "$not": {
...                "$elemMatch": { Marks2: 70, Marks3: 70 }
...             }
...          }
...       },
...       "update": {
...          "$push": { "Details": { Marks1: 94, Marks2: 97,Marks3:99} }
...       }
...    }},
...    { "updateOne": {
...       "filter": { "_id": 101 },
...       "update": {
...          "$setOnInsert": {
...             "Details": [
...             { Marks1: 34, Marks2: 67,Marks3:87 }
...          ]
...       }
...    },
...    "upsert": true
... }}
... ])
{
   "acknowledged" : true,
   "deletedCount" : 0,
   "insertedCount" : 0,
   "matchedCount" : 2,
   "upsertedCount" : 1,
   "insertedIds" : {
   },
   "upsertedIds" : {
      "2" : 101
   }
}

各操作のポイント解説

このbulkWrite()は、3つの操作から構成されています。

• 1つ目のupdateOne（既存要素の更新）： $elemMatchを使って「Marks2が70かつMarks3が70」という条件に一致する要素を検索します。一致した場合は、位置演算子$を利用して、該当する配列要素のMarks3の値だけを96に更新します。
• 2つ目のupdateOne（条件不一致時の挿入）： $notと$elemMatchを組み合わせることで、「条件に一致する要素が存在しない」場合のみマッチさせます。このとき$pushによって、新しいオブジェクトがDetails配列の末尾に追加されます。
• 3つ目のupdateOne（upsertによる新規作成）： ドキュメント自体が存在しないケースに備えた処理です。$setOnInsertとupsert: trueを組み合わせることで、該当ドキュメントが見つからなかった場合にのみ、初期データを持つ新しいドキュメントが作成されます。

注意点： この例では3つ目のフィルターで_id: 101（数値型）を指定しているため、文字列型の"101"を持つ既存ドキュメントとは一致せず、別の新規ドキュメントが作成されています。実際の開発では、_idのデータ型（文字列か数値か）を統一するよう注意しましょう。

実行結果の確認

処理完了後、再度find()メソッドですべてのドキュメントを表示します。

> db.demo105.find().pretty();

実行すると、次の出力が得られます。

{
   "_id" : "101",
   "Name" : "Chris",
   "Details" : [
      {
         "Marks1" : 60,
         "Marks2" : 70,
         "Marks3" : 96
      },
      {
         "Marks1" : 70,
         "Marks2" : 70,
         "Marks3" : 90
      },
      {
         "Marks1" : 94,
         "Marks2" : 97,
         "Marks3" : 99
      }
   ]
}
{
   "_id" : 101,
   "Details" : [
      {
         "Marks1" : 34,
         "Marks2" : 67,
         "Marks3" : 87
      }
   ]
}

まとめ

bulkWrite()を活用すれば、「条件に一致する要素があれば更新、なければ挿入」という処理をアトミックに近い形で一度に実行できます。$elemMatchによる複数条件の絞り込み、位置演算子$による部分更新、$pushによる配列への追加、そしてupsertと$setOnInsertによる新規ドキュメントの作成を組み合わせることで、柔軟な配列操作が可能になります。ぜひ実際のアプリケーション開発でも活用してみてください。