1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.commons.configuration2;
19
20 import java.io.IOException;
21 import java.io.InputStream;
22 import java.io.Reader;
23 import java.io.Writer;
24 import java.util.Map;
25
26 import org.apache.commons.configuration2.ex.ConfigurationException;
27 import org.apache.commons.configuration2.io.InputStreamSupport;
28 import org.apache.commons.configuration2.tree.ImmutableNode;
29 import org.yaml.snakeyaml.DumperOptions;
30 import org.yaml.snakeyaml.LoaderOptions;
31 import org.yaml.snakeyaml.Yaml;
32 import org.yaml.snakeyaml.constructor.SafeConstructor;
33 import org.yaml.snakeyaml.representer.Representer;
34
35
36
37
38
39
40
41
42 public class YAMLConfiguration extends AbstractYAMLBasedConfiguration implements FileBasedConfiguration, InputStreamSupport {
43
44
45
46 public YAMLConfiguration() {
47 }
48
49
50
51
52
53
54 public YAMLConfiguration(final HierarchicalConfiguration<ImmutableNode> c) {
55 super(c);
56 }
57
58 @Override
59 public void read(final Reader in) throws ConfigurationException {
60 try {
61 final Yaml yaml = createYamlForReading(new LoaderOptions());
62 final Map<String, Object> map = yaml.load(in);
63 load(map);
64 } catch (final Exception e) {
65 rethrowException(e);
66 }
67 }
68
69 public void read(final Reader in, final LoaderOptions options) throws ConfigurationException {
70 try {
71 final Yaml yaml = createYamlForReading(options);
72 final Map<String, Object> map = yaml.load(in);
73 load(map);
74 } catch (final Exception e) {
75 rethrowException(e);
76 }
77 }
78
79 @Override
80 public void write(final Writer out) throws ConfigurationException, IOException {
81 final DumperOptions options = new DumperOptions();
82 options.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
83 dump(out, options);
84 }
85
86 public void dump(final Writer out, final DumperOptions options)
87 throws ConfigurationException, IOException {
88 final Yaml yaml = new Yaml(options);
89 yaml.dump(constructMap(getNodeModel().getNodeHandler().getRootNode()), out);
90 }
91
92
93
94
95
96
97
98 @Override
99 public void read(final InputStream in) throws ConfigurationException {
100 try {
101 final Yaml yaml = createYamlForReading(new LoaderOptions());
102 final Map<String, Object> map = yaml.load(in);
103 load(map);
104 } catch (final Exception e) {
105 rethrowException(e);
106 }
107 }
108
109 public void read(final InputStream in, final LoaderOptions options) throws ConfigurationException {
110 try {
111 final Yaml yaml = createYamlForReading(options);
112 final Map<String, Object> map = yaml.load(in);
113 load(map);
114 } catch (final Exception e) {
115 rethrowException(e);
116 }
117 }
118
119
120
121
122
123
124
125 private static Yaml createYamlForReading(final LoaderOptions options) {
126 return new Yaml(new SafeConstructor(options), new Representer(new DumperOptions()), new DumperOptions(), options);
127 }
128
129 }