Fix raising exception when passing number only string. fix #70

Author: repeatedly

fluent/handler.py

@@ -88,7 +88,11 @@ def _structuring(self, data, record):
             self._add_dic(data, msg)
         elif isinstance(msg, basestring):
             try:
-                self._add_dic(data, json.loads(str(msg)))
+                json_msg = json.loads(str(msg))
+                if isinstance(json_msg, dict):
+                    self._add_dic(data, json_msg)
+                else:
+                    self._add_dic(data, {'message': str(json_msg)})
             except ValueError:
                 msg = record.getMessage()
                 self._add_dic(data, {'message': msg})

tests/test_handler.py

@@ -146,6 +146,19 @@ def test_unstructured_formatted_message(self):
         self.assertTrue('message' in data[0][2])
         self.assertEqual('hello world, you!', data[0][2]['message'])
 
+    def test_number_string_simple_message(self):
+        handler = fluent.handler.FluentHandler('app.follow', port=self._port)
+
+        logging.basicConfig(level=logging.INFO)
+        log = logging.getLogger('fluent.test')
+        handler.setFormatter(fluent.handler.FluentRecordFormatter())
+        log.addHandler(handler)
+        log.info("1")
+        handler.close()
+
+        data = self.get_data()
+        self.assertTrue('message' in data[0][2])
+
     def test_non_string_simple_message(self):
         handler = fluent.handler.FluentHandler('app.follow', port=self._port)