Compare commits

...

14 Commits

Author SHA1 Message Date
kohlerpop1
95e352908d Minor fix to proxying requests. Finally preserves state and proxies all requests when enabled. 2025-12-28 22:12:30 -05:00
GitHub Action
3f00256634 Update version in pom.xml 2025-11-16 04:30:57 +00:00
kohlerpop1
089d8d6ed8 Minor changes to new social/gift events! 2025-11-15 23:17:39 -05:00
GosDev
4be74c45ff Implement Social Event instances with custom Users (#151)
* Add .of methods for custom User objects

Adds an extra .of() function that accepts a User object, instead of a username string.

* Use username if profile name not set

* Add combo argument

* Replace WebcastGiftMessage from the new constructor

* Fix argument naming error

* Fix wrong argument order
2025-11-15 22:57:43 -05:00
GitHub Action
db4d382e34 Update version in pom.xml 2025-11-09 05:09:03 +00:00
kohlerpop1
2590200205 Temp fix until Eulerstream passes currentViewers! 2025-11-09 00:06:43 -05:00
GitHub Action
4aefde8a0c Update version in pom.xml 2025-11-09 03:43:26 +00:00
kohlerpop1
6486519876 Fix Eulerstream websocket roomInfo incorrectly being mapped! 2025-11-08 22:41:09 -05:00
GitHub Action
96cf28e5d5 Update version in pom.xml 2025-10-07 03:11:13 +00:00
kohlerpop1
cfdced9645 Add direct method to provide sessionId and ttTargetIdc for sending chats from 1 client. 2025-10-06 23:08:10 -04:00
GitHub Action
7589a2ac4a Update version in pom.xml 2025-10-05 02:40:59 +00:00
kohlerpop1
a0c445656c Slight change of decoding for TikTokWebSocketEulerListener! 2025-10-04 22:38:47 -04:00
GitHub Action
1e78fdda89 Update version in pom.xml 2025-09-24 23:28:34 +00:00
David Kohler
57f33b2efa Change static schedulers to AsyncHandler to hold for heartbeat and reconnect logic. (#148) 2025-09-24 19:26:44 -04:00
28 changed files with 253 additions and 123 deletions

View File

@@ -5,7 +5,7 @@
<parent>
<artifactId>TikTokLiveJava</artifactId>
<groupId>io.github.jwdeveloper.tiktok</groupId>
<version>1.11.2-Release</version>
<version>1.11.8-Release</version>
</parent>
<modelVersion>4.0.0</modelVersion>
<artifactId>API</artifactId>

View File

@@ -39,11 +39,16 @@ public class TikTokGiftComboEvent extends TikTokGiftEvent {
this.comboState = comboState;
}
public TikTokGiftComboEvent(Gift gift, User host, User user, int combo, GiftComboStateType comboState) {
super(gift, user, host, combo);
this.comboState = comboState;
}
public static TikTokGiftComboEvent of(Gift gift, int combo, GiftComboStateType comboState) {
return new TikTokGiftComboEvent(
gift,
new User(0L, "Test", new Picture("")),
WebcastGiftMessage.newBuilder().setComboCount(combo).build(),
comboState);
return new TikTokGiftComboEvent(gift, new User(0L, "Test", new Picture("")), WebcastGiftMessage.newBuilder().setComboCount(combo).build(), comboState);
}
public static TikTokGiftComboEvent of(Gift gift, User host, User user, int combo, GiftComboStateType comboState) {
return new TikTokGiftComboEvent(gift, host, user, combo, comboState);
}
}

View File

@@ -49,17 +49,25 @@ public class TikTokGiftEvent extends TikTokHeaderEvent {
}
combo = msg.getComboCount();
}
public TikTokGiftEvent(Gift gift) {
public TikTokGiftEvent(Gift gift, User user, User toUser, int combo) {
this.gift = gift;
user = new User(0L, "sender", new Picture(""));
toUser = new User(0L, "receiver", new Picture(""));
combo = 1;
this.user = user;
this.toUser = toUser;
this.combo = combo;
}
public static TikTokGiftEvent of(Gift gift) {
return new TikTokGiftEvent(gift);
return new TikTokGiftEvent(
gift,
new User(0L, "sender", new Picture("")),
new User(0L, "reviever", new Picture("")),
1
);
}
public static TikTokGiftEvent of(Gift gift, User user, User toUser) {
return new TikTokGiftEvent(gift, user, toUser, 1) ;
}
public static TikTokGiftEvent of(String name, int id, int diamonds) {

View File

@@ -44,10 +44,20 @@ public class TikTokFollowEvent extends TikTokHeaderEvent {
public static TikTokFollowEvent of(String userName) {
return new TikTokFollowEvent(WebcastSocialMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName)
.setNickname(userName)
.build())
.build());
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName)
.setNickname(userName)
.build())
.build());
}
public static TikTokFollowEvent of(User user) {
return new TikTokFollowEvent(WebcastSocialMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(user.getName())
.setNickname(user.getProfileName() != null ? user.getProfileName() : user.getName())
.setId(user.getId())
.build())
.build());
}
}

