Vedran Pavic
6 years ago
committed by
Sam Brannen
3 changed files with 176 additions and 3 deletions
@ -0,0 +1,159 @@
@@ -0,0 +1,159 @@
|
||||
/* |
||||
* Copyright 2002-2019 the original author or authors. |
||||
* |
||||
* Licensed 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 |
||||
* |
||||
* https://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.springframework.context.index.processor; |
||||
|
||||
import java.io.ByteArrayOutputStream; |
||||
import java.io.IOException; |
||||
import java.io.OutputStream; |
||||
import java.io.StringWriter; |
||||
import java.io.Writer; |
||||
import java.nio.charset.StandardCharsets; |
||||
import java.util.Collections; |
||||
import java.util.Comparator; |
||||
import java.util.Enumeration; |
||||
import java.util.Map.Entry; |
||||
import java.util.Properties; |
||||
import java.util.Set; |
||||
import java.util.TreeSet; |
||||
|
||||
/** |
||||
* Specialization of {@link Properties} that sorts properties alphanumerically |
||||
* based on their keys. |
||||
* |
||||
* <p>This can be useful when storing the {@link Properties} instance in a |
||||
* properties file, since it allows such files to be generated in a repeatable |
||||
* manner with consistent ordering of properties. |
||||
* |
||||
* <p>Comments in generated properties files can also be optionally omitted. |
||||
* |
||||
* @author Sam Brannen |
||||
* @since 5.2 |
||||
* @see java.util.Properties |
||||
*/ |
||||
@SuppressWarnings("serial") |
||||
class SortedProperties extends Properties { |
||||
|
||||
static final String EOL = System.lineSeparator(); |
||||
|
||||
private static final Comparator<Object> keyComparator = //
|
||||
(key1, key2) -> String.valueOf(key1).compareTo(String.valueOf(key2)); |
||||
|
||||
private static final Comparator<Entry<Object, Object>> entryComparator = //
|
||||
Entry.comparingByKey(keyComparator); |
||||
|
||||
private final boolean omitComments; |
||||
|
||||
|
||||
/** |
||||
* Construct a new {@code SortedProperties} instance that honors the supplied |
||||
* {@code omitComments} flag. |
||||
* |
||||
* @param omitComments {@code true} if comments should be omitted when |
||||
* storing properties in a file |
||||
*/ |
||||
SortedProperties(boolean omitComments) { |
||||
this.omitComments = omitComments; |
||||
} |
||||
|
||||
/** |
||||
* Construct a new {@code SortedProperties} instance with properties populated |
||||
* from the supplied {@link Properties} object and honoring the supplied |
||||
* {@code omitComments} flag. |
||||
* |
||||
* <p>Default properties from the supplied {@code Properties} object will |
||||
* not be copied. |
||||
* |
||||
* @param properties the {@code Properties} object from which to copy the |
||||
* initial properties |
||||
* @param omitComments {@code true} if comments should be omitted when |
||||
* storing properties in a file |
||||
*/ |
||||
SortedProperties(Properties properties, boolean omitComments) { |
||||
this(omitComments); |
||||
putAll(properties); |
||||
} |
||||
|
||||
@Override |
||||
public void store(OutputStream out, String comments) throws IOException { |
||||
ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
||||
super.store(baos, (this.omitComments ? null : comments)); |
||||
String contents = new String(baos.toByteArray(), StandardCharsets.ISO_8859_1); |
||||
for (String line : contents.split(EOL)) { |
||||
if (!this.omitComments || !line.startsWith("#")) { |
||||
out.write((line + EOL).getBytes(StandardCharsets.ISO_8859_1)); |
||||
} |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void store(Writer writer, String comments) throws IOException { |
||||
StringWriter stringWriter = new StringWriter(); |
||||
super.store(stringWriter, (this.omitComments ? null : comments)); |
||||
String contents = stringWriter.toString(); |
||||
for (String line : contents.split(EOL)) { |
||||
if (!this.omitComments || !line.startsWith("#")) { |
||||
writer.write(line + EOL); |
||||
} |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void storeToXML(OutputStream out, String comments) throws IOException { |
||||
super.storeToXML(out, (this.omitComments ? null : comments)); |
||||
} |
||||
|
||||
@Override |
||||
public void storeToXML(OutputStream out, String comments, String encoding) throws IOException { |
||||
super.storeToXML(out, (this.omitComments ? null : comments), encoding); |
||||
} |
||||
|
||||
/** |
||||
* Return a sorted enumeration of the keys in this {@link Properties} object. |
||||
* @see #keySet() |
||||
*/ |
||||
@Override |
||||
public synchronized Enumeration<Object> keys() { |
||||
return Collections.enumeration(keySet()); |
||||
} |
||||
|
||||
/** |
||||
* Return a sorted set of the keys in this {@link Properties} object. |
||||
* <p>The keys will be converted to strings if necessary using |
||||
* {@link String#valueOf(Object)} and sorted alphanumerically according to |
||||
* the natural order of strings. |
||||
*/ |
||||
@Override |
||||
public Set<Object> keySet() { |
||||
Set<Object> sortedKeys = new TreeSet<>(keyComparator); |
||||
sortedKeys.addAll(super.keySet()); |
||||
return Collections.synchronizedSet(sortedKeys); |
||||
} |
||||
|
||||
/** |
||||
* Return a sorted set of the entries in this {@link Properties} object. |
||||
* <p>The entries will be sorted based on their keys, and the keys will be |
||||
* converted to strings if necessary using {@link String#valueOf(Object)} |
||||
* and compared alphanumerically according to the natural order of strings. |
||||
*/ |
||||
@Override |
||||
public Set<Entry<Object, Object>> entrySet() { |
||||
Set<Entry<Object, Object>> sortedEntries = new TreeSet<>(entryComparator); |
||||
sortedEntries.addAll(super.entrySet()); |
||||
return Collections.synchronizedSet(sortedEntries); |
||||
} |
||||
|
||||
} |
Loading…
Reference in new issue