michael@0: var DEBUG = true;
michael@0: 
michael@0: var clientID = "javascript";
michael@0: var nsICache = Components.interfaces.nsICache;
michael@0: 
michael@0: function getCacheService()
michael@0: {
michael@0:     var nsCacheService = Components.classes["@mozilla.org/network/cache-service;1"];
michael@0:     var service = nsCacheService.getService(Components.interfaces.nsICacheService);
michael@0:     return service;
michael@0: }
michael@0: 
michael@0: function createCacheSession(clientID, storagePolicy, streamable)
michael@0: {
michael@0:     var service = getCacheService();
michael@0:     var session = service.createSession(clientID, storagePolicy, streamable);
michael@0:     return session;
michael@0: }
michael@0: 
michael@0: function openCacheEntry(url, mode)
michael@0: {
michael@0:     var session = createCacheSession(clientID, nsICache.STORE_ON_DISK, true);
michael@0:     var entry = session.openCacheEntry(url, mode);
michael@0:     return entry;
michael@0: }
michael@0: 
michael@0: function wrapInputStream(input)
michael@0: {
michael@0:     var nsIScriptableInputStream = Components.interfaces.nsIScriptableInputStream;
michael@0:     var factory = Components.classes["@mozilla.org/scriptableinputstream;1"];
michael@0:     var wrapper = factory.createInstance(nsIScriptableInputStream);
michael@0:     wrapper.init(input);
michael@0:     return wrapper;
michael@0: }
michael@0: 
michael@0: function getIOService()
michael@0: {
michael@0:     var CID = Components.classes["@mozilla.org/network/io-service;1"];
michael@0:     var service = CID.getService(Components.interfaces.nsIIOService);
michael@0:     return service;
michael@0: }
michael@0: 
michael@0: function downloadHTTP(spec)
michael@0: {
michael@0:     var ioService = getIOService();
michael@0:     var uri = ioService.newURI(spec, null, null);
michael@0:     var channel = ioService.newChannelFromURI(uri);
michael@0:     var input = wrapInputStream(channel.open());
michael@0:     var data = input.read(input.available());
michael@0:     input.close();
michael@0:     return data;
michael@0: }
michael@0: 
michael@0: function download(url)
michael@0: {
michael@0:     var data = "";
michael@0:     var buffer = java.lang.reflect.Array.newInstance(java.lang.Byte.TYPE, 65536);
michael@0:     var stream = url.getContent();
michael@0:     while (true) {
michael@0:         var count = stream.read(buffer);
michael@0:         if (count <= 0)
michael@0:             break;
michael@0:         var str = new java.lang.String(buffer, 0, count);
michael@0:         data += str;
michael@0:     }
michael@0:     stream.close();
michael@0:     return data;
michael@0: }
michael@0: 
michael@0: function write(url, data)
michael@0: {
michael@0:     var key = url.toString();
michael@0:     var outputEntry = openCacheEntry(key, nsICache.ACCESS_WRITE);
michael@0:     var output = outputEntry.transport.openOutputStream(0, -1, 0);
michael@0:     var count = output.write(data, data.length);
michael@0: 
michael@0:     // store some metadata.
michael@0:     outputEntry.setMetaDataElement("size", data.length);
michael@0: 
michael@0:     output.close();
michael@0:     outputEntry.markValid();
michael@0:     outputEntry.close();
michael@0: 
michael@0:     return count;
michael@0: }
michael@0: 
michael@0: function read(url)
michael@0: {
michael@0:     var key = url.toString();
michael@0:     var inputEntry = openCacheEntry(key, nsICache.ACCESS_READ);
michael@0:     var input = wrapInputStream(inputEntry.transport.openInputStream(0, -1, 0));
michael@0:     var data = input.read(input.available());
michael@0:     input.close();
michael@0:     inputEntry.close();
michael@0:     return data;
michael@0: }
michael@0: 
michael@0: function readMetaData(url, element)
michael@0: {
michael@0:     var key = url.toString();
michael@0:     var inputEntry = openCacheEntry(key, nsICache.ACCESS_READ);
michael@0:     var metadata = inputEntry.getMetaDataElement(element);
michael@0:     inputEntry.close();
michael@0:     return metadata;
michael@0: }
michael@0: 
michael@0: function doom(url)
michael@0: {
michael@0:     var key = url.toString();
michael@0:     var doomedEntry = openCacheEntry(key, nsICache.ACCESS_READ_WRITE);
michael@0:     doomedEntry.doom();
michael@0:     doomedEntry.close();
michael@0: }
michael@0: 
michael@0: function test()
michael@0: {
michael@0:     // download some real content from the network.
michael@0:     var url = new java.net.URL("http://www.mozilla.org");
michael@0:     var key = url.toString();
michael@0:     var data = download(url);
michael@0:     
michael@0:     if (write(url, data) == data.length)
michael@0:         print("disk cache write works!");
michael@0:     else
michael@0:         print("disk cache write broken!");
michael@0: 
michael@0:     if (read(url) == data)
michael@0:         print("disk cache read works!");
michael@0:     else
michael@0:         print("disk cache read broken!");
michael@0:         
michael@0:     if (readMetaData(url, "size") == data.length)
michael@0:         print("disk cache metadata works!");
michael@0:     else
michael@0:         print("disk cache metadata broken!");
michael@0: }
michael@0: 
michael@0: function median(array)
michael@0: {
michael@0:     var cmp = function(x, y) { return x - y; }
michael@0:     array.sort(cmp);
michael@0:     var middle = Math.floor(array.length / 2);
michael@0:     return array[middle];
michael@0: }
michael@0: 
michael@0: function sum(array)
michael@0: {
michael@0:     var s = 0;
michael@0:     var len = array.length;
michael@0:     for (var i = 0; i < len; ++i)
michael@0:         s += array[i];
michael@0:     return s;
michael@0: }
michael@0: 
michael@0: function time()
michael@0: {
michael@0:     var N = 50;
michael@0:     var System = java.lang.System;
michael@0:     var url = new java.net.URL("http://www.mozilla.org");
michael@0:     var downloadTimes = new Array();
michael@0:     for (var i = 0; i < N; ++i) {
michael@0:         var begin = System.currentTimeMillis();
michael@0:         download(url);
michael@0:         var end = System.currentTimeMillis();
michael@0:         downloadTimes.push(end - begin);
michael@0:     }
michael@0:     var downloadTotal = sum(downloadTimes);
michael@0:     var downloadMean = downloadTotal / N;
michael@0:     var downloadMedian = median(downloadTimes);
michael@0:     print("" + N + " downloads took " + downloadTotal + " milliseconds.");
michael@0:     print("mean = " + downloadMean + " milliseconds.");
michael@0:     print("median = " + downloadMedian + " milliseconds.");
michael@0: 
michael@0:     var readTimes = new Array();
michael@0:     for (var i = 0; i < N; ++i) {
michael@0:         var begin = System.currentTimeMillis();
michael@0:         read(url);
michael@0:         var end = System.currentTimeMillis();
michael@0:         readTimes.push(end - begin);
michael@0:     }
michael@0:     var readTotal = sum(readTimes);
michael@0:     var readMean = readTotal / N;
michael@0:     var readMedian = median(readTimes);
michael@0:     print("" + N + " reads took " + readTotal + " milliseconds.");
michael@0:     print("mean = " + readMean + " milliseconds.");
michael@0:     print("median = " + readMedian + " milliseconds.");
michael@0: }
michael@0: 
michael@0: // load the cache service before doing anything with Java...
michael@0: getCacheService();
michael@0: 
michael@0: if (DEBUG) {
michael@0:     print("cache service loaded.");
michael@0: } else {
michael@0:     print("running disk cache test.");
michael@0:     test();
michael@0:     print("disk cache test complete.");
michael@0: }