update FileConfig class...

Signed-off-by: j502647092 <jtb1@163.com>
master
j502647092 2015-05-30 17:39:22 +08:00
parent f6f15f5016
commit b06f262e24
1 changed files with 26 additions and 8 deletions

View File

@ -1,5 +1,6 @@
package cn.citycraft.ResFly.config; package cn.citycraft.ResFly.config;
import java.io.BufferedReader;
import java.io.File; import java.io.File;
import java.io.FileInputStream; import java.io.FileInputStream;
import java.io.FileNotFoundException; import java.io.FileNotFoundException;
@ -7,6 +8,7 @@ import java.io.FileOutputStream;
import java.io.IOException; import java.io.IOException;
import java.io.InputStreamReader; import java.io.InputStreamReader;
import java.io.OutputStreamWriter; import java.io.OutputStreamWriter;
import java.io.Reader;
import java.io.Writer; import java.io.Writer;
import java.util.logging.Level; import java.util.logging.Level;
@ -25,14 +27,15 @@ import com.google.common.base.Charsets;
import com.google.common.io.Files; import com.google.common.io.Files;
/** /**
* An implementation of {@link Configuration} which saves all files in Yaml. * An implementation of {@link Configuration} which saves all files in Yaml. Note that this
* Note that this implementation is not synchronized. * implementation is not synchronized.
*/ */
public class FileConfig extends YamlConfiguration { public class FileConfig extends YamlConfiguration {
public static FileConfig init(File file) { public static FileConfig init(File file) {
return FileConfig.loadConfiguration(file); return FileConfig.loadConfiguration(file);
} }
public static FileConfig loadConfiguration(File file) { public static FileConfig loadConfiguration(File file) {
Validate.notNull(file, "File cannot be null"); Validate.notNull(file, "File cannot be null");
FileConfig config = new FileConfig(); FileConfig config = new FileConfig();
@ -46,28 +49,43 @@ public class FileConfig extends YamlConfiguration {
} }
return config; return config;
} }
protected final DumperOptions yamlOptions = new DumperOptions(); protected final DumperOptions yamlOptions = new DumperOptions();
protected final Representer yamlRepresenter = new YamlRepresenter(); protected final Representer yamlRepresenter = new YamlRepresenter();
protected final Yaml yaml = new Yaml(new YamlConstructor(), protected final Yaml yaml = new Yaml(new YamlConstructor(), yamlRepresenter, yamlOptions);
yamlRepresenter, yamlOptions);
@Override @Override
public void load(File file) throws FileNotFoundException, IOException, public void load(File file) throws FileNotFoundException, IOException, InvalidConfigurationException {
InvalidConfigurationException {
Validate.notNull(file, "File cannot be null"); Validate.notNull(file, "File cannot be null");
final FileInputStream stream = new FileInputStream(file); final FileInputStream stream = new FileInputStream(file);
load(new InputStreamReader(stream, Charsets.UTF_8)); load(new InputStreamReader(stream, Charsets.UTF_8));
} }
@Override
public void load(Reader reader) throws IOException, InvalidConfigurationException {
BufferedReader input = (reader instanceof BufferedReader) ? (BufferedReader) reader
: new BufferedReader(reader);
StringBuilder builder = new StringBuilder();
try {
String line;
while ((line = input.readLine()) != null) {
builder.append(line);
builder.append('\n');
}
} finally {
input.close();
}
loadFromString(builder.toString());
}
@Override @Override
public void save(File file) throws IOException { public void save(File file) throws IOException {
Validate.notNull(file, "File cannot be null"); Validate.notNull(file, "File cannot be null");
Files.createParentDirs(file); Files.createParentDirs(file);
String data = saveToString(); String data = saveToString();
Writer writer = new OutputStreamWriter(new FileOutputStream(file), Writer writer = new OutputStreamWriter(new FileOutputStream(file), Charsets.UTF_8);
Charsets.UTF_8);
try { try {
writer.write(data); writer.write(data);
} finally { } finally {