2727public class UserMetadata {
2828 static final int MAX_ATTRIBUTES = 64 ;
2929 static final int MAX_ATTRIBUTE_SIZE = 1024 ;
30+ static final int MAX_INTERNAL_KEY_SIZE = 8192 ;
3031
3132 private String userId = null ;
3233 private final Map <String , String > attributes = new HashMap <>();
34+ private final Map <String , String > internalKeys = new HashMap <>();
3335
3436 public UserMetadata () {}
3537
@@ -39,7 +41,7 @@ public String getUserId() {
3941 }
4042
4143 public void setUserId (String identifier ) {
42- userId = sanitizeAttribute (identifier );
44+ userId = sanitizeAttribute (identifier , MAX_ATTRIBUTE_SIZE );
4345 }
4446
4547 @ NonNull
@@ -51,18 +53,31 @@ public void setCustomKey(String key, String value) {
5153 setSyncCustomKeys (
5254 new HashMap <String , String >() {
5355 {
54- put (sanitizeKey ( key ), sanitizeAttribute ( value ) );
56+ put (key , value );
5557 }
56- });
58+ }, attributes , MAX_ATTRIBUTE_SIZE );
5759 }
5860
5961 public void setCustomKeys (Map <String , String > keysAndValues ) {
60- setSyncCustomKeys (keysAndValues );
62+ setSyncCustomKeys (keysAndValues , attributes , MAX_ATTRIBUTE_SIZE );
6163 }
6264
63- /** Gatekeeper function for access to attributes */
64- private synchronized void setSyncCustomKeys (Map <String , String > keysAndValues ) {
65- // We want all access to the attributes hashmap to be locked so that there is no way to create
65+ public Map <String , String > getInternalKeys () {
66+ return Collections .unmodifiableMap (internalKeys );
67+ }
68+
69+ public void setInternalKey (String key , String value ) {
70+ setSyncCustomKeys (
71+ new HashMap <String , String >() {
72+ {
73+ put (key , value );
74+ }
75+ }, internalKeys , MAX_INTERNAL_KEY_SIZE );
76+ }
77+
78+ /** Gatekeeper function for access to attributes or internalKeys */
79+ private synchronized void setSyncCustomKeys (Map <String , String > keysAndValues , Map <String , String > keys_map , int maxAttributeSize ) {
80+ // We want all access to the keys_map hashmap to be locked so that there is no way to create
6681 // a race condition and add more than MAX_ATTRIBUTES keys.
6782
6883 // Update any existing keys first, then add any additional keys
@@ -71,42 +86,42 @@ private synchronized void setSyncCustomKeys(Map<String, String> keysAndValues) {
7186
7287 // Split into current and new keys
7388 for (Map .Entry <String , String > entry : keysAndValues .entrySet ()) {
74- String key = sanitizeKey (entry .getKey ());
75- String value = (entry .getValue () == null ) ? "" : sanitizeAttribute (entry .getValue ());
76- if (attributes .containsKey (key )) {
89+ String key = sanitizeKey (entry .getKey (), maxAttributeSize );
90+ String value = (entry .getValue () == null ) ? "" : sanitizeAttribute (entry .getValue (), maxAttributeSize );
91+ if (keys_map .containsKey (key )) {
7792 currentKeys .put (key , value );
7893 } else {
7994 newKeys .put (key , value );
8095 }
8196 }
8297
83- attributes .putAll (currentKeys );
98+ keys_map .putAll (currentKeys );
8499
85100 // Add new keys if there is space
86- if (attributes .size () + newKeys .size () > MAX_ATTRIBUTES ) {
87- int keySlotsLeft = MAX_ATTRIBUTES - attributes .size ();
101+ if (keys_map .size () + newKeys .size () > MAX_ATTRIBUTES ) {
102+ int keySlotsLeft = MAX_ATTRIBUTES - keys_map .size ();
88103 Logger .getLogger ()
89104 .v ("Exceeded maximum number of custom attributes (" + MAX_ATTRIBUTES + ")." );
90105 List <String > newKeyList = new ArrayList <>(newKeys .keySet ());
91106 newKeys .keySet ().retainAll (newKeyList .subList (0 , keySlotsLeft ));
92107 }
93- attributes .putAll (newKeys );
108+ keys_map .putAll (newKeys );
94109 }
95110
96111 /** Checks that the key is not null then sanitizes it. */
97- private static String sanitizeKey (String key ) {
112+ private static String sanitizeKey (String key , int maxAttributeSize ) {
98113 if (key == null ) {
99114 throw new IllegalArgumentException ("Custom attribute key must not be null." );
100115 }
101- return sanitizeAttribute (key );
116+ return sanitizeAttribute (key , maxAttributeSize );
102117 }
103118
104119 /** Trims the string and truncates it to MAX_ATTRIBUTE_SIZE. */
105- private static String sanitizeAttribute (String input ) {
120+ private static String sanitizeAttribute (String input , int maxAttributeSize ) {
106121 if (input != null ) {
107122 input = input .trim ();
108- if (input .length () > MAX_ATTRIBUTE_SIZE ) {
109- input = input .substring (0 , MAX_ATTRIBUTE_SIZE );
123+ if (input .length () > maxAttributeSize ) {
124+ input = input .substring (0 , maxAttributeSize );
110125 }
111126 }
112127 return input ;
0 commit comments