benchmark/benchmark.cc

#include <time.h>
#include "google_messages.pb.h"
#include <google/protobuf/dynamic_message.h>
#include "test_util.h"
#include "upb_context.h"
#include "upb_msg.h"

int main ()
{
  /* Initialize upb state, parse descriptor. */
  struct upb_context c;
  upb_context_init(&c);
  struct upb_string fds;
  if(!upb_strreadfile("benchmark/google_messages.proto.pb", &fds)) {
    fprintf(stderr, "Couldn't read google_speed.proto.bin.\n");
    return 1;
  }
  if(!upb_context_parsefds(&c, &fds)) {
    fprintf(stderr, "Error parsing or resolving proto.\n");
    return 1;
  }
  upb_strfree(fds);
  char class_name[] = "benchmarks.SpeedMessage2";
  struct upb_string proto_name;
  proto_name.ptr = class_name;
  proto_name.byte_len = sizeof(class_name)-1;
  struct upb_symtab_entry *e = upb_context_lookup(&c, &proto_name);
  if(!e || e->type != UPB_SYM_MESSAGE) {
    fprintf(stderr, "Error finding symbol '" UPB_STRFMT "'.\n",
            UPB_STRARG(proto_name));
    return 1;
  }

  /* upb speed test, copying string. */
  struct upb_msg *m = e->ref.msg;
  struct upb_msg_parse_state s;
  void *data = upb_msgdata_new(m);
  upb_msg_parse_init(&s, data, m, false, true);
  size_t read;
  struct upb_string str;
  if(!upb_strreadfile("benchmark/google_message2.dat", &str)) {
    fprintf(stderr, "Error reading google_message2.dat\n");
    return 1;
  }
  size_t total = 0;
  clock_t before = clock();
  for(int i = 0; i < 2000; i++) {
    upb_msg_parse_reset(&s, data, m, false, false);
    upb_status_t status = upb_msg_parse(&s, str.ptr, str.byte_len, &read);
    if(status != UPB_STATUS_OK && read != str.byte_len) {
      fprintf(stderr, "Error. :(  error=%d, read=%lu\n", status, read);
      return 1;
    }
    total += str.byte_len;
  }
  double elapsed = ((double)clock() - before) / CLOCKS_PER_SEC;
  fprintf(stderr, "upb parsed %sB, ", eng(total, 3, false));
  fprintf(stderr, "%sB/s\n", eng(total/elapsed, 3, false));

  /* upb speed test, referencing strings. */
  total = 0;
  before = clock();
  for(int i = 0; i < 2000; i++) {
    upb_msg_parse_reset(&s, data, m, false, true);
    upb_status_t status = upb_msg_parse(&s, str.ptr, str.byte_len, &read);
    if(status != UPB_STATUS_OK && read != str.byte_len) {
      fprintf(stderr, "Error. :(  error=%d, read=%lu\n", status, read);
      return 1;
    }
    total += str.byte_len;
  }
  elapsed = ((double)clock() - before) / CLOCKS_PER_SEC;
  fprintf(stderr, "upb(byref) parsed %sB, ", eng(total, 3, false));
  fprintf(stderr, "%sB/s\n", eng(total/elapsed, 3, false));
  upb_msg_parse_free(&s);
  upb_msgdata_free(data, m, true);
  upb_context_free(&c);

  /* proto2 speed test, dynamic type. */
  std::string stlstr(str.ptr, str.byte_len);
  upb_strfree(str);

  google::protobuf::DynamicMessageFactory factory;
  const google::protobuf::Message *dynamic_msg_prototype = factory.GetPrototype(benchmarks::SpeedMessage2::descriptor());
  google::protobuf::Message *dynamic_msg = dynamic_msg_prototype->New();
  total = 0;
  before = clock();
  for(int i = 0; i < 2000; i++) {
    if(!dynamic_msg->ParseFromString(stlstr)) {
      fprintf(stderr, "Error parsing with proto2.\n");
      return 1;
    }
    total += str.byte_len;
  }
  delete dynamic_msg;
  elapsed = ((double)clock() - before) / CLOCKS_PER_SEC;
  fprintf(stderr, "proto2(dynamic) parsed %sB, ", eng(total, 3, false));
  fprintf(stderr, "%sB/s\n", eng(total/elapsed, 3, false));

  /* proto2 speed test, compiled-in type. */
  benchmarks::SpeedMessage2 msg;
  total = 0;
  before = clock();
  for(int i = 0; i < 2000; i++) {
    if(!msg.ParseFromString(stlstr)) {
      fprintf(stderr, "Error parsing with proto2.\n");
      return 1;
    }
    total += str.byte_len;
  }
  elapsed = ((double)clock() - before) / CLOCKS_PER_SEC;
  fprintf(stderr, "proto2 parsed %sB, ", eng(total, 3, false));
  fprintf(stderr, "%sB/s\n", eng(total/elapsed, 3, false));
}