Update edge endpoint

Signed-off-by: Andy Kwok <[email protected]>

Author: andy-k-improving

solutions/aws-neptune-analytics/README.md

@@ -8,7 +8,7 @@ deployUrl: https://vercel.com/new/clone?repository-url=https://github.com/vercel
 
 # Next.js + AWS Neptune Analytics
 
-This is an example of a Next.js application using AWS Neptune Analytics for creating, reading, updating, and deleting graph nodes with OpenCypher queries.
+This is an example of a Next.js application using AWS Neptune Analytics for creating, reading, updating, and deleting graph nodes and edges with OpenCypher queries.
 
 ## How to Use
 
@@ -55,13 +55,22 @@ const client = new NeptuneGraphClient({})
 
 ## API Endpoints
 
-The application provides a RESTful API for graph node operations:
+The application provides a RESTful API for graph node and edge operations:
+
+### Node Operations
 
 - `GET /api/node?id={id}` - Retrieve a node by ID
 - `POST /api/node` - Create a new node
 - `PUT /api/node` - Update an existing node
 - `DELETE /api/node?id={id}` - Delete a node and its relationships
 
+### Edge Operations
+
+- `GET /api/edge?id={id}` - Retrieve an edge by ID
+- `POST /api/edge` - Create a new edge
+- `PUT /api/edge` - Update an existing edge
+- `DELETE /api/edge?id={id}` - Delete an edge
+
 ## Testing
 
 ### Create Node (POST)
@@ -91,3 +100,31 @@ curl -X PUT http://localhost:3000/api/node \
 ```bash
 curl -X DELETE "http://localhost:3000/api/node?id=user-123"
 ```
+
+### Create Edge (POST)
+
+```bash
+curl -X POST http://localhost:3000/api/edge \
+  -d '{"fromId": "user-123", "toId": "user-456", "type": "FOLLOWS"}' \
+  -H "Content-type: application/json"
+```
+
+### Get Edge (GET)
+
+```bash
+curl "http://localhost:3000/api/edge?id=follows-001"
+```
+
+### Update Edge (PUT)
+
+```bash
+curl -X PUT http://localhost:3000/api/edge \
+  -d '{"id": "follows-001", "since": "2024-01-15", "strength": "strong"}' \
+  -H "Content-type: application/json"
+```
+
+### Delete Edge (DELETE)
+
+```bash
+curl -X DELETE "http://localhost:3000/api/edge?id=follows-001"
+```

solutions/aws-neptune-analytics/app/api/edge/route.ts

@@ -0,0 +1,147 @@
+import { NextResponse } from 'next/server'
+import * as NG from '@aws-sdk/client-neptune-graph'
+
+const client = new NG.NeptuneGraphClient({})
+const GRAPH_ID = process.env.GRAPH_ID
+
+if (!GRAPH_ID) {
+  throw new Error('GRAPH_ID environment variable is required')
+}
+
+/**
+ * Execute Neptune Analytics query with parameters
+ */
+async function executeQuery(
+  queryString: string,
+  parameters: Record<string, any>
+) {
+  const input = {
+    graphIdentifier: GRAPH_ID,
+    queryString,
+    language: NG.QueryLanguage.OPEN_CYPHER,
+    parameters,
+  }
+
+  const cmd = new NG.ExecuteQueryCommand(input)
+  const response = await client.send(cmd)
+  const responseStr = await response.payload.transformToString()
+  return JSON.parse(responseStr)
+}
+
+/**
+ * Handle errors with consistent logging and response format
+ */
+function handleError(error: any, method: string) {
+  console.error(`${method} /api/edge error:`, error)
+  return NextResponse.json({ error: 'Internal server error' }, { status: 500 })
+}
+
+/**
+ * Validate ID parameter from query string
+ */
+function validateId(id: string | null) {
+  if (!id) {
+    return NextResponse.json(
+      { error: 'id parameter is required' },
+      { status: 400 }
+    )
+  }
+  return null
+}
+
+/**
+ * Validate request body contains required fields for edge creation
+ */
+function validateEdgeBody(body: any) {
+  if (!body?.fromId || !body?.toId || !body?.type) {
+    return NextResponse.json(
+      { error: 'Request body with fromId, toId, and type is required' },
+      { status: 400 }
+    )
+  }
+  return null
+}
+
+export async function GET(request: Request) {
+  try {
+    const { searchParams } = new URL(request.url)
+    const id = searchParams.get('id')
+
+    const error = validateId(id)
+    if (error) return error
+
+    const result = await executeQuery(
+      'MATCH()-[r]->() WHERE id(r) = $EDGE_ID RETURN r',
+      { EDGE_ID: id }
+    )
+
+    return NextResponse.json(result)
+  } catch (error) {
+    return handleError(error, 'GET')
+  }
+}
+
+export async function POST(request: Request) {
+  try {
+    const body = await request.json()
+
+    const error = validateEdgeBody(body)
+    if (error) return error
+
+    const { fromId, toId, type, ...properties } = body
+
+    const result = await executeQuery(
+      'MATCH (from), (to) WHERE id(from) = $FROM_ID AND id(to) = $TO_ID CREATE (from)-[r:' +
+        type +
+        ']->(to) SET r += $PROPERTIES RETURN r',
+      { FROM_ID: fromId, TO_ID: toId, PROPERTIES: properties }
+    )
+
+    return NextResponse.json(result, { status: 201 })
+  } catch (error) {
+    return handleError(error, 'POST')
+  }
+}
+
+export async function PUT(request: Request) {
+  try {
+    const body = await request.json()
+
+    if (!body?.id) {
+      return NextResponse.json(
+        { error: 'Request body with id is required' },
+        { status: 400 }
+      )
+    }
+
+    const { id, ...properties } = body
+
+    const result = await executeQuery(
+      'MATCH()-[r]->() WHERE id(r) = $EDGE_ID SET r = $PROPERTIES RETURN r',
+      { EDGE_ID: id, PROPERTIES: properties }
+    )
+
+    return NextResponse.json(result)
+  } catch (error) {
+    return handleError(error, 'PUT')
+  }
+}
+
+export async function DELETE(request: Request) {
+  try {
+    const { searchParams } = new URL(request.url)
+    const id = searchParams.get('id')
+
+    const error = validateId(id)
+    if (error) return error
+
+    const result = await executeQuery(
+      'MATCH()-[r]->() WHERE id(r) = $EDGE_ID DELETE r',
+      { EDGE_ID: id }
+    )
+
+    return NextResponse.json(result)
+  } catch (error) {
+    return handleError(error, 'DELETE')
+  }
+}