[Improvenet: add ZSha2Helper]

Author: mingjian2049

utils/src/main/java/org/zstack/utils/zsha2/ZSha2Helper.java

@@ -0,0 +1,39 @@
+package org.zstack.utils.zsha2;
+
+import org.zstack.utils.ShellResult;
+import org.zstack.utils.ShellUtils;
+import org.zstack.utils.Utils;
+import org.zstack.utils.gson.JSONObjectUtil;
+import org.zstack.utils.logging.CLogger;
+import org.zstack.utils.path.PathUtil;
+
+
+/**
+ * Created by mingjian.deng on 2020/4/2.
+ */
+public class ZSha2Helper {
+    private static final CLogger logger = Utils.getLogger(ZSha2Helper.class);
+
+    public static boolean isMNHaEnvironment() {
+        return PathUtil.exists("/usr/local/bin/zsha2");
+    }
+
+    public static ZSha2Info getInfo() {
+        ShellResult result = ShellUtils.runAndReturn("sudo -i /usr/local/bin/zsha2 status", false);
+        if (!result.isReturnCode(0)) {
+            throw new RuntimeException(String.format("cannot get zsha2 status, because %s", result.getStderr()));
+        }
+
+        result = ShellUtils.runAndReturn("/usr/local/bin/zsha2 show-config");
+        if (!result.isReturnCode(0)) {
+            throw new RuntimeException(String.format("cannot get zsha2 config, because %s, maybe you need upgrade zsha2", result.getStderr()));
+        }
+
+        ZSha2Info info = JSONObjectUtil.toObject(result.getStdout(), ZSha2Info.class);
+
+        info.setMaster(ShellUtils.runAndReturn(String.format(
+                "ip addr show %s | grep -q '[^0-9]%s[^0-9]'", info.getNic(), info.getDbvip())).isReturnCode(0));
+        return info;
+    }
+
+}

utils/src/main/java/org/zstack/utils/zsha2/ZSha2Info.java

@@ -0,0 +1,52 @@
+package org.zstack.utils.zsha2;
+
+/**
+ * Created by mingjian.deng on 2020/4/2.
+ */
+public class ZSha2Info {
+    private String nodeip;
+    private String peerip;
+    private String dbvip;
+    private String nic;
+    private boolean isMaster;
+
+    public String getNodeip() {
+        return nodeip;
+    }
+
+    public void setNodeip(String nodeip) {
+        this.nodeip = nodeip;
+    }
+
+    public String getPeerip() {
+        return peerip;
+    }
+
+    public void setPeerip(String peerip) {
+        this.peerip = peerip;
+    }
+
+    public String getDbvip() {
+        return dbvip;
+    }
+
+    public void setDbvip(String dbvip) {
+        this.dbvip = dbvip;
+    }
+
+    public String getNic() {
+        return nic;
+    }
+
+    public void setNic(String nic) {
+        this.nic = nic;
+    }
+
+    public boolean isMaster() {
+        return isMaster;
+    }
+
+    public void setMaster(boolean master) {
+        isMaster = master;
+    }
+}