[BEAM-12631] Allow Inheritance from KafkaUnboundedSource

• Type: New Feature
• Status: Open
• Priority: P3
• Resolution: Unresolved
• Component/s: io-java-kafka
• Assignee: Unassigned
• Reporter: HongXiang Ren
• Created: 16/Jul/21 18:58
• Updated: 04/Jun/22 20:58

Description

Reason:

We have dataflow to call one internal API (the API call take 60ms). So we want to start more threads to call this API. Kafka IO thread is control by split by default. And split is control by total partitions and system suggestion number. And system suggestion number is 4 * max workers. which is too low for us.  

Solution:

Use facture to allow user inheritance KafkaUnboundedSource.  and override the split function like

 

// code placeholder
    @Override
    public List<KafkaUnboundedSource<K, V>> split(int desiredNumSplits, PipelineOptions options)
            throws Exception {
        int split = this.getMaxSplits();
        if(split <= 0){
            return super.split(desiredNumSplits, options);
        }else{
            return super.split(split, options);
        }
    }

 this may also useful for Override other feature of the KafkaUnboundedSource without copy all the io code.