1/********************************************************************
2 * Copyright (c) 2013 - 2014, Pivotal Inc.
3 * All rights reserved.
4 *
5 * Author: Zhanwei Wang
6 ********************************************************************/
7/********************************************************************
8 * 2014 -
9 * open source under Apache License Version 2.0
10 ********************************************************************/
11/**
12 * Licensed to the Apache Software Foundation (ASF) under one
13 * or more contributor license agreements. See the NOTICE file
14 * distributed with this work for additional information
15 * regarding copyright ownership. The ASF licenses this file
16 * to you under the Apache License, Version 2.0 (the
17 * "License"); you may not use this file except in compliance
18 * with the License. You may obtain a copy of the License at
19 *
20 * http://www.apache.org/licenses/LICENSE-2.0
21 *
22 * Unless required by applicable law or agreed to in writing, software
23 * distributed under the License is distributed on an "AS IS" BASIS,
24 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
25 * See the License for the specific language governing permissions and
26 * limitations under the License.
27 */
28#ifndef _HDFS_LIBHDFS3_SERVER_LOCATEDBLOCKS_H_
29#define _HDFS_LIBHDFS3_SERVER_LOCATEDBLOCKS_H_
30
31#include "LocatedBlock.h"
32#include "Memory.h"
33
34#include <cassert>
35
36namespace Hdfs {
37namespace Internal {
38
39class LocatedBlocks {
40public:
41 virtual ~LocatedBlocks() {}
42
43 virtual int64_t getFileLength() const = 0;
44
45 virtual void setFileLength(int64_t fileLength) = 0;
46
47 virtual bool isLastBlockComplete() const = 0;
48
49 virtual void setIsLastBlockComplete(bool lastBlockComplete) = 0;
50
51 virtual shared_ptr<LocatedBlock> getLastBlock() = 0;
52
53 virtual void setLastBlock(shared_ptr<LocatedBlock> lastBlock) = 0;
54
55 virtual bool isUnderConstruction() const = 0;
56
57 virtual void setUnderConstruction(bool underConstruction) = 0;
58
59 virtual const LocatedBlock * findBlock(int64_t position) = 0;
60
61 virtual std::vector<LocatedBlock> & getBlocks() = 0;
62};
63
64/**
65 * Collection of blocks with their locations and the file length.
66 */
67class LocatedBlocksImpl : public LocatedBlocks {
68public:
69 int64_t getFileLength() const {
70 return fileLength;
71 }
72
73 void setFileLength(int64_t fileLength) {
74 this->fileLength = fileLength;
75 }
76
77 bool isLastBlockComplete() const {
78 return lastBlockComplete;
79 }
80
81 void setIsLastBlockComplete(bool lastBlockComplete) {
82 this->lastBlockComplete = lastBlockComplete;
83 }
84
85 shared_ptr<LocatedBlock> getLastBlock() {
86 assert(!lastBlockComplete);
87 return lastBlock;
88 }
89
90 void setLastBlock(shared_ptr<LocatedBlock> lastBlock) {
91 this->lastBlock = lastBlock;
92 }
93
94 bool isUnderConstruction() const {
95 return underConstruction;
96 }
97
98 void setUnderConstruction(bool underConstruction) {
99 this->underConstruction = underConstruction;
100 }
101
102 const LocatedBlock * findBlock(int64_t position);
103
104 std::vector<LocatedBlock> & getBlocks() {
105 return blocks;
106 }
107
108private:
109 bool lastBlockComplete;
110 bool underConstruction;
111 int64_t fileLength;
112 shared_ptr<LocatedBlock> lastBlock;
113 std::vector<LocatedBlock> blocks;
114
115};
116
117}
118}
119#endif /* _HDFS_LIBHDFS3_SERVER_LOCATEDBLOCKS_H_ */
120