am ef073f40: Use PrioritizedSerialExecutor instead of ReentrantReadWriteLock.

* commit 'ef073f402407b19f5be90ddf68beb874945e82be':
  Use PrioritizedSerialExecutor instead of ReentrantReadWriteLock.
main
Yuichiro Hanada 2013-09-10 03:12:50 -07:00 committed by Android Git Automerger
commit 4754ac4c65
8 changed files with 423 additions and 248 deletions

View File

@ -24,14 +24,14 @@ import com.android.inputmethod.annotations.UsedForTesting;
import com.android.inputmethod.keyboard.ProximityInfo; import com.android.inputmethod.keyboard.ProximityInfo;
import com.android.inputmethod.latin.SuggestedWords.SuggestedWordInfo; import com.android.inputmethod.latin.SuggestedWords.SuggestedWordInfo;
import com.android.inputmethod.latin.personalization.DynamicPersonalizationDictionaryWriter; import com.android.inputmethod.latin.personalization.DynamicPersonalizationDictionaryWriter;
import com.android.inputmethod.latin.personalization.DynamicPredictionDictionaryBase; import com.android.inputmethod.latin.utils.AsyncResultHolder;
import com.android.inputmethod.latin.utils.CollectionUtils; import com.android.inputmethod.latin.utils.CollectionUtils;
import com.android.inputmethod.latin.utils.PrioritizedSerialExecutor;
import java.io.File; import java.io.File;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.HashMap; import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicReference; import java.util.concurrent.atomic.AtomicReference;
import java.util.concurrent.locks.ReentrantReadWriteLock;
/** /**
* Abstract base class for an expandable dictionary that can be created and updated dynamically * Abstract base class for an expandable dictionary that can be created and updated dynamically
@ -53,19 +53,23 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
/** Whether to call binary dictionary dynamically updating methods. */ /** Whether to call binary dictionary dynamically updating methods. */
private static boolean ENABLE_BINARY_DICTIONARY_DYNAMIC_UPDATE = false; private static boolean ENABLE_BINARY_DICTIONARY_DYNAMIC_UPDATE = false;
private static final int TIMEOUT_FOR_READ_OPS_IN_MILLISECONDS = 100;
/** /**
* The maximum length of a word in this dictionary. * The maximum length of a word in this dictionary.
*/ */
protected static final int MAX_WORD_LENGTH = Constants.DICTIONARY_MAX_WORD_LENGTH; protected static final int MAX_WORD_LENGTH = Constants.DICTIONARY_MAX_WORD_LENGTH;
/** /**
* A static map of locks, each of which controls access to a single binary dictionary file. They * A static map of time recorders, each of which records the time of accesses to a single binary
* ensure that only one instance can update the same dictionary at the same time. The key for * dictionary file. The key for this map is the filename and the value is the shared dictionary
* this map is the filename and the value is the shared dictionary controller associated with * time recorder associated with that filename.
* that filename.
*/ */
private static final HashMap<String, DictionaryController> sSharedDictionaryControllers = private static volatile ConcurrentHashMap<String, DictionaryTimeRecorder>
CollectionUtils.newHashMap(); sFilenameDictionaryTimeRecorderMap = CollectionUtils.newConcurrentHashMap();
private static volatile ConcurrentHashMap<String, PrioritizedSerialExecutor>
sFilenameExecutorMap = CollectionUtils.newConcurrentHashMap();
/** The application context. */ /** The application context. */
protected final Context mContext; protected final Context mContext;
@ -83,7 +87,7 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
/** /**
* The name of this dictionary, used as the filename for storing the binary dictionary. Multiple * The name of this dictionary, used as the filename for storing the binary dictionary. Multiple
* dictionary instances with the same filename is supported, with access controlled by * dictionary instances with the same filename is supported, with access controlled by
* DictionaryController. * DictionaryTimeRecorder.
*/ */
private final String mFilename; private final String mFilename;
@ -91,18 +95,19 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
private final boolean mIsUpdatable; private final boolean mIsUpdatable;
// TODO: remove, once dynamic operations is serialized // TODO: remove, once dynamic operations is serialized
/** Controls access to the shared binary dictionary file across multiple instances. */ /** Records access to the shared binary dictionary file across multiple instances. */
private final DictionaryController mSharedDictionaryController; private final DictionaryTimeRecorder mFilenameDictionaryTimeRecorder;
// TODO: remove, once dynamic operations is serialized // TODO: remove, once dynamic operations is serialized
/** Controls access to the local binary dictionary for this instance. */ /** Records access to the local binary dictionary for this instance. */
private final DictionaryController mLocalDictionaryController = new DictionaryController(); private final DictionaryTimeRecorder mPerInstanceDictionaryTimeRecorder =
new DictionaryTimeRecorder();
/* A extension for a binary dictionary file. */ /* A extension for a binary dictionary file. */
public static final String DICT_FILE_EXTENSION = ".dict"; public static final String DICT_FILE_EXTENSION = ".dict";
private final AtomicReference<AsyncWriteBinaryDictionaryTask> mWaitingTask = private final AtomicReference<Runnable> mUnfinishedFlushingTask =
new AtomicReference<AsyncWriteBinaryDictionaryTask>(); new AtomicReference<Runnable>();
/** /**
* Abstract method for loading the unigrams and bigrams of a given dictionary in a background * Abstract method for loading the unigrams and bigrams of a given dictionary in a background
@ -118,16 +123,32 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
protected abstract boolean hasContentChanged(); protected abstract boolean hasContentChanged();
/** /**
* Gets the shared dictionary controller for the given filename. * Gets the dictionary time recorder for the given filename.
*/ */
private static synchronized DictionaryController getSharedDictionaryController( private static DictionaryTimeRecorder getDictionaryTimeRecorder(
String filename) { String filename) {
DictionaryController controller = sSharedDictionaryControllers.get(filename); DictionaryTimeRecorder recorder = sFilenameDictionaryTimeRecorderMap.get(filename);
if (controller == null) { if (recorder == null) {
controller = new DictionaryController(); synchronized(sFilenameDictionaryTimeRecorderMap) {
sSharedDictionaryControllers.put(filename, controller); recorder = new DictionaryTimeRecorder();
sFilenameDictionaryTimeRecorderMap.put(filename, recorder);
}
} }
return controller; return recorder;
}
/**
* Gets the executor for the given filename.
*/
private static PrioritizedSerialExecutor getExecutor(final String filename) {
PrioritizedSerialExecutor executor = sFilenameExecutorMap.get(filename);
if (executor == null) {
synchronized(sFilenameExecutorMap) {
executor = new PrioritizedSerialExecutor();
sFilenameExecutorMap.put(filename, executor);
}
}
return executor;
} }
private static AbstractDictionaryWriter getDictionaryWriter(final Context context, private static AbstractDictionaryWriter getDictionaryWriter(final Context context,
@ -156,7 +177,7 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
mContext = context; mContext = context;
mIsUpdatable = isUpdatable; mIsUpdatable = isUpdatable;
mBinaryDictionary = null; mBinaryDictionary = null;
mSharedDictionaryController = getSharedDictionaryController(filename); mFilenameDictionaryTimeRecorder = getDictionaryTimeRecorder(filename);
// Currently, only dynamic personalization dictionary is updatable. // Currently, only dynamic personalization dictionary is updatable.
mDictionaryWriter = getDictionaryWriter(context, dictType, isUpdatable); mDictionaryWriter = getDictionaryWriter(context, dictType, isUpdatable);
} }
@ -170,35 +191,38 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
*/ */
@Override @Override
public void close() { public void close() {
closeBinaryDictionary(); getExecutor(mFilename).execute(new Runnable() {
mLocalDictionaryController.writeLock().lock(); @Override
try { public void run() {
mDictionaryWriter.close(); if (mBinaryDictionary!= null) {
} finally { mBinaryDictionary.close();
mLocalDictionaryController.writeLock().unlock(); mBinaryDictionary = null;
} }
mDictionaryWriter.close();
}
});
} }
protected void closeBinaryDictionary() { protected void closeBinaryDictionary() {
// Ensure that no other threads are accessing the local binary dictionary. // Ensure that no other threads are accessing the local binary dictionary.
mLocalDictionaryController.writeLock().lock(); getExecutor(mFilename).execute(new Runnable() {
try { @Override
if (mBinaryDictionary != null) { public void run() {
mBinaryDictionary.close(); if (mBinaryDictionary != null) {
mBinaryDictionary = null; mBinaryDictionary.close();
mBinaryDictionary = null;
}
} }
} finally { });
mLocalDictionaryController.writeLock().unlock();
}
} }
protected void clear() { protected void clear() {
mLocalDictionaryController.writeLock().lock(); getExecutor(mFilename).execute(new Runnable() {
try { @Override
mDictionaryWriter.clear(); public void run() {
} finally { mDictionaryWriter.clear();
mLocalDictionaryController.writeLock().unlock(); }
} });
} }
/** /**
@ -227,18 +251,17 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
Log.w(TAG, "addWordDynamically is called for non-updatable dictionary: " + mFilename); Log.w(TAG, "addWordDynamically is called for non-updatable dictionary: " + mFilename);
return; return;
} }
// TODO: Use a queue to reflect what needs to be reflected.
if (mLocalDictionaryController.writeLock().tryLock()) { getExecutor(mFilename).execute(new Runnable() {
try { @Override
public void run() {
if (ENABLE_BINARY_DICTIONARY_DYNAMIC_UPDATE) { if (ENABLE_BINARY_DICTIONARY_DYNAMIC_UPDATE) {
mBinaryDictionary.addUnigramWord(word, frequency); mBinaryDictionary.addUnigramWord(word, frequency);
} }
// TODO: Remove. // TODO: Remove.
mDictionaryWriter.addUnigramWord(word, shortcutTarget, frequency, isNotAWord); mDictionaryWriter.addUnigramWord(word, shortcutTarget, frequency, isNotAWord);
} finally {
mLocalDictionaryController.writeLock().unlock();
} }
} });
} }
/** /**
@ -251,19 +274,18 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
+ mFilename); + mFilename);
return; return;
} }
// TODO: Use a queue to reflect what needs to be reflected.
if (mLocalDictionaryController.writeLock().tryLock()) { getExecutor(mFilename).execute(new Runnable() {
try { @Override
public void run() {
if (ENABLE_BINARY_DICTIONARY_DYNAMIC_UPDATE) { if (ENABLE_BINARY_DICTIONARY_DYNAMIC_UPDATE) {
mBinaryDictionary.addBigramWords(word0, word1, frequency); mBinaryDictionary.addBigramWords(word0, word1, frequency);
} }
// TODO: Remove. // TODO: Remove.
mDictionaryWriter.addBigramWords(word0, word1, frequency, isValid, mDictionaryWriter.addBigramWords(word0, word1, frequency, isValid,
0 /* lastTouchedTime */); 0 /* lastTouchedTime */);
} finally {
mLocalDictionaryController.writeLock().unlock();
} }
} });
} }
/** /**
@ -275,28 +297,30 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
+ mFilename); + mFilename);
return; return;
} }
// TODO: Use a queue to reflect what needs to be reflected.
if (mLocalDictionaryController.writeLock().tryLock()) { getExecutor(mFilename).execute(new Runnable() {
try { @Override
public void run() {
if (ENABLE_BINARY_DICTIONARY_DYNAMIC_UPDATE) { if (ENABLE_BINARY_DICTIONARY_DYNAMIC_UPDATE) {
mBinaryDictionary.removeBigramWords(word0, word1); mBinaryDictionary.removeBigramWords(word0, word1);
} }
// TODO: Remove. // TODO: Remove.
mDictionaryWriter.removeBigramWords(word0, word1); mDictionaryWriter.removeBigramWords(word0, word1);
} finally {
mLocalDictionaryController.writeLock().unlock();
} }
} });
} }
@Override @Override
public ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer composer, public ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer composer,
final String prevWord, final ProximityInfo proximityInfo, final String prevWord, final ProximityInfo proximityInfo,
final boolean blockOffensiveWords, final int[] additionalFeaturesOptions) { final boolean blockOffensiveWords, final int[] additionalFeaturesOptions) {
asyncReloadDictionaryIfRequired(); reloadDictionaryIfRequired();
// Write lock because getSuggestions in native updates session status. final ArrayList<SuggestedWordInfo> suggestions = CollectionUtils.newArrayList();
if (mLocalDictionaryController.writeLock().tryLock()) { final AsyncResultHolder<ArrayList<SuggestedWordInfo>> holder =
try { new AsyncResultHolder<ArrayList<SuggestedWordInfo>>();
getExecutor(mFilename).executePrioritized(new Runnable() {
@Override
public void run() {
final ArrayList<SuggestedWordInfo> inMemDictSuggestion = final ArrayList<SuggestedWordInfo> inMemDictSuggestion =
mDictionaryWriter.getSuggestions(composer, prevWord, proximityInfo, mDictionaryWriter.getSuggestions(composer, prevWord, proximityInfo,
blockOffensiveWords, additionalFeaturesOptions); blockOffensiveWords, additionalFeaturesOptions);
@ -306,38 +330,37 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
mBinaryDictionary.getSuggestions(composer, prevWord, proximityInfo, mBinaryDictionary.getSuggestions(composer, prevWord, proximityInfo,
blockOffensiveWords, additionalFeaturesOptions); blockOffensiveWords, additionalFeaturesOptions);
if (inMemDictSuggestion == null) { if (inMemDictSuggestion == null) {
return binarySuggestion; holder.set(binarySuggestion);
} else if (binarySuggestion == null) { } else if (binarySuggestion == null) {
return inMemDictSuggestion; holder.set(inMemDictSuggestion);
} else { } else {
binarySuggestion.addAll(inMemDictSuggestion); binarySuggestion.addAll(inMemDictSuggestion);
return binarySuggestion; holder.set(binarySuggestion);
} }
} else { } else {
return inMemDictSuggestion; holder.set(inMemDictSuggestion);
} }
} finally {
mLocalDictionaryController.writeLock().unlock();
} }
} });
return null;
return holder.get(null, TIMEOUT_FOR_READ_OPS_IN_MILLISECONDS);
} }
@Override @Override
public boolean isValidWord(final String word) { public boolean isValidWord(final String word) {
asyncReloadDictionaryIfRequired(); reloadDictionaryIfRequired();
return isValidWordInner(word); return isValidWordInner(word);
} }
protected boolean isValidWordInner(final String word) { protected boolean isValidWordInner(final String word) {
if (mLocalDictionaryController.readLock().tryLock()) { final AsyncResultHolder<Boolean> holder = new AsyncResultHolder<Boolean>();
try { getExecutor(mFilename).executePrioritized(new Runnable() {
return isValidWordLocked(word); @Override
} finally { public void run() {
mLocalDictionaryController.readLock().unlock(); holder.set(isValidWordLocked(word));
} }
} });
return false; return holder.get(false, TIMEOUT_FOR_READ_OPS_IN_MILLISECONDS);
} }
protected boolean isValidWordLocked(final String word) { protected boolean isValidWordLocked(final String word) {
@ -355,8 +378,8 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
* dictionary exists, this method will generate one. * dictionary exists, this method will generate one.
*/ */
protected void loadDictionary() { protected void loadDictionary() {
mLocalDictionaryController.mLastUpdateRequestTime = SystemClock.uptimeMillis(); mPerInstanceDictionaryTimeRecorder.mLastUpdateRequestTime = SystemClock.uptimeMillis();
asyncReloadDictionaryIfRequired(); reloadDictionaryIfRequired();
} }
/** /**
@ -366,8 +389,8 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
private void loadBinaryDictionary() { private void loadBinaryDictionary() {
if (DEBUG) { if (DEBUG) {
Log.d(TAG, "Loading binary dictionary: " + mFilename + " request=" Log.d(TAG, "Loading binary dictionary: " + mFilename + " request="
+ mSharedDictionaryController.mLastUpdateRequestTime + " update=" + mFilenameDictionaryTimeRecorder.mLastUpdateRequestTime + " update="
+ mSharedDictionaryController.mLastUpdateTime); + mFilenameDictionaryTimeRecorder.mLastUpdateTime);
} }
final File file = new File(mContext.getFilesDir(), mFilename); final File file = new File(mContext.getFilesDir(), mFilename);
@ -378,20 +401,18 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
final BinaryDictionary newBinaryDictionary = new BinaryDictionary(filename, 0, length, final BinaryDictionary newBinaryDictionary = new BinaryDictionary(filename, 0, length,
true /* useFullEditDistance */, null, mDictType, mIsUpdatable); true /* useFullEditDistance */, null, mDictType, mIsUpdatable);
if (mBinaryDictionary != null) { // Ensure all threads accessing the current dictionary have finished before swapping in
// Ensure all threads accessing the current dictionary have finished before swapping in // the new one.
// the new one. final BinaryDictionary oldBinaryDictionary = mBinaryDictionary;
final BinaryDictionary oldBinaryDictionary = mBinaryDictionary; getExecutor(mFilename).executePrioritized(new Runnable() {
mLocalDictionaryController.writeLock().lock(); @Override
try { public void run() {
mBinaryDictionary = newBinaryDictionary; mBinaryDictionary = newBinaryDictionary;
} finally { if (oldBinaryDictionary != null) {
mLocalDictionaryController.writeLock().unlock(); oldBinaryDictionary.close();
}
} }
oldBinaryDictionary.close(); });
} else {
mBinaryDictionary = newBinaryDictionary;
}
} }
/** /**
@ -406,8 +427,8 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
private void writeBinaryDictionary() { private void writeBinaryDictionary() {
if (DEBUG) { if (DEBUG) {
Log.d(TAG, "Generating binary dictionary: " + mFilename + " request=" Log.d(TAG, "Generating binary dictionary: " + mFilename + " request="
+ mSharedDictionaryController.mLastUpdateRequestTime + " update=" + mFilenameDictionaryTimeRecorder.mLastUpdateRequestTime + " update="
+ mSharedDictionaryController.mLastUpdateTime); + mFilenameDictionaryTimeRecorder.mLastUpdateTime);
} }
if (needsToReloadBeforeWriting()) { if (needsToReloadBeforeWriting()) {
mDictionaryWriter.clear(); mDictionaryWriter.clear();
@ -425,54 +446,42 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
*/ */
protected void setRequiresReload(final boolean requiresRebuild) { protected void setRequiresReload(final boolean requiresRebuild) {
final long time = SystemClock.uptimeMillis(); final long time = SystemClock.uptimeMillis();
mLocalDictionaryController.mLastUpdateRequestTime = time; mPerInstanceDictionaryTimeRecorder.mLastUpdateRequestTime = time;
mSharedDictionaryController.mLastUpdateRequestTime = time; mFilenameDictionaryTimeRecorder.mLastUpdateRequestTime = time;
if (DEBUG) { if (DEBUG) {
Log.d(TAG, "Reload request: " + mFilename + ": request=" + time + " update=" Log.d(TAG, "Reload request: " + mFilename + ": request=" + time + " update="
+ mSharedDictionaryController.mLastUpdateTime); + mFilenameDictionaryTimeRecorder.mLastUpdateTime);
} }
} }
/**
* Reloads the dictionary if required. Reload will occur asynchronously in a separate thread.
*/
public void asyncReloadDictionaryIfRequired() {
if (!isReloadRequired()) return;
if (DEBUG) {
Log.d(TAG, "Starting AsyncReloadDictionaryTask: " + mFilename);
}
new AsyncReloadDictionaryTask().start();
}
/** /**
* Reloads the dictionary if required. * Reloads the dictionary if required.
*/ */
public final void syncReloadDictionaryIfRequired() { public final void reloadDictionaryIfRequired() {
if (!isReloadRequired()) return; if (!isReloadRequired()) return;
syncReloadDictionaryInternal(); reloadDictionary();
} }
/** /**
* Returns whether a dictionary reload is required. * Returns whether a dictionary reload is required.
*/ */
private boolean isReloadRequired() { private boolean isReloadRequired() {
return mBinaryDictionary == null || mLocalDictionaryController.isOutOfDate(); return mBinaryDictionary == null || mPerInstanceDictionaryTimeRecorder.isOutOfDate();
} }
/** /**
* Reloads the dictionary. Access is controlled on a per dictionary file basis and supports * Reloads the dictionary. Access is controlled on a per dictionary file basis and supports
* concurrent calls from multiple instances that share the same dictionary file. * concurrent calls from multiple instances that share the same dictionary file.
*/ */
private final void syncReloadDictionaryInternal() { private final void reloadDictionary() {
// Ensure that only one thread attempts to read or write to the shared binary dictionary // Ensure that only one thread attempts to read or write to the shared binary dictionary
// file at the same time. // file at the same time.
mSharedDictionaryController.writeLock().lock(); getExecutor(mFilename).execute(new Runnable() {
try { @Override
mLocalDictionaryController.writeLock().lock(); public void run() {
try {
final long time = SystemClock.uptimeMillis(); final long time = SystemClock.uptimeMillis();
final boolean dictionaryFileExists = dictionaryFileExists(); final boolean dictionaryFileExists = dictionaryFileExists();
if (mSharedDictionaryController.isOutOfDate() || !dictionaryFileExists) { if (mFilenameDictionaryTimeRecorder.isOutOfDate() || !dictionaryFileExists) {
// If the shared dictionary file does not exist or is out of date, the first // If the shared dictionary file does not exist or is out of date, the first
// instance that acquires the lock will generate a new one. // instance that acquires the lock will generate a new one.
if (hasContentChanged() || !dictionaryFileExists) { if (hasContentChanged() || !dictionaryFileExists) {
@ -480,34 +489,31 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
// rebuild the binary dictionary. Empty dictionaries are supported (in the // rebuild the binary dictionary. Empty dictionaries are supported (in the
// case where loadDictionaryAsync() adds nothing) in order to provide a // case where loadDictionaryAsync() adds nothing) in order to provide a
// uniform framework. // uniform framework.
mSharedDictionaryController.mLastUpdateTime = time; mFilenameDictionaryTimeRecorder.mLastUpdateTime = time;
writeBinaryDictionary(); writeBinaryDictionary();
loadBinaryDictionary(); loadBinaryDictionary();
} else { } else {
// If not, the reload request was unnecessary so revert // If not, the reload request was unnecessary so revert
// LastUpdateRequestTime to LastUpdateTime. // LastUpdateRequestTime to LastUpdateTime.
mSharedDictionaryController.mLastUpdateRequestTime = mFilenameDictionaryTimeRecorder.mLastUpdateRequestTime =
mSharedDictionaryController.mLastUpdateTime; mFilenameDictionaryTimeRecorder.mLastUpdateTime;
} }
} else if (mBinaryDictionary == null || mLocalDictionaryController.mLastUpdateTime } else if (mBinaryDictionary == null ||
< mSharedDictionaryController.mLastUpdateTime) { mPerInstanceDictionaryTimeRecorder.mLastUpdateTime
< mFilenameDictionaryTimeRecorder.mLastUpdateTime) {
// Otherwise, if the local dictionary is older than the shared dictionary, load // Otherwise, if the local dictionary is older than the shared dictionary, load
// the shared dictionary. // the shared dictionary.
loadBinaryDictionary(); loadBinaryDictionary();
} }
if (mBinaryDictionary != null && !mBinaryDictionary.isValidDictionary()) { if (mBinaryDictionary != null && !mBinaryDictionary.isValidDictionary()) {
// Binary dictionary is not valid. Regenerate the dictionary file. // Binary dictionary is not valid. Regenerate the dictionary file.
mSharedDictionaryController.mLastUpdateTime = time; mFilenameDictionaryTimeRecorder.mLastUpdateTime = time;
writeBinaryDictionary(); writeBinaryDictionary();
loadBinaryDictionary(); loadBinaryDictionary();
} }
mLocalDictionaryController.mLastUpdateTime = time; mPerInstanceDictionaryTimeRecorder.mLastUpdateTime = time;
} finally {
mLocalDictionaryController.writeLock().unlock();
} }
} finally { });
mSharedDictionaryController.writeLock().unlock();
}
} }
// TODO: cache the file's existence so that we avoid doing a disk access each time. // TODO: cache the file's existence so that we avoid doing a disk access each time.
@ -516,84 +522,37 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
return file.exists(); return file.exists();
} }
/**
* Thread class for asynchronously reloading and rewriting the binary dictionary.
*/
private class AsyncReloadDictionaryTask extends Thread {
@Override
public void run() {
syncReloadDictionaryInternal();
}
}
/** /**
* Load the dictionary to memory. * Load the dictionary to memory.
*/ */
protected void asyncLoadDictionaryToMemory() { protected void asyncLoadDictionaryToMemory() {
new AsyncLoadDictionaryToMemoryTask().start(); getExecutor(mFilename).executePrioritized(new Runnable() {
} @Override
public void run() {
/** loadDictionaryAsync();
* Thread class for asynchronously loading dictionary to memory.
*/
private class AsyncLoadDictionaryToMemoryTask extends Thread {
@Override
public void run() {
mSharedDictionaryController.readLock().lock();
try {
mLocalDictionaryController.writeLock().lock();
try {
loadDictionaryAsync();
} finally {
mLocalDictionaryController.writeLock().unlock();
}
} finally {
mSharedDictionaryController.readLock().unlock();
} }
} });
} }
/** /**
* Generate binary dictionary using DictionaryWriter. * Generate binary dictionary using DictionaryWriter.
*/ */
protected void asyncWriteBinaryDictionary() { protected void asyncWriteBinaryDictionary() {
final AsyncWriteBinaryDictionaryTask newTask = new AsyncWriteBinaryDictionaryTask(); final Runnable newTask = new Runnable() {
newTask.start(); @Override
final AsyncWriteBinaryDictionaryTask oldTask = mWaitingTask.getAndSet(newTask); public void run() {
if (oldTask != null) { writeBinaryDictionary();
oldTask.interrupt();
}
}
/**
* Thread class for asynchronously writing the binary dictionary.
*/
private class AsyncWriteBinaryDictionaryTask extends Thread {
@Override
public void run() {
mSharedDictionaryController.writeLock().lock();
try {
mLocalDictionaryController.writeLock().lock();
try {
if (isInterrupted()) {
return;
}
writeBinaryDictionary();
} finally {
mLocalDictionaryController.writeLock().unlock();
}
} finally {
mSharedDictionaryController.writeLock().unlock();
} }
} };
final Runnable oldTask = mUnfinishedFlushingTask.getAndSet(newTask);
getExecutor(mFilename).replaceAndExecute(oldTask, newTask);
} }
/** /**
* Lock for controlling access to a given binary dictionary and for tracking whether the * Time recorder for tracking whether the dictionary is out of date.
* dictionary is out of date. Can be shared across multiple dictionary instances that access the * Can be shared across multiple dictionary instances that access the same filename.
* same filename.
*/ */
private static class DictionaryController extends ReentrantReadWriteLock { private static class DictionaryTimeRecorder {
private volatile long mLastUpdateTime = 0; private volatile long mLastUpdateTime = 0;
private volatile long mLastUpdateRequestTime = 0; private volatile long mLastUpdateRequestTime = 0;
@ -608,12 +567,12 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
@UsedForTesting @UsedForTesting
protected void addWordDynamicallyForTests(final String word, final String shortcutTarget, protected void addWordDynamicallyForTests(final String word, final String shortcutTarget,
final int frequency, final boolean isNotAWord) { final int frequency, final boolean isNotAWord) {
mLocalDictionaryController.writeLock().lock(); getExecutor(mFilename).executePrioritized(new Runnable() {
try { @Override
addWordDynamically(word, shortcutTarget, frequency, isNotAWord); public void run() {
} finally { addWordDynamically(word, shortcutTarget, frequency, isNotAWord);
mLocalDictionaryController.writeLock().unlock(); }
} });
} }
/** /**
@ -622,12 +581,12 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
@UsedForTesting @UsedForTesting
protected void addBigramDynamicallyForTests(final String word0, final String word1, protected void addBigramDynamicallyForTests(final String word0, final String word1,
final int frequency, final boolean isValid) { final int frequency, final boolean isValid) {
mLocalDictionaryController.writeLock().lock(); getExecutor(mFilename).executePrioritized(new Runnable() {
try { @Override
addBigramDynamically(word0, word1, frequency, isValid); public void run() {
} finally { addBigramDynamically(word0, word1, frequency, isValid);
mLocalDictionaryController.writeLock().unlock(); }
} });
} }
/** /**
@ -635,42 +594,27 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
*/ */
@UsedForTesting @UsedForTesting
protected void removeBigramDynamicallyForTests(final String word0, final String word1) { protected void removeBigramDynamicallyForTests(final String word0, final String word1) {
mLocalDictionaryController.writeLock().lock(); getExecutor(mFilename).executePrioritized(new Runnable() {
try { @Override
removeBigramDynamically(word0, word1); public void run() {
} finally { removeBigramDynamically(word0, word1);
mLocalDictionaryController.writeLock().unlock(); }
} });
} }
// TODO: Implement native binary methods once the dynamic dictionary implementation is done. // TODO: Implement native binary methods once the dynamic dictionary implementation is done.
@UsedForTesting @UsedForTesting
public boolean isInDictionaryForTests(final String word) { public boolean isInDictionaryForTests(final String word) {
mLocalDictionaryController.writeLock().lock(); final AsyncResultHolder<Boolean> holder = new AsyncResultHolder<Boolean>();
try { getExecutor(mFilename).executePrioritized(new Runnable() {
if (mDictType == Dictionary.TYPE_USER_HISTORY) { @Override
return ((DynamicPersonalizationDictionaryWriter) mDictionaryWriter) public void run() {
.isInDictionaryForTests(word); if (mDictType == Dictionary.TYPE_USER_HISTORY) {
holder.set(((DynamicPersonalizationDictionaryWriter) mDictionaryWriter)
.isInDictionaryForTests(word));
}
} }
} finally { });
mLocalDictionaryController.writeLock().unlock(); return holder.get(false, TIMEOUT_FOR_READ_OPS_IN_MILLISECONDS);
}
return false;
}
// TODO: Remove and use addToPersonalizationPredictionDictionary instead!!!!!!!!!!!!!!!!
@UsedForTesting
public void forceAddWordForTest(
final String word0, final String word1, final boolean isValid) {
mLocalDictionaryController.writeLock().lock();
try {
mDictionaryWriter.addUnigramWord(word1, null /* the "shortcut" parameter is null */,
DynamicPredictionDictionaryBase.FREQUENCY_FOR_TYPED, false /* isNotAWord */);
mDictionaryWriter.addBigramWords(word0, word1,
DynamicPredictionDictionaryBase.FREQUENCY_FOR_TYPED, isValid,
0 /* lastTouchedTime */);
} finally {
mLocalDictionaryController.writeLock().unlock();
}
} }
} }

