This commit is contained in:
2024-11-30 19:03:49 +08:00
commit 1e6763c160
3806 changed files with 737676 additions and 0 deletions

View File

@@ -0,0 +1,208 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.tomcat.util.collections;
import java.util.AbstractMap;
import java.util.AbstractSet;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Locale;
import java.util.Map;
import java.util.Set;
import org.apache.tomcat.util.res.StringManager;
/**
* A Map implementation that uses case-insensitive (using {@link
* Locale#ENGLISH}) strings as keys.
* <p>
* Keys must be instances of {@link String}. Note that this means that
* <code>null</code> keys are not permitted.
* <p>
* This implementation is not thread-safe.
*
* @param <V> Type of values placed in this Map.
*/
public class CaseInsensitiveKeyMap<V> extends AbstractMap<String,V> {
private static final StringManager sm =
StringManager.getManager(CaseInsensitiveKeyMap.class);
private final Map<Key,V> map = new HashMap<>();
@Override
public V get(Object key) {
return map.get(Key.getInstance(key));
}
@Override
public V put(String key, V value) {
Key caseInsensitiveKey = Key.getInstance(key);
if (caseInsensitiveKey == null) {
throw new NullPointerException(sm.getString("caseInsensitiveKeyMap.nullKey"));
}
return map.put(caseInsensitiveKey, value);
}
/**
* {@inheritDoc}
* <p>
* <b>Use this method with caution</b>. If the input Map contains duplicate
* keys when the keys are compared in a case insensitive manner then some
* values will be lost when inserting via this method.
*/
@Override
public void putAll(Map<? extends String, ? extends V> m) {
super.putAll(m);
}
@Override
public boolean containsKey(Object key) {
return map.containsKey(Key.getInstance(key));
}
@Override
public V remove(Object key) {
return map.remove(Key.getInstance(key));
}
@Override
public Set<Entry<String, V>> entrySet() {
return new EntrySet<>(map.entrySet());
}
private static class EntrySet<V> extends AbstractSet<Entry<String,V>> {
private final Set<Entry<Key,V>> entrySet;
public EntrySet(Set<Map.Entry<Key,V>> entrySet) {
this.entrySet = entrySet;
}
@Override
public Iterator<Entry<String,V>> iterator() {
return new EntryIterator<>(entrySet.iterator());
}
@Override
public int size() {
return entrySet.size();
}
}
private static class EntryIterator<V> implements Iterator<Entry<String,V>> {
private final Iterator<Entry<Key,V>> iterator;
public EntryIterator(Iterator<Entry<Key,V>> iterator) {
this.iterator = iterator;
}
@Override
public boolean hasNext() {
return iterator.hasNext();
}
@Override
public Entry<String,V> next() {
Entry<Key,V> entry = iterator.next();
return new EntryImpl<>(entry.getKey().getKey(), entry.getValue());
}
@Override
public void remove() {
iterator.remove();
}
}
private static class EntryImpl<V> implements Entry<String,V> {
private final String key;
private final V value;
public EntryImpl(String key, V value) {
this.key = key;
this.value = value;
}
@Override
public String getKey() {
return key;
}
@Override
public V getValue() {
return value;
}
@Override
public V setValue(V value) {
throw new UnsupportedOperationException();
}
}
private static class Key {
private final String key;
private final String lcKey;
private Key(String key) {
this.key = key;
this.lcKey = key.toLowerCase(Locale.ENGLISH);
}
public String getKey() {
return key;
}
@Override
public int hashCode() {
return lcKey.hashCode();
}
@Override
public boolean equals(Object obj) {
if (this == obj) {
return true;
}
if (obj == null) {
return false;
}
if (getClass() != obj.getClass()) {
return false;
}
Key other = (Key) obj;
return lcKey.equals(other.lcKey);
}
public static Key getInstance(Object o) {
if (o instanceof String) {
return new Key((String) o);
}
return null;
}
}
}

View File

@@ -0,0 +1,59 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.tomcat.util.collections;
import java.util.Map;
import java.util.WeakHashMap;
import java.util.concurrent.ConcurrentHashMap;
public final class ConcurrentCache<K,V> {
private final int size;
private final Map<K,V> eden;
private final Map<K,V> longterm;
public ConcurrentCache(int size) {
this.size = size;
this.eden = new ConcurrentHashMap<>(size);
this.longterm = new WeakHashMap<>(size);
}
public V get(K k) {
V v = this.eden.get(k);
if (v == null) {
synchronized (longterm) {
v = this.longterm.get(k);
}
if (v != null) {
this.eden.put(k, v);
}
}
return v;
}
public void put(K k, V v) {
if (this.eden.size() >= size) {
synchronized (longterm) {
this.longterm.putAll(this.eden);
}
this.eden.clear();
}
this.eden.put(k, v);
}
}

