JOJ3/pkg/healthcheck/reposize.go
zzjc1234 dd40a17476
Some checks failed
build / trigger-build-image (push) Blocked by required conditions
build / build (push) Has been cancelled
build / build (pull_request) Failing after 1m14s
build / trigger-build-image (pull_request) Has been skipped
fix(healthcheck/reposize): output format
2024-10-12 17:59:26 +08:00

41 lines
1.1 KiB
Go

package healthcheck
import (
"fmt"
"log/slog"
"os/exec"
"strconv"
"strings"
)
// RepoSize checks the size of the repository to determine if it is oversized.
// It executes the 'git count-objects -v' command to obtain the size information,
func RepoSize(conf_size float64) error {
// TODO: reimplement here when go-git is available
// https://github.com/go-git/go-git/blob/master/COMPATIBILITY.md
cmd := exec.Command("git", "count-objects", "-v")
output, err := cmd.CombinedOutput()
if err != nil {
slog.Error("running git command:", "err", err)
return fmt.Errorf("error running git command: %w", err)
}
lines := strings.Split(string(output), "\n")
var sum int
for _, line := range lines {
if strings.Contains(line, "size") {
fields := strings.Fields(line)
sizeStr := fields[1]
size, err := strconv.Atoi(sizeStr)
if err != nil {
slog.Error("running git command:", "err", err)
return fmt.Errorf("error running git command: %w", err)
}
sum += size
}
}
if sum > int(conf_size*1024) {
return fmt.Errorf("Repository larger than %.1f Mib. Please clean up or contact the teaching team.", conf_size)
}
return nil
}