cmd/gitaly-debug/main.go

package main

import (
	"fmt"
	"io"
	"os"

	"gitlab.com/gitlab-org/gitaly/v16/internal/log"
)

const (
	usage = `Usage: gitaly-debug SUBCOMMAND ARGS

Subcommands:

simulate-http-clone GIT_DIR
	Simulates the server side workload of serving a full git clone over
	HTTP. The clone data is written to /dev/null. Note that in real life
	the workload also depends on the transport capabilities requested by
	the client; this tool uses a fixed set of capabilities.

analyze-http-clone HTTP_URL
	Clones a Git repository from a public HTTP URL into /dev/null.

list-bitmap-pack IDX_FILE
	List contents of packfile corresponding to IDX_FILE in offset
	order. Only works if there also is a .bitmap file for this
	pack.

map-bitmap-pack IDX_FILE
	Print map visualization of contents of packfile corresponding
	to IDX_FILE in offset order. Only works if there also is a
	.bitmap file for this pack.

list-bitmap-commits IDX_FILE
	List bitmap commits for .idx / .bitmap pair.

list-bitmap-reachable IDX_FILE BITMAP_COMMIT_ID
	List commits reachable from a bitmapped commit in a packfile.
`
)

func main() {
	logger := log.ConfigureCommand()

	if len(os.Args) < 2 {
		fatal(usage)
	}
	extraArgs := os.Args[2:]

	switch os.Args[1] {
	case "simulate-http-clone":
		if len(extraArgs) != 1 {
			fatal(usage)
		}
		simulateHTTPClone(extraArgs[0])
	case "analyze-http-clone":
		if len(extraArgs) != 1 {
			fatal(usage)
		}
		analyzeHTTPClone(extraArgs[0])
	case "list-bitmap-pack":
		if len(extraArgs) != 1 {
			fatal(usage)
		}
		listBitmapPack(logger, extraArgs[0])
	case "map-bitmap-pack":
		if len(extraArgs) != 1 {
			fatal(usage)
		}
		mapBitmapPack(logger, extraArgs[0])
	case "list-bitmap-commits":
		if len(extraArgs) != 1 {
			fatal(usage)
		}
		listBitmapCommits(logger, extraArgs[0])
	case "list-bitmap-reachable":
		if len(extraArgs) != 2 {
			fatal(usage)
		}
		listBitmapReachable(logger, extraArgs[0], extraArgs[1])
	default:
		fatal(usage)
	}
}

func noError(err error) {
	if err != nil {
		fatal(err)
	}
}

func fatal(a interface{}) {
	msg("%v", a)
	os.Exit(1)
}

func msg(format string, a ...interface{}) {
	fmt.Fprintln(os.Stderr, fmt.Sprintf(format, a...))
}

func humanBytes(n int64) string {
	units := []struct {
		size  int64
		label string
	}{
		{size: 1000000000000, label: "TB"},
		{size: 1000000000, label: "GB"},
		{size: 1000000, label: "MB"},
		{size: 1000, label: "KB"},
	}

	for _, u := range units {
		if n > u.size {
			return fmt.Sprintf("%.2f %s", float32(n)/float32(u.size), u.label)
		}
	}

	return fmt.Sprintf("%d bytes", n)
}

func fprintf(writer io.Writer, format string, args ...interface{}) {
	_, err := fmt.Fprintf(writer, format, args...)
	noError(err)
}

func fprintln(writer io.Writer, args ...interface{}) {
	_, err := fmt.Fprintln(writer, args...)
	noError(err)
}

func fprint(w io.Writer, a ...interface{}) {
	_, err := fmt.Fprint(w, a...)
	noError(err)
}

func flush(out interface{ Flush() error }) {
	noError(out.Flush())
}