]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | * Builtin "git count-objects". | |
3 | * | |
4 | * Copyright (c) 2006 Junio C Hamano | |
5 | */ | |
6 | ||
7 | #include "cache.h" | |
8 | #include "config.h" | |
9 | #include "dir.h" | |
10 | #include "repository.h" | |
11 | #include "builtin.h" | |
12 | #include "parse-options.h" | |
13 | #include "quote.h" | |
14 | #include "packfile.h" | |
15 | #include "object-store.h" | |
16 | ||
17 | static unsigned long garbage; | |
18 | static off_t size_garbage; | |
19 | static int verbose; | |
20 | static unsigned long loose, packed, packed_loose; | |
21 | static off_t loose_size; | |
22 | ||
23 | static const char *bits_to_msg(unsigned seen_bits) | |
24 | { | |
25 | switch (seen_bits) { | |
26 | case 0: | |
27 | return "no corresponding .idx or .pack"; | |
28 | case PACKDIR_FILE_GARBAGE: | |
29 | return "garbage found"; | |
30 | case PACKDIR_FILE_PACK: | |
31 | return "no corresponding .idx"; | |
32 | case PACKDIR_FILE_IDX: | |
33 | return "no corresponding .pack"; | |
34 | case PACKDIR_FILE_PACK|PACKDIR_FILE_IDX: | |
35 | default: | |
36 | return NULL; | |
37 | } | |
38 | } | |
39 | ||
40 | static void real_report_garbage(unsigned seen_bits, const char *path) | |
41 | { | |
42 | struct stat st; | |
43 | const char *desc = bits_to_msg(seen_bits); | |
44 | ||
45 | if (!desc) | |
46 | return; | |
47 | ||
48 | if (!stat(path, &st)) | |
49 | size_garbage += st.st_size; | |
50 | warning("%s: %s", desc, path); | |
51 | garbage++; | |
52 | } | |
53 | ||
54 | static void loose_garbage(const char *path) | |
55 | { | |
56 | if (verbose) | |
57 | report_garbage(PACKDIR_FILE_GARBAGE, path); | |
58 | } | |
59 | ||
60 | static int count_loose(const struct object_id *oid, const char *path, void *data) | |
61 | { | |
62 | struct stat st; | |
63 | ||
64 | if (lstat(path, &st) || !S_ISREG(st.st_mode)) | |
65 | loose_garbage(path); | |
66 | else { | |
67 | loose_size += on_disk_bytes(st); | |
68 | loose++; | |
69 | if (verbose && has_object_pack(oid)) | |
70 | packed_loose++; | |
71 | } | |
72 | return 0; | |
73 | } | |
74 | ||
75 | static int count_cruft(const char *basename, const char *path, void *data) | |
76 | { | |
77 | loose_garbage(path); | |
78 | return 0; | |
79 | } | |
80 | ||
81 | static int print_alternate(struct object_directory *odb, void *data) | |
82 | { | |
83 | printf("alternate: "); | |
84 | quote_c_style(odb->path, NULL, stdout, 0); | |
85 | putchar('\n'); | |
86 | return 0; | |
87 | } | |
88 | ||
89 | static char const * const count_objects_usage[] = { | |
90 | N_("git count-objects [-v] [-H | --human-readable]"), | |
91 | NULL | |
92 | }; | |
93 | ||
94 | int cmd_count_objects(int argc, const char **argv, const char *prefix) | |
95 | { | |
96 | int human_readable = 0; | |
97 | struct option opts[] = { | |
98 | OPT__VERBOSE(&verbose, N_("be verbose")), | |
99 | OPT_BOOL('H', "human-readable", &human_readable, | |
100 | N_("print sizes in human readable format")), | |
101 | OPT_END(), | |
102 | }; | |
103 | ||
104 | git_config(git_default_config, NULL); | |
105 | ||
106 | argc = parse_options(argc, argv, prefix, opts, count_objects_usage, 0); | |
107 | /* we do not take arguments other than flags for now */ | |
108 | if (argc) | |
109 | usage_with_options(count_objects_usage, opts); | |
110 | if (verbose) { | |
111 | report_garbage = real_report_garbage; | |
112 | report_linked_checkout_garbage(); | |
113 | } | |
114 | ||
115 | for_each_loose_file_in_objdir(get_object_directory(), | |
116 | count_loose, count_cruft, NULL, NULL); | |
117 | ||
118 | if (verbose) { | |
119 | struct packed_git *p; | |
120 | unsigned long num_pack = 0; | |
121 | off_t size_pack = 0; | |
122 | struct strbuf loose_buf = STRBUF_INIT; | |
123 | struct strbuf pack_buf = STRBUF_INIT; | |
124 | struct strbuf garbage_buf = STRBUF_INIT; | |
125 | ||
126 | for (p = get_all_packs(the_repository); p; p = p->next) { | |
127 | if (!p->pack_local) | |
128 | continue; | |
129 | if (open_pack_index(p)) | |
130 | continue; | |
131 | packed += p->num_objects; | |
132 | size_pack += p->pack_size + p->index_size; | |
133 | num_pack++; | |
134 | } | |
135 | ||
136 | if (human_readable) { | |
137 | strbuf_humanise_bytes(&loose_buf, loose_size); | |
138 | strbuf_humanise_bytes(&pack_buf, size_pack); | |
139 | strbuf_humanise_bytes(&garbage_buf, size_garbage); | |
140 | } else { | |
141 | strbuf_addf(&loose_buf, "%lu", | |
142 | (unsigned long)(loose_size / 1024)); | |
143 | strbuf_addf(&pack_buf, "%lu", | |
144 | (unsigned long)(size_pack / 1024)); | |
145 | strbuf_addf(&garbage_buf, "%lu", | |
146 | (unsigned long)(size_garbage / 1024)); | |
147 | } | |
148 | ||
149 | printf("count: %lu\n", loose); | |
150 | printf("size: %s\n", loose_buf.buf); | |
151 | printf("in-pack: %lu\n", packed); | |
152 | printf("packs: %lu\n", num_pack); | |
153 | printf("size-pack: %s\n", pack_buf.buf); | |
154 | printf("prune-packable: %lu\n", packed_loose); | |
155 | printf("garbage: %lu\n", garbage); | |
156 | printf("size-garbage: %s\n", garbage_buf.buf); | |
157 | foreach_alt_odb(print_alternate, NULL); | |
158 | strbuf_release(&loose_buf); | |
159 | strbuf_release(&pack_buf); | |
160 | strbuf_release(&garbage_buf); | |
161 | } else { | |
162 | struct strbuf buf = STRBUF_INIT; | |
163 | if (human_readable) | |
164 | strbuf_humanise_bytes(&buf, loose_size); | |
165 | else | |
166 | strbuf_addf(&buf, "%lu kilobytes", | |
167 | (unsigned long)(loose_size / 1024)); | |
168 | printf("%lu objects, %s\n", loose, buf.buf); | |
169 | strbuf_release(&buf); | |
170 | } | |
171 | return 0; | |
172 | } |