custom query params

Author: shi-yuan

src/main/assembly/plugin.xml

@@ -20,6 +20,11 @@
                 <exclude>**/dic/**</exclude>
             </excludes>
         </fileSet>
+
+        <fileSet>
+            <directory>${project.basedir}/src/main/plugin-metadata</directory>
+            <outputDirectory>/elasticsearch</outputDirectory>
+        </fileSet>
     </fileSets>
     <files>
         <file>

src/main/java/org/ansj/elasticsearch/cat/AnsjCatAction.java

@@ -3,6 +3,7 @@
 import org.ansj.elasticsearch.action.AnsjAction;
 import org.ansj.elasticsearch.action.AnsjRequest;
 import org.ansj.elasticsearch.action.AnsjResponse;
+import org.ansj.library.*;
 import org.elasticsearch.client.node.NodeClient;
 import org.elasticsearch.common.Table;
 import org.elasticsearch.common.inject.Inject;
@@ -13,6 +14,10 @@
 import org.elasticsearch.rest.action.RestResponseListener;
 import org.elasticsearch.rest.action.cat.AbstractCatAction;
 
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.Set;
+
 /**
  * Created by zhangqinghua on 16/2/2.
  */
@@ -43,6 +48,15 @@ public RestResponse buildResponse(final AnsjResponse ansjResponse) throws Except
         });
     }
 
+    @Override
+    protected Set<String> responseParams() {
+        Set<String> responseParams = new HashSet<>(super.responseParams());
+        responseParams.addAll(Arrays.asList("text", "analyzer", "tokenizer", "type", "key",
+                "isNameRecognition", "isNumRecognition", "isQuantifierRecognition", "isRealName", "isSkipUserDefine",
+                CrfLibrary.DEFAULT, DicLibrary.DEFAULT, AmbiguityLibrary.DEFAULT, StopLibrary.DEFAULT, SynonymsLibrary.DEFAULT));
+        return responseParams;
+    }
+
     @Override
     protected void documentation(StringBuilder stringBuilder) {
 

src/main/java/org/ansj/elasticsearch/index/config/AnsjElasticConfigurator.java

@@ -9,6 +9,7 @@
 import org.ansj.splitWord.analysis.ToAnalysis;
 import org.ansj.util.MyStaticValue;
 import org.apache.logging.log4j.Logger;
+import org.elasticsearch.SpecialPermission;
 import org.elasticsearch.common.inject.Inject;
 import org.elasticsearch.common.logging.Loggers;
 import org.elasticsearch.common.settings.Settings;
@@ -22,6 +23,8 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
+import java.security.AccessController;
+import java.security.PrivilegedAction;
 import java.util.Map;
 
 public class AnsjElasticConfigurator {
@@ -90,25 +93,34 @@ private void flushConfig() {
     }
 
     private void initConfig(String path, boolean printErr) {
-        try (BufferedReader br = IOUtil.getReader(PathToStream.stream(path), "utf-8")) {
-            String temp;
-            int index;
-            while ((temp = br.readLine()) != null) {
-                if (StringUtil.isBlank(temp) || temp.trim().charAt(0) == '#' || !temp.contains("=")) {
-                    continue;
-                }
+        final SecurityManager sm = System.getSecurityManager();
+        if (sm != null) {
+            sm.checkPermission(new SpecialPermission());
+        }
+        AccessController.doPrivileged((PrivilegedAction<Void>) () -> {
+            try (BufferedReader br = IOUtil.getReader(PathToStream.stream(path), "utf-8")) {
+                String temp, key;
+                int index;
+                while ((temp = br.readLine()) != null) {
+                    if (StringUtil.isBlank(temp) || temp.trim().charAt(0) == '#' || !temp.contains("=")) {
+                        continue;
+                    }
 
-                index = temp.indexOf('=');
+                    index = temp.indexOf('=');
 
-                MyStaticValue.ENV.put(temp.substring(0, index).trim(), temp.substring(index + 1, temp.length()).trim());
-            }
-        } catch (Exception e) {
-            if (printErr) {
-                LOG.error("{} load err: {}", path, e);
-            } else {
-                LOG.warn("{} load err", path);
+                    MyStaticValue.ENV.put(key = temp.substring(0, index).trim(), temp.substring(index + 1, temp.length()).trim());
+
+                    DicLibrary.get(key);
+                }
+            } catch (Exception e) {
+                if (printErr) {
+                    LOG.error("{} load err: {}", path, e);
+                } else {
+                    LOG.warn("{} load err", path);
+                }
             }
-        }
+            return null;
+        });
     }
 
     private void preheat() {

src/main/plugin-metadata/plugin-security.policy

@@ -0,0 +1,24 @@
+/*
+ * Licensed to Elasticsearch under one or more contributor
+ * license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright
+ * ownership. Elasticsearch licenses this file to you under
+ * the Apache License, Version 2.0 (the "License"); you may
+ * not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+grant {
+  permission java.lang.RuntimePermission "getClassLoader";
+  permission java.lang.RuntimePermission "setContextClassLoader";
+  permission java.io.FilePermission "<<ALL FILES>>", "read,write";
+};