minor fixes

wizzardo · Oct 9, 2023 · 2bbafa0 · 2bbafa0
1 parent 7c45ad6
commit 2bbafa0
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 10 deletions.
diff --git a/src/main/java/com/wizzardo/http/AbstractHttpServer.java b/src/main/java/com/wizzardo/http/AbstractHttpServer.java
@@ -61,11 +61,6 @@ protected IOThread<? extends T> createIOThread(int number, int divider) {
             };
         } else {
             server = new FallbackServerSocket<T>(host, port, this) {
-                @Override
-                public void onRead(T connection, ByteBufferProvider bufferProvider) throws IOException {
-                    process(connection, bufferProvider);
-                }
-
                 @Override
                 protected SelectorConnectionWrapper createConnection(SocketChannel client) throws IOException {
                     SelectorConnectionWrapper connection = super.createConnection(client);

diff --git a/src/main/java/com/wizzardo/http/FallbackServerSocket.java b/src/main/java/com/wizzardo/http/FallbackServerSocket.java
@@ -308,6 +308,10 @@ public void close() {
                     IOTools.close(data);
 
             IOTools.close(channel);
+            try {
+                onDisconnect(ByteBufferProvider.current());
+            } catch (IOException ignored) {
+            }
         }
 
         @Override
@@ -420,6 +424,7 @@ protected SelectorConnectionWrapper createConnection(SocketChannel client) throw
     }
 
     public void onRead(T connection, ByteBufferProvider bufferProvider) throws IOException {
+        connection.onRead(bufferProvider);
     }
 
     @Override

diff --git a/src/main/java/com/wizzardo/http/HttpWorker.java b/src/main/java/com/wizzardo/http/HttpWorker.java
@@ -20,8 +20,10 @@ protected void process(T connection) throws IOException {
         if (!connection.processingBy.compareAndSet(null, this))
             return;
 
-        connection.process(this);
-
-        connection.processingBy.set(null);
+        try {
+            connection.process(this);
+        } finally {
+            connection.processingBy.set(null);
+        }
     }
 }
diff --git a/src/main/java/com/wizzardo/http/framework/parameters/ParametersHelper.java b/src/main/java/com/wizzardo/http/framework/parameters/ParametersHelper.java
@@ -382,8 +382,11 @@ protected static Mapper<Request, Object> createPojoMapper(Class type, String par
 
             if (request.isMultipart()) {
                 MultiPartEntry entry = request.entry(parameterName);
-                if (entry != null && entry.contentType().toLowerCase().startsWith(Header.VALUE_APPLICATION_JSON.value)) {
-                    return JsonTools.parse(entry.asBytes(), type);
+                if (entry != null) {
+                    String entryContentType = entry.contentType();
+                    if (entryContentType != null && entryContentType.toLowerCase().startsWith(Header.VALUE_APPLICATION_JSON.value)) {
+                        return JsonTools.parse(entry.asBytes(), type);
+                    }
                 }
             }