Go 파일 사이즈 구하기

Go 파일 사이즈 구하기

파일사이즈 구하는 소스

package main

import (
    "os"
    "strings"
    "fmt"
    "log"
)

const (
    BYTE     = 1.0
    KILOBYTE = 1024 * BYTE
    MEGABYTE = 1024 * KILOBYTE
    GIGABYTE = 1024 * MEGABYTE
    TERABYTE = 1024 * GIGABYTE
)

func ByteSize(bytes uint64) string {
    unit := ""
    value := float32(bytes)

    switch {
    case bytes >= TERABYTE:
        unit = "T"
        value = value / TERABYTE
    case bytes >= GIGABYTE:
        unit = "G"
        value = value / GIGABYTE
    case bytes >= MEGABYTE:
        unit = "M"
        value = value / MEGABYTE
    case bytes >= KILOBYTE:
        unit = "K"
        value = value / KILOBYTE
    case bytes >= BYTE:
        unit = "B"
    case bytes == 0:
        return "0"
    }

    stringValue := fmt.Sprintf("%.1f", value)
    stringValue = strings.TrimSuffix(stringValue, ".0")
    return fmt.Sprintf("%s%s", stringValue, unit)
}

func main() {
    stat, err := os.Stat(filePath)
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println(ByteSize(uint64(stat.Size()))
}