Merge branch 'opengoofy:develop' into develop

pull/1356/head
时间邮递员 2 years ago committed by GitHub
commit a5dabf7748
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -0,0 +1,107 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package cn.hippo4j.common.executor.support;
import cn.hippo4j.common.toolkit.ThreadUtil;
import lombok.extern.slf4j.Slf4j;
import org.junit.Assert;
import org.junit.Test;
import java.util.Arrays;
import java.util.concurrent.RejectedExecutionHandler;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.IntStream;
@Slf4j
public class ResizableCapacityLinkedBlockingQueueTest {
@Test
public void testResizableCapacityLinkedBlockingQueueSize() throws InterruptedException {
ResizableCapacityLinkedBlockingQueue<Integer> queue1 = new ResizableCapacityLinkedBlockingQueue(10);
queue1.setCapacity(20);
Assert.assertEquals(20, queue1.remainingCapacity());
queue1.add(1);
Assert.assertEquals(19, queue1.remainingCapacity());
ResizableCapacityLinkedBlockingQueue<Integer> queue2 = new ResizableCapacityLinkedBlockingQueue(Arrays.asList(1, 2, 3, 4));
queue2.setCapacity(5);
Assert.assertEquals(1, queue2.remainingCapacity());
}
@Test
public void testIncreaseResizableCapacityLinkedBlockingQueue() throws InterruptedException {
MyRejectedExecutionHandler myRejectedExecutionHandler = new MyRejectedExecutionHandler();
ResizableCapacityLinkedBlockingQueue<Runnable> queue = new ResizableCapacityLinkedBlockingQueue();
ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(1, 1,
60, TimeUnit.SECONDS, queue, myRejectedExecutionHandler);
Assert.assertSame(queue, threadPoolExecutor.getQueue());
threadPoolExecutor.prestartAllCoreThreads();
queue.setCapacity(6);
IntStream.range(0, 4).forEach(s -> {
threadPoolExecutor.execute(() -> ThreadUtil.sleep(0L));
});
threadPoolExecutor.shutdown();
while (!threadPoolExecutor.isTerminated()) {
}
Assert.assertEquals(4, threadPoolExecutor.getCompletedTaskCount());
Assert.assertEquals(0, myRejectedExecutionHandler.getCount());
}
@Test
public void testDecreaseResizableCapacityLinkedBlockingQueue() throws InterruptedException {
MyRejectedExecutionHandler myRejectedExecutionHandler = new MyRejectedExecutionHandler();
ResizableCapacityLinkedBlockingQueue<Runnable> queue = new ResizableCapacityLinkedBlockingQueue<>(4);
ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(1, 1,
60, TimeUnit.SECONDS, queue, myRejectedExecutionHandler);
Assert.assertSame(queue, threadPoolExecutor.getQueue());
threadPoolExecutor.prestartAllCoreThreads();
queue.setCapacity(0);
IntStream.range(0, 4).forEach(s -> {
threadPoolExecutor.execute(() -> ThreadUtil.sleep(0L));
});
threadPoolExecutor.shutdown();
while (!threadPoolExecutor.isTerminated()) {
}
Assert.assertEquals(0, threadPoolExecutor.getCompletedTaskCount());
Assert.assertEquals(4, myRejectedExecutionHandler.getCount());
}
}
class MyRejectedExecutionHandler implements RejectedExecutionHandler {
public AtomicInteger count = new AtomicInteger(0);
@Override
public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
if (executor.isShutdown()) {
return;
}
if (!executor.getQueue().offer(r)) {
count.incrementAndGet();
}
}
public int getCount() {
return count.get();
}
}

