fix set_mode

add_mode

visualdl/logic/sdk.h

@@ -10,7 +10,7 @@ const static std::string kDefaultMode{"default"};
 
 class LogWriter {
 public:
-  LogWriter(const std::string& dir, int sync_cycle)  {
+  LogWriter(const std::string& dir, int sync_cycle) {
     storage_.SetDir(dir);
     storage_.meta.cycle = sync_cycle;
   }
@@ -20,7 +20,10 @@ public:
     storage_ = other.storage_;
   }
 
-  void SetMode(const std::string& mode) { mode_ = mode; }
+  void SetMode(const std::string& mode) {
+    mode_ = mode;
+    storage_.AddMode(mode);
+  }
 
   LogWriter AsMode(const std::string& mode) {
     LogWriter writer = *this;

visualdl/logic/sdk_test.cc

@@ -31,6 +31,8 @@ TEST(Scalar, write) {
   // check the first entry of first record
   ASSERT_EQ(record.front(), 12);
 
+  ASSERT_TRUE(!reader.storage().modes().empty());
+
   // check tags
   ASSERT_EQ(reader.all_tags().size(), 1);
   auto tags = reader.tags("scalar");