Index: llvm/trunk/include/llvm/Support/YAMLTraits.h =================================================================== --- llvm/trunk/include/llvm/Support/YAMLTraits.h +++ llvm/trunk/include/llvm/Support/YAMLTraits.h @@ -647,6 +647,8 @@ if ( this->preflightKey(Key, Required, false, UseDefault, SaveInfo) ) { yamlize(*this, Val, Required); this->postflightKey(SaveInfo); + } else if (UseDefault) { + Val = T(); } } Index: llvm/trunk/unittests/Support/YAMLIOTest.cpp =================================================================== --- llvm/trunk/unittests/Support/YAMLIOTest.cpp +++ llvm/trunk/unittests/Support/YAMLIOTest.cpp @@ -68,6 +68,21 @@ } } +struct FooBarOptional { + int Foo; + int Bar; +}; + +namespace llvm { +namespace yaml { +template <> struct MappingTraits { + static void mapping(IO &YamlIO, FooBarOptional &Obj) { + YamlIO.mapRequired("foo", Obj.Foo); + YamlIO.mapOptional("bar", Obj.Bar); + } +}; +} +} // // Test the reading of a yaml mapping @@ -93,6 +108,19 @@ } } +TEST(YAMLIO, TestMapReadOptional) { + FooBarOptional Doc; + Doc.Bar = 42; + { + Input In("---\nfoo: 3\n...\n"); + In >> Doc; + + EXPECT_FALSE(In.error()); + EXPECT_EQ(Doc.Foo, 3); + EXPECT_EQ(Doc.Bar, 0); + } +} + TEST(YAMLIO, TestMalformedMapRead) { FooBar doc; Input yin("{foo: 3; bar: 5}", nullptr, suppressErrorMessages);