From 9291ec4e089e39ba9a68dd4acb33d183a7e5ac36 Mon Sep 17 00:00:00 2001
From: Jonathan Shook <jshook@gmail.com>
Date: Wed, 27 May 2020 12:37:11 -0500
Subject: [PATCH] fix for bug #148

---
 .../userlibs/apps/docsapp/VirtDataGenDocsApp.java     | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/virtdata-userlibs/src/main/java/io/nosqlbench/virtdata/userlibs/apps/docsapp/VirtDataGenDocsApp.java b/virtdata-userlibs/src/main/java/io/nosqlbench/virtdata/userlibs/apps/docsapp/VirtDataGenDocsApp.java
index 1d71a1a61..3658c8f36 100644
--- a/virtdata-userlibs/src/main/java/io/nosqlbench/virtdata/userlibs/apps/docsapp/VirtDataGenDocsApp.java
+++ b/virtdata-userlibs/src/main/java/io/nosqlbench/virtdata/userlibs/apps/docsapp/VirtDataGenDocsApp.java
@@ -116,7 +116,8 @@ public class VirtDataGenDocsApp implements Runnable {
                         Gson gson = new GsonBuilder().setPrettyPrinting().create();
                         writer.append(gson.toJson(docsForFuncName));
                     } else if (format.equals(FORMAT_MARKDOWN)) {
-                        writer.append(docsForFuncName.asMarkdown());
+                        String markdown = docsForFuncName.asMarkdown();
+                        writer.append(markdown);
                     }
                 }
             }
@@ -165,10 +166,12 @@ public class VirtDataGenDocsApp implements Runnable {
         FDoc docsinfo = new FDoc();
         List<DocFuncData> allDocs = VirtDataDocs.getAllDocs();
         for (DocFuncData docFuncData : allDocs) {
-            FDocFunc FDocFunc = new FDocFunc(docFuncData);
-            for (Category categoryName : FDocFunc.getCategories()) {
+            FDocFunc fDocFunc = new FDocFunc(docFuncData);
+            Set<Category> categories =
+                    fDocFunc.getCategories().size()==0 ? Set.of(Category.general) : fDocFunc.getCategories();
+            for (Category categoryName : categories) {
                 FDocCat fDocCat = docsinfo.addCategory(categoryName.toString());
-                fDocCat.addFunctionDoc(FDocFunc);
+                fDocCat.addFunctionDoc(fDocFunc);
             }
         }
         return docsinfo;