View File

@ -35,14 +35,14 @@ public final class SynchronouslyLoadedContactsBinaryDictionary extends ContactsB
public synchronized ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer codes, public synchronized ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer codes,
final String prevWordForBigrams, final ProximityInfo proximityInfo, final String prevWordForBigrams, final ProximityInfo proximityInfo,
final boolean blockOffensiveWords, final int[] additionalFeaturesOptions) { final boolean blockOffensiveWords, final int[] additionalFeaturesOptions) {
syncReloadDictionaryIfRequired(); reloadDictionaryIfRequired();
return super.getSuggestions(codes, prevWordForBigrams, proximityInfo, blockOffensiveWords, return super.getSuggestions(codes, prevWordForBigrams, proximityInfo, blockOffensiveWords,
additionalFeaturesOptions); additionalFeaturesOptions);
} }
@Override @Override
public synchronized boolean isValidWord(final String word) { public synchronized boolean isValidWord(final String word) {
syncReloadDictionaryIfRequired(); reloadDictionaryIfRequired();
return isValidWordInner(word); return isValidWordInner(word);
} }

View File

@ -38,14 +38,14 @@ public final class SynchronouslyLoadedUserBinaryDictionary extends UserBinaryDic
public synchronized ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer codes, public synchronized ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer codes,
final String prevWordForBigrams, final ProximityInfo proximityInfo, final String prevWordForBigrams, final ProximityInfo proximityInfo,
final boolean blockOffensiveWords, final int[] additionalFeaturesOptions) { final boolean blockOffensiveWords, final int[] additionalFeaturesOptions) {
syncReloadDictionaryIfRequired(); reloadDictionaryIfRequired();
return super.getSuggestions(codes, prevWordForBigrams, proximityInfo, blockOffensiveWords, return super.getSuggestions(codes, prevWordForBigrams, proximityInfo, blockOffensiveWords,
additionalFeaturesOptions); additionalFeaturesOptions);
} }
@Override @Override
public synchronized boolean isValidWord(final String word) { public synchronized boolean isValidWord(final String word) {
syncReloadDictionaryIfRequired(); reloadDictionaryIfRequired();
return isValidWordInner(word); return isValidWordInner(word);
} }
} }

