-
Notifications
You must be signed in to change notification settings - Fork 2.3k
/
InMemoryTokenCache.java
110 lines (95 loc) · 3.39 KB
/
InMemoryTokenCache.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/*
* Copyright (c) 2002-2018 "Neo Technology,"
* Network Engine for Objects in Lund AB [http://neotechnology.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.neo4j.kernel.impl.core;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.neo4j.kernel.impl.util.CopyOnWriteHashMap;
import org.neo4j.storageengine.api.Token;
import static java.util.Collections.unmodifiableCollection;
/**
* Token cache that provide id -> TOKEN and name -> id mappings.
* Name -> id mapping will be updated last since it's used as part of the check for token existence in a cache.
* As soon as token visible through it - it's considered added into a cache.
*
* Implementation does not provide any atomicity guarantees. Mapping updates will be visible independently from each
* other.
* Implementation is not thread safe.
*
* @param <TOKEN> token type
*/
public class InMemoryTokenCache<TOKEN extends Token>
{
private final Map<String, Integer> nameToId = new CopyOnWriteHashMap<>();
private final Map<Integer, TOKEN> idToToken = new CopyOnWriteHashMap<>();
private final String tokenType;
public InMemoryTokenCache( String tokenType )
{
this.tokenType = tokenType;
}
public void clear()
{
nameToId.clear();
idToToken.clear();
}
private void putAndEnsureUnique( Map<String,Integer> nameToId, TOKEN token, String tokenType )
{
Integer previous = nameToId.putIfAbsent( token.name(), token.id() );
if ( previous != null && previous != token.id() )
{
// since we optimistically put token into a map before, now we need to remove it.
idToToken.remove( token.id(), token );
throw new NonUniqueTokenException( tokenType, token.name(), token.id(), previous );
}
}
public void putAll( List<TOKEN> tokens ) throws NonUniqueTokenException
{
Map<String, Integer> newNameToId = new HashMap<>();
Map<Integer, TOKEN> newIdToToken = new HashMap<>();
for ( TOKEN token : tokens )
{
newIdToToken.put( token.id(), token );
putAndEnsureUnique( newNameToId, token, tokenType );
}
idToToken.putAll( newIdToToken );
nameToId.putAll( newNameToId );
}
public void put( TOKEN token ) throws NonUniqueTokenException
{
idToToken.put( token.id(), token );
putAndEnsureUnique( nameToId, token, tokenType );
}
public Integer getId( String name )
{
return nameToId.get( name );
}
public TOKEN getToken( int id )
{
return idToToken.get( id );
}
public Iterable<TOKEN> allTokens()
{
return unmodifiableCollection( idToToken.values() );
}
public int size()
{
return nameToId.size();
}
}