Merge pull request #5 from druids/feat/sentry-support

feat: support for logging to Sentry

Author: radimsuckr

functions/lambda.py

@@ -4,6 +4,8 @@
 from os import environ
 
 import boto3
+import sentry_sdk
+from sentry_sdk.integrations.aws_lambda import AwsLambdaIntegration
 
 
 class LambdaError(RuntimeError):
@@ -25,6 +27,10 @@ class InstanceRefreshInProgress(LambdaError):
 REFRESH_INSTANCE_WARMUP = int(environ['REFRESH_INSTANCE_WARMUP'])
 REFRESH_MIN_HEALTHY_PERCENTAGE = int(environ['REFRESH_MIN_HEALTHY_PERCENTAGE'])
 SSM_PARAMETER_NAME = environ['SSM_PARAMETER_NAME']
+SENTRY_DSN = environ.get('SENTRY_DSN')
+
+if SENTRY_DSN:
+    sentry_sdk.init(SENTRY_DSN, integrations=[AwsLambdaIntegration()])
 
 logging.config.dictConfig({
     'version': 1,

lambda.tf

@@ -10,12 +10,18 @@ resource "aws_lambda_function" "refresh" {
 
   description = var.lambda_description
 
+  layers = [
+    "arn:aws:lambda:${data.aws_region.current.name}:943013980633:layer:SentryPythonServerlessSDK:7",
+  ]
+
   environment {
     variables = {
       AUTO_SCALING_GROUP_NAME                 = var.autoscaling_group_name
       DESCRIBE_INSTANCE_REFRESHES_MAX_RECORDS = var.describe_instance_refreshes_max_records
       REFRESH_INSTANCE_WARMUP                 = var.instance_refresh_instance_warmup
       REFRESH_MIN_HEALTHY_PERCENTAGE          = var.instance_refresh_min_healthy_percentage
+      SENTRY_DSN                              = var.sentry_dsn
+      SENTRY_ENVIRONMENT                      = var.sentry_environment
       SSM_PARAMETER_NAME                      = var.ami_ssm_parameter
     }
   }

variables.tf

@@ -85,3 +85,13 @@ variable "launch_template_source_version" {
   default     = "$Default"
   type        = string
 }
+
+variable "sentry_dsn" {
+  default = null
+  type    = string
+}
+
+variable "sentry_environment" {
+  default = null
+  type    = string
+}