fapaul commented on a change in pull request #16023: URL: https://github.com/apache/flink/pull/16023#discussion_r644714868
########## File path: flink-connectors/flink-connector-rabbitmq/src/main/java/org/apache/flink/streaming/connectors/rabbitmq/common/RMQConnectionConfig.java ########## @@ -29,21 +29,24 @@ import java.security.KeyManagementException; import java.security.NoSuchAlgorithmException; import java.util.Optional; +import java.util.concurrent.TimeUnit; /** * Connection Configuration for RMQ. If {@link Builder#setUri(String)} has been set then {@link * RMQConnectionConfig#RMQConnectionConfig(String, Integer, Boolean, Boolean, Integer, Integer, - * Integer, Integer, Integer)} will be used for initialize the RMQ connection or {@link + * Integer, Integer, Integer, Long)} will be used for initialize the RMQ connection or {@link * RMQConnectionConfig#RMQConnectionConfig(String, Integer, String, String, String, Integer, - * Boolean, Boolean, Integer, Integer, Integer, Integer, Integer)} will be used for initialize the - * RMQ connection + * Boolean, Boolean, Integer, Integer, Integer, Integer, Integer, Long)} will be used for initialize + * the RMQ connection */ public class RMQConnectionConfig implements Serializable { private static final long serialVersionUID = 1L; private static final Logger LOG = LoggerFactory.getLogger(RMQConnectionConfig.class); + private static final long DEFAULT_DELIVERY_TIMEOUT = 30000; Review comment: I'd prefer to have this value as part of the `RMQSource` class. This way this class does not hold any logic and it can be easier reused. ########## File path: flink-connectors/flink-connector-rabbitmq/src/main/java/org/apache/flink/streaming/connectors/rabbitmq/common/RMQConnectionConfig.java ########## @@ -506,16 +532,45 @@ public Builder setPrefetchCount(int prefetchCount) { return this; } + /** + * Enables setting the next message delivery timeout in the queueing consumer. Only + * applicable to the {@link RMQSource}. Set to 0 for unlimited (the consumer will be blocked + * until an element becomes available). Default is 30000. Review comment: Nit: If you move the constant variable out of this class please also update the comments. ########## File path: flink-connectors/flink-connector-rabbitmq/src/main/java/org/apache/flink/streaming/connectors/rabbitmq/common/RMQConnectionConfig.java ########## @@ -506,16 +532,45 @@ public Builder setPrefetchCount(int prefetchCount) { return this; } + /** + * Enables setting the next message delivery timeout in the queueing consumer. Only + * applicable to the {@link RMQSource}. Set to 0 for unlimited (the consumer will be blocked + * until an element becomes available). Default is 30000. + * + * @param deliveryTimeout maximum wait time, in milliseconds, for the next message delivery + * @return the Builder + */ + public Builder setDeliveryTimeout(long deliveryTimeout) { + Preconditions.checkArgument( + deliveryTimeout >= 0, "deliveryTimeout can not be negative"); + this.deliveryTimeout = deliveryTimeout; + return this; + } + + /** + * Enables setting the next message delivery timeout in the queueing consumer. Only + * applicable to the {@link RMQSource}. Set to 0 for unlimited (the consumer will be blocked + * until an element becomes available). Default is 30 seconds. Review comment: Nit: update comment if constant is moved. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org