Pour créer un fichier Zip lors du regroupement des résultats de recherche de base de données en Java

Environnement: Java8

La description

ZipEntry est ajouté pour chaque clé en fonction d'un tableau bidimensionnel trié avec des valeurs répétées par une simple instruction SELECT, le contenu de l'entrée est sous forme de fichier texte et le contenu du fichier texte est également groupé par indentation. Créez un fichier Zip.

production

1. Créez une entrée de fichier Zip pour chaque changement de clé.
2. Chaque entrée contient un fichier texte.
3. Regroupez également le contenu du fichier texte. Regrouper par retrait.

contribution

1. L'instruction SELECT est un tableau bidimensionnel plat avec la répétition des mêmes valeurs.

Le point

1. Utilisez ByteArrayOutputStream comme zone temporaire pour l'édition de texte de longueur variable.
2. Utilisez le constructeur Formatter (OutputStream, String) pour gérer simultanément la conversion et la mise en forme du code de caractère.
3. Après avoir mis en forme l'enregistrement lu, appelez Formatter # flush () pour transférer le contenu de texte modifié vers la mémoire tampon.
4. Après avoir transféré le contenu du tampon vers ZipOutputStream, appelez ByteArrayOutputStream # reset () pour revenir au début du tampon.
5. Ne fermez pas.

prime

1. Créez une classe Row simple pour représenter des lignes, faites une boucle avec des instructions for et accédez aux valeurs de colonne par nom. C'est pour plus de clarté.
2. eq (cur.a, prev) même dans le premier processus où la valeur précédente n'existe pas comme pour (Row prev = new Row (null) afin que la comparaison entre la valeur actuelle (cur) et la valeur précédente (prev) soit toujours valable. Rendre .a) valide.
3. Pour la comparaison de chaînes, préparez une méthode pratique qui autorise null et peut comparer null avec une chaîne. Cela permet d'éviter que NullPointerException et IllegalArgumentException ne se produisent dans null.equals (s) et s.equals (null).
4. La valeur initiale de prev est cohérente avec les données réelles. Cette fois, il est supposé qu'il n'y a pas de valeur null comme données dans la table. Cela devrait être bien dans la plupart des cas. Les valeurs de colonne pour le regroupement ne doivent jamais être nulles.

Code source

Charset cs = Charset.defaultCharset();
try (ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(file), cs)) {
  try (ByteArrayOutputStream buffer = new ByteArrayOutputStream(1024)) {
    try (Formatter formatter = new Formatter(buffer, cs.name())) {
      try (PreparedStatement select = con.prepareStatement("SELECT DISTINCT a, b, c, d, e, f FROM foo ORDER BY a, b, c desc, d, e, f")) {
        try (ResultSet rs = select.executeQuery()) {
          for (Row prev = new Row(null), cur = Row.next(rs); cur != null; prev = cur, cur = Row.next(rs)) {
            if (!eq(cur.a, prev.a) || !eq(cur.b, prev.b)) {
              zos.putNextEntry(new ZipEntry(String.format("%s-%s.txt", cur.a, cur.b)));
              formatter.format("%s%n", cur.c);
              formatter.format("  %s%n", cur.d);
            } else if (!eq(cur.c, prev.c)) {
              formatter.format("%s%n", cur.c);
              formatter.format("  %s%n", cur.d);
            } else if (!eq(cur.d, prev.d)) {
              formatter.format("  %s%n", cur.d);
            }
            formatter.format("    %s: %s%n", cur.e, cur.f);
            formatter.flush();
            zos.write(buffer.toByteArray());
            buffer.reset();
          }
        }
      }
    }
  }
}

Méthode pratique qui permet la comparaison avec null

public static boolean eq(String l, String r) {
  if (l == r) {
    return true;
  } else if (l != null && r == null) {
    return false;
  } else if (l == null && r != null) {
    return false;
  } else {
    return l.equals(r);
  }
}

Un mécanisme simple pour boucler ResultSet

public class Row {
  public static Row next(ResultSet rs) throws SQLException {
    return (rs.next()) ? new Row(rs.getString(1), rs.getString(2), rs.getString(3), rs.getString(4), rs.getString(5), rs.getString(6)) : null;
  }

  String a;
  String b;
  String c;
  String d;
  String e;
  String f;
  
  public Row(String v) {
    this(v, v, v, v, v, v);
  }

  public Row(String a, String b, String c, String d, String e, String f) {
    this.a = a;
    this.b = b;
    this.c = c;
    this.d = d;
    this.e = e;
    this.f = f;
  }
}

point important

Le tampon et le formateur doivent être visibles pour le code qu'ils contiennent. Si vous écrivez comme suit, vous aurez des problèmes car vous ne pouvez pas récupérer les données docilement à cause de ZipOutputStream # write (byte []). Il existe un moyen de le faire, mais vous n'êtes pas obligé de l'écrire comme ça.

Formatter formatter = new Formatter(new ByteArrayOutputStream(1024), cs.name());
...
zos.write(???);//Que faire?