git.sesse.net Git - remoteglot-book/blob - build-book.sh

   1 #! /bin/bash
   2
   3 set -e
   4
   5 export MAPPERS=40
   6 export REDUCERS=40
   7 export SHARDS=40
   8
   9 rm -f pgnnames.txt tmp.mtbl* open.mtbl.part????.new 2>/dev/null
  10
  11 for FILE in "$@"; do
  12         echo "$FILE" >> pgnnames.txt
  13 done
  14
  15 TS_BEGIN=$( date +%s%3N )
  16
  17 for X in $( seq 0 $(( MAPPERS - 1 )) ); do
  18         (
  19                 (
  20                         PGNNUM=0
  21                         for FILE in "$@"; do
  22                                 START=$( ./find-pgn-split-point.sh "$FILE" $X $MAPPERS )
  23                                 END=$( ./find-pgn-split-point.sh "$FILE" $(( X + 1 )) $MAPPERS )
  24                                 ~/nmu/pgn-extract/pgn-extract --startpos $START --endpos $END --startfilenum $PGNNUM -e -Wsessebin "$FILE" 2>/dev/null
  25                                 PGNNUM=$(( PGNNUM + 1 ))
  26                         done
  27                 ) | ./binloader - tmp.mtbl.mapper$X $SHARDS 500000
  28         ) &
  29 done
  30 wait
  31
  32 TS_MAPDONE=$( date +%s%3N )
  33 MAPOUTPUT=$( du -sch tmp.mtbl.mapper* | tail -n 1 | cut -f1 | sed 's/\([^0-9]\)$/ \1/' )
  34
  35 echo "Reducing..."
  36
  37 parallel -i -j $REDUCERS bash -c 'X={}; ./binmerger tmp.mtbl.mapper*.part$( printf %04d $X ).subshard* open.mtbl.part$( printf %04d $X ).new' -- $( seq 0 $(( SHARDS - 1 )) )
  38
  39 TS_REDUCEDONE=$( date +%s%3N )
  40 REDUCEOUTPUT=$( du -sch open.mtbl.part*.new | tail -n 1 | cut -f1 | sed 's/\([^0-9]\)$/ \1/' )
  41
  42 for X in $( seq 0 $(( SHARDS - 1 )) ); do
  43         mv open.mtbl.part$( printf %04d $X ).new open.mtbl.part$( printf %04d $X )
  44 done
  45 rm -f tmp.mtbl*
  46
  47 TS_END=$( date +%s%3N )
  48
  49 printf "\n"
  50 printf "  Map time:       %7d ms\n" $(( TS_MAPDONE - TS_BEGIN ))
  51 printf "  Reduce time:    %7d ms\n" $(( TS_REDUCEDONE - TS_MAPDONE ))
  52 printf "  Total time:     %7d ms\n" $(( TS_END - TS_BEGIN ))
  53 printf "\n"
  54 printf "  Map output:       %7sB\n" "$MAPOUTPUT"
  55 printf "  Reduce output:    %7sB\n" "$REDUCEOUTPUT"
  56 printf "\n"