Create test-list-objects helper program to output a random sample of
OIDs present in the repo.

If no command line arguments are provided, all OIDs are output.

The last command line argument specifies how many samples to output.
Samples are collected across the entire set of available OIDs to avoid
clustering and therefore quite uniformly distributed.

If a command line argument "--missing" is given before the sample count,
then a list of OIDs is generated without examining the repo.

    Signed-off-by: Derrick Stolee <dsto...@microsoft.com>
---
 Makefile                     |  1 +
 t/helper/.gitignore          |  1 +
 t/helper/test-list-objects.c | 87 ++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 89 insertions(+)
 create mode 100644 t/helper/test-list-objects.c

diff --git a/Makefile b/Makefile
index ed4ca438b..50a2eab80 100644
--- a/Makefile
+++ b/Makefile
@@ -652,6 +652,7 @@ TEST_PROGRAMS_NEED_X += test-hashmap
 TEST_PROGRAMS_NEED_X += test-index-version
 TEST_PROGRAMS_NEED_X += test-lazy-init-name-hash
 TEST_PROGRAMS_NEED_X += test-line-buffer
+TEST_PROGRAMS_NEED_X += test-list-objects
 TEST_PROGRAMS_NEED_X += test-match-trees
 TEST_PROGRAMS_NEED_X += test-mergesort
 TEST_PROGRAMS_NEED_X += test-mktemp
diff --git a/t/helper/.gitignore b/t/helper/.gitignore
index 7c9d28a83..9696f54bb 100644
--- a/t/helper/.gitignore
+++ b/t/helper/.gitignore
@@ -13,6 +13,7 @@
 /test-index-version
 /test-lazy-init-name-hash
 /test-line-buffer
+/test-list-objects
 /test-match-trees
 /test-mergesort
 /test-mktemp
diff --git a/t/helper/test-list-objects.c b/t/helper/test-list-objects.c
new file mode 100644
index 000000000..22bc9b4e6
--- /dev/null
+++ b/t/helper/test-list-objects.c
@@ -0,0 +1,87 @@
+#include "cache.h"
+#include "packfile.h"
+
+struct count {
+       int total;
+       int select_mod;
+};
+
+int count_loose(const struct object_id *oid,
+               const char *path,
+               void *data)
+{
+       ((struct count*)data)->total++;
+       return 0;
+}
+
+int count_packed(const struct object_id *oid,
+                struct packed_git *pack,
+                uint32_t pos,
+                void* data)
+{
+       ((struct count*)data)->total++;
+       return 0;
+}
+
+void output(const struct object_id *oid,
+           struct count *c)
+{
+       if (!(c->total % c->select_mod))
+               printf("%s\n", oid_to_hex(oid));
+       c->total--;
+}
+
+int output_loose(const struct object_id *oid,
+                const char *path,
+                void *data)
+{
+       output(oid, (struct count*)data);
+       return 0;
+}
+
+int output_packed(const struct object_id *oid,
+                 struct packed_git *pack,
+                 uint32_t pos,
+                 void* data)
+{
+       output(oid, (struct count*)data);
+       return 0;
+}
+
+int cmd_main(int ac, const char **av)
+{
+       uint32_t hash_delt = 0xFDB97531;
+       uint32_t hash_base = 0x01020304;
+       int i, n = -1;
+       struct count c;
+       const int u_per_oid = sizeof(struct object_id) / sizeof(uint32_t);
+
+       c.total = 0;
+       if (ac > 1)
+               n = atoi(av[ac - 1]);
+
+       if (ac > 2 && !strcmp(av[1], "--missing")) {
+               while (c.total++ < n) {
+                       for (i = 0; i < u_per_oid; i++) {
+                               printf("%08x", hash_base);
+                               hash_base += hash_delt;
+                       }
+                       printf("\n");
+               }
+       } else {
+               setup_git_directory();
+
+               if (n > 0) {
+                       for_each_loose_object(count_loose, &c, 0);
+                       for_each_packed_object(count_packed, &c, 0);
+                       c.select_mod = 1 + c.total / n;
+               } else {
+                       c.select_mod = 1;
+               }
+
+               for_each_loose_object(output_loose, &c, 0);
+               for_each_packed_object(output_packed, &c, 0);
+       }
+
+       exit(0);
+}
-- 
2.14.1.538.g56ec8fc98.dirty

Reply via email to