github.com/ethw3/go-ethereuma@v0.0.0-20221013053120-c14602a4c23c/cmd/utils/diskusage.go (about)

     1  // Copyright 2021 The go-ethereum Authors
     2  // This file is part of go-ethereum.
     3  //
     4  // go-ethereum 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, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // go-ethereum is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    12  // GNU General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU General Public License
    15  // along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  //go:build !windows && !openbsd
    18  // +build !windows,!openbsd
    19  
    20  package utils
    21  
    22  import (
    23  	"fmt"
    24  
    25  	"golang.org/x/sys/unix"
    26  )
    27  
    28  func getFreeDiskSpace(path string) (uint64, error) {
    29  	var stat unix.Statfs_t
    30  	if err := unix.Statfs(path, &stat); err != nil {
    31  		return 0, fmt.Errorf("failed to call Statfs: %v", err)
    32  	}
    33  
    34  	// Available blocks * size per block = available space in bytes
    35  	var bavail = stat.Bavail
    36  	// nolint:staticcheck
    37  	if stat.Bavail < 0 {
    38  		// FreeBSD can have a negative number of blocks available
    39  		// because of the grace limit.
    40  		bavail = 0
    41  	}
    42  	//nolint:unconvert
    43  	return uint64(bavail) * uint64(stat.Bsize), nil
    44  }