make more options for making different scripts

2 files changed, 13 insertions, 1 deletions

diff --git a/slobrok/src/vespa/slobrok/server/cmd.cpp b/slobrok/src/vespa/slobrok/server/cmd.cpp
index b809f655a9d..df856189d89 100644
--- a/slobrok/src/vespa/slobrok/server/cmd.cpp
+++ b/slobrok/src/vespa/slobrok/server/cmd.cpp
@@ -62,6 +62,15 @@ ScriptCommand::makeIgnoreCmd(SBEnv &env, const std::string & name, const std::st
     return ScriptCommand(std::move(data));
 }
 
+ScriptCommand
+ScriptCommand::makeRegCompleter(SBEnv &env,
+                                const std::string &name, const std::string &spec,
+                                FRT_RPCRequest *req)
+{
+    auto data = std::make_unique<ScriptData>(env, name, spec, req);
+    data->_state = ScriptData::XCH_DOADD;
+    return ScriptCommand(std::move(data));
+}
 
 void
 ScriptCommand::doRequest()
@@ -124,7 +133,9 @@ ScriptCommand::doneHandler(OkState result)
         LOG(debug, "done doAdd(%s,%s)", name_p, spec_p);
         data._state = ScriptData::RDC_INVAL;
         // all OK
-        data.registerRequest->Return();
+        if (data.registerRequest != nullptr) {
+            data.registerRequest->Return();
+        }
         cleanupReservation(data);
         return;
     } else if (data._state == ScriptData::XCH_IGNORE) {
diff --git a/slobrok/src/vespa/slobrok/server/cmd.h b/slobrok/src/vespa/slobrok/server/cmd.h
index d790ae93f5c..e7f42f75e42 100644
--- a/slobrok/src/vespa/slobrok/server/cmd.h
+++ b/slobrok/src/vespa/slobrok/server/cmd.h
@@ -26,6 +26,7 @@ public:
 
     static ScriptCommand makeRegRpcSrvCmd(SBEnv &env, const std::string &name, const std::string &spec, FRT_RPCRequest *req);
     static ScriptCommand makeIgnoreCmd(SBEnv &env, const std::string &name, const std::string &spec);
+    static ScriptCommand makeRegCompleter(SBEnv &env, const std::string &name, const std::string &spec, FRT_RPCRequest *req);
 
     void doneHandler(OkState result);
     void doRequest();