@@ -564,17 +564,34 @@ private void updateCassandraConsistencyLevel(EntityStore entityStore, String rea
564564 updateCassandraConsistencyLevel (shorthandKeyFinder , "/[KPSRoot]name=Key Property Stores/[KPSPackage]name=OAuth/[KPSDataSourceGroup]name=DataSources/[KPSCassandraDataSource]name=Cassandra Storage" ,
565565 "readConsistencyLevel" , readConsistencyLevel , "writeConsistencyLevel" , writeConsistencyLevel );
566566 // Update Quota table consistency level
567- updateCassandraConsistencyLevel (shorthandKeyFinder , "/[KPSRoot ]name=Key Property Stores/[KPSPackage]name=OAuth/[KPSDataSourceGroup]name=DataSources/[KPSCassandraDataSource]name=Cassandra Storage " ,
567+ updateCassandraConsistencyLevel (shorthandKeyFinder , "/[PortalConfiguration ]name=Portal Config " ,
568568 "quotaReadConsistency" , readConsistencyLevel , "quotaWriteConsistency" , writeConsistencyLevel );
569+ //Update throttling consistency level
570+ updateCassandraConsistencyLevel (shorthandKeyFinder , "/[CassandraSettings]name=Cassandra Settings" ,
571+ "throttlingReadConsistencyLevel" , readConsistencyLevel , "throttlingWriteConsistencyLevel" , writeConsistencyLevel );
572+
573+ //Update access token consistency level
574+ updateCassandraConsistencyLevel (shorthandKeyFinder , "/[OAuth2StoresGroup]name=OAuth2 Stores/[AccessTokenStoreGroup]name=Access Token Stores/[AccessTokenPersist]**" ,
575+ "readConsistencyLevel" , readConsistencyLevel , "writeConsistencyLevel" , writeConsistencyLevel );
576+
577+ //Update auth code consistency level
578+ updateCassandraConsistencyLevel (shorthandKeyFinder , "/[OAuth2StoresGroup]name=OAuth2 Stores/[AuthzCodeStoreGroup]name=Authorization Code Stores/[AuthzCodePersist]**" ,
579+ "readConsistencyLevel" , readConsistencyLevel , "writeConsistencyLevel" , writeConsistencyLevel );
580+ //update client access token consistency level
581+ updateCassandraConsistencyLevel (shorthandKeyFinder , "/[OAuth2StoresGroup]name=OAuth2 Stores/[ClientAccessTokenStoreGroup]name=Client Access Token Stores/[ClientAccessTokenPersist]**" ,
582+ "readConsistencyLevel" , readConsistencyLevel , "writeConsistencyLevel" , writeConsistencyLevel );
583+
569584 }
570585
571586 private void updateCassandraConsistencyLevel (ShorthandKeyFinder shorthandKeyFinder , String shorthandKey , String readConsistencyLevelFieldName , String readConsistencyLevel , String writeConsistencyLevelFieldName , String writeConsistencyLevel ) {
572587 List <Entity > kpsEntities = shorthandKeyFinder .getEntities (shorthandKey );
573588 if (kpsEntities != null ) {
574- Trace .info ("Total number of KPS Store: " + kpsEntities .size ());
589+ Trace .info ("Total number of KPS Store: " + kpsEntities .size () + "in entity : " + shorthandKey );
575590 for (Entity entity : kpsEntities ) {
576591 entity .setStringField (readConsistencyLevelFieldName , readConsistencyLevel );
577592 entity .setStringField (writeConsistencyLevelFieldName , writeConsistencyLevel );
593+ shorthandKeyFinder .getEntityStore ().updateEntity (entity );
594+
578595 }
579596 }
580597 }
0 commit comments