View File

@ -69,7 +69,7 @@ public abstract class DynamicPredictionDictionaryBase extends ExpandableBinaryDi
mPrefs = sp; mPrefs = sp;
if (mLocale != null && mLocale.length() > 1) { if (mLocale != null && mLocale.length() > 1) {
asyncLoadDictionaryToMemory(); asyncLoadDictionaryToMemory();
asyncReloadDictionaryIfRequired(); reloadDictionaryIfRequired();
} }
} }

View File

@ -52,7 +52,7 @@ public class PersonalizationHelper {
if (DEBUG) { if (DEBUG) {
Log.w(TAG, "Use cached UserHistoryPredictionDictionary for " + locale); Log.w(TAG, "Use cached UserHistoryPredictionDictionary for " + locale);
} }
dict.asyncReloadDictionaryIfRequired(); dict.reloadDictionaryIfRequired();
return dict; return dict;
} }
} }

View File

@ -0,0 +1,126 @@
/*
* Copyright (C) 2013 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.inputmethod.latin.utils;
import java.util.ArrayDeque;
import java.util.Queue;
/**
* An object that executes submitted tasks using a thread.
*/
public class PrioritizedSerialExecutor {
public static final String TAG = PrioritizedSerialExecutor.class.getSimpleName();
private final Object mLock = new Object();
// The default value of capacities of task queues.
private static final int TASK_QUEUE_CAPACITY = 1000;
private final Queue<Runnable> mTasks;
private final Queue<Runnable> mPrioritizedTasks;
// The task which is running now.
private Runnable mActive;
public PrioritizedSerialExecutor() {
mTasks = new ArrayDeque<Runnable>(TASK_QUEUE_CAPACITY);
mPrioritizedTasks = new ArrayDeque<Runnable>(TASK_QUEUE_CAPACITY);
}
/**
* Clears all queued tasks.
*/
public void clearAllTasks() {
synchronized(mLock) {
mTasks.clear();
mPrioritizedTasks.clear();
}
}
/**
* Enqueues the given task into the task queue.
* @param r the enqueued task
*/
public void execute(final Runnable r) {
synchronized(mLock) {
mTasks.offer(r);
if (mActive == null) {
scheduleNext();
}
}
}
/**
* Enqueues the given task into the prioritized task queue.
* @param r the enqueued task
*/
public void executePrioritized(final Runnable r) {
synchronized(mLock) {
mPrioritizedTasks.offer(r);
if (mActive == null) {
scheduleNext();
}
}
}
private boolean fetchNextTasks() {
synchronized(mLock) {
mActive = mPrioritizedTasks.poll();
if (mActive == null) {
mActive = mTasks.poll();
}
return mActive != null;
}
}
private void scheduleNext() {
synchronized(mLock) {
if (!fetchNextTasks()) {
return;
}
new Thread(new Runnable() {
@Override
public void run() {
try {
do {
synchronized(mLock) {
if (mActive != null) {
mActive.run();
}
}
} while (fetchNextTasks());
} finally {
scheduleNext();
}
}
}).start();
}
}
public void remove(final Runnable r) {
synchronized(mLock) {
mTasks.remove(r);
mPrioritizedTasks.remove(r);
}
}
public void replaceAndExecute(final Runnable oldTask, final Runnable newTask) {
synchronized(mLock) {
if (oldTask != null) remove(oldTask);
execute(newTask);
}
}
}

