Skip to content

Commit

Permalink
Create sampling templates when creating sampling store
Browse files Browse the repository at this point in the history
Signed-off-by: Jaegoo Kim <rlaworn1993@gmail.com>
  • Loading branch information
JaeguKim committed Apr 11, 2024
1 parent aa9cefc commit 1f2d4c6
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions plugin/storage/es/factory.go
Original file line number Diff line number Diff line change
Expand Up @@ -317,6 +317,30 @@ func (f *Factory) CreateSamplingStore(maxBuckets int) (samplingstore.Store, erro
Lookback: f.primaryConfig.AdaptiveSamplingLookback,
MaxDocCount: f.primaryConfig.MaxDocCount,
})

if f.primaryConfig.CreateIndexTemplates && !f.primaryConfig.UseILM {
mappingBuilder := mappings.MappingBuilder{
TemplateBuilder: es.TextTemplateBuilder{},
Shards: f.primaryConfig.NumShards,
Replicas: f.primaryConfig.NumReplicas,
EsVersion: f.primaryConfig.Version,
IndexPrefix: f.primaryConfig.IndexPrefix,
UseILM: f.primaryConfig.UseILM,
PrioritySpanTemplate: f.primaryConfig.PrioritySpanTemplate,
PriorityServiceTemplate: f.primaryConfig.PriorityServiceTemplate,
PriorityDependenciesTemplate: f.primaryConfig.PriorityDependenciesTemplate,
}

sampleMapping, err := mappingBuilder.GetSamplingMappings()
if err != nil {
return nil, err
}

if err := store.CreateTemplates(sampleMapping); err != nil {
return nil, err
}
}

return store, nil
}

Expand Down

0 comments on commit 1f2d4c6

Please sign in to comment.