FWD » Core Development » Database

package test;

import java.io.*;

import java.nio.charset.*;

import java.util.*;

public class GenDumpFiles

{

   private final static String UTF_8  = "UTF-8";

   private final static String KOI8_R = "KOI8-R";

   private final static String WINDOWS_1251 = "WINDOWS-1251";

   private final static String ISO8859_1 = "ISO-8859-1";

   private final static String ISO8859_5 = "ISO-8859-5";

   private final static String IBM866  = "IBM866";

   private final static String USASCII = "US-ASCII";

   private final static String TEMPLATE_PATH = "dump_file_template.txt";

   private final String VALUE = "КОШКА";

   private String[] encodings = new String[] {UTF_8,

                                              KOI8_R,

                                              WINDOWS_1251,

                                              ISO8859_1,

                                              ISO8859_5,

                                              IBM866,

                                              USASCII};

   private Map<String, Charset> charsets = new HashMap<>(encodings.length);

   private Map<Charset, byte[]> valuesMap = new HashMap<>(encodings.length);

   private List<String> lines;

   private String dumpFileName;

   public GenDumpFiles()

   {

      for(String encoding : encodings)

      {

         Charset charset = Charset.forName(encoding);

         charsets.put(encoding, charset);

         valuesMap.put(charset, VALUE.getBytes(charset));

      }

      lines = readTemplate(TEMPLATE_PATH, charsets.get(USASCII));

      dumpFileName = lines.get(3).substring("filename=".length());

   }

   private List<String> readTemplate(String filePath, Charset charset)

   {

      List<String> lines = new ArrayList<>(13);

      String line;

      try(BufferedReader reader = new BufferedReader(

               new InputStreamReader(new FileInputStream(filePath), charset));)

      {

         while((line = reader.readLine()) != null)

         {

            lines.add(line);

         }

      }

      catch (IOException e)

      {

         e.printStackTrace();

      }

      return lines;

   }

   public void buildDumpFiles()

   {

      for(String encoding : encodings)

      {

         StringBuilder dumpFilePath = new StringBuilder(dumpFileName);

         dumpFilePath.append("_").append(encoding.replace("-", "_")).append(".d");

         buildDumpFile(dumpFilePath.toString(), encoding);

      }

   }

   private void buildDumpFile(String filePath, String encoding)

   {

      Charset charset = charsets.get(encoding);

      try (ByteArrayOutputStream out = new ByteArrayOutputStream(256);

           FileOutputStream fout = new FileOutputStream(filePath);

           BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(out, charset));)

      {

         writer.append(lines.get(0)).append('"')

         .append(new String(valuesMap.get(charset), charset)).append('"');

         writer.newLine();

         writer.append(lines.get(1));

         writer.newLine();

         writer.flush();

         int offset = out.size();

         for (int i = 2; i < 10; i++)

         {

            writer.append(lines.get(i));

            writer.newLine();

         }

         writer.append(lines.get(10)).append(encoding);

         writer.newLine();

         writer.append(lines.get(11));

         writer.newLine();

         writer.append(String.format("%1$010d", offset));

         writer.newLine();

         writer.flush();

         fout.write(out.toByteArray());

         fout.flush();

      }

      catch(IOException e)

      {

         e.printStackTrace();

      }

   }

   public static void main(String[] args)

   {

      GenDumpFiles gdf = new GenDumpFiles();

      gdf.buildDumpFiles();

   }

}