Attachment 16772 Details for Bug 37200 – Split task source code.

Split task source code.

SplitTask.java (text/plain), 9.23 KB, created by Dale Anson on 2005-10-21 21:13:57 UTC

(hide)

Description:

Filename:

MIME Type:

Creator: Dale Anson

Created: 2005-10-21 21:13:57 UTC

Size: 9.23 KB

patch

obsolete

>
>package ise.antelope.tasks;
>
>import java.io.*;
>
>import org.apache.tools.ant.*;
>import org.apache.tools.ant.taskdefs.*;
>
>/**
> * This task is similar to the Unix/Linux split utility, it splits a file into
> * a number of smaller pieces.  It will also split a property value or a string.
> *
> * @author    Dale Anson
> * @version   $Revision: 1.1 $
> * @since     Ant 1.6
> */
>public class SplitTask extends Task {
>
>    private String prefix = "x";
>    private int bytes = -1;
>    private int lines = 1000;
>    private String value = null;
>    private File file = null;
>    private File outputDir = null;
>    private boolean failOnError = true;
>
>
>    /**
>     * The start of the file names to write.  Files are named using this string,
>     * followed by a "." and a number, e.g. x.0, x.1, etc.  The Unix/Linux split
>     * uses a letter scheme for the suffix, that is not supported here. 
>     *
>     * Should the dot go away? If the user wants a dot, it could be part of this
>     * attribute.  Right now, the dot is there, and there is no way for the user
>     * to make it go away.
>     *
>     * @param x  The new prefix value
>     */
>    public void setPrefix( String x ) {
>        prefix = x;
>    }
>
>    /**
>     * Set the number of bytes per part.  This is not a required parameter,
>     * the default is to use lines rather than bytes.
>     *
>     * Use bytes or lines, not both. In general, use bytes or size for binary 
>     * files, lines for text files.
>     *
>     * @param b  number of bytes per part.
>     */
>    public void setBytes( int b ) {
>        bytes = b;
>        lines = -1;
>    }
>
>    /**
>     * The linux split command allows modifiers: b for 512, k for 1K, m for 1
>     * Meg.  Use this method for a similar effect.  This is not a required
>     * parameter, the default is to use lines rather than size.
>     *
>     * Use bytes or lines, not both. In general, use bytes or size for binary 
>     * files, lines for text files.
>     *
>     * @param b  the number of bytes per part, with an optional modifier. If
>     *      there is no modifier, treat same as setBytes(int).  For example,
>     *      setSize("100k") is the same as setBytes(100 * 1024).
>     */
>    public void setSize( String b ) {
>        if ( b == null || b.length() == 0 )
>            return ;
>        b = b.toLowerCase();
>        String modifier = b.substring( b.length() - 1 );
>        int multiplier = 1;
>        b = b.substring( 0, b.length() - 1 );
>        if ( modifier.equals( "b" ) ) {
>            multiplier = 512;
>        }
>        else if ( modifier.equals( "k" ) ) {
>            multiplier = 1024;
>        }
>        else if ( modifier.equals( "m" ) ) {
>            multiplier = 1024 * 1024;
>        }
>        else {
>            // modifier is not recognized, so put it back, maybe it's a number
>            b = b + modifier;
>        }
>        try {
>            int size = Integer.parseInt( b ) * multiplier;
>            setBytes( size );
>        }
>        catch ( NumberFormatException e ) {
>            throw new BuildException( "Invalid size parameter." );
>        }
>    }
>
>    /**
>     * Set the number of lines per part, default is 1000.  This is not a required
>     * parameter, but is the default setting for splitting.
>     *
>     * Use bytes or lines, not both. In general, use bytes or size for binary 
>     * files, lines for text files.
>     *
>     * @param x  The number of lines per part.
>     */
>    public void setLines( int x ) {
>        lines = x;
>        bytes = -1;
>    }
>
>    /**
>     * Split the text value of the given property.
>     *
>     * One of property, value, or file are required.
>     *
>     * @param p  the name of the property whose value will be split.
>     */
>    public void setProperty( String p ) {
>        String v = getProject().getProperty( p );
>        if ( v == null || v.equals( "" ) )
>            throw new BuildException( "Property " + p + " has no value." );
>        setValue( v );
>    }
>
>    /**
>     * Split the given string.
>     *
>     * One of property, value, or file are required.
>     *
>     * @param v  a string
>     */
>    public void setValue( String v ) {
>        if ( v == null || v.equals( "" ) )
>            throw new BuildException( "Value is null or empty." );
>        value = v;
>    }
>
>    /**
>     * Split the contents of the given file.
>     *
>     * One of property, value, or file are required.
>     *
>     * @param f  the name of the file
>     */
>    public void setFile( File f ) {
>        file = f;
>    }
>
>    /**
>     * Where to put the parts. If file has been set and output directory has not 
>     * set, output to directory containing file.
>     *
>     * @param d  the output directory
>     */
>    public void setOutputdir( File d ) {
>        outputDir = d;
>    }
>
>    /**
>     * Determines whether the build should fail if there is an error. Default is
>     * true.
>     *
>     * @param fail  true or false
>     */
>    public void setFailonerror( boolean fail ) {
>        failOnError = fail;
>    }
>
>
>    /**
>     * Split the given property, value, or file into pieces.
>     *
>     * @exception BuildException  only if failOnError is true
>     */
>    public void execute() throws BuildException {
>        // check params --
>        // must have value or file
>        if ( value == null && file == null )
>            throw new BuildException( "Must have property, value, or file." );
>        // if no file, must have outputDir
>        if ( file == null && outputDir == null )
>            throw new BuildException( "Must have output directory." );
>        // must have only one of value or file
>        if ( value != null && file != null )
>            throw new BuildException( "Must not have more than one of property, value, or file." );
>
>        try {
>            if ( value != null )
>                splitValue();
>            else
>                splitFile();
>        }
>        catch ( Exception e ) {
>            if ( failOnError )
>                throw new BuildException( e.getMessage() );
>            else
>                log( e.getMessage() );
>        }
>    }
>
>    /**
>     * Split a string value into several files.
>     *
>     * @exception IOException  if there is an i/o problem
>     */
>    private void splitValue() throws Exception {
>        if ( !outputDir.exists() && !outputDir.mkdirs() ) {
>            throw new IOException( "Unable to create output directory." );
>        }
>
>        StringReader reader = new StringReader( value );
>        int bytes_read = 0;
>        int suffix = 0;
>        if ( bytes > 0 ) {
>            // make files all the same number of bytes
>            char[] buffer = new char[ bytes ];
>            while ( bytes_read > -1 ) {
>                bytes_read = reader.read( buffer, 0, bytes );
>                if ( bytes_read == -1 )
>                    break;
>                FileWriter fw = new FileWriter( new File( outputDir, prefix + "." + String.valueOf( suffix ) ) );
>                fw.write( buffer, 0, bytes_read );
>                fw.flush();
>                fw.close();
>                ++suffix;
>            }
>        }
>        else {
>            // make files all the same number of lines
>            splitByLines( reader );
>        }
>    }
>
>    /**
>     * Split a file into several files.
>     *
>     * @exception IOException  if there is an i/o problem
>     */
>    private void splitFile() throws IOException {
>        if ( !file.exists() )
>            throw new FileNotFoundException( file.toString() );
>        if ( file.length() == 0 )
>            throw new BuildException( "Zero length file." );
>        if ( outputDir == null )
>            outputDir = file.getParentFile();
>        if ( !outputDir.exists() && !outputDir.mkdirs() ) {
>            throw new IOException( "Unable to create output directory." );
>        }
>
>        int suffix = 0;
>        if ( bytes > 0 ) {
>            // make files all the same number of bytes
>            BufferedInputStream bis = new BufferedInputStream( new FileInputStream( file ) );
>            int bytes_read = 0;
>            byte[] buffer = new byte[ bytes ];
>            while ( bytes_read > -1 ) {
>                bytes_read = bis.read( buffer, 0, bytes );
>                if ( bytes_read == -1 )
>                    break;
>                FileOutputStream fos = new FileOutputStream( new File( outputDir, prefix + "." + String.valueOf( suffix ) ) );
>                fos.write( buffer, 0, bytes_read );
>                fos.flush();
>                fos.close();
>                ++suffix;
>            }
>        }
>        else {
>            // make files all the same number of lines
>            splitByLines( new FileReader( file ) );
>        }
>    }
>    
>    private void splitByLines( Reader reader ) throws IOException {
>        int suffix = 0;
>        LineNumberReader lnr = new LineNumberReader( reader );
>        String line = lnr.readLine();
>        BufferedWriter writer = new BufferedWriter( new FileWriter( new File( outputDir, prefix + "." + String.valueOf( suffix ) ) ) );
>        while ( line != null ) {
>            writer.write( line );
>            writer.newLine();
>            if ( lnr.getLineNumber() % lines == 0 ) {
>                writer.flush();
>                writer.close();
>                ++suffix;
>                writer = new BufferedWriter( new FileWriter( new File( outputDir, prefix + "." + String.valueOf( suffix ) ) ) );
>            }
>            line = lnr.readLine();
>        }
>        writer.flush();
>        writer.close();
>    }
>
>}
>
>

Actions: View

Attachments on bug 37200: 16772 | 16774 | 16775