Add method checkImage and fix in ImageServer and implement in InitFrame

Author: devparada

src/main/java/com/devparada/frame/InitFrame.java

@@ -20,6 +20,7 @@
 import com.devparada.logic.StatusMCServer;
 import com.devparada.persistency.DBManager;
 import java.awt.GridBagConstraints;
+import java.awt.image.BufferedImage;
 import javax.swing.ImageIcon;
 import javax.swing.JLabel;
 import javax.swing.JPanel;
@@ -381,10 +382,13 @@ private void addPanel() {
         jTxtIMG.setPreferredSize(new java.awt.Dimension(64, 64));
 
         ImageServer image = new ImageServer();
-        // Create ImageIcon with base64 Image without "data:image/png;base64"
         System.out.println(image);
-        ImageIcon imageIcon = new ImageIcon(image.showImage(statusServer.showDataSection(JTxtText, "icon")));
-        jTxtIMG.setIcon(imageIcon);
+        BufferedImage imageServer = image.showImage(statusServer.showDataSection(JTxtText, "icon"));
+        if (image.checkImage(imageServer)) {
+            // Create ImageIcon with base64 Image without "data:image/png;base64"
+            ImageIcon imageIcon = new ImageIcon(imageServer);
+            jTxtIMG.setIcon(imageIcon);
+        }
 
         gridBagConstraints = new GridBagConstraints();
         gridBagConstraints.gridx = 0;

src/main/java/com/devparada/logic/ImageServer.java

@@ -34,12 +34,23 @@ public BufferedImage showImage(String base64Image) {
 
         BufferedImage img = null;
         try {
-            ByteArrayInputStream bis = new ByteArrayInputStream(imageBytes);
-            img = ImageIO.read(bis);
+            try (ByteArrayInputStream bis = new ByteArrayInputStream(imageBytes)) {
+                img = ImageIO.read(bis);
+            }
         } catch (IOException e) {
             JOptionPane.showMessageDialog(null, e, "An error occurred", JOptionPane.WARNING_MESSAGE);
         }
 
         return img;
     }
+
+    /**
+     * Check if the server have a image
+     *
+     * @param imageServer image of the server
+     * @return Return true if image is not null
+     */
+    public boolean checkImage(BufferedImage imageServer) {
+        return imageServer != null;
+    }
 }