terraform: support TBS in standalone_apm_server

Author: carsonip

testing/benchmark/main.tf

@@ -160,6 +160,9 @@ module "standalone_apm_server" {
   apm_server_bin_path = var.apm_server_bin_path
   ea_managed          = false
 
+  apm_server_tail_sampling               = var.apm_server_tail_sampling
+  apm_server_tail_sampling_storage_limit = var.apm_server_tail_sampling_storage_limit
+
   aws_provisioner_key_name = var.private_key
 
   elasticsearch_url      = module.moxy[0].moxy_url

testing/infra/terraform/modules/standalone_apm_server/README.md

@@ -33,14 +33,15 @@ This module can be used to create a standalone APM Server deployment against a s
 |------|-------------|------|---------|:--------:|
 | <a name="input_apm_instance_type"></a> [apm\_instance\_type](#input\_apm\_instance\_type) | Optional apm server instance type overide | `string` | `""` | no |
 | <a name="input_apm_server_bin_path"></a> [apm\_server\_bin\_path](#input\_apm\_server\_bin\_path) | Optionally use the apm-server binary from the specified path instead | `string` | `""` | no |
+| <a name="input_apm_server_tail_sampling"></a> [apm\_server\_tail\_sampling](#input\_apm\_server\_tail\_sampling) | Whether or not to enable APM Server tail-based sampling. Defaults to false | `bool` | `false` | no |
+| <a name="input_apm_server_tail_sampling_storage_limit"></a> [apm\_server\_tail\_sampling\_storage\_limit](#input\_apm\_server\_tail\_sampling\_storage\_limit) | Storage size limit of APM Server tail-based sampling. Defaults to 10GB | `string` | `"10GB"` | no |
 | <a name="input_aws_os"></a> [aws\_os](#input\_aws\_os) | Optional aws EC2 instance OS | `string` | `""` | no |
 | <a name="input_aws_provisioner_key_name"></a> [aws\_provisioner\_key\_name](#input\_aws\_provisioner\_key\_name) | ssh key name to create the aws key pair and remote provision the EC2 instance | `string` | n/a | yes |
 | <a name="input_ea_managed"></a> [ea\_managed](#input\_ea\_managed) | Whether or not install Elastic Agent managed APM Server | `bool` | `false` | no |
 | <a name="input_elasticsearch_password"></a> [elasticsearch\_password](#input\_elasticsearch\_password) | The Elasticsearch password | `string` | n/a | yes |
 | <a name="input_elasticsearch_url"></a> [elasticsearch\_url](#input\_elasticsearch\_url) | The secure Elasticsearch URL | `string` | n/a | yes |
 | <a name="input_elasticsearch_username"></a> [elasticsearch\_username](#input\_elasticsearch\_username) | The Elasticsearch username | `string` | n/a | yes |
 | <a name="input_stack_version"></a> [stack\_version](#input\_stack\_version) | Optional stack version | `string` | `"latest"` | no |
-| <a name="input_tags"></a> [tags](#input\_tags) | Optional set of tags to use for all deployments | `map(string)` | `{}` | no |
 | <a name="input_vpc_id"></a> [vpc\_id](#input\_vpc\_id) | VPC ID to provision the EC2 instance | `string` | n/a | yes |
 
 ## Outputs

testing/infra/terraform/modules/standalone_apm_server/apm-server.yml.tftpl

@@ -10,7 +10,10 @@ apm-server:
     enabled: true
   pprof:
     enabled: true
-  
+  sampling:
+    tail:
+      enabled: ${apm_server_tail_sampling}
+      storage_limit: ${apm_server_tail_sampling_storage_limit}
 output:
   elasticsearch:
     hosts: [ ${elasticsearch_url} ]

testing/infra/terraform/modules/standalone_apm_server/elastic-agent.yml.tftpl

@@ -25,6 +25,10 @@ inputs:
       rum:
         enabled: true
       host: '0.0.0.0:${apm_port}'
+      sampling:
+        tail:
+          enabled: ${apm_server_tail_sampling}
+          storage_limit: ${apm_server_tail_sampling_storage_limit}
     logging.level: debug
     logging.to_files: true
     logging.files:

testing/infra/terraform/modules/standalone_apm_server/main.tf

@@ -185,12 +185,14 @@ resource "aws_instance" "apm" {
   provisioner "file" {
     destination = local.conf_path
     content = templatefile(var.ea_managed ? "${path.module}/elastic-agent.yml.tftpl" : "${path.module}/apm-server.yml.tftpl", {
-      elasticsearch_url      = "${var.elasticsearch_url}",
-      elasticsearch_username = "${var.elasticsearch_username}",
-      elasticsearch_password = "${var.elasticsearch_password}",
-      apm_version            = "${var.stack_version}"
-      apm_secret_token       = "${random_password.apm_secret_token.result}"
-      apm_port               = "${local.apm_port}"
+      elasticsearch_url                      = "${var.elasticsearch_url}",
+      elasticsearch_username                 = "${var.elasticsearch_username}",
+      elasticsearch_password                 = "${var.elasticsearch_password}",
+      apm_version                            = "${var.stack_version}"
+      apm_secret_token                       = "${random_password.apm_secret_token.result}"
+      apm_port                               = "${local.apm_port}"
+      apm_server_tail_sampling               = "${var.apm_server_tail_sampling}"
+      apm_server_tail_sampling_storage_limit = "${var.apm_server_tail_sampling_storage_limit}"
     })
   }
 

testing/infra/terraform/modules/standalone_apm_server/variables.tf

@@ -54,3 +54,15 @@ variable "apm_server_bin_path" {
   type        = string
   description = "Optionally use the apm-server binary from the specified path instead"
 }
+
+variable "apm_server_tail_sampling" {
+  default     = false
+  description = "Whether or not to enable APM Server tail-based sampling. Defaults to false"
+  type        = bool
+}
+
+variable "apm_server_tail_sampling_storage_limit" {
+  default     = "10GB"
+  description = "Storage size limit of APM Server tail-based sampling. Defaults to 10GB"
+  type        = string
+}