Add jsonnet for ingester StatefulSet with WAL

Signed-off-by: Ganesh Vernekar <[email protected]>

cortex/config.libsonnet

@@ -63,6 +63,10 @@
 
     store_gateway_replication_factor: 3,
 
+    // By default ingesters will be run as StatefulSet with WAL.
+    // If this is set to true, ingesters will use staless deployments without WAL.
+    ingester_deployment_without_wal: false,
+
     // Blocks storage engine doesn't require the table manager.
     // When running blocks with chunks as secondary storage engine for querier only, we need table-manager to apply
     // retention policies.

cortex/ingester.libsonnet

@@ -1,6 +1,4 @@
 {
-  local container = $.core.v1.container,
-
   ingester_args::
     $._config.ringConfig +
     $._config.storeConfig +
@@ -40,9 +38,28 @@
       else {}
     ),
 
+  ingester_statefulset_args:: {
+    'ingester.wal-enabled': true,
+    'ingester.checkpoint-enabled': true,
+    'ingester.recover-from-wal': true,
+    'ingester.wal-dir': $._config.wal_dir,
+    'ingester.checkpoint-duration': '15m',
+    '-log.level': 'info',
+    'ingester.tokens-file-path': $._config.wal_dir + '/tokens',
+  },
+
+  _config+:: {
+    wal_dir: '/wal_data',
+    ingester+: {
+      statefulset_replicas: 3,
+      statefulset_disk: '150Gi',
+    },
+  },
+
   ingester_ports:: $.util.defaultPorts,
 
   local name = 'ingester',
+  local container = $.core.v1.container,
 
   ingester_container::
     container.new(name, $._images.ingester) +
@@ -53,11 +70,53 @@
     $.util.readinessProbe +
     $.jaeger_mixin,
 
-  local deployment = $.apps.v1.deployment,
+  local volumeMount = $.core.v1.volumeMount,
+
+  ingester_statefulset_container::
+    $.ingester_container +
+    container.withArgsMixin($.util.mapToFlags($.ingester_statefulset_args)) +
+    container.withVolumeMountsMixin([
+      volumeMount.new('ingester-pvc', $._config.wal_dir),
+    ]),
 
   ingester_deployment_labels:: {},
 
+  local pvc = $.core.v1.persistentVolumeClaim,
+  local volume = $.core.v1.volume,
+  local statefulSet = $.apps.v1.statefulSet,
+
+  local ingester_pvc =
+    pvc.new('ingester-pvc') +
+    pvc.mixin.spec.resources.withRequests({ storage: $._config.ingester.statefulset_disk }) +
+    pvc.mixin.spec.withAccessModes(['ReadWriteOnce']) +
+    pvc.mixin.spec.withStorageClassName('fast'),
+
+  statefulset_storage_config_mixin::
+    statefulSet.mixin.spec.template.metadata.withAnnotationsMixin({ schemaID: $._config.schemaID },) +
+    $.util.configVolumeMount('schema-' + $._config.schemaID, '/etc/cortex/schema'),
+
+  ingester_statefulset:
+  if $._config.ingester_deployment_without_wal == false then
+    statefulSet.new('ingester', $._config.ingester.statefulset_replicas, [$.ingester_statefulset_container], ingester_pvc) +
+    statefulSet.mixin.spec.withServiceName('ingester') +
+    statefulSet.mixin.spec.template.spec.withVolumes([volume.fromPersistentVolumeClaim('ingester-pvc', 'ingester-pvc')]) +
+    statefulSet.mixin.metadata.withNamespace($._config.namespace) +
+    statefulSet.mixin.metadata.withLabels({ name: 'ingester' }) +
+    statefulSet.mixin.spec.template.metadata.withLabels({ name: 'ingester' } + $.ingester_deployment_labels) +
+    statefulSet.mixin.spec.selector.withMatchLabels({ name: 'ingester' }) +
+    statefulSet.mixin.spec.template.spec.securityContext.withRunAsUser(0) +
+    statefulSet.mixin.spec.template.spec.withTerminationGracePeriodSeconds(4800) +
+    statefulSet.mixin.spec.updateStrategy.withType('RollingUpdate') +
+    $.statefulset_storage_config_mixin +
+    $.util.configVolumeMount('overrides', '/etc/cortex') +
+    $.util.podPriority('high') +
+    $.util.antiAffinityStatefulSet
+  else null,
+
+  local deployment = $.apps.v1.deployment,
+
   ingester_deployment:
+  if $._config.ingester_deployment_without_wal then
     deployment.new(name, 3, [$.ingester_container], $.ingester_deployment_labels) +
     $.util.antiAffinity +
     $.util.configVolumeMount('overrides', '/etc/cortex') +
@@ -67,12 +126,16 @@
     deployment.mixin.spec.strategy.rollingUpdate.withMaxUnavailable(1) +
     deployment.mixin.spec.template.spec.withTerminationGracePeriodSeconds(4800) +
     $.storage_config_mixin +
-    $.util.podPriority('high'),
+    $.util.podPriority('high')
+  else null,
 
   ingester_service_ignored_labels:: [],
 
   ingester_service:
-    $.util.serviceFor($.ingester_deployment, $.ingester_service_ignored_labels),
+  if $._config.ingester_deployment_without_wal then
+    $.util.serviceFor($.ingester_deployment, $.ingester_service_ignored_labels)
+  else
+    $.util.serviceFor($.ingester_statefulset, $.ingester_service_ignored_labels),
 
   local podDisruptionBudget = $.policy.v1beta1.podDisruptionBudget,