index.go

   1 // glocate -- ZFS-diff-friendly locate-like utility
   2 // Copyright (C) 2022-2024 Sergey Matveev <stargrave@stargrave.org>
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU General Public License as published by
   6 // the Free Software Foundation, version 3 of the License.
   7 //
   8 // This program is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 // GNU General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU General Public License
  14 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15
  16 package main
  17
  18 import (
  19         "fmt"
  20         "log"
  21         "os"
  22 )
  23
  24 func index() *os.File {
  25         tmp0, err := os.CreateTemp(TmpDir, "glocate-idx")
  26         if err != nil {
  27                 log.Fatalln(err)
  28         }
  29         defer os.Remove(tmp0.Name())
  30
  31         entsWalker := make(chan Ent, 1<<10)
  32         entsWriter := make(chan Ent, 1<<10)
  33         dirSizerJob := make(chan struct{})
  34         var dirSizes []int64
  35         entsWalker <- Ent{name: []string{"./"}}
  36         sinkBack := make(chan Ent, 1)
  37         go func() {
  38                 dirSizer(&dirSizes, 1, sinkBack, entsWalker, entsWriter)
  39                 close(dirSizerJob)
  40         }()
  41
  42         writerJob := make(chan struct{})
  43         go func() {
  44                 writer(tmp0, entsWriter)
  45                 close(writerJob)
  46         }()
  47
  48         walkerStatusStop := make(chan struct{})
  49         go walkerStatus(walkerStatusStop)
  50         err = walker(entsWalker, []string{"./"})
  51         walkerStatusStop <- struct{}{}
  52         <-walkerStatusStop
  53         fmt.Print("\r")
  54         if err != nil {
  55                 log.Fatalln(err)
  56         }
  57         close(entsWalker)
  58         <-dirSizerJob
  59         close(entsWriter)
  60         <-writerJob
  61
  62         tmp1 := applyDirSizes(tmp0, dirSizes)
  63         tmp0.Close()
  64         return tmp1
  65 }