private Function maybeTombstone()

in src/main/java/com/amazonaws/mskdatagen/core/CompileGeneratorStrategies.java [96:110]


    private Function<DepsParameters, ResultGen> maybeTombstone(Context context, String topic, String ns, Function<DepsParameters, ResultGen> depsResultGenFunction) {
        Double topicRate = context.getTopicConfigs()
                .filter(t -> topic.equals(t.getTopic()))
                .findFirst().map(TopicConfigs::getTombstoneRate).orElse(null);

        if ("value".equals(ns) && topicRate != null) {
            return input -> {
                if (Math.random() >= topicRate) {
                    return depsResultGenFunction.apply(input);
                }
                return new ResultGen(false);
            };
        }
        return depsResultGenFunction;
    }