周赵嘉程521432910016
901290d263
Co-authored-by: zzjc1234 <2359047351@qq.com> Co-authored-by: Boming Zhang <bomingzh@sjtu.edu.cn> Reviewed-on: #56 Reviewed-by: 张泊明518370910136 <bomingzh@sjtu.edu.cn> Co-authored-by: 周赵嘉程521432910016 <zzjc123@sjtu.edu.cn> Co-committed-by: 周赵嘉程521432910016 <zzjc123@sjtu.edu.cn>
41 lines
1.1 KiB
Go
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(confSize 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(confSize*1024) {
|
|
return fmt.Errorf("Repository larger than %.1f MiB. Please clean up or contact the teaching team.", confSize)
|
|
}
|
|
return nil
|
|
}
|