View Javadoc
1   package org.apache.maven.surefire.booter;
2   
3   /*
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *     http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing,
15   * software distributed under the License is distributed on an
16   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17   * KIND, either express or implied.  See the License for the
18   * specific language governing permissions and limitations
19   * under the License.
20   */
21  
22  import java.io.File;
23  import java.io.FileInputStream;
24  import java.io.FileOutputStream;
25  import java.io.IOException;
26  import java.io.InputStream;
27  import java.util.Map;
28  import java.util.Properties;
29  import java.util.concurrent.ConcurrentHashMap;
30  
31  /**
32   * @author Kristian Rosenvold
33   */
34  public class SystemPropertyManager
35  {
36  
37      /**
38       * Loads the properties, closes the stream
39       *
40       * @param inStream The stream to read from, will be closed
41       * @return The properties
42       * @throws java.io.IOException If something bad happens
43       */
44      public static PropertiesWrapper loadProperties( InputStream inStream )
45          throws IOException
46      {
47          try ( final InputStream stream = inStream )
48          {
49              Properties p = new Properties();
50              p.load( stream );
51              Map<String, String> map = new ConcurrentHashMap<>( p.size() );
52              for ( String key : p.stringPropertyNames() )
53              {
54                  map.put( key, p.getProperty( key ) );
55              }
56              return new PropertiesWrapper( map );
57          }
58      }
59  
60      private static PropertiesWrapper loadProperties( File file )
61          throws IOException
62      {
63          return loadProperties( new FileInputStream( file ) );
64      }
65  
66      public static void setSystemProperties( File file )
67          throws IOException
68      {
69          PropertiesWrapper p = loadProperties( file );
70          p.setAsSystemProperties();
71      }
72  
73      public static File writePropertiesFile( Properties properties, File tempDirectory, String name,
74                                              boolean keepForkFiles )
75          throws IOException
76      {
77          File file = File.createTempFile( name, "tmp", tempDirectory );
78          if ( !keepForkFiles )
79          {
80              file.deleteOnExit();
81          }
82  
83          writePropertiesFile( file, name, properties );
84  
85          return file;
86      }
87  
88      public static void writePropertiesFile( File file, String name, Properties properties )
89          throws IOException
90      {
91          try ( FileOutputStream out = new FileOutputStream( file ) )
92          {
93              properties.store( out, name );
94          }
95      }
96  }