1 /***********************************************************************
2  * Copyright (c) 2013-2025 General Atomics Integrated Intelligence, Inc.
3  * All rights reserved. This program and the accompanying materials
4  * are made available under the terms of the Apache License, Version 2.0
5  * which accompanies this distribution and is available at
6  * https://www.apache.org/licenses/LICENSE-2.0
7  ***********************************************************************/
8 
9 package org.locationtech.geomesa.accumulo.util
10 
11 import com.typesafe.scalalogging.LazyLogging
12 import org.apache.accumulo.core.client.BatchWriterConfig
13 import org.locationtech.geomesa.accumulo.AccumuloProperties
14 
15 import java.util.concurrent.TimeUnit
16 
17 object GeoMesaBatchWriterConfig extends LazyLogging {
18 
19   def apply(threads: Option[Int] = None,
20             memory: Option[Long] = None,
21             latency: Option[Long] = None,
22             timeout: Option[Long] = None): BatchWriterConfig = {
23     import AccumuloProperties.BatchWriterProperties
24 
25     val bwc = new BatchWriterConfig
26 
27     threads.orElse(BatchWriterProperties.WRITER_THREADS.option.map(_.toInt)).foreach { threads =>
28       logger.trace(s"GeoMesaBatchWriter config: maxWriteThreads set to $threads")
29       bwc.setMaxWriteThreads(threads)
30     }
31 
32     memory.orElse(BatchWriterProperties.WRITER_MEMORY_BYTES.toBytes).foreach { memory =>
33       logger.trace(s"GeoMesaBatchWriter config: maxMemory set to $memory bytes")
34       bwc.setMaxMemory(memory)
35     }
36 
37     latency.orElse(BatchWriterProperties.WRITER_LATENCY.toDuration.map(_.toMillis)).foreach { latency =>
38       logger.trace(s"GeoMesaBatchWriter config: maxLatency set to $latency millis")
39       bwc.setMaxLatency(latency, TimeUnit.MILLISECONDS)
40     }
41 
42     timeout.orElse(BatchWriterProperties.WRITE_TIMEOUT.toDuration.map(_.toMillis)).foreach { timeout =>
43       logger.trace(s"GeoMesaBatchWriter config: maxTimeout set to $timeout millis")
44       bwc.setTimeout(timeout, TimeUnit.MILLISECONDS)
45     }
46 
47     bwc
48   }
49 }