View File

@ -77,7 +77,7 @@ public class UserHistoryDictionaryTests extends AndroidTestCase {
private void addToDict(final UserHistoryPredictionDictionary dict, final List<String> words) { private void addToDict(final UserHistoryPredictionDictionary dict, final List<String> words) {
String prevWord = null; String prevWord = null;
for (String word : words) { for (String word : words) {
dict.forceAddWordForTest(prevWord, word, true); dict.addToPersonalizationPredictionDictionary(prevWord, word, true);
prevWord = word; prevWord = word;
} }
} }

View File

@ -0,0 +1,105 @@
/*
* Copyright (C) 2013 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.inputmethod.latin.utils;
import android.test.AndroidTestCase;
import android.test.suitebuilder.annotation.MediumTest;
import android.util.Log;
import java.util.concurrent.atomic.AtomicInteger;
/**
* Unit tests for PrioritizedSerialExecutor.
* TODO: Add more detailed tests to make use of priorities, etc.
*/
@MediumTest
public class PrioritizedSerialExecutorTests extends AndroidTestCase {
private static final String TAG = PrioritizedSerialExecutorTests.class.getSimpleName();
private static final int NUM_OF_TASKS = 10;
private static final int DELAY_FOR_WAITING_TASKS_MILLISECONDS = 500;
public void testExecute() {
final PrioritizedSerialExecutor executor = new PrioritizedSerialExecutor();
final AtomicInteger v = new AtomicInteger(0);
for (int i = 0; i < NUM_OF_TASKS; ++i) {
executor.execute(new Runnable() {
@Override
public void run() {
v.incrementAndGet();
}
});
}
try {
Thread.sleep(DELAY_FOR_WAITING_TASKS_MILLISECONDS);
} catch (InterruptedException e) {
Log.d(TAG, "Exception while sleeping.", e);
}
assertEquals(NUM_OF_TASKS, v.get());
}
public void testExecutePrioritized() {
final PrioritizedSerialExecutor executor = new PrioritizedSerialExecutor();
final AtomicInteger v = new AtomicInteger(0);
for (int i = 0; i < NUM_OF_TASKS; ++i) {
executor.executePrioritized(new Runnable() {
@Override
public void run() {
v.incrementAndGet();
}
});
}
try {
Thread.sleep(DELAY_FOR_WAITING_TASKS_MILLISECONDS);
} catch (InterruptedException e) {
Log.d(TAG, "Exception while sleeping.", e);
}
assertEquals(NUM_OF_TASKS, v.get());
}
public void testExecuteCombined() {
final PrioritizedSerialExecutor executor = new PrioritizedSerialExecutor();
final AtomicInteger v = new AtomicInteger(0);
for (int i = 0; i < NUM_OF_TASKS; ++i) {
executor.execute(new Runnable() {
@Override
public void run() {
v.incrementAndGet();
}
});
}
for (int i = 0; i < NUM_OF_TASKS; ++i) {
executor.executePrioritized(new Runnable() {
@Override
public void run() {
v.incrementAndGet();
}
});
}
try {
Thread.sleep(DELAY_FOR_WAITING_TASKS_MILLISECONDS);
} catch (InterruptedException e) {
Log.d(TAG, "Exception while sleeping.", e);
}
assertEquals(2 * NUM_OF_TASKS, v.get());
}
}