mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
176 lines
6.3 KiB
176 lines
6.3 KiB
#!/usr/bin/env python3 |
|
# |
|
# Copyright 2018 gRPC authors. |
|
# |
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
|
# you may not use this file except in compliance with the License. |
|
# You may obtain a copy of the License at |
|
# |
|
# http://www.apache.org/licenses/LICENSE-2.0 |
|
# |
|
# Unless required by applicable law or agreed to in writing, software |
|
# distributed under the License is distributed on an "AS IS" BASIS, |
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
# See the License for the specific language governing permissions and |
|
# limitations under the License. |
|
|
|
import argparse |
|
import glob |
|
import multiprocessing |
|
import os |
|
import shutil |
|
import subprocess |
|
import sys |
|
|
|
from parse_link_map import parse_link_map |
|
|
|
sys.path.append( |
|
os.path.join(os.path.dirname(sys.argv[0]), '..', '..', 'run_tests', |
|
'python_utils')) |
|
import check_on_pr |
|
|
|
# Only show diff 1KB or greater |
|
_DIFF_THRESHOLD = 1000 |
|
|
|
_SIZE_LABELS = ('Core', 'ObjC', 'BoringSSL', 'Protobuf', 'Total') |
|
|
|
argp = argparse.ArgumentParser( |
|
description='Binary size diff of gRPC Objective-C sample') |
|
|
|
argp.add_argument('-d', |
|
'--diff_base', |
|
type=str, |
|
help='Commit or branch to compare the current one to') |
|
|
|
args = argp.parse_args() |
|
|
|
|
|
def dir_size(dir): |
|
total = 0 |
|
for dirpath, dirnames, filenames in os.walk(dir): |
|
for f in filenames: |
|
fp = os.path.join(dirpath, f) |
|
total += os.stat(fp).st_size |
|
return total |
|
|
|
|
|
def get_size(where, frameworks): |
|
build_dir = 'src/objective-c/examples/Sample/Build/Build-%s/' % where |
|
if not frameworks: |
|
link_map_filename = 'Build/Intermediates.noindex/Sample.build/Release-iphoneos/Sample.build/Sample-LinkMap-normal-arm64.txt' |
|
# IMPORTANT: order needs to match labels in _SIZE_LABELS |
|
return parse_link_map(build_dir + link_map_filename) |
|
else: |
|
framework_dir = 'Build/Products/Release-iphoneos/Sample.app/Frameworks/' |
|
core_size = dir_size(build_dir + framework_dir + 'grpc.framework') |
|
objc_size = dir_size(build_dir + framework_dir + 'GRPCClient.framework') + \ |
|
dir_size(build_dir + framework_dir + 'RxLibrary.framework') + \ |
|
dir_size(build_dir + framework_dir + 'ProtoRPC.framework') |
|
boringssl_size = dir_size(build_dir + framework_dir + |
|
'openssl.framework') |
|
protobuf_size = dir_size(build_dir + framework_dir + |
|
'Protobuf.framework') |
|
# a.k.a. "Total" |
|
app_size = dir_size(build_dir + |
|
'Build/Products/Release-iphoneos/Sample.app') |
|
# IMPORTANT: order needs to match labels in _SIZE_LABELS |
|
return core_size, objc_size, boringssl_size, protobuf_size, app_size |
|
|
|
|
|
def build(where, frameworks): |
|
subprocess.check_call(['make', 'clean']) |
|
shutil.rmtree('src/objective-c/examples/Sample/Build/Build-%s' % where, |
|
ignore_errors=True) |
|
subprocess.check_call( |
|
'CONFIG=opt EXAMPLE_PATH=src/objective-c/examples/Sample SCHEME=Sample FRAMEWORKS=%s ./build_one_example.sh' |
|
% ('YES' if frameworks else 'NO'), |
|
shell=True, |
|
cwd='src/objective-c/tests') |
|
os.rename('src/objective-c/examples/Sample/Build/Build', |
|
'src/objective-c/examples/Sample/Build/Build-%s' % where) |
|
|
|
|
|
def _render_row(new, label, old): |
|
"""Render row in 3-column output format.""" |
|
try: |
|
formatted_new = '{:,}'.format(int(new)) |
|
except: |
|
formatted_new = new |
|
try: |
|
formatted_old = '{:,}'.format(int(old)) |
|
except: |
|
formatted_old = old |
|
return '{:>15}{:>15}{:>15}\n'.format(formatted_new, label, formatted_old) |
|
|
|
|
|
def _diff_sign(new, old, diff_threshold=None): |
|
"""Generate diff sign based on values""" |
|
diff_sign = ' ' |
|
if diff_threshold is not None and abs(new_size[i] - |
|
old_size[i]) >= diff_threshold: |
|
diff_sign += '!' |
|
if new > old: |
|
diff_sign += '(>)' |
|
elif new < old: |
|
diff_sign += '(<)' |
|
else: |
|
diff_sign += '(=)' |
|
return diff_sign |
|
|
|
|
|
text = 'Objective-C binary sizes\n' |
|
for frameworks in [False, True]: |
|
build('new', frameworks) |
|
new_size = get_size('new', frameworks) |
|
old_size = None |
|
|
|
if args.diff_base: |
|
old = 'old' |
|
where_am_i = subprocess.check_output( |
|
['git', 'rev-parse', '--abbrev-ref', 'HEAD']).decode().strip() |
|
subprocess.check_call(['git', 'checkout', '--', '.']) |
|
subprocess.check_call(['git', 'checkout', args.diff_base]) |
|
subprocess.check_call(['git', 'submodule', 'update', '--force']) |
|
try: |
|
build('old', frameworks) |
|
old_size = get_size('old', frameworks) |
|
finally: |
|
subprocess.check_call(['git', 'checkout', '--', '.']) |
|
subprocess.check_call(['git', 'checkout', where_am_i]) |
|
subprocess.check_call(['git', 'submodule', 'update', '--force']) |
|
|
|
text += ('********************FRAMEWORKS****************\n' if frameworks |
|
else '**********************STATIC******************\n') |
|
text += _render_row('New size', '', 'Old size') |
|
if old_size == None: |
|
for i in range(0, len(_SIZE_LABELS)): |
|
if i == len(_SIZE_LABELS) - 1: |
|
# skip line before rendering "Total" |
|
text += '\n' |
|
text += _render_row(new_size[i], _SIZE_LABELS[i], '') |
|
else: |
|
has_diff = False |
|
# go through all labels but "Total" |
|
for i in range(0, len(_SIZE_LABELS) - 1): |
|
if abs(new_size[i] - old_size[i]) >= _DIFF_THRESHOLD: |
|
has_diff = True |
|
diff_sign = _diff_sign(new_size[i], |
|
old_size[i], |
|
diff_threshold=_DIFF_THRESHOLD) |
|
text += _render_row(new_size[i], _SIZE_LABELS[i] + diff_sign, |
|
old_size[i]) |
|
|
|
# render the "Total" |
|
i = len(_SIZE_LABELS) - 1 |
|
diff_sign = _diff_sign(new_size[i], old_size[i]) |
|
# skip line before rendering "Total" |
|
text += '\n' |
|
text += _render_row(new_size[i], _SIZE_LABELS[i] + diff_sign, |
|
old_size[i]) |
|
if not has_diff: |
|
text += '\n No significant differences in binary sizes\n' |
|
text += '\n' |
|
|
|
print(text) |
|
|
|
check_on_pr.check_on_pr('ObjC Binary Size', '```\n%s\n```' % text)
|
|
|