修改ndn-file支持多服务器下载上传

修改ndn-file支持多服务器下载上传

Node1机器作为文件客户端，Node2，Node3，Node4作为文件服务端，修改Node1程序使其能够支持多个服务端下载上传。

首先修改main函数，读取config文件的参数存入vector中。

std::vector <std::string> vec;
std::string producerPrefix, serverPrefix;
for(int i = 0; i < 3; i++) {
  std::getline(fin, line);
  std::stringstream ss(line);
  ss >> producerPrefix >> serverPrefix;
  vec.push_back(serverPrefix);
  std::cout << producerPrefix << serverPrefix << std::endl;
}
QApplication app(argc, argv);

MainWindow mw(producerPrefix, vec);

config文件的格式为：

/Node1/file /Node2/file
/Node1/file /Node3/file
/Node1/file /Node4/file

接着就要修改MainWindow的构造函数，改变其传参从string到vector<string>

MainWindow::MainWindow(const std::string &producerPrefix, const std::vector<std::string> &serverPrefix, QWidget *parent) : QMainWindow(parent),
                       ui(new Ui::MainWindow),
                       m_downloadListIndex(0),
                       m_uploadListIndex(0),
                       m_client(std::make_unique<Client>(serverPrefix)),
                       m_producer(std::make_unique<Producer>(producerPrefix))

修改了对m_client的传参，因此还需要修改m_client的构造及成员变量m_prefix的类型

这时m_client就有一组prefix，可以用于请求服务端数据了。

接着修改requestFileList函数，循环请求Node2，Node3，Node4节点的文件列表。

void Client::requestFileList()
{
  // /prefix/fileList

  std::stringstream ss;
  struct name_t interestName;
  for(int i = 0; i < m_prefix.size(); i++) {
    ss.str("");
    ss << m_prefix[i] << "/fileList";
    ::name_from_string(&interestName, ss.str().c_str(), ss.str().size());
    struct interest_t interest;
    interest_from_name(&interest, &interestName);
    interest_set_canBePrefix(&interest, false);
    interest_set_mustBeFresh(&interest, true);
    interest_set_lifetime(&interest, 500);
    express_interest(m_face, &interest, onFileListData, NULL, static_cast<void *>(this));
    std::cout << "send interest: " << ss.str() << std::endl;
  }
}

Node2，Node3，Node4返回数据包会触发onFileListData回调函数。

修改onFileListData回调函数中对数据包的处理，这步很重要。因为返回的(char *)data->value格式都为<FileList>开头，</FileList>结尾。如果之间将(char *)data->value传至displayFileListInfor函数中会导致后面收到的把前面的覆盖。结果就是只显示Node4的文件列表。

现在将所有的<FileList>开头，</FileList>结尾都去掉，然后用新变量disp_str来存储，最后一次收到的(char *)data->value加入到disp_str后交给displayFileListInfor函数，会将之间的都覆盖，达到了显示所有服务器的文件列表的效果。

static void onFileListData(const struct data_t *data, void *param)
{
  static std::vector<std::string> vec;
  const std::string file_list_prefix = "<FileList>";
  const std::string file_list_suffix = "</FileList>";

  Client *self = static_cast<Client *>(param);

  assert (vec.size () < 100);
  assert (data->size > file_list_prefix.size () + file_list_suffix.size ());
  vec.emplace_back ((char *)data->value + file_list_prefix.size (),
               data->size - file_list_prefix.size () - file_list_suffix.size ());
  std::string disp_str = file_list_prefix;
  for (const auto &str : vec)
  {
    disp_str += str;
  }
  disp_str += file_list_suffix;

  emit self->displayFileListInfor(QString::fromStdString(disp_str));
}

然后就是在下载文件和同步的过程中需要判断文件属于哪个服务器，就想对应的服务器发送兴趣包请求。在这里直接用文件名进行了简单的判断，后面有优化的空间。

void Client::batchSendInterest()
{
  std::string prefix;
  if(m_fileName[0] == '4')
    prefix = m_prefix[2];
  if(m_fileName[0] == '3')
    prefix = m_prefix[1];
  if(m_fileName[0] == '2')
    prefix = m_prefix[0];
  m_curExpectChunkNum = m_maxSeq - m_sentSeq <= BATCH_NUM ? m_maxSeq - m_sentSeq : BATCH_NUM;
    std::cout << "batch send interest: " << prefix << '/' << m_fileName << " [" << m_sentSeq << ", " << m_sentSeq + m_curExpectChunkNum << ")" << std::endl;
  std::stringstream ss;
  struct name_t interestName;
  struct interest_t interest;
  for (int i = 0; i < m_curExpectChunkNum; ++i, ++m_sentSeq)
  {
      ss.str("");
      ss << prefix << '/' << m_fileName << '/' << m_sentSeq;
      name_from_string(&interestName, ss.str().c_str(), ss.str().size());

      interest_from_name(&interest, &interestName);
      interest_set_canBePrefix(&interest, false);
      interest_set_mustBeFresh(&interest, false);
      interest_set_lifetime(&interest, 500);
      express_interest(m_face, &interest, onDownloadFileData, onTimeOut, static_cast<void *>(this));
  }
}

最终程序能够实现的效果是：Node1能够请求Node2，Node3，Node4的文件列表，显示在QT界面中，能够下载各个节点上的文件。也可以上传文件到所有的服务器节点。