View File

@@ -0,0 +1,271 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.tomcat.util.collections;
import java.lang.ref.Reference;
import java.lang.ref.ReferenceQueue;
import java.lang.ref.WeakReference;
import java.util.AbstractMap;
import java.util.AbstractSet;
import java.util.Collection;
import java.util.Iterator;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
/**
* Concurrent hash map that holds its keys via weak references. Unlike
* <code>WeakHashMap</code> this class does not handle dead keys during common
* access operations, but expects you to call its {@link #maintain()} method
* periodically. Both keys and values are expected to be not-<code>null</code>.
*
* @param <K> The type of keys used with the Map instance
* @param <V> The type of values used with the Map instance
*/
public class ManagedConcurrentWeakHashMap<K, V> extends AbstractMap<K, V> implements
ConcurrentMap<K, V> {
private final ConcurrentMap<Key, V> map = new ConcurrentHashMap<>();
private final ReferenceQueue<Object> queue = new ReferenceQueue<>();
/**
* Method, that has to be invoked periodically to clean dead keys from the
* map.
*/
public void maintain() {
Key key;
while ((key = (Key) queue.poll()) != null) {
if (key.isDead()) {
// No need to lookup if the key is not in the map
continue;
}
key.ackDeath();
map.remove(key);
}
}
private static class Key extends WeakReference<Object> {
private final int hash;
private boolean dead;
public Key(Object key, ReferenceQueue<Object> queue) {
super(key, queue);
hash = key.hashCode();
}
@Override
public int hashCode() {
return hash;
}
@Override
public boolean equals(Object obj) {
if (this == obj) {
return true;
}
if (dead) {
// Post-mortem cleanup looks for this specific Reference
// instance
return false;
}
if (!(obj instanceof Reference<?>)) {
return false;
}
Object oA = get();
Object oB = ((Reference<?>) obj).get();
if (oA == oB) {
return true;
}
if (oA == null || oB == null) {
return false;
}
return oA.equals(oB);
}
public void ackDeath() {
this.dead = true;
}
public boolean isDead() {
return dead;
}
}
/**
* Creates Key instance to be used to store values in the map. It is
* registered with the ReferenceQueue.
*/
private Key createStoreKey(Object key) {
return new Key(key, queue);
}
/**
* Creates Key instance to be used only to lookup values in the map. It is
* not registered with the ReferenceQueue.
*/
private Key createLookupKey(Object key) {
return new Key(key, null);
}
@Override
public int size() {
return map.size();
}
@Override
public boolean isEmpty() {
return map.isEmpty();
}
@Override
public boolean containsValue(Object value) {
if (value == null) {
return false;
}
return map.containsValue(value);
}
@Override
public boolean containsKey(Object key) {
if (key == null) {
return false;
}
return map.containsKey(createLookupKey(key));
}
@Override
public V get(Object key) {
if (key == null) {
return null;
}
return map.get(createLookupKey(key));
}
@Override
public V put(K key, V value) {
Objects.requireNonNull(value);
return map.put(createStoreKey(key), value);
}
@Override
public V remove(Object key) {
return map.remove(createLookupKey(key));
}
@Override
public void clear() {
map.clear();
// maintain() clears the queue, though it is not 100% reliable,
// as queue is populated by GC asynchronously
maintain();
}
@Override
public V putIfAbsent(K key, V value) {
Objects.requireNonNull(value);
Key storeKey = createStoreKey(key);
V oldValue = map.putIfAbsent(storeKey, value);
if (oldValue != null) { // ack that key has not been stored
storeKey.ackDeath();
}
return oldValue;
}
@Override
public boolean remove(Object key, Object value) {
if (value == null) {
return false;
}
return map.remove(createLookupKey(key), value);
}
@Override
public boolean replace(K key, V oldValue, V newValue) {
Objects.requireNonNull(newValue);
return map.replace(createLookupKey(key), oldValue, newValue);
}
@Override
public V replace(K key, V value) {
Objects.requireNonNull(value);
return map.replace(createLookupKey(key), value);
}
@Override
public Collection<V> values() {
return map.values();
}
@Override
public Set<Map.Entry<K, V>> entrySet() {
return new AbstractSet<Map.Entry<K, V>>() {
@Override
public boolean isEmpty() {
return map.isEmpty();
}
@Override
public int size() {
return map.size();
}
@Override
public Iterator<Map.Entry<K, V>> iterator() {
return new Iterator<Map.Entry<K, V>>() {
private final Iterator<Map.Entry<Key, V>> it = map
.entrySet().iterator();
@Override
public boolean hasNext() {
return it.hasNext();
}
@Override
public Map.Entry<K, V> next() {
return new Map.Entry<K, V>() {
private final Map.Entry<Key, V> en = it.next();
@SuppressWarnings("unchecked")
@Override
public K getKey() {
return (K) en.getKey().get();
}
@Override
public V getValue() {
return en.getValue();
}
@Override
public V setValue(V value) {
Objects.requireNonNull(value);
return en.setValue(value);
}
};
}
@Override
public void remove() {
it.remove();
}
};
}
};
}
}

