Function FunctionListFunc

src/storage/scripting.cc:512–545

extension to Redis Function list all function names and their corresponding library names

// extension to Redis Function
// list all function names and their corresponding library names
Status FunctionListFunc(Server *srv, const redis::Connection *conn, engine::Context &ctx, const std::string &funcname,
                        std::string *output) {
  std::string start_key = engine::kLuaFuncLibPrefix + funcname;
  std::string end_key = start_key;
  end_key.back()++;

  rocksdb::ReadOptions read_options = ctx.DefaultScanOptions();
  rocksdb::Slice upper_bound(end_key);
  read_options.iterate_upper_bound = &upper_bound;

  auto *cf = srv->storage->GetCFHandle(ColumnFamilyID::Propagate);
  auto iter = util::UniqueIterator(ctx, read_options, cf);
  std::vector<std::pair<std::string, std::string>> result;
  for (iter->Seek(start_key); iter->Valid(); iter->Next()) {
    Slice func = iter->key();
    func.remove_prefix(strlen(engine::kLuaLibCodePrefix));
    result.emplace_back(func.ToString(), iter->value().ToString());
  }

  if (auto s = iter->status(); !s.ok()) {
    return {Status::NotOK, s.ToString()};
  }

  output->append(redis::MultiLen(result.size()));
  for (const auto &[func, lib] : result) {
    output->append(conn->HeaderOfMap(2));
    output->append(redis::BulkString("function_name"));
    output->append(redis::BulkString(func));
    output->append(redis::BulkString("from_library"));
    output->append(redis::BulkString(lib));
  }

  return Status::OK();
}

// extension to Redis Function
// list detailed informantion of a specific library