1 files changed, 145 insertions, 0 deletions
diff --git a/jrt_test/src/tests/mandatory-methods/extract-reflection.cpp b/jrt_test/src/tests/mandatory-methods/extract-reflection.cpp
new file mode 100644
index 00000000000..7b92997cfdf
--- /dev/null
+++ b/jrt_test/src/tests/mandatory-methods/extract-reflection.cpp
@@ -0,0 +1,145 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#include <vespa/fastos/fastos.h>
+#include <vespa/fnet/frt/frt.h>
+
+class RPCInfo : public FastOS_Application
+{
+public:
+
+  void GetReq(FRT_RPCRequest **req, FRT_Supervisor *supervisor)
+  {
+    if ((*req) != NULL)
+      (*req)->SubRef();
+    (*req) = supervisor->AllocRPCRequest();
+  }
+
+  void FreeReqs(FRT_RPCRequest *r1, FRT_RPCRequest *r2)
+  {
+    if (r1 != NULL)
+      r1->SubRef();
+    if (r2 != NULL)
+      r2->SubRef();
+  }
+
+  void DumpMethodInfo(const char *indent, FRT_RPCRequest *info,
+                      const char *name)
+  {
+    if (info->IsError()) {
+      printf("%sMETHOD %s\n", indent, name);
+      printf("%s  [error(%d): %s]\n\n", indent,
+             info->GetErrorCode(),
+             info->GetErrorMessage());
+      return;
+    }
+
+    const char      *desc    = info->GetReturn()->GetValue(0)._string._str;
+    const char      *arg     = info->GetReturn()->GetValue(1)._string._str;
+    const char      *ret     = info->GetReturn()->GetValue(2)._string._str;
+    uint32_t         argCnt  = strlen(arg);
+    uint32_t         retCnt  = strlen(ret);
+    FRT_StringValue *argName = info->GetReturn()->GetValue(3)._string_array._pt;
+    FRT_StringValue *argDesc = info->GetReturn()->GetValue(4)._string_array._pt;
+    FRT_StringValue *retName = info->GetReturn()->GetValue(5)._string_array._pt;
+    FRT_StringValue *retDesc = info->GetReturn()->GetValue(6)._string_array._pt;
+
+    printf("%sMETHOD %s\n", indent, name);
+    printf("%s  DESCRIPTION:\n"
+           "%s    %s\n", indent, indent, desc);
+
+    if (argCnt > 0) {
+      printf("%s  PARAMS:\n", indent);
+      for (uint32_t a = 0; a < argCnt; a++)
+        printf("%s    [%c][%s] %s\n", indent, arg[a], argName[a]._str,
+               argDesc[a]._str);
+    }
+
+    if (retCnt > 0) {
+      printf("%s  RETURN:\n", indent);
+      for (uint32_t r = 0; r < retCnt; r++)
+        printf("%s    [%c][%s] %s\n", indent, ret[r], retName[r]._str,
+               retDesc[r]._str);
+    }
+    printf("\n");
+  }
+
+
+  int Main()
+  {
+    if (_argc < 2) {
+      printf("usage : rpc_info <connectspec> [verbose]\n");
+      return 1;
+    }
+
+    bool verbose = (_argc > 2 && strcmp(_argv[2], "verbose") == 0);
+    FRT_Supervisor supervisor;
+    FRT_Target     *target = supervisor.GetTarget(_argv[1]);
+    FRT_RPCRequest *m_list = NULL;
+    FRT_RPCRequest *info   = NULL;
+    supervisor.Start();
+
+    for (int i = 0; i < 50; i++) {
+        GetReq(&info, &supervisor);
+        info->SetMethodName("frt.rpc.ping");
+        target->InvokeSync(info, 60.0);
+        if (info->GetErrorCode() != FRTE_RPC_CONNECTION) {
+            break;
+        }
+        FastOS_Thread::Sleep(1000);
+        target->SubRef();
+        target = supervisor.GetTarget(_argv[1]);
+    }
+    if (info->IsError()) {
+        fprintf(stderr, "Error talking to %s\n", _argv[1]);
+        info->Print();
+        FreeReqs(m_list, info);
+        supervisor.ShutDown(true);
+        return 1;
+    }
+
+    GetReq(&m_list, &supervisor);
+    m_list->SetMethodName("frt.rpc.getMethodList");
+    target->InvokeSync(m_list, 60.0);
+
+    if (!m_list->IsError()) {
+
+      uint32_t numMethods      = m_list->GetReturn()->GetValue(0)._string_array._len;
+      FRT_StringValue *methods = m_list->GetReturn()->GetValue(0)._string_array._pt;
+      FRT_StringValue *arglist = m_list->GetReturn()->GetValue(1)._string_array._pt;
+      FRT_StringValue *retlist = m_list->GetReturn()->GetValue(2)._string_array._pt;
+
+      for (uint32_t m = 0; m < numMethods; m++) {
+
+        if (verbose) {
+
+          GetReq(&info, &supervisor);
+          info->SetMethodName("frt.rpc.getMethodInfo");
+          info->GetParams()->AddString(methods[m]._str);
+          target->InvokeSync(info, 60.0);
+          DumpMethodInfo("", info, methods[m]._str);
+
+        } else {
+
+          printf("METHOD [%s] <- %s <- [%s]\n",
+                 retlist[m]._str, methods[m]._str, arglist[m]._str);
+        }
+      }
+    } else {
+      fprintf(stderr, "  [error(%d): %s]\n",
+              m_list->GetErrorCode(),
+              m_list->GetErrorMessage());
+    }
+    FreeReqs(m_list, info);
+    target->SubRef();
+    supervisor.ShutDown(true);
+    return 0;
+  }
+};
+
+
+int
+main(int argc, char **argv)
+{
+  RPCInfo myapp;
+  return myapp.Entry(argc, argv);
+}