1 files changed, 12 insertions, 1 deletions
diff --git a/src/main/DataFiller.java b/src/main/DataFiller.java
index e6ac6f4..a400975 100644
--- a/src/main/DataFiller.java
+++ b/src/main/DataFiller.java
@@ -29,13 +29,16 @@ public class DataFiller {
     private final NamedPreparedStatement m_insertUserUrl;
     private final NamedPreparedStatement m_insertMentions;
 
+    private final int m_category;
+    
     /**
      * Create the datafiller object.
      *
      * @param connection The database connection to use.
      * @throws java.sql.SQLException on error preparing the database connection.
      */
-    public DataFiller(Connection connection) throws SQLException {
+    public DataFiller(Connection connection, int category) throws SQLException {
+        m_category = category;
         m_connection = connection;
         m_insertTweet = new NamedPreparedStatement(m_connection, QueryUtils.insertTweet);
         m_insertProfile = new NamedPreparedStatement(m_connection, QueryUtils.insertProfile);
@@ -76,6 +79,13 @@ public class DataFiller {
         if (tweet.text.contains("\0")) {
             tweet.text = tweet.text.replaceAll("\0", "");
         }
+        
+        User user = tweet.user;
+        
+        if (user.name.contains("\0")) {
+            user.name = user.name.replaceAll("\0", "");
+        }
+        
 
         try {
             // assume that no previous transaction was started.
@@ -92,6 +102,7 @@ public class DataFiller {
         String text = sanitizeTweetText(tweet.text);
 
         // ensure that the user and tweet are known before adding relations
+        m_insertTweet.setInt("category", m_category);
         QueryUtils.setInsertParams(m_insertTweet, m_insertProfile, tweet, text);
         m_insertProfile.executeUpdate();
         m_insertTweet.executeUpdate();