@@ -306,8 +306,12 @@ private void configureInsertConcurrencyAndBatchSize(int totalRequests) {
306306 if (userProvidedBatchSize != null ) {
307307 try {
308308 int batchSize = Integer .parseInt (userProvidedBatchSize );
309+ if (batchSize > Constants .MAX_DETOKENIZE_BATCH_SIZE ) {
310+ LogUtil .printWarningLog (WarningLogs .BATCH_SIZE_EXCEEDS_MAX_LIMIT .getLog ());
311+ }
309312 int maxBatchSize = Math .min (batchSize , Constants .MAX_INSERT_BATCH_SIZE );
310313 if (maxBatchSize > 0 ) {
314+ // LogUtil.printInfoLog(InfoLogs.SET_BATCH_SIZE.getLog() + maxBatchSize);
311315 this .insertBatchSize = maxBatchSize ;
312316 } else {
313317 LogUtil .printWarningLog (WarningLogs .INVALID_BATCH_SIZE_PROVIDED .getLog ());
@@ -326,8 +330,11 @@ private void configureInsertConcurrencyAndBatchSize(int totalRequests) {
326330 try {
327331 int concurrencyLimit = Integer .parseInt (userProvidedConcurrencyLimit );
328332 int maxConcurrencyLimit = Math .min (concurrencyLimit , Constants .MAX_INSERT_CONCURRENCY_LIMIT );
329-
333+ if (concurrencyLimit > Constants .MAX_DETOKENIZE_CONCURRENCY_LIMIT ) {
334+ LogUtil .printWarningLog (WarningLogs .CONCURRENCY_EXCEEDS_MAX_LIMIT .getLog ());
335+ }
330336 if (maxConcurrencyLimit > 0 ) {
337+ // LogUtil.printInfoLog(InfoLogs.SET_CONCURRENCY_LIMIT.getLog() + maxConcurrencyLimit);
331338 this .insertConcurrencyLimit = Math .min (maxConcurrencyLimit , maxConcurrencyNeeded );
332339 } else {
333340 LogUtil .printWarningLog (WarningLogs .INVALID_CONCURRENCY_LIMIT_PROVIDED .getLog ());
@@ -357,8 +364,12 @@ private void configureDetokenizeConcurrencyAndBatchSize(int totalRequests) {
357364 if (userProvidedBatchSize != null ) {
358365 try {
359366 int batchSize = Integer .parseInt (userProvidedBatchSize );
367+ if (batchSize > Constants .MAX_DETOKENIZE_BATCH_SIZE ) {
368+ LogUtil .printWarningLog (WarningLogs .BATCH_SIZE_EXCEEDS_MAX_LIMIT .getLog ());
369+ }
360370 int maxBatchSize = Math .min (batchSize , Constants .MAX_DETOKENIZE_BATCH_SIZE );
361371 if (maxBatchSize > 0 ) {
372+ // LogUtil.printInfoLog(InfoLogs.SET_BATCH_SIZE.getLog() + maxBatchSize);
362373 this .detokenizeBatchSize = maxBatchSize ;
363374 } else {
364375 LogUtil .printWarningLog (WarningLogs .INVALID_BATCH_SIZE_PROVIDED .getLog ());
@@ -376,9 +387,13 @@ private void configureDetokenizeConcurrencyAndBatchSize(int totalRequests) {
376387 if (userProvidedConcurrencyLimit != null ) {
377388 try {
378389 int concurrencyLimit = Integer .parseInt (userProvidedConcurrencyLimit );
390+ if (concurrencyLimit > Constants .MAX_DETOKENIZE_CONCURRENCY_LIMIT ) {
391+ LogUtil .printWarningLog (WarningLogs .CONCURRENCY_EXCEEDS_MAX_LIMIT .getLog ());
392+ }
379393 int maxConcurrencyLimit = Math .min (concurrencyLimit , Constants .MAX_DETOKENIZE_CONCURRENCY_LIMIT );
380394
381395 if (maxConcurrencyLimit > 0 ) {
396+ // LogUtil.printInfoLog(InfoLogs.SET_CONCURRENCY_LIMIT.getLog() + maxConcurrencyLimit);
382397 this .detokenizeConcurrencyLimit = Math .min (maxConcurrencyLimit , maxConcurrencyNeeded );
383398 } else {
384399 LogUtil .printWarningLog (WarningLogs .INVALID_CONCURRENCY_LIMIT_PROVIDED .getLog ());
0 commit comments