Exemplo n.º 1
0
        /// <exception cref="System.IO.IOException"/>
        /// <exception cref="System.Exception"/>
        private void TestPBDelimitedWriter(string db)
        {
            string Delimiter             = "\t";
            ByteArrayOutputStream output = new ByteArrayOutputStream();

            using (TextWriter o = new TextWriter(output))
            {
                PBImageDelimitedTextWriter v = new PBImageDelimitedTextWriter(o, Delimiter, db);
                v.Visit(new RandomAccessFile(originalFsimage, "r"));
            }
            ICollection <string> fileNames = new HashSet <string>();

            using (ByteArrayInputStream input = new ByteArrayInputStream(output.ToByteArray()
                                                                         ))
            {
                using (BufferedReader reader = new BufferedReader(new InputStreamReader(input)))
                {
                    string line;
                    while ((line = reader.ReadLine()) != null)
                    {
                        System.Console.Out.WriteLine(line);
                        string[] fields = line.Split(Delimiter);
                        NUnit.Framework.Assert.AreEqual(12, fields.Length);
                        fileNames.AddItem(fields[0]);
                    }
                }
            }
            // writtenFiles does not contain root directory and "invalid XML char" dir.
            for (IEnumerator <string> it = fileNames.GetEnumerator(); it.HasNext();)
            {
                string filename = it.Next();
                if (filename.StartsWith("/dirContainingInvalidXMLChar"))
                {
                    it.Remove();
                }
                else
                {
                    if (filename.Equals("/"))
                    {
                        it.Remove();
                    }
                }
            }
            NUnit.Framework.Assert.AreEqual(writtenFiles.Keys, fileNames);
        }
Exemplo n.º 2
0
        /// <exception cref="System.Exception"/>
        public static int Run(string[] args)
        {
            Options options = BuildOptions();

            if (args.Length == 0)
            {
                PrintUsage();
                return(0);
            }
            CommandLineParser parser = new PosixParser();
            CommandLine       cmd;

            try
            {
                cmd = parser.Parse(options, args);
            }
            catch (ParseException)
            {
                System.Console.Out.WriteLine("Error parsing command-line options: ");
                PrintUsage();
                return(-1);
            }
            if (cmd.HasOption("h"))
            {
                // print help and exit
                PrintUsage();
                return(0);
            }
            string inputFile  = cmd.GetOptionValue("i");
            string processor  = cmd.GetOptionValue("p", "Web");
            string outputFile = cmd.GetOptionValue("o", "-");
            string delimiter  = cmd.GetOptionValue("delimiter", PBImageDelimitedTextWriter.DefaultDelimiter
                                                   );
            string        tempPath = cmd.GetOptionValue("t", string.Empty);
            Configuration conf     = new Configuration();

            try
            {
                using (TextWriter @out = outputFile.Equals("-") ? System.Console.Out : new TextWriter
                                             (outputFile, "UTF-8"))
                {
                    switch (processor)
                    {
                    case "FileDistribution":
                    {
                        long maxSize = long.Parse(cmd.GetOptionValue("maxSize", "0"));
                        int  step    = System.Convert.ToInt32(cmd.GetOptionValue("step", "0"));
                        new FileDistributionCalculator(conf, maxSize, step, @out).Visit(new RandomAccessFile
                                                                                            (inputFile, "r"));
                        break;
                    }

                    case "XML":
                    {
                        new PBImageXmlWriter(conf, @out).Visit(new RandomAccessFile(inputFile, "r"));
                        break;
                    }

                    case "Web":
                    {
                        string addr = cmd.GetOptionValue("addr", "localhost:5978");
                        using (WebImageViewer viewer = new WebImageViewer(NetUtils.CreateSocketAddr(addr)
                                                                          ))
                        {
                            viewer.Start(inputFile);
                        }
                        break;
                    }

                    case "Delimited":
                    {
                        using (PBImageDelimitedTextWriter writer = new PBImageDelimitedTextWriter(@out, delimiter
                                                                                                  , tempPath))
                        {
                            writer.Visit(new RandomAccessFile(inputFile, "r"));
                        }
                        break;
                    }
                    }
                    return(0);
                }
            }
            catch (EOFException)
            {
                System.Console.Error.WriteLine("Input file ended unexpectedly. Exiting");
            }
            catch (IOException e)
            {
                System.Console.Error.WriteLine("Encountered exception.  Exiting: " + e.Message);
            }
            return(-1);
        }