mirror of https://github.com/harness/drone.git
65 lines
1.8 KiB
Go
65 lines
1.8 KiB
Go
// Copyright 2022 Harness Inc. All rights reserved.
|
|
// Use of this source code is governed by the Polyform Free Trial License
|
|
// that can be found in the LICENSE.md file for this repository.
|
|
|
|
package server
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/harness/gitness/cache"
|
|
"github.com/harness/gitness/gitrpc/internal/gitea"
|
|
"github.com/harness/gitness/gitrpc/internal/service"
|
|
"github.com/harness/gitness/gitrpc/internal/types"
|
|
|
|
"github.com/go-redis/redis/v8"
|
|
"github.com/google/wire"
|
|
)
|
|
|
|
// WireSet provides a wire set for this package.
|
|
var WireSet = wire.NewSet(
|
|
ProvideServer,
|
|
ProvideHTTPServer,
|
|
ProvideGITAdapter,
|
|
ProvideGoGitRepoProvider,
|
|
ProvideLastCommitCache,
|
|
)
|
|
|
|
func ProvideGoGitRepoProvider() *gitea.GoGitRepoProvider {
|
|
const objectCacheSize = 16 << 20 // 16MiB
|
|
return gitea.NewGoGitRepoProvider(objectCacheSize, 15*time.Minute)
|
|
}
|
|
|
|
func ProvideLastCommitCache(
|
|
config Config,
|
|
redisClient redis.UniversalClient,
|
|
repoProvider *gitea.GoGitRepoProvider,
|
|
) cache.Cache[gitea.CommitEntryKey, *types.Commit] {
|
|
cacheDuration := time.Duration(config.LastCommitCache.DurationSeconds) * time.Second
|
|
|
|
if config.LastCommitCache.Mode == ModeNone || cacheDuration < time.Second {
|
|
return gitea.NoLastCommitCache(repoProvider)
|
|
}
|
|
|
|
if config.LastCommitCache.Mode == ModeRedis && redisClient != nil {
|
|
return gitea.NewRedisLastCommitCache(redisClient, cacheDuration, repoProvider)
|
|
}
|
|
|
|
return gitea.NewInMemoryLastCommitCache(cacheDuration, repoProvider)
|
|
}
|
|
|
|
func ProvideGITAdapter(
|
|
repoProvider *gitea.GoGitRepoProvider,
|
|
lastCommitCache cache.Cache[gitea.CommitEntryKey, *types.Commit],
|
|
) (service.GitAdapter, error) {
|
|
return gitea.New(repoProvider, lastCommitCache)
|
|
}
|
|
|
|
func ProvideServer(config Config, adapter service.GitAdapter) (*GRPCServer, error) {
|
|
return NewServer(config, adapter)
|
|
}
|
|
|
|
func ProvideHTTPServer(config Config) (*HTTPServer, error) {
|
|
return NewHTTPServer(config)
|
|
}
|