Choose the storoge provider with the least latency on initial configuration

Author: faza

cmd/git-gitopia/lfs/init.go

@@ -59,6 +59,15 @@ func InitCommand() *cobra.Command {
 					return err
 				}
 			}
+			gitServerHost, _ := config.GitConfigGet(config.GitopiaConfigGitServerHostOption)
+			if gitServerHost == "" {
+				gitServerHost = api.GetBestGitServerHost(grpcHost)
+				if gitServerHost != "" {
+					if err := api.SetConfiguredGitServerHost(gitServerHost); err != nil {
+						return err
+					}
+				}
+			}
 
 			grpcConn, err := grpc.Dial(grpcHost,
 				grpc.WithTransportCredentials(insecure.NewCredentials()),
@@ -81,7 +90,11 @@ func InitCommand() *cobra.Command {
 			}
 
 			remoteRepository := *res.Repository
-			lfsURL := fmt.Sprintf("%v/%v.git", config.GitServerHost, remoteRepository.Id)
+			gitServerHost, _ = config.GitConfigGet(config.GitopiaConfigGitServerHostOption)
+			if gitServerHost == "" {
+				gitServerHost = config.GitServerHost
+			}
+			lfsURL := fmt.Sprintf("%v/%v.git", gitServerHost, remoteRepository.Id)
 
 			c = core.GitCommand("git", "config",
 				fmt.Sprintf("--file=%s", lfsConfigPath),

cmd/git-remote-gitopia/gitopia.go

@@ -68,6 +68,16 @@ func (h *GitopiaHandler) Initialize(remote *core.Remote) error {
 		}
 	}
 
+	gitServerHost, _ := config.GitConfigGet(config.GitopiaConfigGitServerHostOption)
+	if gitServerHost == "" {
+		gitServerHost = api.GetBestGitServerHost(grpcHost)
+		if gitServerHost != "" {
+			if err := api.SetConfiguredGitServerHost(gitServerHost); err != nil {
+				return err
+			}
+		}
+	}
+
 	h.grpcConn, err = grpc.Dial(grpcHost, grpc.WithTransportCredentials(insecure.NewCredentials()))
 	if err != nil {
 		return err
@@ -138,7 +148,11 @@ func (h *GitopiaHandler) List(remote *core.Remote, forPush bool) ([]string, erro
 }
 
 func (h *GitopiaHandler) Fetch(remote *core.Remote, refsToFetch []core.RefToFetch) error {
-	remoteURL := fmt.Sprintf("%v/%v.git", config.GitServerHost, h.remoteRepository.Id)
+	gitServerHost, _ := config.GitConfigGet(config.GitopiaConfigGitServerHostOption)
+	if gitServerHost == "" {
+		gitServerHost = config.GitServerHost
+	}
+	remoteURL := fmt.Sprintf("%v/%v.git", gitServerHost, h.remoteRepository.Id)
 
 	if !remote.Force {
 		args := []string{
@@ -214,7 +228,11 @@ func (h *GitopiaHandler) Push(remote *core.Remote, refsToPush []core.RefToPush)
 		return nil, fmt.Errorf("fatal: you don't have write permissions to this repository")
 	}
 
-	remoteURL := fmt.Sprintf("%v/%v.git", config.GitServerHost, h.remoteRepository.Id)
+	gitServerHost, _ := config.GitConfigGet(config.GitopiaConfigGitServerHostOption)
+	if gitServerHost == "" {
+		gitServerHost = config.GitServerHost
+	}
+	remoteURL := fmt.Sprintf("%v/%v.git", gitServerHost, h.remoteRepository.Id)
 
 	var newRemoteRefSha string
 	var setBranches []gitopiatypes.MsgMultiSetBranch_Branch

core/api/gitserver.go

@@ -0,0 +1,42 @@
+package api
+
+import (
+	"net/http"
+	"time"
+
+	"github.com/gitopia/git-remote-gitopia/core"
+)
+
+func checkHttpHostLatency(host string) time.Duration {
+	start := time.Now()
+	_, err := http.Get(host)
+	if err != nil {
+		return time.Hour
+	}
+	return time.Since(start)
+}
+
+func GetBestGitServerHost(grpcHost string) string {
+	providers := GetActiveStorageProviders(grpcHost)
+	if len(providers) == 0 {
+		return ""
+	}
+
+	var bestHost string
+	bestLatency := time.Hour
+
+	for _, p := range providers {
+		latency := checkHttpHostLatency(p.ApiUrl)
+		if latency < bestLatency {
+			bestHost = p.ApiUrl
+			bestLatency = latency
+		}
+	}
+
+	return bestHost
+}
+
+func SetConfiguredGitServerHost(host string) error {
+	cmd := core.GitCommand("git", "config", "--global", "gitopia.gitServerHost", host)
+	return cmd.Run()
+}

core/api/grpc.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/cosmos/cosmos-sdk/client/grpc/tmservice"
 	"github.com/gitopia/git-remote-gitopia/core"
+	storagetypes "github.com/gitopia/gitopia/v6/x/storage/types"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials/insecure"
 )
@@ -50,3 +51,18 @@ func SetConfiguredGRPCHost(host string) error {
 	cmd := core.GitCommand("git", "config", "--global", "gitopia.grpcHost", host)
 	return cmd.Run()
 }
+
+func GetActiveStorageProviders(host string) []storagetypes.Provider {
+	conn, err := grpc.Dial(host, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	if err != nil {
+		return nil
+	}
+	defer conn.Close()
+
+	client := storagetypes.NewQueryClient(conn)
+	res, err := client.ActiveProviders(context.Background(), &storagetypes.QueryActiveProvidersRequest{})
+	if err != nil {
+		return nil
+	}
+	return res.Providers
+}