View File

@@ -48,13 +48,22 @@ public class TikTokJoinEvent extends TikTokHeaderEvent {
totalUsers = msg.getMemberCount();
}
public static TikTokJoinEvent of(String userName)
{
public static TikTokJoinEvent of(String userName) {
return new TikTokJoinEvent(WebcastMemberMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName)
.setNickname(userName)
.build())
.build());
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName)
.setNickname(userName)
.build())
.build());
}
public static TikTokJoinEvent of(User user) {
return new TikTokJoinEvent(WebcastMemberMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(user.getName())
.setNickname(user.getProfileName())
.setId(user.getId())
.build())
.build());
}
}

View File

@@ -56,15 +56,25 @@ public class TikTokLikeEvent extends TikTokHeaderEvent
totalLikes = msg.getTotal();
}
public static TikTokLikeEvent of(String userName, int likes)
{
public static TikTokLikeEvent of(String userName, int likes) {
return new TikTokLikeEvent(WebcastLikeMessage.newBuilder()
.setCount(likes)
.setTotal(likes)
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName)
.setNickname(userName)
.build())
.build());
.setCount(likes)
.setTotal(likes)
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName)
.setNickname(userName)
.build())
.build());
}
public static TikTokLikeEvent of(User user, int likes) {
return new TikTokLikeEvent(WebcastLikeMessage.newBuilder()
.setCount(likes)
.setTotal(likes)
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(user.getName())
.setNickname(user.getProfileName())
.build())
.build());
}
}

View File

@@ -49,10 +49,19 @@ public class TikTokShareEvent extends TikTokHeaderEvent {
public static TikTokShareEvent of(String userName, int shaders) {
return new TikTokShareEvent(WebcastSocialMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName)
.setNickname(userName)
.build())
.build(), shaders);
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName)
.setNickname(userName)
.build())
.build(), shaders);
}
public static TikTokShareEvent of(User user, int shaders) {
return new TikTokShareEvent(WebcastSocialMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(user.getName())
.setNickname(user.getProfileName())
.build())
.build(), shaders);
}
}

View File

@@ -95,6 +95,9 @@ public class ProxyClientSettings implements Iterator<ProxyData>, Iterable<ProxyD
public ProxyClientSettings clone() {
ProxyClientSettings settings = new ProxyClientSettings();
settings.setEnabled(enabled);
settings.setAutoDiscard(autoDiscard);
settings.setFallback(fallback);
settings.setAllowWebsocket(allowWebsocket);
settings.setRotation(rotation);
settings.setIndex(index);
settings.setType(type);

View File

@@ -66,5 +66,5 @@ public interface LiveHttpClient
LiveConnectionData.Response fetchLiveConnectionData(LiveConnectionData.Request request);
boolean sendChat(LiveRoomInfo roomInfo, String content);
boolean sendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc);
}

View File

@@ -102,4 +102,12 @@ public interface LiveClient {
* <p>We cannot fix this as it is a TikTok issue, not a library issue.
*/
boolean sendChat(String content);
/**
* Send a chat message to the connected room
* @return true if successful, otherwise false
* @apiNote This is known to return true on some sessionIds despite failing!
* <p>We cannot fix this as it is a TikTok issue, not a library issue.
*/
boolean sendChat(String content, String sessionId, String ttTargetIdc);
}

View File