View File

@@ -0,0 +1,105 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.tomcat.util.collections;
/**
* This is intended as a (mostly) GC-free alternative to
* {@link java.util.concurrent.ConcurrentLinkedQueue} when the requirement is to
* create an unbounded queue with no requirement to shrink the queue. The aim is
* to provide the bare minimum of required functionality as quickly as possible
* with minimum garbage.
*
* @param <T> The type of object managed by this queue
*/
public class SynchronizedQueue<T> {
public static final int DEFAULT_SIZE = 128;
private Object[] queue;
private int size;
private int insert = 0;
private int remove = 0;
public SynchronizedQueue() {
this(DEFAULT_SIZE);
}
public SynchronizedQueue(int initialSize) {
queue = new Object[initialSize];
size = initialSize;
}
public synchronized boolean offer(T t) {
queue[insert++] = t;
// Wrap
if (insert == size) {
insert = 0;
}
if (insert == remove) {
expand();
}
return true;
}
public synchronized T poll() {
if (insert == remove) {
// empty
return null;
}
@SuppressWarnings("unchecked")
T result = (T) queue[remove];
queue[remove] = null;
remove++;
// Wrap
if (remove == size) {
remove = 0;
}
return result;
}
private void expand() {
int newSize = size * 2;
Object[] newQueue = new Object[newSize];
System.arraycopy(queue, insert, newQueue, 0, size - insert);
System.arraycopy(queue, 0, newQueue, size - insert, insert);
insert = size;
remove = 0;
queue = newQueue;
size = newSize;
}
public synchronized int size() {
int result = insert - remove;
if (result < 0) {
result += size;
}
return result;
}
public synchronized void clear() {
queue = new Object[size];
insert = 0;
remove = 0;
}
}

View File

@@ -0,0 +1,105 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.tomcat.util.collections;
/**
* This is intended as a (mostly) GC-free alternative to
* {@link java.util.concurrent.ConcurrentLinkedQueue} when the requirement is to
* create a pool of re-usable objects with no requirement to shrink the pool.
* The aim is to provide the bare minimum of required functionality as quickly
* as possible with minimum garbage.
*
* @param <T> The type of object managed by this stack
*/
public class SynchronizedStack<T> {
public static final int DEFAULT_SIZE = 128;
private static final int DEFAULT_LIMIT = -1;
private int size;
private final int limit;
/*
* Points to the next available object in the stack
*/
private int index = -1;
private Object[] stack;
public SynchronizedStack() {
this(DEFAULT_SIZE, DEFAULT_LIMIT);
}
public SynchronizedStack(int size, int limit) {
if (limit > -1 && size > limit) {
this.size = limit;
} else {
this.size = size;
}
this.limit = limit;
stack = new Object[size];
}
public synchronized boolean push(T obj) {
index++;
if (index == size) {
if (limit == -1 || size < limit) {
expand();
} else {
index--;
return false;
}
}
stack[index] = obj;
return true;
}
@SuppressWarnings("unchecked")
public synchronized T pop() {
if (index == -1) {
return null;
}
T result = (T) stack[index];
stack[index--] = null;
return result;
}
public synchronized void clear() {
if (index > -1) {
for (int i = 0; i < index + 1; i++) {
stack[i] = null;
}
}
index = -1;
}
private void expand() {
int newSize = size * 2;
if (limit != -1 && newSize > limit) {
newSize = limit;
}
Object[] newStack = new Object[newSize];
System.arraycopy(stack, 0, newStack, 0, size);
// This is the only point where garbage is created by throwing away the
// old array. Note it is only the array, not the contents, that becomes
// garbage.
stack = newStack;
size = newSize;
}
}