1 7a4ee46d 2003-11-23 devnull #include "stdinc.h"
2 7a4ee46d 2003-11-23 devnull #include "dat.h"
3 7a4ee46d 2003-11-23 devnull #include "fns.h"
5 7a4ee46d 2003-11-23 devnull static int
6 7a4ee46d 2003-11-23 devnull writebucket(Index *ix, u32int buck, IBucket *ib, ZBlock *b)
8 7a4ee46d 2003-11-23 devnull ISect *is;
10 9ffbb5ad 2004-03-12 devnull is = findibucket(ix, buck, &buck);
11 9ffbb5ad 2004-03-12 devnull if(is == nil)
12 7a4ee46d 2003-11-23 devnull return -1;
13 7a4ee46d 2003-11-23 devnull qlock(&stats.lock);
14 7a4ee46d 2003-11-23 devnull stats.indexwrites++;
15 7a4ee46d 2003-11-23 devnull qunlock(&stats.lock);
16 7a4ee46d 2003-11-23 devnull packibucket(ib, b->data);
17 7a4ee46d 2003-11-23 devnull return writepart(is->part, is->blockbase + ((u64int)buck << is->blocklog), b->data, is->blocksize);
20 7a4ee46d 2003-11-23 devnull static int
21 7a4ee46d 2003-11-23 devnull buildindex(Index *ix, Part *part, u64int off, u64int clumps, int zero)
23 7a4ee46d 2003-11-23 devnull IEStream *ies;
24 7a4ee46d 2003-11-23 devnull IBucket ib, zib;
25 7a4ee46d 2003-11-23 devnull ZBlock *z, *b;
26 7a4ee46d 2003-11-23 devnull u32int next, buck;
28 7a4ee46d 2003-11-23 devnull u64int found = 0;
30 7a4ee46d 2003-11-23 devnull //ZZZ make buffer size configurable
31 7a4ee46d 2003-11-23 devnull b = alloczblock(ix->blocksize, 0);
32 7a4ee46d 2003-11-23 devnull z = alloczblock(ix->blocksize, 1);
33 7a4ee46d 2003-11-23 devnull ies = initiestream(part, off, clumps, 64*1024);
34 7a4ee46d 2003-11-23 devnull if(b == nil || z == nil || ies == nil){
36 7a4ee46d 2003-11-23 devnull goto breakout;
37 7a4ee46d 2003-11-23 devnull return -1;
40 7a4ee46d 2003-11-23 devnull next = 0;
41 7a4ee46d 2003-11-23 devnull ib.data = b->data + IBucketSize;
42 7a4ee46d 2003-11-23 devnull zib.data = z->data + IBucketSize;
43 7a4ee46d 2003-11-23 devnull zib.n = 0;
44 9ffbb5ad 2004-03-12 devnull zib.depth = 0;
46 7a4ee46d 2003-11-23 devnull buck = buildbucket(ix, ies, &ib);
47 7a4ee46d 2003-11-23 devnull found += ib.n;
48 7a4ee46d 2003-11-23 devnull if(zero){
49 7a4ee46d 2003-11-23 devnull for(; next != buck; next++){
50 7a4ee46d 2003-11-23 devnull if(next == ix->buckets){
51 7a4ee46d 2003-11-23 devnull if(buck != TWID32){
52 7a4ee46d 2003-11-23 devnull fprint(2, "bucket out of range\n");
55 7a4ee46d 2003-11-23 devnull goto breakout;
57 7a4ee46d 2003-11-23 devnull if(writebucket(ix, next, &zib, z) < 0){
58 7a4ee46d 2003-11-23 devnull fprint(2, "can't write zero bucket to buck=%d: %r", next);
63 7a4ee46d 2003-11-23 devnull if(buck >= ix->buckets){
64 7a4ee46d 2003-11-23 devnull if(buck == TWID32)
66 7a4ee46d 2003-11-23 devnull fprint(2, "bucket out of range\n");
68 7a4ee46d 2003-11-23 devnull goto breakout;
70 7a4ee46d 2003-11-23 devnull if(writebucket(ix, buck, &ib, b) < 0){
71 7a4ee46d 2003-11-23 devnull fprint(2, "bad bucket found=%lld: %r\n", found);
74 7a4ee46d 2003-11-23 devnull next = buck + 1;
76 7a4ee46d 2003-11-23 devnull breakout:;
77 7a4ee46d 2003-11-23 devnull fprint(2, "constructed index with %lld entries\n", found);
78 7a4ee46d 2003-11-23 devnull freeiestream(ies);
79 7a4ee46d 2003-11-23 devnull freezblock(z);
80 7a4ee46d 2003-11-23 devnull freezblock(b);
81 7a4ee46d 2003-11-23 devnull return ok;
85 7a4ee46d 2003-11-23 devnull usage(void)
87 7a4ee46d 2003-11-23 devnull fprint(2, "usage: buildindex [-Z] [-B blockcachesize] config tmppart\n");
88 7a4ee46d 2003-11-23 devnull threadexitsall(0);
92 7a4ee46d 2003-11-23 devnull threadmain(int argc, char *argv[])
94 7a4ee46d 2003-11-23 devnull Part *part;
95 7a4ee46d 2003-11-23 devnull u64int clumps, base;
96 7a4ee46d 2003-11-23 devnull u32int bcmem;
97 7a4ee46d 2003-11-23 devnull int zero;
99 7a4ee46d 2003-11-23 devnull zero = 1;
100 7a4ee46d 2003-11-23 devnull bcmem = 0;
101 7a4ee46d 2003-11-23 devnull ARGBEGIN{
102 7a4ee46d 2003-11-23 devnull case 'B':
103 7a4ee46d 2003-11-23 devnull bcmem = unittoull(ARGF());
105 7a4ee46d 2003-11-23 devnull case 'Z':
106 7a4ee46d 2003-11-23 devnull zero = 0;
108 7a4ee46d 2003-11-23 devnull default:
109 7a4ee46d 2003-11-23 devnull usage();
113 7a4ee46d 2003-11-23 devnull if(argc != 2)
114 7a4ee46d 2003-11-23 devnull usage();
116 7a4ee46d 2003-11-23 devnull if(initventi(argv[0]) < 0)
117 7a4ee46d 2003-11-23 devnull sysfatal("can't init venti: %r");
119 7a4ee46d 2003-11-23 devnull if(bcmem < maxblocksize * (mainindex->narenas + mainindex->nsects * 4 + 16))
120 7a4ee46d 2003-11-23 devnull bcmem = maxblocksize * (mainindex->narenas + mainindex->nsects * 4 + 16);
121 7a4ee46d 2003-11-23 devnull fprint(2, "initialize %d bytes of disk block cache\n", bcmem);
122 7a4ee46d 2003-11-23 devnull initdcache(bcmem);
124 7a4ee46d 2003-11-23 devnull fprint(2, "building a new index %s using %s for temporary storage\n", mainindex->name, argv[1]);
126 7a4ee46d 2003-11-23 devnull part = initpart(argv[1], 1);
127 7a4ee46d 2003-11-23 devnull if(part == nil)
128 7a4ee46d 2003-11-23 devnull sysfatal("can't initialize temporary partition: %r");
130 7a4ee46d 2003-11-23 devnull clumps = sortrawientries(mainindex, part, &base);
131 7a4ee46d 2003-11-23 devnull if(clumps == TWID64)
132 7a4ee46d 2003-11-23 devnull sysfatal("can't build sorted index: %r");
133 7a4ee46d 2003-11-23 devnull fprint(2, "found and sorted index entries for clumps=%lld at %lld\n", clumps, base);
135 7a4ee46d 2003-11-23 devnull if(buildindex(mainindex, part, base, clumps, zero) < 0)
136 7a4ee46d 2003-11-23 devnull sysfatal("can't build new index: %r");
138 7a4ee46d 2003-11-23 devnull threadexitsall(0);
