Read sample document by json id using sql

Author: harishkannarao

app/dao/sample_jsonb_sql_dao.py

@@ -11,6 +11,12 @@
 
 READ_SAMPLE_DOCUMENT_BY_ID = "select * from sample_documents where id=:id"
 
+READ_SAMPLE_DOCUMENT_BY_JSON_ID = """
+    SELECT *
+    FROM sample_documents
+    WHERE (cast(json_data->>'id' as text)) = :json_id
+    """
+
 INSERT_SAMPLE_DOCUMENT = """
     INSERT INTO
     sample_documents(
@@ -32,6 +38,16 @@ async def read_sample_document_by_id(sample_document_id: UUID) -> SampleDocument
         return None
 
 
+async def read_sample_document_by_json_id(json_id: UUID) -> SampleDocument | None:
+    row: Record = await database.fetch_one(
+        query=READ_SAMPLE_DOCUMENT_BY_JSON_ID, values={"json_id": str(json_id)}
+    )
+    if row:
+        return map_from_db_row(row)
+    else:
+        return None
+
+
 async def create_sample_document(input_document: SampleDocumentCreate) -> UUID:
     inserted_row: Record = await database.fetch_one(
         query=INSERT_SAMPLE_DOCUMENT, values=map_to_db_row(input_document)

tests_integration/dao/test_sample_json_sql_dao.py

@@ -78,6 +78,34 @@ async def test_sample_document_create_and_read():
     assert_that(read_document_by_id.created_datetime).is_between(start_time, end_time)
 
 
+@pytest.mark.asyncio
+async def test_sample_document_read_by_json_id():
+    sample_id: UUID = await create_random_sample()
+
+    create_request: SampleDocumentCreate = SampleDocumentCreate(
+        sample_id=sample_id,
+        json_data=DocumentMetadata(id=uuid.uuid4(), tags=("tag-1", "tag-2")),
+        secondary_json_dict={
+            "test": "value",
+            "nested": {"sub": "value"},
+            "array": [{"key": "test-value"}],
+        },
+    )
+
+    sample_document_id: UUID = await sample_jsonb_sql_dao.create_sample_document(
+        create_request
+    )
+    read_document_by_id: SampleDocument = (
+        await sample_jsonb_sql_dao.read_sample_document_by_id(sample_document_id)
+    )
+    read_document_by_json_id: SampleDocument = (
+        await sample_jsonb_sql_dao.read_sample_document_by_json_id(
+            create_request.json_data.id
+        )
+    )
+    assert_that(read_document_by_json_id).is_equal_to(read_document_by_id)
+
+
 async def create_random_sample() -> UUID:
     create_sample_request: SampleCreate = SampleCreate(
         username=f"usr-{uuid.uuid4()}",