From 3b1a908e08d82911082b100a29b94b7ca569f2b5 Mon Sep 17 00:00:00 2001
From: jadamcrain <jadamcrain@gmail.com>
Date: Thu, 25 Apr 2024 13:28:51 -0700
Subject: [PATCH] Add java master udp example

---
 .../stepfunc/dnp3/examples/MasterExample.java | 66 ++++++++++++++-----
 1 file changed, 51 insertions(+), 15 deletions(-)

diff --git a/ffi/bindings/java/examples/src/main/java/io/stepfunc/dnp3/examples/MasterExample.java b/ffi/bindings/java/examples/src/main/java/io/stepfunc/dnp3/examples/MasterExample.java
index 15a1e52a..65403225 100644
--- a/ffi/bindings/java/examples/src/main/java/io/stepfunc/dnp3/examples/MasterExample.java
+++ b/ffi/bindings/java/examples/src/main/java/io/stepfunc/dnp3/examples/MasterExample.java
@@ -381,6 +381,35 @@ private static void runTls(Runtime runtime, TlsClientConfig tlsConfig) throws Ex
     }
   }
 
+  private static void runUdp(Runtime runtime) throws Exception {
+    // ANCHOR: create_udp_channel
+    MasterChannel channel =
+            MasterChannel.createUdpChannel(
+                    runtime,
+                    getMasterChannelConfig(),
+                    "127.0.0.1:20001",
+                    LinkReadMode.DATAGRAM,
+                    Duration.ofSeconds(5)
+            );
+    // ANCHOR_END: create_udp_channel
+
+    try {
+      AssociationId association =
+              channel.addUdpAssociation(
+                      ushort(1024),
+                      "127.0.0.1:20000",
+                      getAssociationConfig(),
+                      new TestReadHandler(),
+                      new TestAssociationHandler(),
+                      new TestAssociationInformation());
+
+      runAssociation(channel, association);
+    }
+    finally {
+      channel.shutdown();
+    }
+  }
+
   private static void runSerial(Runtime runtime) throws Exception {
     // ANCHOR: create_serial_channel
     MasterChannel channel =
@@ -434,6 +463,9 @@ private static void run(Runtime runtime, String[] args) throws Exception {
       case "tcp":
         runTcp(runtime);
         break;
+      case "udp":
+        runUdp(runtime);
+        break;
       case "serial":
         runSerial(runtime);
         break;
@@ -612,24 +644,12 @@ private static List<UByte> getFileLine() {
     return bytes;
   }
 
-  private static void runChannel(MasterChannel channel) {
-
-    // Create the association
-    // ANCHOR: association_create
-    AssociationId association =
-        channel.addAssociation(
-            ushort(1024),
-            getAssociationConfig(),
-            new TestReadHandler(),
-            new TestAssociationHandler(),
-            new TestAssociationInformation());
-    // ANCHOR_END: association_create
-
+  private static void runAssociation(MasterChannel channel, AssociationId association) {
     // Create a periodic poll
     // ANCHOR: add_poll
     PollId poll =
-        channel.addPoll(
-            association, Request.classRequest(false, true, true, true), Duration.ofSeconds(5));
+            channel.addPoll(
+                    association, Request.classRequest(false, true, true, true), Duration.ofSeconds(5));
     // ANCHOR_END: add_poll
 
     // start communications
@@ -651,6 +671,22 @@ private static void runChannel(MasterChannel channel) {
     }
   }
 
+  private static void runChannel(MasterChannel channel) {
+
+    // Create the association
+    // ANCHOR: association_create
+    AssociationId association =
+        channel.addAssociation(
+            ushort(1024),
+            getAssociationConfig(),
+            new TestReadHandler(),
+            new TestAssociationHandler(),
+            new TestAssociationInformation());
+    // ANCHOR_END: association_create
+
+    runAssociation(channel, association);
+  }
+
   private static void printFileInfo(FileInfo info) {
     System.out.println("file name: " + info.fileName);
     System.out.println("     type: " + info.fileType);