g3fcgen: support get udp listen addr from env

bytedance · Aug 16, 2023 · 9ae5810 · 9ae5810
1 parent aedc823
commit 9ae5810
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/g3fcgen/service/[email protected] b/g3fcgen/service/[email protected]
@@ -3,7 +3,7 @@ Description=G3 X.509 Certificates Generation Service at %i
 
 [Service]
 Type=simple
-EnvironmentFile=-/etc/g3fcgen/env
+EnvironmentFile=-/etc/g3fcgen/%i/env
 ExecStart=/usr/bin/g3fcgen -c /etc/g3fcgen/%i/main.yml -s -G %i
 
 [Install]

diff --git a/g3fcgen/src/opts.rs b/g3fcgen/src/opts.rs
@@ -172,5 +172,11 @@ pub fn parse_clap() -> anyhow::Result<Option<ProcArgs>> {
         }
     }
 
+    if let Some(s) = option_env!("UDP_LISTEN_ADDR") {
+        if let Ok(addr) = SocketAddr::from_str(s) {
+            proc_args.udp_addr = Some(addr);
+        }
+    }
+
     Ok(Some(proc_args))
 }