Support customized plugin when init store

Fix: #359 Change-Id: Iae7cb4694b2db22a187f74dc9b5408785e15806c
apache · Feb 18, 2019 · 8ee8e50 · 8ee8e50
1 parent 0239d49
commit 8ee8e50
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/hugegraph-dist/src/assembly/static/bin/init-store.sh b/hugegraph-dist/src/assembly/static/bin/init-store.sh
@@ -12,8 +12,13 @@ abs_path() {
 
 BIN=`abs_path`
 TOP="$(cd $BIN/../ && pwd)"
-CONF=$TOP/conf
-LIB=$TOP/lib
+CONF="$TOP/conf"
+LIB="$TOP/lib"
+PLUGINS="$TOP/plugins"
+
+. ${BIN}/util.sh
+
+ensure_path_writable $PLUGINS
 
 if [ -n "$JAVA_HOME" ]; then
     JAVA="$JAVA_HOME"/bin/java
@@ -25,5 +30,5 @@ cd $TOP
 
 echo "Initing HugeGraph Store..."
 
-exec $JAVA -cp $LIB/hugegraph-dist-*.jar -Djava.ext.dirs=$LIB/ \
+exec $JAVA -cp $LIB/hugegraph-dist-*.jar -Djava.ext.dirs=$LIB:$PLUGINS \
 com.baidu.hugegraph.cmd.InitStore $CONF/gremlin-server.yaml | grep "com.baidu.hugegraph"
diff --git a/hugegraph-dist/src/main/java/com/baidu/hugegraph/cmd/InitStore.java b/hugegraph-dist/src/main/java/com/baidu/hugegraph/cmd/InitStore.java
@@ -66,6 +66,7 @@ public static void main(String[] args)
 
         String confFile = args[0];
         RegisterUtil.registerBackends();
+        RegisterUtil.registerPlugins();
 
         YamlConfiguration config = new YamlConfiguration();
         config.load(confFile);