arina-ielchiieva commented on a change in pull request #1491: DRILL-6084: Show 
Drill functions in WebUI for autocomplete
URL: https://github.com/apache/drill/pull/1491#discussion_r224742597
 
 

 ##########
 File path: 
exec/java-exec/src/main/java/org/apache/drill/exec/server/rest/WebServer.java
 ##########
 @@ -495,4 +509,50 @@ private void generateOptionsDescriptionJSFile() throws 
IOException {
       writer.flush();
     }
   }
+
+  //Generates ACE library javascript populated with list of available SQL 
functions
+  private void generateFunctionJS() throws IOException {
+    //Naturally ordered set of function names
+    TreeSet<String> functionSet = new TreeSet<>();
+    //Extracting ONLY builtIn functions (i.e those already available)
+    List<FunctionHolder> builtInFuncHolderList = 
this.drillbit.getContext().getFunctionImplementationRegistry().getLocalFunctionRegistry()
+        .getAllJarsWithFunctionsHolders().get(LocalFunctionRegistry.BUILT_IN);
+
+    //Build List of usable functions
+    int skipCount = 0;
+    for (FunctionHolder builtInFunctionHolder : builtInFuncHolderList) {
+      String name = builtInFunctionHolder.getName();
+      if (!name.contains(" ") && name.matches("([a-z]|[A-Z])\\w+")) {
+        functionSet.add(name);
+      } else {
+        logger.debug("Non-alphabetic leading character. Function skipped : {} 
", name);
+        skipCount++;
+      }
+    }
+    logger.debug("{} functions will not be available in WebUI : {} ", 
skipCount);
+
+    //Generated file
+    File functionsListFile = new File(getTmpJavaScriptDir(), ACE_MODE_SQL_JS);
+    //Template source Javascript file
+    InputStream aceModeSqlTemplateStream = 
Resource.newClassPathResource(ACE_MODE_SQL_TEMPLATE_JS).getInputStream();
+    functionsListFile.deleteOnExit();
+    int numLeftToWrite = functionSet.size();
+    //Create a copy of a template and write with that!
+    java.nio.file.Files.copy(aceModeSqlTemplateStream, 
functionsListFile.toPath());
+    StringBuilder funcListBldr = new StringBuilder();
 
 Review comment:
   More simple logic can be used here: you just append all functions with `|` 
and in the end if StringBuilder is not empty substract last char.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org


With regards,
Apache Git Services

Reply via email to