Skip to content

Commit 7c95bbd

Browse files
committed
refactoring
1 parent 40d229f commit 7c95bbd

File tree

2 files changed

+24
-19
lines changed

2 files changed

+24
-19
lines changed

src/main/java/com/github/jlangch/venice/util/ipc/TcpClient.java

Lines changed: 20 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -200,23 +200,11 @@ public void open() {
200200

201201
// handle config
202202
final VncMap config = (VncMap)((Message)response).getVeniceData();
203-
maxMessageSize.set(
204-
Coerce.toVncLong(
205-
config.get(
206-
new VncKeyword("max-msg-size"),
207-
new VncLong(Message.MESSAGE_LIMIT_MAX))).toJavaLong());
208-
compressor.set(
209-
new Compressor(
210-
Coerce.toVncLong(
211-
config.get(
212-
new VncKeyword("compress-cutoff-size"),
213-
new VncLong(-1))).toJavaLong()));
203+
maxMessageSize.set(getLong(config, "max-msg-size", Message.MESSAGE_LIMIT_MAX));
204+
compressor.set(new Compressor(getLong(config, "compress-cutoff-size", -1)));
214205
encrypt.set(
215-
encrypt.get() // client side encrypt request
216-
|| VncBoolean.isTrue( // server side encrypt request
217-
config.get(
218-
new VncKeyword(":encrypt"),
219-
VncBoolean.False)));
206+
encrypt.get() // client side encrypt request
207+
|| getBoolean(config, "encrypt", false)); // server side encrypt request
220208
}
221209
catch(Exception ex) {
222210
IO.safeClose(ch);
@@ -1196,6 +1184,22 @@ private static byte[] toBytes(final String s, final String charset) {
11961184
return s.getBytes(Charset.forName(charset));
11971185
}
11981186

1187+
private static long getLong(final VncMap map, final String entryName, final long defaulValue) {
1188+
return Coerce.toVncLong(
1189+
map.get(
1190+
new VncKeyword(entryName),
1191+
new VncLong(defaulValue))).toJavaLong();
1192+
}
1193+
1194+
private static boolean getBoolean(final VncMap map, final String entryName, final boolean defaulValue) {
1195+
return VncBoolean.isTrue(
1196+
map.get(
1197+
new VncKeyword(entryName),
1198+
VncBoolean.of(defaulValue)));
1199+
}
1200+
1201+
1202+
11991203

12001204
private final Semaphore sendSemaphore = new Semaphore(1);
12011205

src/test/java/com/github/jlangch/venice/util/ipc/TcpServerTest.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434

3535
import com.github.jlangch.venice.EofException;
3636
import com.github.jlangch.venice.VncException;
37+
import com.github.jlangch.venice.util.ipc.impl.Message;
3738

3839

3940
public class TcpServerTest {
@@ -234,19 +235,19 @@ public void test_server_threadpool_stats() throws Exception {
234235
@Test
235236
public void test_server_max_size() throws Exception {
236237
try (TcpServer server = new TcpServer(33333)) {
237-
assertEquals(TcpServer.MESSAGE_LIMIT_MAX, server.getMaxMessageSize());
238+
assertEquals(Message.MESSAGE_LIMIT_MAX, server.getMaxMessageSize());
238239

239240
// below minimum
240241
server.setMaxMessageSize(100L);
241-
assertEquals(TcpServer.MESSAGE_LIMIT_MIN, server.getMaxMessageSize());
242+
assertEquals(Message.MESSAGE_LIMIT_MIN, server.getMaxMessageSize());
242243

243244
// in range
244245
server.setMaxMessageSize(100L * 1024L);
245246
assertEquals(100L * 1024L, server.getMaxMessageSize());
246247

247248
// above maximum
248249
server.setMaxMessageSize(800L * 1024L * 1024L);
249-
assertEquals(TcpServer.MESSAGE_LIMIT_MAX, server.getMaxMessageSize());
250+
assertEquals(Message.MESSAGE_LIMIT_MAX, server.getMaxMessageSize());
250251
}
251252
}
252253

0 commit comments

Comments
 (0)