apache · cgivre · Mar 28, 2022 · Mar 28, 2022 · Apr 26, 2022
diff --git a/contrib/storage-http/src/main/resources/storage_logos/Http.png b/contrib/storage-http/src/main/resources/storage_logos/Http.png
diff --git a/exec/java-exec/src/main/java/org/apache/drill/exec/server/rest/StorageResources.java b/exec/java-exec/src/main/java/org/apache/drill/exec/server/rest/StorageResources.java
@@ -490,6 +490,27 @@ public String getPassword() {
       }
     }
 
+    /**
+     * Returns the path for the image used in the list view.
+     * @return
+     */
+    public String getImagePath() {
+      if (getCleanType().contentEquals("FileSystem")) {
+        return "/static/img/storage_logos/FileSystem.png";
+      }
+
+      // If the plugin is in contrib, get the full path
+      String path = plugin.getConfig().getClass().getResource("/storage_logos/" + getCleanType() + ".png").getPath();
+      return path;
+    }
+
+    public String getCleanType() {
+      String pluginName = type.replace("Config", "");
+      pluginName = pluginName.replace("Plugin", "");
+      pluginName = pluginName.replace("Storage", "");
+      return pluginName;
+    }
+
     public String getType() {
       return type;
     }

diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Cassandra.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Cassandra.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Druid.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Druid.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Elasticsearch.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Elasticsearch.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/FileSystem.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/FileSystem.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/HBase.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/HBase.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Hive.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Hive.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Jdbc.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Jdbc.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Kafka.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Kafka.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Kudu.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Kudu.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Mongo.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Mongo.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/OpenTSDB.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/OpenTSDB.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Phoenix.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Phoenix.png
diff --git a/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Splunk.png b/exec/java-exec/src/main/resources/rest/static/img/storage_logos/Splunk.png
diff --git a/exec/java-exec/src/main/resources/rest/storage/list.ftl b/exec/java-exec/src/main/resources/rest/storage/list.ftl
@@ -59,6 +59,10 @@
             <#list model as pluginModel>
               <#if pluginModel.getPlugin()?? && pluginModel.getPlugin().enabled() == true>
                 <tr>
+                  <td style="border:none;">
+                    <img src="/static/img/storage_logos/${pluginModel.getCleanType()}.png" alt="${pluginModel.getCleanType()}" height=40 width=40 />
+                ${pluginModel.getImagePath()}
+                  </td>
                   <td style="border:none; max-width: 200px; overflow: hidden; text-overflow: ellipsis;">
                     ${pluginModel.getPlugin().getName()}
                   </td>
@@ -88,6 +92,9 @@
             <#list model as pluginModel>
               <#if pluginModel.getPlugin()?? && pluginModel.getPlugin().enabled() == false>
                 <tr>
+                  <td style="border:none;">
+                    <img src="/static/img/storage_logos/${pluginModel.getCleanType()}.png" alt="${pluginModel.getCleanType()}" height=40 width=40 />
+                  </td>
                   <td style="border:none; max-width: 200px; overflow: hidden; text-overflow: ellipsis;">
                     ${pluginModel.getPlugin().getName()}
                   </td>