@@ -62,12 +62,11 @@ message Text {
int32 type = 1;
TextFormat format = 2;
string stringValue = 11;
oneof textPieceType
{
TextPieceUser userValue = 21;
TextPieceGift giftValue = 22;
}
TextPieceUser userValue = 21;
TextPieceGift giftValue = 22;
TextPieceHeart heartValue = 23;
TextPiecePatternRef patternRefValue = 24;
TextPieceImage imageValue = 25;
}
message TextFormat {
@@ -83,7 +82,7 @@ message Text {
message TextPieceGift {
int32 giftId = 1;
PatternRef nameRef = 2;
TextPiecePatternRef nameRef = 2;
ShowType showType = 3; // Enum
int64 colorId = 4;
}
@@ -98,16 +97,19 @@ message Text {
bool withColon = 2;
}
message PatternRef {
string key = 1;
string default_pattern = 2;
}
enum ShowType {
SHOW_TYPE_NORMAL = 0;
SHOW_TYPE_FADE_IN_OUT = 1;
}
message TextPieceHeart {
string color = 1;
}
message TextPieceImage {
Image image_model = 1;
}
}
// @Image
@@ -151,7 +153,7 @@ message BadgeStruct {
bool is_customized = 24;
message CombineBadge {
int32 badge_display_type = 1;
BadgeDisplayType badge_display_type = 1;
Image icon = 2;
TextBadge text = 3;
string str = 4;

View File

@@ -5,7 +5,7 @@
<parent>
<artifactId>TikTokLiveJava</artifactId>
<groupId>io.github.jwdeveloper.tiktok</groupId>
<version>1.11.2-Release</version>
<version>1.11.8-Release</version>
</parent>
<modelVersion>4.0.0</modelVersion>

View File

@@ -23,6 +23,7 @@
package io.github.jwdeveloper.tiktok;
import com.google.protobuf.ByteString;
import io.github.jwdeveloper.tiktok.common.AsyncHandler;
import io.github.jwdeveloper.tiktok.data.events.*;
import io.github.jwdeveloper.tiktok.data.events.common.TikTokEvent;
import io.github.jwdeveloper.tiktok.data.events.control.*;
@@ -39,7 +40,7 @@ import io.github.jwdeveloper.tiktok.websocket.*;
import lombok.Getter;
import java.util.Base64;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.*;
import java.util.function.Consumer;
import java.util.logging.Logger;
@@ -89,12 +90,11 @@ public class TikTokLiveClient implements LiveClient
tikTokEventHandler.publish(this, new TikTokDisconnectedEvent("Exception: " + e.getMessage()));
if (e instanceof TikTokLiveOfflineHostException && clientSettings.isRetryOnConnectionFailure()) {
try {
Thread.sleep(clientSettings.getRetryConnectionTimeout().toMillis());
} catch (Exception ignored) {}
logger.info("Reconnecting");
tikTokEventHandler.publish(this, new TikTokReconnectingEvent());
this.connect();
AsyncHandler.getReconnectScheduler().schedule(() -> {
logger.info("Reconnecting");
tikTokEventHandler.publish(this, new TikTokReconnectingEvent());
this.connect();
}, clientSettings.getRetryConnectionTimeout().toMillis(), TimeUnit.MILLISECONDS);
}
throw e;
} catch (Exception e) {
@@ -199,7 +199,12 @@ public class TikTokLiveClient implements LiveClient
@Override
public boolean sendChat(String content) {
return httpClient.sendChat(roomInfo, content);
return sendChat(content, clientSettings.getSessionId(), clientSettings.getTtTargetIdc());
}
@Override
public boolean sendChat(String content, String sessionId, String ttTargetIdc) {
return httpClient.sendChat(roomInfo, content, sessionId, ttTargetIdc);
}
public void connectAsync(Consumer<LiveClient> onConnection) {

View File

@@ -133,7 +133,7 @@ public class TikTokLiveClientBuilder implements LiveClientBuilder {
//networking
dependance.registerSingleton(HttpClientFactory.class);
dependance.registerSingleton(WebSocketHeartbeatTask.class); // True global singleton - Static objects are located to serve as global
dependance.registerSingleton(WebSocketHeartbeatTask.class);
if (clientSettings.isOffline()) {
dependance.registerSingleton(LiveSocketClient.class, TikTokWebSocketOfflineClient.class);
dependance.registerSingleton(LiveHttpClient.class, TikTokLiveHttpOfflineClient.class);

View File

@@ -182,23 +182,23 @@ public class TikTokLiveHttpClient implements LiveHttpClient
}
@Override
public boolean sendChat(LiveRoomInfo roomInfo, String content) {
public boolean sendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc) {
var proxyClientSettings = clientSettings.getHttpSettings().getProxyClientSettings();
if (proxyClientSettings.isEnabled()) {
while (proxyClientSettings.hasNext()) {
try {
return requestSendChat(roomInfo, content);
return requestSendChat(roomInfo, content, sessionId, ttTargetIdc);
} catch (TikTokProxyRequestException ignored) {}
}
}
return requestSendChat(roomInfo, content);
return requestSendChat(roomInfo, content, sessionId, ttTargetIdc);
}
public boolean requestSendChat(LiveRoomInfo roomInfo, String content) {
public boolean requestSendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc) {
JsonObject body = new JsonObject();
body.addProperty("content", content);
body.addProperty("sessionId", clientSettings.getSessionId());
body.addProperty("ttTargetIdc", clientSettings.getTtTargetIdc());
body.addProperty("sessionId", sessionId);
body.addProperty("ttTargetIdc", ttTargetIdc);
body.addProperty("roomId", roomInfo.getRoomId());
HttpClientBuilder builder = httpFactory.client(clientSettings.isUseEulerstreamEnterprise() ? TIKTOK_CHAT_ENTERPRISE_URL : TIKTOK_CHAT_URL)
.withHeader("Content-Type", "application/json");

View File

@@ -64,7 +64,7 @@ public class TikTokLiveHttpOfflineClient implements LiveHttpClient {
}
@Override
public boolean sendChat(LiveRoomInfo roomInfo, String content) {
public boolean sendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc) {
// DO NOTHING
return false;
}

View File

@@ -0,0 +1,22 @@
package io.github.jwdeveloper.tiktok.common;
import lombok.Getter;
import java.util.concurrent.*;
public class AsyncHandler
{
@Getter
private static final ScheduledExecutorService heartBeatScheduler = Executors.newScheduledThreadPool(1, r -> {
Thread t = new Thread(r, "heartbeat-pool");
t.setDaemon(true);
return t;
});
@Getter
private static final ScheduledExecutorService reconnectScheduler = Executors.newScheduledThreadPool(0, r -> {
Thread t = new Thread(r, "reconnect-pool");
t.setDaemon(true);
return t;
});
}

View File

@@ -69,7 +69,7 @@ public class HttpClient {
}
}
public <T> ActionResult<T> toResponse(HttpResponse.BodyHandler<T> handler) {
protected <T> ActionResult<T> toResponse(HttpResponse.BodyHandler<T> handler) {
return toHttpResponse(handler).map(HttpResponse::body);
}

View File

@@ -31,9 +31,11 @@ import java.io.IOException;
import java.net.*;
import java.net.http.*;
import java.net.http.HttpResponse.ResponseInfo;
import java.nio.ByteBuffer;
import java.security.*;
import java.security.cert.X509Certificate;
import java.util.*;
import java.util.concurrent.Flow;
import java.util.stream.Collectors;
public class HttpProxyClient extends HttpClient {
@@ -45,14 +47,14 @@ public class HttpProxyClient extends HttpClient {
this.proxySettings = httpClientSettings.getProxyClientSettings();
}
public ActionResult<HttpResponse<byte[]>> toResponse() {
public <T> ActionResult<HttpResponse<T>> toHttpResponse(HttpResponse.BodyHandler<T> handler) {
return switch (proxySettings.getType()) {
case HTTP, DIRECT -> handleHttpProxyRequest();
default -> handleSocksProxyRequest();
case HTTP, DIRECT -> handleHttpProxyRequest(handler);
default -> handleSocksProxyRequest(handler);
};
}
public ActionResult<HttpResponse<byte[]>> handleHttpProxyRequest() {
public <T> ActionResult<HttpResponse<T>> handleHttpProxyRequest(HttpResponse.BodyHandler<T> handler) {
var builder = java.net.http.HttpClient.newBuilder()
.followRedirects(java.net.http.HttpClient.Redirect.NORMAL)
.cookieHandler(new CookieManager())
@@ -67,7 +69,7 @@ public class HttpProxyClient extends HttpClient {
var client = builder.build();
var request = prepareRequest();
var response = client.send(request, HttpResponse.BodyHandlers.ofByteArray());
var response = client.send(request, handler);
if (response.statusCode() != 200)
continue;
return ActionResult.success(response);
@@ -77,7 +79,7 @@ public class HttpProxyClient extends HttpClient {
throw new TikTokProxyRequestException(e);
} catch (IOException e) {
if (e.getMessage().contains("503") && proxySettings.isFallback()) // Indicates proxy protocol is not supported
return super.toHttpResponse(HttpResponse.BodyHandlers.ofByteArray());
return super.toHttpResponse(handler);
throw new TikTokProxyRequestException(e);
} catch (Exception e) {
throw new TikTokLiveRequestException(e);
@@ -86,7 +88,7 @@ public class HttpProxyClient extends HttpClient {
throw new TikTokLiveRequestException("No more proxies available!");
}
private ActionResult<HttpResponse<byte[]>> handleSocksProxyRequest() {
private <T> ActionResult<HttpResponse<T>> handleSocksProxyRequest(HttpResponse.BodyHandler<T> handler) {
try {
SSLContext sc = SSLContext.getInstance("SSL");
sc.init(null, new TrustManager[]{ new X509TrustManager() {
@@ -95,7 +97,8 @@ public class HttpProxyClient extends HttpClient {
public X509Certificate[] getAcceptedIssuers() { return null; }
}}, null);
URL url = toUri().toURL();
URI uri = toUri();
URL url = uri.toURL();
if (proxySettings.hasNext()) {
try {
@@ -117,12 +120,22 @@ public class HttpProxyClient extends HttpClient {
var responseInfo = createResponseInfo(socksConnection.getResponseCode(), headers);
var response = createHttpResponse(body, toUri(), responseInfo);
HttpResponse.BodySubscriber<T> subscriber = handler.apply(responseInfo);
subscriber.onSubscribe(new Flow.Subscription() {
@Override public void request(long n) {}
@Override public void cancel() {}
});
subscriber.onNext(List.of(ByteBuffer.wrap(body)));
subscriber.onComplete();
var response = createHttpResponse(subscriber.getBody().toCompletableFuture().join(), uri, responseInfo);
return ActionResult.success(response);
} catch (IOException e) {
if (e.getMessage().contains("503") && proxySettings.isFallback()) // Indicates proxy protocol is not supported
return super.toHttpResponse(HttpResponse.BodyHandlers.ofByteArray());
return super.toHttpResponse(handler);
if (proxySettings.isAutoDiscard())
proxySettings.remove();
throw new TikTokProxyRequestException(e);
@@ -160,7 +173,7 @@ public class HttpProxyClient extends HttpClient {
};
}
private HttpResponse<byte[]> createHttpResponse(byte[] body,
private <T> HttpResponse<T> createHttpResponse(T body,
URI uri,
ResponseInfo info) {
return new HttpResponse<>()
@@ -176,7 +189,7 @@ public class HttpProxyClient extends HttpClient {
}
@Override
public Optional<HttpResponse<byte[]>> previousResponse() {
public Optional<HttpResponse<T>> previousResponse() {
return Optional.empty();
}
@@ -186,7 +199,7 @@ public class HttpProxyClient extends HttpClient {
}
@Override
public byte[] body() {
public T body() {
return body;
}

View File

@@ -29,7 +29,7 @@ import io.github.jwdeveloper.tiktok.data.models.users.User;
import io.github.jwdeveloper.tiktok.data.requests.LiveUserData;
import io.github.jwdeveloper.tiktok.exceptions.TikTokLiveRequestException;
import java.util.List;
import java.util.*;
import java.util.logging.Logger;
public class LiveUserDataMapper
@@ -66,7 +66,6 @@ public class LiveUserDataMapper
roomInfo.setTitle(liveRoom.get("title").getAsString());
roomInfo.setStartTime(liveRoom.get("startTime").getAsLong());
roomInfo.setTitle(liveRoom.get("title").getAsString());
roomInfo.setViewersCount(liveRoom.getAsJsonObject("liveRoomStats").get("userCount").getAsInt());
roomInfo.setTotalViewersCount(liveRoom.getAsJsonObject("liveRoomStats").get("enterCount").getAsInt());
roomInfo.setAgeRestricted(jsonObject.get("statusCode").getAsInt() == TikTokLiveHttpClient.TIKTOK_AGE_RESTRICTED_CODE);
@@ -92,9 +91,51 @@ public class LiveUserDataMapper
roomInfo.setHostName(foundUser.getName());
return new LiveUserData.Response(json, statusEnum, roomInfo);
} catch (JsonSyntaxException | IllegalStateException e) {
} catch (JsonSyntaxException | IllegalStateException | NullPointerException e) {
logger.warning("Malformed Json: '"+json+"' - Error Message: "+e.getMessage());
return new LiveUserData.Response(json, LiveUserData.UserStatus.NotFound, null);
}
}
public static LiveUserData.Response mapEulerstream(JsonObject jsonObject, Logger logger) {
try {
JsonObject roomInfoJson = jsonObject.getAsJsonObject("roomInfo");
JsonObject userJson = jsonObject.getAsJsonObject("user");
var roomId = roomInfoJson.get("id").getAsString();
var status = roomInfoJson.get("status").getAsInt();
TikTokRoomInfo roomInfo = new TikTokRoomInfo();
roomInfo.setRoomId(roomId);
roomInfo.setTitle(roomInfoJson.get("title").getAsString());
roomInfo.setStartTime(roomInfoJson.get("startTime").getAsLong());
roomInfo.setViewersCount(Optional.ofNullable(roomInfoJson.get("currentViewers")).filter(JsonElement::isJsonPrimitive).map(JsonElement::getAsInt).orElse(0));
roomInfo.setTotalViewersCount(roomInfoJson.get("totalViewers").getAsInt());
var statusEnum = switch (status) {
case 2 -> LiveUserData.UserStatus.Live;
case 3 -> LiveUserData.UserStatus.LivePaused;
case 4 -> LiveUserData.UserStatus.Offline;
default -> LiveUserData.UserStatus.NotFound;
};
User foundUser = new User(
Long.parseLong(userJson.get("numericUid").getAsString()),
userJson.get("uniqueId").getAsString(),
userJson.get("nickname").getAsString(),
userJson.get("signature").getAsString(),
new Picture(userJson.get("avatarUrl").getAsString()),
userJson.get("following").getAsLong(),
userJson.get("followers").getAsLong(),
List.of());
roomInfo.setHost(foundUser);
roomInfo.setHostName(foundUser.getName());
return new LiveUserData.Response(jsonObject.toString(), statusEnum, roomInfo);
} catch (JsonSyntaxException | IllegalStateException | NullPointerException e) {
logger.warning("Malformed Json: '"+jsonObject.toString()+"' - Error Message: "+e.getMessage());
return new LiveUserData.Response(jsonObject.toString(), LiveUserData.UserStatus.NotFound, null);
}
}
}

View File

@@ -142,7 +142,7 @@ public class TikTokWebSocketClient implements LiveSocketClient {
case DISCONNECT -> webSocketClient.closeConnection(CloseFrame.NORMAL, "");
default -> webSocketClient.close();
}
heartbeatTask.stop(webSocketClient);
heartbeatTask.stop();
}
webSocketClient = null;
}

View File

@@ -22,6 +22,7 @@
*/
package io.github.jwdeveloper.tiktok.websocket;
import io.github.jwdeveloper.tiktok.common.AsyncHandler;
import org.java_websocket.WebSocket;
import java.util.*;
@@ -29,47 +30,30 @@ import java.util.concurrent.*;
public class WebSocketHeartbeatTask
{
// Single shared pool for all heartbeat tasks
private static final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1, r -> {
Thread t = new Thread(r, "heartbeat-pool");
t.setDaemon(true);
return t;
});
private static final Map<WebSocket, ScheduledFuture<?>> tasks = new ConcurrentHashMap<>();
private static final Map<WebSocket, Long> commTime = new ConcurrentHashMap<>();
private ScheduledFuture<?> task;
private Long commTime;
private final byte[] heartbeatBytes = Base64.getDecoder().decode("MgJwYjoCaGI="); // Used to be '3A026862' aka ':\x02hb', now is '2\x02pb:\x02hb'.
private final static byte[] heartbeatBytes = Base64.getDecoder().decode("MgJwYjoCaGI="); // Used to be '3A026862' aka ':\x02hb', now is '2\x02pb:\x02hb'.
public void run(WebSocket webSocket, long pingTaskTime) {
stop(webSocket); // remove existing task if any
stop(); // remove existing task if any
tasks.put(webSocket, scheduler.scheduleAtFixedRate(() -> {
task = AsyncHandler.getHeartBeatScheduler().scheduleAtFixedRate(() -> {
try {
if (webSocket.isOpen()) {
webSocket.send(heartbeatBytes);
commTime.put(webSocket, System.currentTimeMillis());
} else {
Long time = commTime.get(webSocket);
if (time != null && System.currentTimeMillis() - time >= 60_000) // Stop if disconnected longer than 60s
stop(webSocket);
}
commTime = System.currentTimeMillis();
} else if (commTime != null && System.currentTimeMillis() - commTime >= 60_000) // Stop if disconnected longer than 60s
stop();
} catch (Exception e) {
e.printStackTrace();
stop(webSocket);
stop();
}
}, 0, pingTaskTime, TimeUnit.MILLISECONDS));
}, 0, pingTaskTime, TimeUnit.MILLISECONDS);
}
public void stop(WebSocket webSocket) {
ScheduledFuture<?> future = tasks.remove(webSocket);
if (future != null)
future.cancel(true);
commTime.remove(webSocket);
}
public void shutdown() {
tasks.values().forEach(f -> f.cancel(true));
commTime.clear();
scheduler.shutdownNow();
public void stop() {
if (task != null)
task.cancel(true);
}
}

View File

@@ -56,7 +56,7 @@ public class TikTokWebSocketEulerListener extends TikTokWebSocketListener
switch (oMsg.get("type").getAsString()) { // Should only receive these 2 types ever
case "workerInfo" -> liveClient.getLogger().info(oMsg.toString()); // Always 1st message
case "roomInfo" -> { // Always 2nd message
LiveUserData.Response data = LiveUserDataMapper.map(oMsg.getAsJsonObject("data").getAsJsonObject("data").getAsJsonObject("raw").toString(), liveClient.getLogger());
LiveUserData.Response data = LiveUserDataMapper.mapEulerstream(oMsg.getAsJsonObject("data"), liveClient.getLogger());
liveClient.getRoomInfo().copy(data.getRoomInfo());
eventHandler.publish(liveClient, new TikTokRoomInfoEvent(liveClient.getRoomInfo()));
}
@@ -67,6 +67,7 @@ public class TikTokWebSocketEulerListener extends TikTokWebSocketListener
} else
throw new IllegalArgumentException("Invalid JsonObject: "+element);
} catch (Exception e) {
e.printStackTrace();
eventHandler.publish(liveClient, new TikTokErrorEvent(e));
}
if (isOpen()) {

View File

@@ -41,7 +41,7 @@
<parent>
<artifactId>TikTokLiveJava</artifactId>
<groupId>io.github.jwdeveloper.tiktok</groupId>
<version>1.11.2-Release</version>
<version>1.11.8-Release</version>
</parent>
<modelVersion>4.0.0</modelVersion>

View File

@@ -6,7 +6,7 @@
<parent>
<groupId>io.github.jwdeveloper.tiktok</groupId>
<artifactId>TikTokLiveJava</artifactId>
<version>1.11.2-Release</version>
<version>1.11.8-Release</version>
</parent>

View File

@@ -5,7 +5,7 @@
<parent>
<artifactId>TikTokLiveJava</artifactId>
<groupId>io.github.jwdeveloper.tiktok</groupId>
<version>1.11.2-Release</version>
<version>1.11.8-Release</version>
</parent>
<modelVersion>4.0.0</modelVersion>
<artifactId>extension-recorder</artifactId>

View File

@@ -7,7 +7,7 @@
<groupId>io.github.jwdeveloper.tiktok</groupId>
<artifactId>TikTokLiveJava</artifactId>
<packaging>pom</packaging>
<version>1.11.2-Release</version>
<version>1.11.8-Release</version>
<modules>
<module>API</module>
<module>Client</module>

View File

@@ -5,7 +5,7 @@
<parent>
<artifactId>TikTokLiveJava</artifactId>
<groupId>io.github.jwdeveloper.tiktok</groupId>
<version>1.11.2-Release</version>
<version>1.11.8-Release</version>
</parent>
<modelVersion>4.0.0</modelVersion>