Скопировать содержимое одного файла в другой, но что бы строки были отсортированы по длине

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
 
public class MainClazz {
 
    public static void main(String[] args) {
        if (args.length != 3) {
            System.exit(0);
        }
 
        final String INPUT_FILE = args[0];
        final String OUTPUT_FILE = args[1];
        final String SORT_TYPE = args[2];
 
        List<String> list;
 
        list = readFromFile(INPUT_FILE);
        switch (SORT_TYPE) {
        case "abc":
            sortListABC(list);
            break;
        case "cba":
            sortListCBA(list);
            break;
        }
 
        writeToFile(list, OUTPUT_FILE);
    }
 
    public static List<String> readFromFile(String fileName) {
        List<String> listWithStrings = new ArrayList<>();
        String str = null;
 
        try (BufferedReader reader = new BufferedReader(
                new FileReader(fileName))) {
            while ((str = reader.readLine()) != null) {
                listWithStrings.add(str);
            }
            
        } catch (IOException e) {
            e.printStackTrace();
        }
 
        return listWithStrings;
    }
 
    public static void sortListABC(List<String> list) {
        Collections.sort(list, new Comparator<String>() {
 
            @Override
            public int compare(String o1, String o2) {
                if (o1.length() > o2.length()) {
                    return 1;
                }
                if (o1.length() < o2.length()) {
                    return -1;
                }
                return 0;
            }
        });
    }
 
    public static void sortListCBA(List<String> list) {
        Collections.sort(list, new Comparator<String>() {
 
            @Override
            public int compare(String o1, String o2) {
                if (o1.length() < o2.length()) {
                    return 1;
                }
                if (o1.length() > o2.length()) {
                    return -1;
                }
                return 0;
            }
        });
    }
 
    public static void writeToFile(List<String> list, String fileName) {
        try (PrintWriter writer = new PrintWriter(new File(fileName))) {
            for (String str : list) {
                writer.write(str);
                writer.write("\n");
            }
            
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
 
}

package easybudda.examples.sortstringsbylength;
 
import java.io.*;
import java.util.*;
 
public class SortFile {
    private static final String INPUT_FILE = "input.txt";
    private static final String OUTPUT_FILE = "output.txt";
    public static void main(String[] args) {
        Set<String> set = new TreeSet<>(new Comparator<String>() {
            @Override
            public int compare(String a, String b) {
                int diff = a.length() - b.length();
                if ( diff == 0 )
                    diff = a.compareTo(b);
                
                return ( diff == 0 ) ? 1 : diff;
            }
        });
        
        try ( Scanner scanner = new Scanner(new File(INPUT_FILE)) ) {
            while ( scanner.hasNext() )
                set.add(scanner.nextLine());
        }
        catch ( FileNotFoundException fnfe ) {
            System.err.println("Can't open input file!");
            return;
        }
        catch ( IllegalStateException ise ) {
            System.err.println("Can't read from input file!");
            return;
        }
        
        if ( set.isEmpty() ) {
            System.err.println("Input file is empty!");
            return;
        }
        
        try ( PrintStream ps = new PrintStream(OUTPUT_FILE) ) {
            for ( String str : set )
                ps.println(str);
        }
        catch ( IOException ioe ) {
            System.err.println("Can't write to output file!");
            return;
        }
        
        System.err.println("Done.");
    }
 
}