From 8abb17312f788fb2b510145e22d35f31c2a96c01 Mon Sep 17 00:00:00 2001
From: bsorrentino <bartolomeo.sorrentino@gmail.com>
Date: Thu, 29 Aug 2024 21:55:00 +0200
Subject: [PATCH] feat: add factory methods to create custom Channel

---
 .../org/bsc/langgraph4j/state/Channel.java    | 34 +++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/core-jdk8/src/main/java/org/bsc/langgraph4j/state/Channel.java b/core-jdk8/src/main/java/org/bsc/langgraph4j/state/Channel.java
index dce7ac5..404242d 100644
--- a/core-jdk8/src/main/java/org/bsc/langgraph4j/state/Channel.java
+++ b/core-jdk8/src/main/java/org/bsc/langgraph4j/state/Channel.java
@@ -3,6 +3,30 @@
 import java.util.Optional;
 import java.util.function.Supplier;
 
+import static java.util.Optional.ofNullable;
+
+/**
+ *
+ * @param <T>
+ */
+class BaseChannel<T> implements Channel<T> {
+    final Supplier<T> defaultProvider;
+    final Reducer<T> reducer;
+
+    BaseChannel(Reducer<T> reducer, Supplier<T> defaultProvider ) {
+        this.defaultProvider = defaultProvider;
+        this.reducer = reducer;
+    }
+
+    public Optional<Supplier<T>> getDefault() {
+        return ofNullable(defaultProvider);
+    }
+
+    public Optional<Reducer<T>> getReducer() {
+        return ofNullable(reducer);
+    }
+}
+
 /**
  * A Channel is a mechanism used to maintain a state property.
  * <p>
@@ -28,6 +52,16 @@
  */
 public interface Channel<T> {
 
+    static <T> Channel<T> of( Supplier<T> defaultProvider) {
+        return new BaseChannel<T>(null, defaultProvider);
+    }
+    static <T> Channel<T> of( Reducer<T> reducer ) {
+        return new BaseChannel<T>(reducer, null);
+    }
+    static <T> Channel<T> of( Reducer<T> reducer, Supplier<T> defaultProvider ) {
+        return new BaseChannel<T>(reducer, defaultProvider);
+    }
+
     /**
      * The Reducer, if provided, is invoked for each state property to compute value.
      *