107 lines
2.9 KiB
D
107 lines
2.9 KiB
D
|
|
||
|
|
//Benchmark that loads, and optionally extracts data from and/or emits a YAML file.
|
||
|
|
|
||
|
|
import std.datetime;
|
||
|
|
import std.stdio;
|
||
|
|
import yaml;
|
||
|
|
|
||
|
|
///Print help information.
|
||
|
|
void help()
|
||
|
|
{
|
||
|
|
string help =
|
||
|
|
"D:YAML benchmark\n"
|
||
|
|
"Copyright (C) 2011 Ferdinand Majerech\n"
|
||
|
|
"Usage: yaml_bench [OPTION ...] [YAML_FILE]\n"
|
||
|
|
"\n"
|
||
|
|
"Loads and optionally extracts data and/or dumps a YAML file.\n"
|
||
|
|
"\n"
|
||
|
|
"Available options:\n"
|
||
|
|
" -h --help Show this help information.\n"
|
||
|
|
" -g --get Extract data from the file (using Node.get()).\n"
|
||
|
|
" -d --dump Dump the loaded data (to YAML_FILE.dump).\n";
|
||
|
|
writeln(help);
|
||
|
|
}
|
||
|
|
|
||
|
|
///Get data out of every node.
|
||
|
|
void extract(ref Node document)
|
||
|
|
{
|
||
|
|
void crawl(ref Node root)
|
||
|
|
{
|
||
|
|
if(root.isScalar) switch(root.tag)
|
||
|
|
{
|
||
|
|
case "tag:yaml.org,2002:null": auto value = root.get!YAMLNull; break;
|
||
|
|
case "tag:yaml.org,2002:bool": auto value = root.get!bool; break;
|
||
|
|
case "tag:yaml.org,2002:int": auto value = root.get!long; break;
|
||
|
|
case "tag:yaml.org,2002:float": auto value = root.get!real; break;
|
||
|
|
case "tag:yaml.org,2002:binary": auto value = root.get!(ubyte[]); break;
|
||
|
|
case "tag:yaml.org,2002:timestamp": auto value = root.get!SysTime; break;
|
||
|
|
case "tag:yaml.org,2002:str": auto value = root.get!string; break;
|
||
|
|
default: writeln("Unrecognozed tag: ", root.tag);
|
||
|
|
}
|
||
|
|
else if(root.isSequence) foreach(ref Node node; root)
|
||
|
|
{
|
||
|
|
crawl(node);
|
||
|
|
}
|
||
|
|
else if(root.isMapping) foreach(ref Node key, ref Node value; root)
|
||
|
|
{
|
||
|
|
crawl(key);
|
||
|
|
crawl(value);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
crawl(document);
|
||
|
|
}
|
||
|
|
|
||
|
|
void main(string[] args)
|
||
|
|
{
|
||
|
|
bool get = false;
|
||
|
|
bool dump = false;
|
||
|
|
string file = null;
|
||
|
|
|
||
|
|
//Parse command line args
|
||
|
|
foreach(arg; args[1 .. $])
|
||
|
|
{
|
||
|
|
if(arg[0] == '-') switch(arg)
|
||
|
|
{
|
||
|
|
case "--help", "-h": help(); return;
|
||
|
|
case "--get", "-g": get = true; break;
|
||
|
|
case "--dump", "-d": dump = true; break;
|
||
|
|
default: writeln("\nUnknown argument: ", arg, "\n\n"); help(); return;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
if(file !is null)
|
||
|
|
{
|
||
|
|
writeln("\nUnknown argument or file specified twice: ", arg, "\n\n");
|
||
|
|
help();
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
file = arg;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if(file is null)
|
||
|
|
{
|
||
|
|
writeln("\nFile not specified.\n\n");
|
||
|
|
help();
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
try
|
||
|
|
{
|
||
|
|
auto nodes = Loader(file).loadAll();
|
||
|
|
if(dump)
|
||
|
|
{
|
||
|
|
Dumper(file ~ ".dump").dump(nodes);
|
||
|
|
}
|
||
|
|
if(get) foreach(ref node; nodes)
|
||
|
|
{
|
||
|
|
extract(node);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
catch(YAMLException e)
|
||
|
|
{
|
||
|
|
writeln("ERROR: ", e.msg);
|
||
|
|
}
|
||
|
|
}
|