rosetta-lbry/services/block_service.go
Thomas Zarebczan f92c407370
LBRY changes
more


more
2020-11-17 03:20:36 -05:00

103 lines
2.7 KiB
Go

// Copyright 2020 Coinbase, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package services
import (
"context"
"github.com/lbryio/rosetta-lbry/configuration"
"github.com/lbryio/rosetta-sdk-go/server"
"github.com/lbryio/rosetta-sdk-go/types"
)
// BlockAPIService implements the server.BlockAPIServicer interface.
type BlockAPIService struct {
config *configuration.Configuration
i Indexer
}
// NewBlockAPIService creates a new instance of a BlockAPIService.
func NewBlockAPIService(
config *configuration.Configuration,
i Indexer,
) server.BlockAPIServicer {
return &BlockAPIService{
config: config,
i: i,
}
}
// Block implements the /block endpoint.
func (s *BlockAPIService) Block(
ctx context.Context,
request *types.BlockRequest,
) (*types.BlockResponse, *types.Error) {
if s.config.Mode != configuration.Online {
return nil, wrapErr(ErrUnavailableOffline, nil)
}
blockResponse, err := s.i.GetBlockLazy(ctx, request.BlockIdentifier)
if err != nil {
return nil, wrapErr(ErrBlockNotFound, err)
}
// Direct client to fetch transactions individually if
// more than inlineFetchLimit.
if len(blockResponse.OtherTransactions) > inlineFetchLimit {
return blockResponse, nil
}
txs := make([]*types.Transaction, len(blockResponse.OtherTransactions))
for i, otherTx := range blockResponse.OtherTransactions {
transaction, err := s.i.GetBlockTransaction(
ctx,
blockResponse.Block.BlockIdentifier,
otherTx,
)
if err != nil {
return nil, wrapErr(ErrTransactionNotFound, err)
}
txs[i] = transaction
}
blockResponse.Block.Transactions = txs
blockResponse.OtherTransactions = nil
return blockResponse, nil
}
// BlockTransaction implements the /block/transaction endpoint.
func (s *BlockAPIService) BlockTransaction(
ctx context.Context,
request *types.BlockTransactionRequest,
) (*types.BlockTransactionResponse, *types.Error) {
if s.config.Mode != configuration.Online {
return nil, wrapErr(ErrUnavailableOffline, nil)
}
transaction, err := s.i.GetBlockTransaction(
ctx,
request.BlockIdentifier,
request.TransactionIdentifier,
)
if err != nil {
return nil, wrapErr(ErrTransactionNotFound, err)
}
return &types.BlockTransactionResponse{
Transaction: transaction,
}, nil
}