@ -0,0 +1,260 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package cn.hippo4j.common.toolkit;
import org.apache.commons.io.input.BrokenInputStream;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.jupiter.api.Assertions;
import java.io.BufferedReader;
import java.io.PrintWriter;
import java.io.Reader;
import java.io.Writer;
import java.io.File;
import java.io.StringReader;
import java.io.StringWriter;
import java.io.Closeable;
import java.io.FileReader;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.ByteArrayOutputStream;
import java.io.ByteArrayInputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.List;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;
/**
* test for {@link IoUtil}
*/
public class IoUtilTest {
private Path tempDir;
private File sourceFile;
@Before
public void setUp() throws IOException {
tempDir = Files.createTempDirectory("test");
sourceFile = new File(tempDir.toFile(), "source.txt");
try (FileOutputStream fos = new FileOutputStream(sourceFile)) {
fos.write("Hello, World!".getBytes());
}
}
@After
public void tearDown() throws IOException {
Files.walk(tempDir)
.sorted((a, b) -> -a.compareTo(b))
.forEach(path -> {
try {
Files.deleteIfExists(path);
} catch (IOException e) {
e.printStackTrace();
}
});
}
@Test
public void testTryDecompressInputStream() throws IOException {
byte[] inputBytes = "This is a test string.".getBytes("UTF-8");
ByteArrayOutputStream compressedOutput = new ByteArrayOutputStream();
try (GZIPOutputStream gzipOutputStream = new GZIPOutputStream(compressedOutput)) {
gzipOutputStream.write(inputBytes);
}
byte[] compressedBytes = compressedOutput.toByteArray();
ByteArrayInputStream compressedInput = new ByteArrayInputStream(compressedBytes);
byte[] decompressedBytes = IoUtil.tryDecompress(compressedInput);
Assert.assertNotNull(decompressedBytes);
Assert.assertTrue(decompressedBytes.length > 0);
Assert.assertArrayEquals(inputBytes, decompressedBytes);
}
@Test
public void testTryDecompressByteArray() throws Exception {
byte[] inputBytes = "This is a test string.".getBytes("UTF-8");
ByteArrayOutputStream compressedOutput = new ByteArrayOutputStream();
try (GZIPOutputStream gzipOutputStream = new GZIPOutputStream(compressedOutput)) {
gzipOutputStream.write(inputBytes);
}
byte[] compressedBytes = compressedOutput.toByteArray();
byte[] decompressedBytes = IoUtil.tryDecompress(compressedBytes);
Assert.assertNotNull(decompressedBytes);
Assert.assertTrue(decompressedBytes.length > 0);
Assert.assertArrayEquals(inputBytes, decompressedBytes);
}
@Test
public void testTryCompress() throws IOException {
String inputString = "This is a test string.";
String encoding = "UTF-8";
byte[] compressedBytes = IoUtil.tryCompress(inputString, encoding);
Assert.assertNotNull(compressedBytes);
Assert.assertTrue(compressedBytes.length > 0);
try (
GZIPInputStream gzipInputStream = new GZIPInputStream(new ByteArrayInputStream(compressedBytes))) {
byte[] decompressedBytes = new byte[1024];
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
int readBytes;
while ((readBytes = gzipInputStream.read(decompressedBytes)) > 0) {
outputStream.write(decompressedBytes, 0, readBytes);
}
String decompressedString = outputStream.toString(encoding);
Assert.assertEquals(inputString, decompressedString);
}
}
@Test
public void testWriteStringToFile() throws IOException {
File tempFile = new File(tempDir.toFile(), "testWriteStringToFile.txt");
String testString = "test string";
IoUtil.writeStringToFile(tempFile, testString, "UTF-8");
BufferedReader reader = new BufferedReader(new FileReader(tempFile));
String fileContent = reader.readLine();
reader.close();
Assert.assertEquals(testString, fileContent);
}
@Test
public void testReadLines() throws IOException {
File tempFile = new File(tempDir.toFile(), "testReadLines.txt");
try (
PrintWriter writer = new PrintWriter(tempFile)) {
writer.println("test string 1");
writer.println("test string 2");
writer.println("test string 3");
}
FileReader fileReader = new FileReader(tempFile);
List<String> lines = IoUtil.readLines(fileReader);
fileReader.close();
Assert.assertEquals(3, lines.size());
Assert.assertEquals("test string 1", lines.get(0));
Assert.assertEquals("test string 2", lines.get(1));
Assert.assertEquals("test string 3", lines.get(2));
}
@Test
public void testToStringInputStream() {
String testString = "test string";
InputStream inputStream = new ByteArrayInputStream(testString.getBytes());
String result = IoUtil.toString(inputStream, "UTF-8");
Assert.assertEquals(testString, result);
}
@Test
public void testToStringReader() throws IOException {
String testString = "test string";
Reader reader = new StringReader(testString);
String result = IoUtil.toString(reader);
Assert.assertEquals(testString, result);
}
@Test
public void testCopyReaderWriter() throws IOException {
String inputString = "testCopyReaderWriter";
Reader reader = new StringReader(inputString);
Writer writer = new StringWriter();
IoUtil.copy(reader, writer);
Assert.assertEquals(inputString, writer.toString());
}
@Test
public void testCopyInputStreamOutputStream() throws IOException {
String inputString = "testCopyInputStreamOutputStream";
InputStream inputStream = new ByteArrayInputStream(inputString.getBytes());
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
IoUtil.copy(inputStream, outputStream);
Assert.assertEquals(inputString, outputStream.toString());
}
@Test
public void testDelete() throws IOException {
File deleteFile = new File(tempDir.toFile(), "delete.txt");
deleteFile.createNewFile();
Assert.assertTrue(deleteFile.exists());
IoUtil.delete(deleteFile);
Assert.assertFalse(deleteFile.exists());
File deleteDir = new File(tempDir.toFile(), "delete");
deleteDir.mkdirs();
Assert.assertTrue(deleteDir.exists());
File deleteDirFile = new File(deleteDir, "delete.txt");
deleteDirFile.createNewFile();
Assert.assertTrue(deleteDirFile.exists());
IoUtil.delete(deleteDir);
Assert.assertTrue(deleteDir.exists());
Assert.assertFalse(deleteDirFile.exists());
}
@Test
public void testCleanDirectory() throws IOException {
File cleanDir = new File(tempDir.toFile(), "clean");
cleanDir.mkdirs();
File cleanFile1 = new File(cleanDir, "clean1.txt");
File cleanFile2 = new File(cleanDir, "clean2.txt");
Assert.assertTrue(cleanDir.exists());
Assert.assertTrue(cleanFile1.createNewFile());
Assert.assertTrue(cleanFile2.createNewFile());
IoUtil.cleanDirectory(cleanDir);
Assert.assertFalse(cleanFile1.exists());
Assert.assertFalse(cleanFile2.exists());
}
@Test
public void testCopyFile() throws Exception {
String sourcePath = sourceFile.getAbsolutePath();
String targetPath = sourceFile.getParent() + File.separator + "copy" + File.separator + "target.txt";
IoUtil.copyFile(sourcePath, targetPath);
File targetFile = new File(targetPath);
Assert.assertTrue(targetFile.exists());
Assert.assertEquals(sourceFile.length(), targetFile.length());
byte[] sourceBytes = Files.readAllBytes(sourceFile.toPath());
byte[] targetBytes = Files.readAllBytes(targetFile.toPath());
Assert.assertArrayEquals(sourceBytes, targetBytes);
}
@Test
public void testIsGzipStream() {
byte[] gzipBytes = new byte[2];
gzipBytes[0] = (byte) GZIPInputStream.GZIP_MAGIC;
gzipBytes[1] = (byte) (GZIPInputStream.GZIP_MAGIC >> 8);
byte[] invalidGzipBytes = new byte[2];
invalidGzipBytes[0] = (byte) (GZIPInputStream.GZIP_MAGIC + 1);
invalidGzipBytes[1] = (byte) ((GZIPInputStream.GZIP_MAGIC >> 8) + 1);
byte[] invalidGzipBytes2 = new byte[1];
byte[] normalBytes = new byte[2];
Assert.assertTrue(IoUtil.isGzipStream(gzipBytes));
Assert.assertFalse(IoUtil.isGzipStream(invalidGzipBytes));
Assert.assertFalse(IoUtil.isGzipStream(invalidGzipBytes2));
Assert.assertFalse(IoUtil.isGzipStream(null));
Assert.assertFalse(IoUtil.isGzipStream(normalBytes));
}
@Test
public void testCloseQuietly() throws IOException {
Closeable closeable = new BrokenInputStream();
URL url = new URL("https://www.baidu.com");
HttpURLConnection httpURLConnection = (HttpURLConnection) url.openConnection();
Assertions.assertDoesNotThrow(() -> IoUtil.closeQuietly(closeable));
Assertions.assertDoesNotThrow(() -> IoUtil.closeQuietly(closeable, closeable, closeable));
Assertions.assertDoesNotThrow(() -> IoUtil.closeQuietly(httpURLConnection));
}
}

