KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > jboss > test > aop > declare > SystemOutDecorator


1 /*
2   * JBoss, Home of Professional Open Source
3   * Copyright 2005, JBoss Inc., and individual contributors as indicated
4   * by the @authors tag. See the copyright.txt in the distribution for a
5   * full listing of individual contributors.
6   *
7   * This is free software; you can redistribute it and/or modify it
8   * under the terms of the GNU Lesser General Public License as
9   * published by the Free Software Foundation; either version 2.1 of
10   * the License, or (at your option) any later version.
11   *
12   * This software is distributed in the hope that it will be useful,
13   * but WITHOUT ANY WARRANTY; without even the implied warranty of
14   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15   * Lesser General Public License for more details.
16   *
17   * You should have received a copy of the GNU Lesser General Public
18   * License along with this software; if not, write to the Free
19   * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20   * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21   */

22 package org.jboss.test.aop.declare;
23
24 import java.io.OutputStream JavaDoc;
25 import java.io.PrintStream JavaDoc;
26 import java.util.ArrayList JavaDoc;
27
28 /**
29  * Wraps calls to System.out.println(String), and stores all Strings beginning
30  * with "WARNING:"
31  *
32  * @author <a HREF="mailto:kabir.khan@jboss.org">Kabir Khan</a>
33  * @version $Revision: 37406 $
34  */

35 public class SystemOutDecorator extends PrintStream JavaDoc
36 {
37    static PrintStream JavaDoc sysout = null;
38    ArrayList JavaDoc warnings = new ArrayList JavaDoc();
39    public SystemOutDecorator(OutputStream JavaDoc out)
40    {
41       super(out);
42       sysout = System.out;
43    }
44    
45    public ArrayList JavaDoc getWarnings()
46    {
47       return warnings;
48    }
49    
50    public void println(String JavaDoc msg)
51    {
52       super.println(msg);
53       if (msg.startsWith("WARNING:"))
54       {
55          System.out.println(">>>>>");
56          super.println(msg);
57          System.out.println("<<<<<");
58          msg = getRidOfAllWhiteSpace(msg);
59          warnings.add(msg);
60       }
61    }
62    
63    public static SystemOutDecorator initialise()
64    {
65       SystemOutDecorator sys = new SystemOutDecorator(System.out);
66       System.setOut(sys);
67       return sys;
68    }
69    
70    public void kill()
71    {
72       System.setOut(sysout);
73    }
74
75    String JavaDoc getRidOfAllWhiteSpace(String JavaDoc msg)
76    {
77       StringBuffer JavaDoc sb = new StringBuffer JavaDoc();
78       
79       for (int i = 0 ; i < msg.length() ; i++)
80       {
81          char ch = msg.charAt(i);
82          if (ch != '\n' && ch != '\t' && ch != ' ' && ch != '\r')
83          {
84             sb.append(ch);
85          }
86       }
87       
88       return sb.toString();
89    }
90 }
91
Popular Tags