some db code rewriting

[mir.git] / source / mircoders / module / ModuleMediaType.java

diff --git a/source/mircoders/module/ModuleMediaType.java b/source/mircoders/module/ModuleMediaType.java
index dc82fef..95e86c4 100755 (executable)
--- a/source/mircoders/module/ModuleMediaType.java
+++ b/source/mircoders/module/ModuleMediaType.java
@@ -38,6 +38,7 @@ import mir.log.LoggerWrapper;
 import mir.module.AbstractModule;
 import mir.util.JDBCStringRoutines;
 import mir.util.StringRoutines;
+import mir.storage.DatabaseExc;
 import mircoders.storage.DatabaseMediaType;
 
 public class ModuleMediaType extends AbstractModule {
@@ -57,11 +58,16 @@ public class ModuleMediaType extends AbstractModule {
 
     EntityList mediaTypes;
 
-    mediaTypes = DatabaseMediaType.getInstance().selectByWhereClause("mime_type = '"+JDBCStringRoutines.escapeStringLiteral(aMimeType)+"'");
-    if (mediaTypes.size() == 0) {
-      mediaTypes = DatabaseMediaType.getInstance().selectByWhereClause("mime_type = '"+JDBCStringRoutines.escapeStringLiteral(mimeTypeMajor+"/*")+"'");
+    try {
+      mediaTypes = DatabaseMediaType.getInstance().selectByWhereClause("mime_type = '"+JDBCStringRoutines.escapeStringLiteral(aMimeType)+"'");
+      if (mediaTypes.size() == 0) {
+        mediaTypes = DatabaseMediaType.getInstance().selectByWhereClause("mime_type = '"+JDBCStringRoutines.escapeStringLiteral(mimeTypeMajor+"/*")+"'");
+      }
+      if (mediaTypes.size() == 0) {
+        return null;
+      }
     }
-    if (mediaTypes.size() == 0) {
+    catch (DatabaseExc e) {
       return null;
     }
 
@@ -69,7 +75,14 @@ public class ModuleMediaType extends AbstractModule {
   }
 
   public Entity findMediaTypeForExtension(String anExtension) {
-    EntityList mediaTypes = DatabaseMediaType.getInstance().selectByWhereClause("name = '"+JDBCStringRoutines.escapeStringLiteral(anExtension)+"'");
+    EntityList mediaTypes;
+
+    try {
+      mediaTypes = DatabaseMediaType.getInstance().selectByWhereClause("name = '"+JDBCStringRoutines.escapeStringLiteral(anExtension)+"'");
+    }
+    catch (DatabaseExc e) {
+      return null;
+    }
 
     if (mediaTypes.size() == 0) {
       return null;