hbase源码系列（八）从Snapshot恢复表
在看这一章之前，建议大家先去看一下snapshot的使用。这一章是上一章snapshot的续集，上一章了讲了怎么做snapshot的原理，这一章就怎么从snapshot恢复表。
restoreSnapshot方法位于HMaster当中，这个方法没几行代码，调用了SnapshotManager的restoreSnapshot方法。
// 检查meta表当中是否存在该表
if (MetaReader.tableExists(master.getCatalogTracker(), tableName)) {
//不能对在线的表进行恢复操作
if (master.getAssignmentManager().getZKTable().isEnabledTable(
TableName.valueOf(fsSnapshot.getTable()))) {
throw new UnsupportedOperationException("Table '" +
TableName.valueOf(fsSnapshot.getTable()) + "' must be disabled in order to " +
"perform a restore operation" +
".");
}
//从snapshot恢复表，通过提交RestoreSnapshotHandler
restoreSnapshot(fsSnapshot, snapshotTableDesc);
} else {
//如果meta表当中没有这个表（可能这个表被删除了，还是咋地），就克隆出来一张新表
HTableDescriptor htd = RestoreSnapshotHelper.cloneTableSchema(snapshotTableDesc, tableName);
//克隆snapshot到一个新的表，通过提交CloneSnapshotHandler
cloneSnapshot(fsSnapshot, htd);
}恢复之前先判断这个表还在不在，有可能表都被删除掉了，分开两种情况处理，但是我们也可以看到它只是通过两个handler去处理了，走的是线程池提交handler。我们直接去RestoreSnapshotHandler和CloneSnapshotHandler的handleTableOperation方法。先说RestoreSnapshotHandler吧。
protected void handleTableOperation(List<HRegionInfo> hris) throws IOException {
MasterFileSystem fileSystemManager = masterServices.getMasterFileSystem();
CatalogTracker catalogTracker = masterServices.getCatalogTracker();
FileSystem fs = fileSystemManager.getFileSystem();
Path rootDir = fileSystemManager.getRootDir();
TableName tableName = hTableDescriptor.getTableName();
try {
// 1. 用snapshot当中的表定义来覆盖现在的表定义
this.masterServices.getTableDescriptors().add(hTableDescriptor);
// 2. 找到snapshot的地址，使用restoreHelper开始恢复
Path snapshotDir = SnapshotDescriptionUtils.getCompletedSnapshotDir(snapshot, rootDir);
RestoreSnapshotHelper restoreHelper = new RestoreSnapshotHelper(
masterServices.getConfiguration(), fs,
snapshot, snapshotDir, hTableDescriptor, rootDir, monitor, status);
RestoreSnapshotHelper.RestoreMetaChanges metaChanges = restoreHelper.restoreHdfsRegions();
// 3. 更改的变化的region的RegionStates为offline状态
forceRegionsOffline(metaChanges);
// 4.1 把那些删除了的region在meta表里面也删除掉
List<HRegionInfo> hrisToRemove = new LinkedList<HRegionInfo>();
if (metaChanges.hasRegionsToRemove()) hrisToRemove.addAll(metaChanges.getRegionsToRemove());
if (metaChanges.hasRegionsToRestore()) hrisToRemove.addAll(metaChanges.getRegionsToRestore());
//删除meta表当中的region
MetaEditor.deleteRegions(catalogTracker, hrisToRemove);
// 4.2 添加新增的region到META表
hris.clear();
//再把新的加进去
if (metaChanges.hasRegionsToAdd()) hris.addAll(metaChanges.getRegionsToAdd());
//删掉旧的，再添加回来
if (metaChanges.hasRegionsToRestore()) hris.addAll(metaChanges.getRegionsToRestore());
MetaEditor.addRegionsToMeta(catalogTracker, hris);
metaChanges.updateMetaParentRegions(catalogTracker, hris);
} catch (IOException e) {
String msg = "restore snapshot=" + ClientSnapshotDescriptionUtils.toString(snapshot)
+ " failed. Try re-running the restore command.";throw new RestoreSnapshotException(msg, e);
}
}从代码上看上面主要包括4个步骤：
（1）更新表的定义
（2）恢复region
（3）把变化了的region在RS端的RegionStates里面强制下线，否则会出现region在恢复之前是split状态的再也无法被分配的情况
（4）修改meta表当中的region记录，根据新增和删除的两种情况来处理
恢复region的过程
我们接下来看RestoreSnapshotHelper的restoreHdfsRegions这个方法吧。
public RestoreMetaChanges restoreHdfsRegions() throws IOException {
LOG.debug("starting restore");
//遍历一下Snapshot目录下的region，没有region就退出了
Set<String> snapshotRegionNames = SnapshotReferenceUtil.getSnapshotRegionNames(fs, snapshotDir);
RestoreMetaChanges metaChanges = new RestoreMetaChanges(parentsMap);
List<HRegionInfo> tableRegions = getTableRegions();
if (tableRegions != null) {
//for (HRegionInfo regionInfo: tableRegions) {
String regionName = regionInfo.getEncodedName();
//snapshot当中包含的region就要恢复，snapshot当中不包括的region就要删除
if (snapshotRegionNames.contains(regionName)) {
snapshotRegionNames.remove(regionName);
metaChanges.addRegionToRestore(regionInfo);
} else {
metaChanges.addRegionToRemove(regionInfo);
}
}
// 恢复需要的恢复region
restoreHdfsRegions(metaChanges.getRegionsToRestore());
// 删除掉不属于snapshot的region
removeHdfsRegions(metaChanges.getRegionsToRemove());
}
// 以前有，现在没有的region也要做恢复
if (snapshotRegionNames.size() > 0) {
List<HRegionInfo> regionsToAdd = new LinkedList<HRegionInfo>();
for (String regionName: snapshotRegionNames) {
Path regionDir = new Path(snapshotDir, regionName);
regionsToAdd.add(HRegionFileSystem.loadRegionInfoFileContent(fs, regionDir));
}
// 要新增的region
HRegionInfo[] clonedRegions = cloneHdfsRegions(regionsToAdd);
metaChanges.setNewRegions(clonedRegions);
}
// Restore WALs
restoreWALs();
return metaChanges;
}首先要拿snapshot的region和现在的table的region逐个对比，分为三种情况：
（1）以前没有的region，现在有的region，这个region是要删掉的
（2）以前有，现在也有的region，这个region要被恢复
（3）以前有，现在没有了，这个region也要恢复，这个情况和前面的有点儿区别，要创建新的region目录和定义
接下来我们看restoreHdfsRegions这个方法吧，对region挨个恢复。
private void restoreRegion(HRegionInfo regionInfo) throws IOException {
Path snapshotRegionDir = new Path(snapshotDir, regionInfo.getEncodedName());
//获得要恢复<family,storeFiles>列表
Map<String, List<String>> snapshotFiles =
SnapshotReferenceUtil.getRegionHFileReferences(fs, snapshotRegionDir);
Path regionDir = new Path(tableDir, regionInfo.getEncodedName());
String tableName = tableDesc.getTableName().getNameAsString();
// 恢复当前在表里面的列族
for (Path familyDir: FSUtils.getFamilyDirs(fs, regionDir)) {
byte[] family = Bytes.toBytes(familyDir.getName());
Set<String> familyFiles = getTableRegionFamilyFiles(familyDir);
List<String> snapshotFamilyFiles = snapshotFiles.remove(familyDir.getName());
if (snapshotFamilyFiles != null) {
List<String> hfilesToAdd = new LinkedList<String>();
for (String hfileName: snapshotFamilyFiles) {
//snapshot中的文件，现有的文件当中已经有的就留着，多了的删除，缺少的就要添加
if (familyFiles.contains(hfileName)) {
// 已经存在的hfile，从这里删除之后，后面就不用处理了
familyFiles.remove(hfileName);
} else {
// 缺少的hfile
hfilesToAdd.add(hfileName);
}
}
// 归档那些不在snapshot当中的hfile
for (String hfileName: familyFiles) {
Path hfile = new Path(familyDir, hfileName);
HFileArchiver.archiveStoreFile(conf, fs, regionInfo, tableDir, family, hfile);
}
// 现在缺少的文件就添加
for (String hfileName: hfilesToAdd) {
restoreStoreFile(familyDir, regionInfo, hfileName);
}
} else {// 在snapshot当中不存在，直接把这个列族的文件归档并删掉
HFileArchiver.archiveFamily(fs, conf, regionInfo, tableDir, family);
fs.delete(familyDir, true);
}
}
// 添加不在当前表里的列族，然后恢复
for (Map.Entry<String, List<String>> familyEntry: snapshotFiles.entrySet()) {
Path familyDir = new Path(regionDir, familyEntry.getKey());
if (!fs.mkdirs(familyDir)) {
throw new IOException("Unable to create familyDir=" + familyDir);
}
for (String hfileName: familyEntry.getValue()) {
restoreStoreFile(familyDir, regionInfo, hfileName);
}
}
}恢复这块的逻辑也差不多，首先先把hfile和列族挂钩，弄成一个<family, List<hfiles>>的map，一个一个列族去恢复，列族这块也存在上面region的3种情况，这里就不说了。
下面有3点是我们要注意的:
（1）相信看了上一章的朋友都有印象，它给hfile创建引用的时候，并未实际保存文件，而是创建了一个同名的空文件。在上面的情况当中，已经存在的同名的hfile，就不需要管了，为什么不要管了？因为hfile一旦写入到文件，writer关闭之后就不会修改了，即使是做compaction的时候，是把多个hfile合成一个新的hfile，把旧的文件删除来一个新的文件。
（2）对于那些后来新增的，在snapshot当前没有的文件，它们不是被直接删除，而是被移到了另外一个地方，归档的位置是archive目录，归档的操作是用HFileArchiver类来归档。碰到极端的情况，该文件已经存在了，就在文件后面加上".当前时间戳"。
（3）对于缺少的文件走的restoreStoreFile方法，下面是它的代码。
private void restoreStoreFile(final Path familyDir, final HRegionInfo regionInfo,
final String hfileName) throws IOException {
if (HFileLink.isHFileLink(hfileName)) {
//是HFileLink的情况
HFileLink.createFromHFileLink(conf, fs, familyDir, hfileName);
} else if (StoreFileInfo.isReference(hfileName)) {
//是Reference的情况
restoreReferenceFile(familyDir, regionInfo, hfileName);
} else {
//是hfile的情况
HFileLink.create(conf, fs, familyDir, regionInfo, hfileName);
}
}在hbase里面文件分3种，HFileLink、ReferenceFile、Hfile3种，所以恢复的时候需要按照这种方式。
HFileLink是一个链接文件，名字形式是table=region-hfile，在读取hfile的时候，如果是HFileLink它会做自动处理，去读取真正的hfile。
ReferenceFile不同于上一章的引用文件，那个文件只是只是用来记录名字的，它是split产生的文件，分Top和Bottom两种，也是一种链接文件，读取的时候会创建一个以分割点为中点的Reader，只读一半的文件。
这个怎么读取链接，之后再介绍，到时候在放链接过来，下面我们回到restoreStoreFile方法上来。
比如一个叫abc的hfile文件，根据这三种情况来恢复，"->"左边是原来的文件名，右边是新的文件名。
（a）Hfile3: abc -> table=region-abc
（b）ReferenceFile: abc.1234 -> table=region-abc.1234
（c）HFileLink: table=region-abc -> table=region-abc
可以看得出来，它并没有把一个真正的hfile文件恢复回去，都是在创建类似桌面快捷方式，这样可以节省空间。
恢复hfile这块就结束了，然后到restoreWALs方法看看，它是怎么恢复日志的。
private void restoreWALs() throws IOException {
final SnapshotLogSplitter logSplitter = new SnapshotLogSplitter(conf, fs, tableDir,
snapshotTable, regionsMap);
try {
// Recover.Edits 遍历snapshot目录下的edits日志
SnapshotReferenceUtil.visitRecoveredEdits(fs, snapshotDir,
new FSVisitor.RecoveredEditsVisitor() {
public void recoveredEdits (final String region, final String logfile) throws IOException {
Path path = SnapshotReferenceUtil.getRecoveredEdits(snapshotDir, region, logfile);
logSplitter.splitRecoveredEdit(path);
}
});
// 前面那个是基于region的日志，这个是基于Region Server的日志WALs日志
SnapshotReferenceUtil.visitLogFiles(fs, snapshotDir, new FSVisitor.LogFileVisitor() {
public void logFile (final String server, final String logfile) throws IOException {
logSplitter.splitLog(server, logfile);
}
});
} finally {
logSplitter.close();
}
}logSplitter.splitRecoveredEdit和logSplitter.splitLog的最后都调用了一个叫做splitLog的方法（editPath）的方法，区别的地方在于splitLog传了一个HLogLink（HLog的快捷方式。。。）
下面看看splitLog这个方法吧public void splitLog(final Path logPath) throws IOException {
HLog.Reader log = HLogFactory.createReader(fs, logPath, conf);
try {
HLog.Entry entry;
LogWriter writer = null;
byte[] regionName = null;
byte[] newRegionName = null;
while ((entry = log.next()) != null) {
HLogKey key = entry.getKey();
// 只处理要snapshot的表的
if (!key.getTablename().equals(snapshotTableName)) continue;
// 为每一个新的region实例化一个Writer，但奇怪的是旧的没有close，就直接切换引用了
if (!Bytes.equals(regionName, key.getEncodedRegionName())) {
regionName = key.getEncodedRegionName().clone();
// Get the new region name in case of clone, or use the original one
newRegionName = regionsMap.get(regionName);
if (newRegionName == null) newRegionName = regionName;
writer = getOrCreateWriter(newRegionName, key.getLogSeqNum());
}
//一个一个追加，没啥好说的
key = new HLogKey(newRegionName, tableName,
key.getLogSeqNum(), key.getWriteTime(), key.getClusterIds());
writer.append(new HLog.Entry(key, entry.getEdit()));
}
} catch (IOException e) {
LOG.warn("Something wrong during the log split", e);
} finally {
log.close();
}
}上面这段代码也没干啥，创建一个HLog.Reader读取日志文件，然后迭代一下，把属于我们要做snapshot的表的日志读取出来，它为每一个region的实例化一个Writer，调用的Writer的Append方法追加HLog。
Writer写入的目录在recovered.edits，还是这个目录，之前是hmaster启动的时候，对那些挂了的region，也是把日志split到这个目录，可能在Region Server恢复的时候直接去找这个目录吧，后面讲到Region Server的时候关注一下。额，到这里为止，恢复的过程就到此结束了。
后面还有两步，强制更新变化的region的Region States为offline和修改meta表中的region都比较简单，这里就不讲了。
对于被删除了的表，处理起来就简单一些了，直接从走了restoreHdfsRegions的方法，这里的可能有点儿疑惑，为啥没建表，原来在它继承的CreateTableHandler的prepare方法里面把这活给干了。
总结一下：从上面的过程我们可以看出来，snapshot还是很轻量级的，除了归档删除的情况外，它的备份和恢复大多数都是创建的链接文件，而不是直接大规模复制、替换HFile的方式，可能也就是因为这样才叫snapshot。