dyaml/examples/yaml_bench/yaml_bench.d

107 lines
2.9 KiB
D
Raw Normal View History

//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);
}
}