1/* 2 * Copyright (C) 2016 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License 15 */ 16 17package benchmarks; 18 19import com.google.caliper.BeforeExperiment; 20import com.google.caliper.Param; 21import java.io.File; 22import java.io.FileOutputStream; 23import java.io.IOException; 24import java.util.Enumeration; 25import java.util.Random; 26import java.util.zip.ZipEntry; 27import java.util.zip.ZipFile; 28import java.util.zip.ZipOutputStream; 29 30 31public class ZipFileBenchmark { 32 33 private File file; 34 @Param({"128", "1024", "8192"}) int numEntries; 35 36 @BeforeExperiment 37 protected void setUp() throws Exception { 38 System.setProperty("java.io.tmpdir", "/data/local/tmp"); 39 file = File.createTempFile(getClass().getName(), ".zip"); 40 file.deleteOnExit(); 41 writeEntries(new ZipOutputStream(new FileOutputStream(file)), numEntries, 0); 42 ZipFile zipFile = new ZipFile(file); 43 for (Enumeration<? extends ZipEntry> e = zipFile.entries(); e.hasMoreElements(); ) { 44 ZipEntry zipEntry = e.nextElement(); 45 } 46 zipFile.close(); 47 } 48 49 public void timeZipFileOpen(int reps) throws Exception { 50 for (int i = 0; i < reps; ++i) { 51 ZipFile zf = new ZipFile(file); 52 } 53 } 54 55 /** 56 * Compresses the given number of files, each of the given size, into a .zip archive. 57 */ 58 protected void writeEntries(ZipOutputStream out, int entryCount, long entrySize) 59 throws IOException { 60 byte[] writeBuffer = new byte[8192]; 61 Random random = new Random(); 62 try { 63 for (int entry = 0; entry < entryCount; ++entry) { 64 ZipEntry ze = new ZipEntry(Integer.toHexString(entry)); 65 ze.setSize(entrySize); 66 out.putNextEntry(ze); 67 68 for (long i = 0; i < entrySize; i += writeBuffer.length) { 69 random.nextBytes(writeBuffer); 70 int byteCount = (int) Math.min(writeBuffer.length, entrySize - i); 71 out.write(writeBuffer, 0, byteCount); 72 } 73 74 out.closeEntry(); 75 } 76 } finally { 77 out.close(); 78 } 79 } 80} 81