[incubator-iotdb] branch watermark-demo updated: save user when useWaterMarkUpdated

[ji...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

jiangtian pushed a commit to branch watermark-demo
in repository https://gitbox.apache.org/repos/asf/incubator-iotdb.git


The following commit(s) were added to refs/heads/watermark-demo by this push:
     new a509890  save user when useWaterMarkUpdated
a509890 is described below

commit a5098904c4f3abf1f881394b3ed8bc292252f980
Author: 江天 <jt...@163.com>
AuthorDate: Thu May 30 15:33:39 2019 +0800

    save user when useWaterMarkUpdated
---
 .../java/org/apache/iotdb/db/auth/user/BasicUserManager.java   | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/iotdb/src/main/java/org/apache/iotdb/db/auth/user/BasicUserManager.java b/iotdb/src/main/java/org/apache/iotdb/db/auth/user/BasicUserManager.java
index a97ffed..5573d18 100644
--- a/iotdb/src/main/java/org/apache/iotdb/db/auth/user/BasicUserManager.java
+++ b/iotdb/src/main/java/org/apache/iotdb/db/auth/user/BasicUserManager.java
@@ -297,6 +297,16 @@ public abstract class BasicUserManager implements IUserManager {
     if (user == null) {
       throw new AuthException(String.format("No such user %s", username));
     }
+    boolean oldFlag = user.isUseWaterMark();
+    if (oldFlag == useWaterMark) {
+      return;
+    }
     user.setUseWaterMark(useWaterMark);
+    try {
+      accessor.saveUser(user);
+    } catch (IOException e) {
+      user.setUseWaterMark(oldFlag);
+      throw new AuthException(e);
+    }
   }
 }