Skip to content

Commit 7502099

Browse files
committed
feat: ability to set custom probe config file
1 parent ce70ca5 commit 7502099

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

control/src/main/kotlin/spp/probe/ProbeConfiguration.kt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,11 @@ object ProbeConfiguration {
2929

3030
private var rawProperties: Map<String, Map<String, Any>>? = null
3131
private var localProperties: JsonObject? = null
32+
var customProbeFile: String? = null
3233

33-
init {
34+
fun load() {
3435
var localFile = File("spp-probe.yml")
36+
customProbeFile?.let { localFile = File(it) }
3537
try {
3638
//working directory?
3739
val mapper = ObjectMapper(YAMLFactory())

control/src/main/kotlin/spp/probe/SourceProbe.kt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,9 @@ object SourceProbe {
111111

112112
@JvmStatic
113113
fun premain(args: String?, inst: Instrumentation) {
114+
ProbeConfiguration.customProbeFile = args
115+
ProbeConfiguration.load()
116+
if (ProbeConfiguration.isNotQuite) println("SourceProbe initiated via premain. args: $args")
114117
if (isAgentInitialized) {
115118
if (ProbeConfiguration.isNotQuite) println("SourceProbe is already initialized")
116119
return

0 commit comments

Comments
 (0)