Android/frameworks_base
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add clear-association-memory-cache shell command

Browse Source 
Test: atest CtsCompanionDeviceManagerCoreTestCases
Test: atest CtsCompanionDeviceManagerUiAutomationTestCases
Change-Id: I8643eb318ed4e4bdd0bafe80eca02141c0f10aa4
This commit is contained in:
Nikolas Havrikov
2022-01-14 13:29:26 +01:00
parent e074a73ddc
commit d7f3452d9d
3 changed files with 58 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
services/companion/java/com/android/server/companion/AssociationStoreImpl.java
View File

@@ -37,6 +37,7 @@ import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.StringJoiner;
/**
* Implementation of the {@link AssociationStore}, with addition of the methods for modification.
@@ -58,33 +59,15 @@ class AssociationStoreImpl implements AssociationStore {
private final Object mLock = new Object();
@GuardedBy("mLock")
private final Map<Integer, AssociationInfo> mIdMap;
private final Map<Integer, AssociationInfo> mIdMap = new HashMap<>();
@GuardedBy("mLock")
private final Map<MacAddress, Set<Integer>> mAddressMap;
private final Map<MacAddress, Set<Integer>> mAddressMap = new HashMap<>();
@GuardedBy("mLock")
private final SparseArray<List<AssociationInfo>> mCachedPerUser = new SparseArray<>();
@GuardedBy("mListeners")
private final Set<OnChangeListener> mListeners = new LinkedHashSet<>();
AssociationStoreImpl(Collection<AssociationInfo> associations) {
synchronized (mLock) {
final int size = associations.size();
mIdMap = new HashMap<>(size);
mAddressMap = new HashMap<>(size);
for (AssociationInfo association : associations) {
final int id = association.getId();
mIdMap.put(id, association);
final MacAddress address = association.getDeviceMacAddress();
if (address != null) {
mAddressMap.computeIfAbsent(address, it -> new HashSet<>()).add(id);
}
}
}
}
void addAssociation(@NonNull AssociationInfo association) {
final int id = association.getId();
@@ -301,4 +284,38 @@ class AssociationStoreImpl implements AssociationStore {
}
}
}
void setAssociations(Collection<AssociationInfo> allAssociations) {
if (DEBUG) {
Log.i(TAG, "setAssociations() n=" + allAssociations.size());
final StringJoiner stringJoiner = new StringJoiner(", ");
allAssociations.forEach(assoc -> stringJoiner.add(assoc.toShortString()));
Log.v(TAG, " associations=" + stringJoiner);
}
synchronized (mLock) {
setAssociationsLocked(allAssociations);
}
}
@GuardedBy("mLock")
private void setAssociationsLocked(Collection<AssociationInfo> associations) {
clearLocked();
for (AssociationInfo association : associations) {
final int id = association.getId();
mIdMap.put(id, association);
final MacAddress address = association.getDeviceMacAddress();
if (address != null) {
mAddressMap.computeIfAbsent(address, it -> new HashSet<>()).add(id);
}
}
}
@GuardedBy("mLock")
private void clearLocked() {
mIdMap.clear();
mAddressMap.clear();
mCachedPerUser.clear();
}
}

24
services/companion/java/com/android/server/companion/CompanionDeviceManagerService.java
View File

@@ -159,7 +159,7 @@ public class CompanionDeviceManagerService extends SystemService
// Persistent data store for all Associations.
private PersistentDataStore mPersistentStore;
private AssociationStoreImpl mAssociationStore;
private final AssociationStoreImpl mAssociationStore = new AssociationStoreImpl();
private AssociationRequestsProcessor mAssociationRequestsProcessor;
private PowerWhitelistManager mPowerWhitelistManager;
@@ -219,16 +219,8 @@ public class CompanionDeviceManagerService extends SystemService
@Override
public void onStart() {
mPersistentStore = new PersistentDataStore();
final Set<AssociationInfo> allAssociations = new ArraySet<>();
synchronized (mPreviouslyUsedIds) {
// The data is stored in DE directories, so we can read the data for all users now
// (which would not be possible if the data was stored to CE directories).
mPersistentStore.readStateForUsers(
mUserManager.getAliveUsers(), allAssociations, mPreviouslyUsedIds);
}
mAssociationStore = new AssociationStoreImpl(allAssociations);
loadAssociationsFromDisk();
mAssociationStore.registerListener(this);
mCompanionDevicePresenceController = new CompanionDevicePresenceController(this);
@@ -239,6 +231,18 @@ public class CompanionDeviceManagerService extends SystemService
publishBinderService(Context.COMPANION_DEVICE_SERVICE, impl);
}
void loadAssociationsFromDisk() {
final Set<AssociationInfo> allAssociations = new ArraySet<>();
synchronized (mPreviouslyUsedIds) {
// The data is stored in DE directories, so we can read the data for all users now
// (which would not be possible if the data was stored to CE directories).
mPersistentStore.readStateForUsers(
mUserManager.getAliveUsers(), allAssociations, mPreviouslyUsedIds);
}
mAssociationStore.setAssociations(allAssociations);
}
@Override
public void onBootPhase(int phase) {
if (phase == SystemService.PHASE_SYSTEM_SERVICES_READY) {

8
services/companion/java/com/android/server/companion/CompanionDeviceShellCommand.java
View File

@@ -82,7 +82,10 @@ class CompanionDeviceShellCommand extends android.os.ShellCommand {
mService.onDeviceDisconnected(getNextArgRequired());
}
break;
case "clear-association-memory-cache": {
mService.loadAssociationsFromDisk();
}
break;
default:
return handleDefaultCommands(cmd);
}
@@ -110,5 +113,8 @@ class CompanionDeviceShellCommand extends android.os.ShellCommand {
pw.println(" Create a new Association.");
pw.println(" disassociate USER_ID PACKAGE MAC_ADDRESS");
pw.println(" Remove an existing Association.");
pw.println(" clear-association-memory-cache");
pw.println(" Clear the in-memory association cache and reload all association "
+ "information from persistent storage. USE FOR DEBUGGING PURPOSES ONLY.");
}
}