Added basic command handler to fast-import.

Moved the new_blob logic off into a new subroutine and
invoked it when getting the 'blob' command.

Added statistics dump to STDERR when the program terminates listing
what it did at a high level.  This is somewhat interesting.

Signed-off-by: Shawn O. Pearce <spearce@spearce.org>

diff --git a/fast-import.c b/fast-import.c
index 8b4be28f606b812e27d513e7907f3e8032a96056..c9c48c5869a7726b0ceb4954385a330fd7653ebd 100644 (file)
--- a/fast-import.c
+++ b/fast-import.c
@@ -34,6 +34,8 @@ static int max_depth = 10;
 static unsigned long alloc_count;
 static unsigned long object_count;
 static unsigned long duplicate_count;
+static unsigned long object_count_by_type[9];
+static unsigned long duplicate_count_by_type[9];
 
 /* The .pack file */
 static int pack_fd;
@@ -173,10 +175,12 @@ static int store_object(
        e = insert_object(sha1);
        if (e->offset) {
                duplicate_count++;
+               duplicate_count_by_type[type]++;
                return 0;
        }
        e->offset = pack_offset;
        object_count++;
+       object_count_by_type[type]++;
 
        if (last->data && last->depth < max_depth)
                delta = diff_delta(last->data, last->len,
@@ -232,7 +236,7 @@ static int store_object(
 static void init_pack_header()
 {
        const char* magic = "PACK";
-       unsigned long version = 2;
+       unsigned long version = 3;
        unsigned long zero = 0;
 
        version = htonl(version);
@@ -331,12 +335,29 @@ static void write_index(const char *idx_name)
        free(idx);
 }
 
+static void new_blob()
+{
+       unsigned long datlen;
+       void *dat;
+
+       if (yread(0, &datlen, 4) != 4)
+               die("Can't obtain blob length");
+
+       dat = xmalloc(datlen);
+       if (yread(0, dat, datlen) != datlen)
+               die("Con't obtain %lu bytes of blob data", datlen);
+
+       if (!store_object(OBJ_BLOB, dat, datlen, &last_blob))
+               free(dat);
+}
+
 int main(int argc, const char **argv)
 {
        const char *base_name = argv[1];
        int est_obj_cnt = atoi(argv[2]);
        char *pack_name;
        char *idx_name;
+       struct stat sb;
 
        pack_name = xmalloc(strlen(base_name) + 6);
        sprintf(pack_name, "%s.pack", base_name);
@@ -345,30 +366,41 @@ int main(int argc, const char **argv)
 
        pack_fd = open(pack_name, O_RDWR|O_CREAT|O_EXCL, 0666);
        if (pack_fd < 0)
-               die("Can't create pack file %s: %s", pack_name, strerror(errno));
+               die("Can't create %s: %s", pack_name, strerror(errno));
 
        alloc_objects(est_obj_cnt);
        init_pack_header();
        for (;;) {
-               unsigned long datlen;
-               void *dat;
-
-               if (yread(0, &datlen, 4) != 4)
+               unsigned long cmd;
+               if (yread(0, &cmd, 4) != 4)
                        break;
 
-               dat = xmalloc(datlen);
-               if (yread(0, dat, datlen) != datlen)
-                       break;
-
-               if (!store_object(OBJ_BLOB, dat, datlen, &last_blob))
-                       free(dat);
+               switch (cmd) {
+               case 'blob': new_blob(); break;
+               default:
+                       die("Invalid command %lu", cmd);
+               }
        }
        fixup_header_footer();
        close(pack_fd);
        write_index(idx_name);
 
-       fprintf(stderr, "%lu objects, %lu duplicates, %lu allocated (%lu overflow)\n",
-               object_count, duplicate_count, alloc_count, alloc_count - est_obj_cnt);
+       fprintf(stderr, "%s statistics:\n", argv[0]);
+       fprintf(stderr, "---------------------------------------------------\n");
+       fprintf(stderr, "Alloc'd objects: %10lu (%10lu overflow  )\n", alloc_count, alloc_count - est_obj_cnt);
+       fprintf(stderr, "Total objects:   %10lu (%10lu duplicates)\n", object_count, duplicate_count);
+       fprintf(stderr, "      blobs  :   %10lu (%10lu duplicates)\n", object_count_by_type[OBJ_BLOB], duplicate_count_by_type[OBJ_BLOB]);
+       fprintf(stderr, "      trees  :   %10lu (%10lu duplicates)\n", object_count_by_type[OBJ_TREE], duplicate_count_by_type[OBJ_TREE]);
+       fprintf(stderr, "      commits:   %10lu (%10lu duplicates)\n", object_count_by_type[OBJ_COMMIT], duplicate_count_by_type[OBJ_COMMIT]);
+       fprintf(stderr, "      tags   :   %10lu (%10lu duplicates)\n", object_count_by_type[OBJ_TAG], duplicate_count_by_type[OBJ_TAG]);
+       fprintf(stderr, "---------------------------------------------------\n");
+
+       stat(pack_name, &sb);
+       fprintf(stderr, "Pack size:       %10lu KiB\n", (unsigned long)(sb.st_size/1024));
+       stat(idx_name, &sb);
+       fprintf(stderr, "Index size:      %10lu KiB\n", (unsigned long)(sb.st_size/1024));
+
+       fprintf(stderr, "\n");
 
        return 0;
 }