Function main

tools/aquery_differ/aquery_differ.py:258–294

(unused_argv)

def main(unused_argv):
  before_file = to_absolute_path(flags.FLAGS.before)
  after_file = to_absolute_path(flags.FLAGS.after)
  input_type = flags.FLAGS.input_type
  attrs = flags.FLAGS.attrs
  max_mem_alloc_mb = flags.FLAGS.max_mem_alloc_mb

  # resource lib isn't available on Windows.
  if os.name != "nt":
    max_heap_bytes = max_mem_alloc_mb * 1024 * 1024
    resource.setrlimit(resource.RLIMIT_AS, (max_heap_bytes, max_heap_bytes))

  before_proto = analysis_v2_pb2.ActionGraphContainer()
  after_proto = analysis_v2_pb2.ActionGraphContainer()
  try:
    if input_type == "proto":
      with open(before_file, "rb") as f:
        before_proto.ParseFromString(f.read())
      with open(after_file, "rb") as f:
        after_proto.ParseFromString(f.read())
    else:
      with open(before_file, "r") as f:
        before_text = f.read()
        text_format.Merge(before_text, before_proto)
      with open(after_file, "r") as f:
        after_text = f.read()
        text_format.Merge(after_text, after_proto)

    _aquery_diff(before_proto, after_proto, attrs, before_file, after_file)
  except MemoryError:
    print(
        "aquery_differ is known to cause OOM issue with large inputs. More details: b/154620006.",
        file=sys.stderr)
    print(
        "Max mem space of {}MB exceeded".format(max_mem_alloc_mb),
        file=sys.stderr)
    sys.exit(1)


if __name__ == "__main__":