diff --git a/Examples/src/main/java/io/github/jwdeveloper/tiktok/RecorderExample.java b/Examples/src/main/java/io/github/jwdeveloper/tiktok/RecorderExample.java index 7f8f8cb..9c10e7b 100644 --- a/Examples/src/main/java/io/github/jwdeveloper/tiktok/RecorderExample.java +++ b/Examples/src/main/java/io/github/jwdeveloper/tiktok/RecorderExample.java @@ -41,7 +41,7 @@ public class RecorderExample { { event.getException().printStackTrace(); }) - .addListener(TikTokLiveRecorder.use(recorderSettings -> + .addListener(TikTokLiveRecorder.use((recorderSettings, liveClient) -> { recorderSettings.setFfmpegPath("C:\\Users\\ja\\IdeaProjects\\TikTokLiveJava\\extension-recorder\\libs\\ffmpeg.exe"); recorderSettings.setOutputFile(new File("C:\\Users\\ja\\IdeaProjects\\TikTokLiveJava\\extension-recorder\\out", "test.flv")); diff --git a/extension-recorder/src/main/java/io/github/jwdeveloper/tiktok/extension/recorder/TikTokLiveRecorder.java b/extension-recorder/src/main/java/io/github/jwdeveloper/tiktok/extension/recorder/TikTokLiveRecorder.java index 4db5f38..a54e786 100644 --- a/extension-recorder/src/main/java/io/github/jwdeveloper/tiktok/extension/recorder/TikTokLiveRecorder.java +++ b/extension-recorder/src/main/java/io/github/jwdeveloper/tiktok/extension/recorder/TikTokLiveRecorder.java @@ -25,18 +25,19 @@ package io.github.jwdeveloper.tiktok.extension.recorder; import io.github.jwdeveloper.tiktok.extension.recorder.api.LiveRecorder; import io.github.jwdeveloper.tiktok.extension.recorder.impl.RecorderListener; import io.github.jwdeveloper.tiktok.extension.recorder.impl.data.RecorderSettings; +import io.github.jwdeveloper.tiktok.live.LiveClient; -import java.util.function.Consumer; +import java.util.function.*; public class TikTokLiveRecorder { - public static LiveRecorder use(Consumer consumer) + public static LiveRecorder use(BiConsumer consumer) { return new RecorderListener(consumer); } public static LiveRecorder use() { - return use(x ->{}); + return use((x,y) ->{}); } } \ No newline at end of file diff --git a/extension-recorder/src/main/java/io/github/jwdeveloper/tiktok/extension/recorder/impl/RecorderListener.java b/extension-recorder/src/main/java/io/github/jwdeveloper/tiktok/extension/recorder/impl/RecorderListener.java index 45eadae..66b1caf 100644 --- a/extension-recorder/src/main/java/io/github/jwdeveloper/tiktok/extension/recorder/impl/RecorderListener.java +++ b/extension-recorder/src/main/java/io/github/jwdeveloper/tiktok/extension/recorder/impl/RecorderListener.java @@ -37,23 +37,23 @@ import io.github.jwdeveloper.tiktok.models.ConnectionState; import javax.net.ssl.HttpsURLConnection; import java.io.*; import java.net.URL; -import java.util.function.Consumer; +import java.util.function.*; public class RecorderListener implements LiveRecorder { - private final Consumer consumer; + private final BiConsumer consumer; private RecorderSettings settings; private DownloadData downloadData; private Thread liveDownloadThread; - public RecorderListener(Consumer consumer) { + public RecorderListener(BiConsumer consumer) { this.consumer = consumer; } @TikTokEventObserver private void onResponse(LiveClient liveClient, TikTokRoomDataResponseEvent event) { settings = RecorderSettings.DEFAULT(); - consumer.accept(settings); + consumer.accept(settings, liveClient); var json = event.getLiveData().getJson();