setup the basic app

Author: HarshCasper

.env

@@ -0,0 +1,3 @@
+export NEO4J_PASSWORD=mySecretPassword
+export NEO4J_URI=bolt://neo4j:7687
+export NEO4J_USERNAME=neo4j

.gitignore

@@ -120,7 +120,6 @@ celerybeat.pid
 *.sage.py
 
 # Environments
-.env
 .venv
 env/
 venv/

docker-compose.yml

@@ -0,0 +1,40 @@
+version: '3.9'
+
+services:
+  localstack:
+    container_name: "${LOCALSTACK_DOCKER_NAME:-localstack-main}"
+    image: localstack/localstack-pro:latest  # required for Pro
+    ports:
+      - "127.0.0.1:4566:4566"            # LocalStack Gateway
+      - "127.0.0.1:4510-4559:4510-4559"  # external services port range
+      - "127.0.0.1:443:443"              # LocalStack HTTPS Gateway (Pro)
+    environment:
+      - LOCALSTACK_AUTH_TOKEN=${LOCALSTACK_AUTH_TOKEN}
+      - DEBUG=1
+      - LAMBDA_RUNTIME_ENVIRONMENT_TIMEOUT=600
+    volumes:
+      - "${LOCALSTACK_VOLUME_DIR:-./volume}:/var/lib/localstack"
+      - "/var/run/docker.sock:/var/run/docker.sock"
+    networks:
+      ls:
+        # Set the container IP address in the 10.0.2.0/24 subnet
+        ipv4_address: 10.0.2.20
+  neo4j:
+    container_name: neo4j
+    image: neo4j:latest
+    ports:
+      - 7474:7474
+      - 7687:7687
+    volumes:  
+      - ./neo4j_db/data:/data
+      - ./neo4j_db/logs:/logs
+      - ./neo4j_db/import:/var/lib/neo4j/import
+    networks:
+      - ls
+
+networks:
+  ls:
+    ipam:
+      config:
+        # Specify the subnet range for IP address allocation
+        - subnet: 10.0.2.0/24

event.json

@@ -0,0 +1,36 @@
+{
+    "Records": [
+        {
+            "kinesis": {
+                "kinesisSchemaVersion": "1.0",
+                "partitionKey": "1",
+                "sequenceNumber": "49590338271490256608559692538361571095921575989136588898",
+                "data": "SGVsbG8sIHRoaXMgaXMgYSB0ZXN0Lg==",
+                "approximateArrivalTimestamp": 1545084650.987
+            },
+            "eventSource": "aws:kinesis",
+            "eventVersion": "1.0",
+            "eventID": "shardId-000000000006:49590338271490256608559692538361571095921575989136588898",
+            "eventName": "aws:kinesis:record",
+            "invokeIdentityArn": "arn:aws:iam::123456789012:role/lambda-role",
+            "awsRegion": "us-east-2",
+            "eventSourceARN": "arn:aws:kinesis:us-east-2:123456789012:stream/lambda-stream"
+        },
+        {
+            "kinesis": {
+                "kinesisSchemaVersion": "1.0",
+                "partitionKey": "1",
+                "sequenceNumber": "49590338271490256608559692540925702759324208523137515618",
+                "data": "VGhpcyBpcyBvbmx5IGEgdGVzdC4=",
+                "approximateArrivalTimestamp": 1545084711.166
+            },
+            "eventSource": "aws:kinesis",
+            "eventVersion": "1.0",
+            "eventID": "shardId-000000000006:49590338271490256608559692540925702759324208523137515618",
+            "eventName": "aws:kinesis:record",
+            "invokeIdentityArn": "arn:aws:iam::123456789012:role/lambda-role",
+            "awsRegion": "us-east-2",
+            "eventSourceARN": "arn:aws:kinesis:us-east-2:123456789012:stream/lambda-stream"
+        }
+    ]
+  }

function/lambda_function.py

@@ -0,0 +1,44 @@
+import os
+import logging
+import jsonpickle
+import boto3
+from aws_xray_sdk.core import patch_all
+import os
+from neo4j import GraphDatabase
+
+neo4j_uri = os.environ.get('NEO4J_URI')
+neo4j_user = os.environ.get('NEO4J_USERNAME')
+neo4j_password = os.environ.get('NEO4J_PASSWORD')
+
+logger = logging.getLogger()
+logger.setLevel(logging.INFO)
+patch_all()
+
+if neo4j_uri is None or neo4j_user is None or neo4j_password is None:
+    logger.info('## ENVIRONMENT VARIABLES\r' + jsonpickle.encode(dict(**os.environ)))
+    raise Exception("Missing environment variables for connecting to Aura; double check!")
+
+driver = GraphDatabase.driver(neo4j_uri, auth=(neo4j_user, neo4j_password),
+    max_connection_lifetime=5*60,
+    keep_alive=True)
+
+client = boto3.client('lambda')
+client.get_account_settings()
+
+def write_graph_record(tx, event):
+    val = jsonpickle.encode(event)
+    tx.run("""
+        MERGE (e:LambdaEvent { created: datetime(), event: $event })       
+    """, event=jsonpickle.encode(event))
+
+def lambda_handler(event, context):
+    logger.info('## ENVIRONMENT VARIABLES\r' + jsonpickle.encode(dict(**os.environ)))
+    logger.info('## EVENT\r' + jsonpickle.encode(event))
+    logger.info('## CONTEXT\r' + jsonpickle.encode(context))
+    logger.info("## CONNECTING TO %s" % neo4j_uri)
+
+    with driver.session() as session:
+        session.write_transaction(write_graph_record, event)
+
+    response = client.get_account_settings()
+    return response['AccountUsage']

function/requirements.txt

@@ -0,0 +1,3 @@
+jsonpickle==1.3
+neo4j==4.3.5
+aws-xray-sdk==2.8.0

template.yml

@@ -0,0 +1,27 @@
+AWSTemplateFormatVersion: '2010-09-09'
+Transform: 'AWS::Serverless-2016-10-31'
+Description: An AWS Lambda application that calls the Lambda API.
+Resources:
+  function:
+    Type: AWS::Serverless::Function
+    Properties:
+      Handler: lambda_function.lambda_handler
+      Runtime: python3.10
+      CodeUri: function/.
+      Description: Call the AWS Lambda API
+      Timeout: 600
+      Policies:
+        - AWSLambdaBasicExecutionRole
+        - AWSLambda_ReadOnlyAccess
+        - AWSXrayWriteOnlyAccess
+      Tracing: Active
+      Layers:
+        - !Ref libs
+  libs:
+    Type: AWS::Serverless::LayerVersion
+    Properties:
+      LayerName: aura-lambda-python-lib
+      Description: Dependencies for the aura-lambda sample app.
+      ContentUri: package/.
+      CompatibleRuntimes:
+        - python3.10