001/* 002 * Copyright (c) 2007-2017 Xplenty, Inc. All Rights Reserved. 003 * 004 * Project and contact information: http://www.cascading.org/ 005 * 006 * This file is part of the Cascading project. 007 * 008 * Licensed under the Apache License, Version 2.0 (the "License"); 009 * you may not use this file except in compliance with the License. 010 * You may obtain a copy of the License at 011 * 012 * http://www.apache.org/licenses/LICENSE-2.0 013 * 014 * Unless required by applicable law or agreed to in writing, software 015 * distributed under the License is distributed on an "AS IS" BASIS, 016 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 017 * See the License for the specific language governing permissions and 018 * limitations under the License. 019 */ 020 021package cascading.tap.local.io; 022 023import java.io.File; 024import java.io.FileNotFoundException; 025import java.io.FileOutputStream; 026 027import cascading.tap.Tap; 028import cascading.tap.TapException; 029 030/** 031 * 032 */ 033public class TapFileOutputStream extends FileOutputStream 034 { 035 public TapFileOutputStream( String path, boolean append ) throws FileNotFoundException 036 { 037 super( prepare( null, path ), append ); 038 } 039 040 public TapFileOutputStream( Tap parent, String path, boolean update ) throws FileNotFoundException 041 { 042 super( prepare( parent, path ), update ); 043 } 044 045 private static String prepare( Tap parent, String path ) 046 { 047 File file; 048 049 if( parent == null ) 050 file = new File( path ); 051 else 052 file = new File( parent.getIdentifier(), path ); 053 054 // ignore the output. will catch the failure downstream if any. 055 // not ignoring the output causes race conditions with other systems writing to the same directory. 056 File parentFile = file.getAbsoluteFile().getParentFile(); 057 058 if( parentFile != null && parentFile.exists() && parentFile.isFile() ) 059 throw new TapException( "cannot create parent directory, it already exists as a file: " + parentFile.getAbsolutePath() ); 060 061 // don't test for success, just fighting a race condition otherwise 062 // will get caught downstream 063 if( parentFile != null ) 064 parentFile.mkdirs(); 065 066 return file.getAbsolutePath(); 067 } 068 }