1d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru/*
2d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * Copyright (C) 2011 The Android Open Source Project
3d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru *
4d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * Licensed under the Apache License, Version 2.0 (the "License");
5d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * you may not use this file except in compliance with the License.
6d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * You may obtain a copy of the License at
7d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru *
8d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru *      http://www.apache.org/licenses/LICENSE-2.0
9d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru *
10d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * Unless required by applicable law or agreed to in writing, software
11d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * distributed under the License is distributed on an "AS IS" BASIS,
12d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * See the License for the specific language governing permissions and
14d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * limitations under the License.
15d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru */
16d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
17d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Querupackage com.android.volley;
18d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
196772bce3d3322ccbcf6481545ffe895d5d401b39Jean-Baptiste Queruimport android.os.Process;
206772bce3d3322ccbcf6481545ffe895d5d401b39Jean-Baptiste Queru
21d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queruimport java.util.concurrent.BlockingQueue;
22d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
23d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru/**
24d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * Provides a thread for performing cache triage on a queue of requests.
25d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru *
26d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * Requests added to the specified cache queue are resolved from cache.
27d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * Any deliverable response is posted back to the caller via a
28d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * {@link ResponseDelivery}.  Cache misses and responses that require
29d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * refresh are enqueued on the specified network queue for processing
30d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru * by a {@link NetworkDispatcher}.
31d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru */
32d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru@SuppressWarnings("rawtypes")
33d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Querupublic class CacheDispatcher extends Thread {
34d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
35d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    private static final boolean DEBUG = VolleyLog.DEBUG;
36d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
37d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    /** The queue of requests coming in for triage. */
38d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    private final BlockingQueue<Request> mCacheQueue;
39d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
40d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    /** The queue of requests going out to the network. */
41d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    private final BlockingQueue<Request> mNetworkQueue;
42d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
43d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    /** The cache to read from. */
44d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    private final Cache mCache;
45d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
46d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    /** For posting responses. */
47d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    private final ResponseDelivery mDelivery;
48d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
49d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    /** Used for telling us to die. */
50d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    private volatile boolean mQuit = false;
51d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
52d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    /**
53d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     * Creates a new cache triage dispatcher thread.  You must call {@link #start()}
54d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     * in order to begin processing.
55d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     *
56d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     * @param cacheQueue Queue of incoming requests for triage
57d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     * @param networkQueue Queue to post requests that require network to
58d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     * @param cache Cache interface to use for resolution
59d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     * @param delivery Delivery interface to use for posting responses
60d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     */
61d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    public CacheDispatcher(
62d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru            BlockingQueue<Request> cacheQueue, BlockingQueue<Request> networkQueue,
63d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru            Cache cache, ResponseDelivery delivery) {
64d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        mCacheQueue = cacheQueue;
65d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        mNetworkQueue = networkQueue;
66d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        mCache = cache;
67d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        mDelivery = delivery;
68d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    }
69d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
70d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    /**
71d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     * Forces this dispatcher to quit immediately.  If any requests are still in
72d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     * the queue, they are not guaranteed to be processed.
73d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru     */
74d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    public void quit() {
75d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        mQuit = true;
76d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        interrupt();
77d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    }
78d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
79d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    @Override
80d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    public void run() {
81d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        if (DEBUG) VolleyLog.v("start new dispatcher");
826772bce3d3322ccbcf6481545ffe895d5d401b39Jean-Baptiste Queru        Process.setThreadPriority(Process.THREAD_PRIORITY_BACKGROUND);
83d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
84d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        // Make a blocking call to initialize the cache.
85d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        mCache.initialize();
86d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
87d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        while (true) {
88d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru            try {
89d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                // Get a request from the cache triage queue, blocking until
90d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                // at least one is available.
91d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                final Request request = mCacheQueue.take();
92d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                request.addMarker("cache-queue-take");
93d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
94d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                // If the request has been canceled, don't bother dispatching it.
95d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                if (request.isCanceled()) {
96d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    request.finish("cache-discard-canceled");
97d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    continue;
98d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                }
99d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
100d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                // Attempt to retrieve this item from cache.
101d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                Cache.Entry entry = mCache.get(request.getCacheKey());
102d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                if (entry == null) {
103d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    request.addMarker("cache-miss");
104d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    // Cache miss; send off to the network dispatcher.
105d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    mNetworkQueue.put(request);
106d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    continue;
107d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                }
108d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
109d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                // If it is completely expired, just send it to the network.
110d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                if (entry.isExpired()) {
111d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    request.addMarker("cache-hit-expired");
112d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    request.setCacheEntry(entry);
113d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    mNetworkQueue.put(request);
114d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    continue;
115d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                }
116d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
117d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                // We have a cache hit; parse its data for delivery back to the request.
118d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                request.addMarker("cache-hit");
119d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                Response<?> response = request.parseNetworkResponse(
120e48f4430bfd3030350aa5ba827b449c37e2fadc9Jean-Baptiste Queru                        new NetworkResponse(entry.data, entry.responseHeaders));
121d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                request.addMarker("cache-hit-parsed");
122d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
123d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                if (!entry.refreshNeeded()) {
124d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    // Completely unexpired cache hit. Just deliver the response.
125d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    mDelivery.postResponse(request, response);
126d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                } else {
127d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    // Soft-expired cache hit. We can deliver the cached response,
128d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    // but we need to also send the request to the network for
129d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    // refreshing.
130d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    request.addMarker("cache-hit-refresh-needed");
131d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    request.setCacheEntry(entry);
132d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
133d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    // Mark the response as intermediate.
134d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    response.intermediate = true;
135d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
136d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    // Post the intermediate response back to the user and have
137d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    // the delivery then forward the request along to the network.
138d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    mDelivery.postResponse(request, response, new Runnable() {
139d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                        @Override
140d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                        public void run() {
141d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                            try {
142d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                                mNetworkQueue.put(request);
143d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                            } catch (InterruptedException e) {
144d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                                // Not much we can do about this.
145d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                            }
146d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                        }
147d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    });
148d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                }
149d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru
150d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru            } catch (InterruptedException e) {
151d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                // We may have been interrupted because it was time to quit.
152d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                if (mQuit) {
153d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                    return;
154d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                }
155d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru                continue;
156d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru            }
157d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru        }
158d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru    }
159d56b88ae161057e848e7410d1b9ce5b0b8c427fcJean-Baptiste Queru}
160