1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
package objectpool
import (
"errors"
"gitlab.com/gitlab-org/gitaly/v16/internal/git/objectpool"
"gitlab.com/gitlab-org/gitaly/v16/internal/structerr"
"gitlab.com/gitlab-org/gitaly/v16/proto/go/gitalypb"
)
var (
errInvalidPoolDir = structerr.NewInvalidArgument("%w", objectpool.ErrInvalidPoolDir)
// errMissingPool is returned when the request is missing the object pool.
errMissingPool = structerr.NewInvalidArgument("no object pool repository")
)
// PoolRequest is the interface of a gRPC request that carries an object pool.
type PoolRequest interface {
GetObjectPool() *gitalypb.ObjectPool
}
// ExtractPool returns the pool repository from the request or an error if the
// request did no contain a pool.
func ExtractPool(req PoolRequest) (*gitalypb.Repository, error) {
poolRepo := req.GetObjectPool().GetRepository()
if poolRepo == nil {
return nil, errMissingPool
}
return poolRepo, nil
}
func (s *server) poolForRequest(req PoolRequest) (*objectpool.ObjectPool, error) {
pool, err := objectpool.FromProto(s.locator, s.gitCmdFactory, s.catfileCache, s.txManager, s.housekeepingManager, req.GetObjectPool())
if err != nil {
if err == objectpool.ErrInvalidPoolDir {
return nil, errInvalidPoolDir
}
if errors.Is(err, objectpool.ErrInvalidPoolRepository) {
return nil, structerr.NewFailedPrecondition("%w", err)
}
return nil, structerr.NewInternal("%w", err)
}
return pool, nil
}
|