| /* |
| * Copyright (C) 2011 The Android Open Source Project |
| * |
| * 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. |
| */ |
| |
| package com.android.providers.downloads; |
| |
| import android.content.res.Resources; |
| import android.util.Log; |
| import android.util.LongSparseArray; |
| |
| import com.android.internal.annotations.GuardedBy; |
| |
| import java.util.ArrayList; |
| import java.util.HashMap; |
| import java.util.Iterator; |
| import java.util.LinkedHashMap; |
| |
| public class DownloadHandler { |
| private static final String TAG = "DownloadHandler"; |
| |
| @GuardedBy("this") |
| private final LinkedHashMap<Long, DownloadInfo> mDownloadsQueue = |
| new LinkedHashMap<Long, DownloadInfo>(); |
| @GuardedBy("this") |
| private final HashMap<Long, DownloadInfo> mDownloadsInProgress = |
| new HashMap<Long, DownloadInfo>(); |
| @GuardedBy("this") |
| private final LongSparseArray<Long> mRemainingMillis = new LongSparseArray<Long>(); |
| |
| private final int mMaxConcurrentDownloadsAllowed = Resources.getSystem().getInteger( |
| com.android.internal.R.integer.config_MaxConcurrentDownloadsAllowed); |
| |
| private static final DownloadHandler sDownloadHandler = new DownloadHandler(); |
| |
| public static DownloadHandler getInstance() { |
| return sDownloadHandler; |
| } |
| |
| public synchronized void enqueueDownload(DownloadInfo info) { |
| if (!mDownloadsQueue.containsKey(info.mId)) { |
| if (Constants.LOGV) { |
| Log.i(TAG, "enqueued download. id: " + info.mId + ", uri: " + info.mUri); |
| } |
| mDownloadsQueue.put(info.mId, info); |
| startDownloadThreadLocked(); |
| } |
| } |
| |
| private void startDownloadThreadLocked() { |
| Iterator<Long> keys = mDownloadsQueue.keySet().iterator(); |
| ArrayList<Long> ids = new ArrayList<Long>(); |
| while (mDownloadsInProgress.size() < mMaxConcurrentDownloadsAllowed && keys.hasNext()) { |
| Long id = keys.next(); |
| DownloadInfo info = mDownloadsQueue.get(id); |
| info.startDownloadThread(); |
| ids.add(id); |
| mDownloadsInProgress.put(id, mDownloadsQueue.get(id)); |
| if (Constants.LOGV) { |
| Log.i(TAG, "started download for : " + id); |
| } |
| } |
| for (Long id : ids) { |
| mDownloadsQueue.remove(id); |
| } |
| } |
| |
| public synchronized boolean hasDownloadInQueue(long id) { |
| return mDownloadsQueue.containsKey(id) || mDownloadsInProgress.containsKey(id); |
| } |
| |
| public synchronized void dequeueDownload(long id) { |
| mDownloadsInProgress.remove(id); |
| mRemainingMillis.remove(id); |
| startDownloadThreadLocked(); |
| if (mDownloadsInProgress.size() == 0 && mDownloadsQueue.size() == 0) { |
| notifyAll(); |
| } |
| } |
| |
| public synchronized void setRemainingMillis(long id, long millis) { |
| mRemainingMillis.put(id, millis); |
| } |
| |
| /** |
| * Return remaining time until given {@link DownloadInfo} finishes, in |
| * milliseconds, or -1 if unknown. |
| */ |
| public synchronized long getRemainingMillis(long id) { |
| return mRemainingMillis.get(id, -1L); |
| } |
| |
| // right now this is only used by tests. but there is no reason why it can't be used |
| // by any module using DownloadManager (TODO add API to DownloadManager.java) |
| public synchronized void waitUntilDownloadsTerminate() throws InterruptedException { |
| if (mDownloadsInProgress.size() == 0 && mDownloadsQueue.size() == 0) { |
| if (Constants.LOGVV) { |
| Log.i(TAG, "nothing to wait on"); |
| } |
| return; |
| } |
| if (Constants.LOGVV) { |
| for (DownloadInfo info : mDownloadsInProgress.values()) { |
| Log.i(TAG, "** progress: " + info.mId + ", " + info.mUri); |
| } |
| for (DownloadInfo info : mDownloadsQueue.values()) { |
| Log.i(TAG, "** in Q: " + info.mId + ", " + info.mUri); |
| } |
| } |
| if (Constants.LOGVV) { |
| Log.i(TAG, "waiting for 5 sec"); |
| } |
| // wait upto 5 sec |
| wait(5 * 1000); |
| } |
| } |