@ -0,0 +1,140 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package cn.hippo4j.common.toolkit;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.exc.MismatchedInputException;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.NoArgsConstructor;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
/**
* test for {@link JacksonHandler}
*/
public class JacksonHandlerTest {
private final static JacksonHandler JACKSON_HANDLER = new JacksonHandler();
private static final Entity EXPECTED_ENTITY =
new Entity(1, "entity1", new Entity(2, "entity2", null));
private static final List<Entity> EXPECTED_ENTITY_ARRAY = Arrays.asList(EXPECTED_ENTITY, EXPECTED_ENTITY);
private static final String EXPECTED_ENTITY_JSON = "{" +
"\"id\":1," +
"\"name\":\"entity1\"," +
"\"entity\":{" +
"\"id\":2," +
"\"name\":\"entity2\"" +
"}}";
private static final String EXPECTED_ENTITY_ARRAY_JSON = "[" +
EXPECTED_ENTITY_JSON + "," +
EXPECTED_ENTITY_JSON +
"]";
@Test
public void testToJSONString() {
// boolean to json
Assertions.assertEquals("true", JACKSON_HANDLER.toJSONString(true));
// double to json
Assertions.assertEquals("0.01", JACKSON_HANDLER.toJSONString(0.01));
// integer to json
Assertions.assertEquals("1", JACKSON_HANDLER.toJSONString(1));
// string to json
Assertions.assertEquals("\"hello world\"", JACKSON_HANDLER.toJSONString("hello world"));
// array to json
Assertions.assertEquals("[0,1,2,3,4]", JACKSON_HANDLER.toJSONString(new int[]{0, 1, 2, 3, 4}));
// object to json
Assertions.assertEquals(EXPECTED_ENTITY_JSON, JACKSON_HANDLER.toJSONString(EXPECTED_ENTITY));
}
@Test
public void testParseObject() {
// normal json to boolean
Assertions.assertEquals(true, JACKSON_HANDLER.parseObject("true", Boolean.class));
// normal json to double
Assertions.assertEquals(0.01, JACKSON_HANDLER.parseObject("0.01", Double.class));
// normal json to integer
Assertions.assertEquals(1, JACKSON_HANDLER.parseObject("1", Integer.class));
// normal json to string
Assertions.assertEquals("hello world",
JACKSON_HANDLER.parseObject("\"hello world\"", String.class));
// normal json to object
Assertions.assertEquals(EXPECTED_ENTITY, JACKSON_HANDLER.parseObject(EXPECTED_ENTITY_JSON, Entity.class));
Assertions.assertEquals(
EXPECTED_ENTITY,
JACKSON_HANDLER.parseObject(EXPECTED_ENTITY_JSON, new TypeReference<Entity>() {
})
);
// illegal json
Assertions.assertThrows(MismatchedInputException.class,
() -> JACKSON_HANDLER.parseObject(" ", Entity.class));
// null json
Assertions.assertThrows(IllegalArgumentException.class,
() -> JACKSON_HANDLER.parseObject(null, Entity.class));
// illegal type
Assertions.assertThrows(MismatchedInputException.class,
() -> JACKSON_HANDLER.parseObject(EXPECTED_ENTITY_JSON, String.class));
}
@Test
public void testParseArray() {
// normal json to array
Assertions.assertEquals(EXPECTED_ENTITY_ARRAY, JSONUtil.parseArray(EXPECTED_ENTITY_ARRAY_JSON, Entity.class));
// null json
Assertions.assertEquals(Collections.emptyList(), JSONUtil.parseArray(null, Entity.class));
// illegal json
Assertions.assertEquals(Collections.emptyList(), JSONUtil.parseArray(" ", Entity.class));
}
@Test
public void testIsJson() {
// normal json
Assertions.assertTrue(JACKSON_HANDLER.isJson(EXPECTED_ENTITY_JSON));
Assertions.assertTrue(JACKSON_HANDLER.isJson(EXPECTED_ENTITY_ARRAY_JSON));
Assertions.assertTrue(JACKSON_HANDLER.isJson(" "));
// illegal json
Assertions.assertFalse(JACKSON_HANDLER.isJson("{" +
"\"id\":1," +
"\"name\":\"entity1\"," +
"\"entity\":{\"id\":2,\"name\":\"entity2\"}"));
// null json
Assertions.assertThrows(IllegalArgumentException.class, () -> JACKSON_HANDLER.isJson(null));
}
@EqualsAndHashCode
@AllArgsConstructor
@NoArgsConstructor
@Data
private static class Entity {
private Integer id;
private String name;
private Entity entity;
}
}
Loading…
Cancel
Save