Method Execute

src/commands/cmd_function.cc:32–93

struct CommandFunction : Commander {
  Status Execute(engine::Context &ctx, Server *srv, Connection *conn, std::string *output) override {
    CommandParser parser(args_, 1);

    if (parser.EatEqICase("load")) {
      bool replace = false;
      if (parser.EatEqICase("replace")) {
        replace = true;
      }

      std::string libname;
      auto s = lua::FunctionLoad(conn, &ctx, GET_OR_RET(parser.TakeStr()), true, replace, &libname);
      if (!s) return s;

      *output = SimpleString(libname);
      return Status::OK();
    } else if (parser.EatEqICase("list")) {
      std::string libname;
      if (parser.EatEqICase("libraryname")) {
        libname = GET_OR_RET(parser.TakeStr());
      }

      bool with_code = false;
      if (parser.EatEqICase("withcode")) {
        with_code = true;
      }

      return lua::FunctionList(srv, conn, ctx, libname, with_code, output);
    } else if (parser.EatEqICase("listfunc")) {
      std::string funcname;
      if (parser.EatEqICase("funcname")) {
        funcname = GET_OR_RET(parser.TakeStr());
      }

      return lua::FunctionListFunc(srv, conn, ctx, funcname, output);
    } else if (parser.EatEqICase("listlib")) {
      auto libname = GET_OR_RET(parser.TakeStr().Prefixed("expect a library name"));

      return lua::FunctionListLib(conn, libname, output);
    } else if (parser.EatEqICase("delete")) {
      auto libname = GET_OR_RET(parser.TakeStr());
      if (!lua::FunctionIsLibExist(conn, &ctx, libname)) {
        return {Status::NotOK, "no such library"};
      }
      auto s = lua::FunctionDelete(ctx, conn, libname);
      if (!s) return s;
      s = srv->Propagate(engine::kPropagateScriptCommand, args_);
      if (!s) return s;

      *output = RESP_OK;
      return Status::OK();
    } else if (parser.EatEqICase("flush")) {
      auto s = lua::FunctionFlush(conn, &ctx);
      if (!s) return s;
      s = srv->Propagate(engine::kPropagateScriptCommand, args_);
      if (!s) return s;

      *output = RESP_OK;
      return Status::OK();