lib/td/logger/td_logger.rb in td-logger-0.3.2 vs lib/td/logger/td_logger.rb in td-logger-0.3.3

@@ -45,11 +45,11 @@
     require 'json'
     require 'time'
     require 'net/http'
     require 'cgi'
     require 'logger'
-    require 'td/client'
+    require 'td-client'
 
     @logger = ::Logger.new(STDERR)
     if debug
       @logger.level = ::Logger::DEBUG
     else
@@ -154,9 +154,22 @@
       JSON.load(JSON.dump(msg)).to_msgpack
     end
   end
 
   def add(db, table, msg)
+    begin
+      TreasureData::API.validate_database_name(db)
+    rescue
+      @logger.error("TreasureDataLogger: Invalid database name #{db.inspect}: #{$!}")
+      return false
+    end
+    begin
+      TreasureData::API.validate_table_name(table)
+      @logger.error("TreasureDataLogger: Invalid table name #{table.inspect}: #{$!}")
+    rescue
+      return false
+    end
+
     begin
       data = to_msgpack(msg)
     rescue
       @logger.error("TreasureDataLogger: Can't convert to msgpack: #{msg.inspect}: #{$!}")
       return false