57 lines
1.4 KiB
Ruby
57 lines
1.4 KiB
Ruby
module Gitlab
|
|
module GitalyClient
|
|
class BlobService
|
|
def initialize(repository)
|
|
@gitaly_repo = repository.gitaly_repository
|
|
end
|
|
|
|
def get_blob(oid:, limit:)
|
|
request = Gitaly::GetBlobRequest.new(
|
|
repository: @gitaly_repo,
|
|
oid: oid,
|
|
limit: limit
|
|
)
|
|
response = GitalyClient.call(@gitaly_repo.storage_name, :blob_service, :get_blob, request)
|
|
|
|
data = ''
|
|
blob = nil
|
|
response.each do |msg|
|
|
if blob.nil?
|
|
blob = msg
|
|
end
|
|
|
|
data << msg.data
|
|
end
|
|
|
|
return nil if blob.oid.blank?
|
|
|
|
Gitlab::Git::Blob.new(
|
|
id: blob.oid,
|
|
size: blob.size,
|
|
data: data,
|
|
binary: Gitlab::Git::Blob.binary?(data)
|
|
)
|
|
end
|
|
|
|
def batch_lfs_pointers(blob_ids)
|
|
request = Gitaly::GetLFSPointersRequest.new(
|
|
repository: @gitaly_repo,
|
|
blob_ids: blob_ids
|
|
)
|
|
|
|
response = GitalyClient.call(@gitaly_repo.storage_name, :blob_service, :get_lfs_pointers, request)
|
|
|
|
response.flat_map do |message|
|
|
message.lfs_pointers.map do |lfs_pointer|
|
|
Gitlab::Git::Blob.new(
|
|
id: lfs_pointer.oid,
|
|
size: lfs_pointer.size,
|
|
data: lfs_pointer.data,
|
|
binary: Gitlab::Git::Blob.binary?(lfs_pointer.data)
|
|
)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|