http://git-wip-us.apache.org/repos/asf/hbase-site/blob/68eae623/devapidocs/src-html/org/apache/hadoop/hbase/snapshot/RestoreSnapshotHelper.RestoreMetaChanges.html ---------------------------------------------------------------------- diff --git a/devapidocs/src-html/org/apache/hadoop/hbase/snapshot/RestoreSnapshotHelper.RestoreMetaChanges.html b/devapidocs/src-html/org/apache/hadoop/hbase/snapshot/RestoreSnapshotHelper.RestoreMetaChanges.html index cbae28e..aec5920 100644 --- a/devapidocs/src-html/org/apache/hadoop/hbase/snapshot/RestoreSnapshotHelper.RestoreMetaChanges.html +++ b/devapidocs/src-html/org/apache/hadoop/hbase/snapshot/RestoreSnapshotHelper.RestoreMetaChanges.html @@ -64,825 +64,827 @@ <span class="sourceLineNo">056</span>import org.apache.hadoop.hbase.regionserver.HRegionFileSystem;<a name="line.56"></a> <span class="sourceLineNo">057</span>import org.apache.hadoop.hbase.regionserver.StoreFileInfo;<a name="line.57"></a> <span class="sourceLineNo">058</span>import org.apache.hadoop.hbase.security.access.AccessControlClient;<a name="line.58"></a> -<span class="sourceLineNo">059</span>import org.apache.hadoop.hbase.security.access.ShadedAccessControlUtil;<a name="line.59"></a> -<span class="sourceLineNo">060</span>import org.apache.hadoop.hbase.security.access.TablePermission;<a name="line.60"></a> -<span class="sourceLineNo">061</span>import org.apache.hadoop.hbase.util.Bytes;<a name="line.61"></a> -<span class="sourceLineNo">062</span>import org.apache.hadoop.hbase.util.FSUtils;<a name="line.62"></a> -<span class="sourceLineNo">063</span>import org.apache.hadoop.hbase.util.ModifyRegionUtils;<a name="line.63"></a> -<span class="sourceLineNo">064</span>import org.apache.hadoop.hbase.util.Pair;<a name="line.64"></a> -<span class="sourceLineNo">065</span>import org.apache.hadoop.io.IOUtils;<a name="line.65"></a> -<span class="sourceLineNo">066</span>import org.apache.yetus.audience.InterfaceAudience;<a name="line.66"></a> -<span class="sourceLineNo">067</span>import org.slf4j.Logger;<a name="line.67"></a> -<span class="sourceLineNo">068</span>import org.slf4j.LoggerFactory;<a name="line.68"></a> -<span class="sourceLineNo">069</span>import org.apache.hbase.thirdparty.com.google.common.collect.ListMultimap;<a name="line.69"></a> -<span class="sourceLineNo">070</span>import org.apache.hadoop.hbase.shaded.protobuf.ProtobufUtil;<a name="line.70"></a> -<span class="sourceLineNo">071</span>import org.apache.hadoop.hbase.shaded.protobuf.generated.SnapshotProtos.SnapshotDescription;<a name="line.71"></a> -<span class="sourceLineNo">072</span>import org.apache.hadoop.hbase.shaded.protobuf.generated.SnapshotProtos.SnapshotRegionManifest;<a name="line.72"></a> -<span class="sourceLineNo">073</span><a name="line.73"></a> -<span class="sourceLineNo">074</span>/**<a name="line.74"></a> -<span class="sourceLineNo">075</span> * Helper to Restore/Clone a Snapshot<a name="line.75"></a> -<span class="sourceLineNo">076</span> *<a name="line.76"></a> -<span class="sourceLineNo">077</span> * <p>The helper assumes that a table is already created, and by calling restore()<a name="line.77"></a> -<span class="sourceLineNo">078</span> * the content present in the snapshot will be restored as the new content of the table.<a name="line.78"></a> -<span class="sourceLineNo">079</span> *<a name="line.79"></a> -<span class="sourceLineNo">080</span> * <p>Clone from Snapshot: If the target table is empty, the restore operation<a name="line.80"></a> -<span class="sourceLineNo">081</span> * is just a "clone operation", where the only operations are:<a name="line.81"></a> -<span class="sourceLineNo">082</span> * <ul><a name="line.82"></a> -<span class="sourceLineNo">083</span> * <li>for each region in the snapshot create a new region<a name="line.83"></a> -<span class="sourceLineNo">084</span> * (note that the region will have a different name, since the encoding contains the table name)<a name="line.84"></a> -<span class="sourceLineNo">085</span> * <li>for each file in the region create a new HFileLink to point to the original file.<a name="line.85"></a> -<span class="sourceLineNo">086</span> * <li>restore the logs, if any<a name="line.86"></a> -<span class="sourceLineNo">087</span> * </ul><a name="line.87"></a> -<span class="sourceLineNo">088</span> *<a name="line.88"></a> -<span class="sourceLineNo">089</span> * <p>Restore from Snapshot:<a name="line.89"></a> -<span class="sourceLineNo">090</span> * <ul><a name="line.90"></a> -<span class="sourceLineNo">091</span> * <li>for each region in the table verify which are available in the snapshot and which are not<a name="line.91"></a> -<span class="sourceLineNo">092</span> * <ul><a name="line.92"></a> -<span class="sourceLineNo">093</span> * <li>if the region is not present in the snapshot, remove it.<a name="line.93"></a> -<span class="sourceLineNo">094</span> * <li>if the region is present in the snapshot<a name="line.94"></a> -<span class="sourceLineNo">095</span> * <ul><a name="line.95"></a> -<span class="sourceLineNo">096</span> * <li>for each file in the table region verify which are available in the snapshot<a name="line.96"></a> -<span class="sourceLineNo">097</span> * <ul><a name="line.97"></a> -<span class="sourceLineNo">098</span> * <li>if the hfile is not present in the snapshot, remove it<a name="line.98"></a> -<span class="sourceLineNo">099</span> * <li>if the hfile is present, keep it (nothing to do)<a name="line.99"></a> -<span class="sourceLineNo">100</span> * </ul><a name="line.100"></a> -<span class="sourceLineNo">101</span> * <li>for each file in the snapshot region but not in the table<a name="line.101"></a> -<span class="sourceLineNo">102</span> * <ul><a name="line.102"></a> -<span class="sourceLineNo">103</span> * <li>create a new HFileLink that point to the original file<a name="line.103"></a> -<span class="sourceLineNo">104</span> * </ul><a name="line.104"></a> -<span class="sourceLineNo">105</span> * </ul><a name="line.105"></a> -<span class="sourceLineNo">106</span> * </ul><a name="line.106"></a> -<span class="sourceLineNo">107</span> * <li>for each region in the snapshot not present in the current table state<a name="line.107"></a> -<span class="sourceLineNo">108</span> * <ul><a name="line.108"></a> -<span class="sourceLineNo">109</span> * <li>create a new region and for each file in the region create a new HFileLink<a name="line.109"></a> -<span class="sourceLineNo">110</span> * (This is the same as the clone operation)<a name="line.110"></a> -<span class="sourceLineNo">111</span> * </ul><a name="line.111"></a> -<span class="sourceLineNo">112</span> * <li>restore the logs, if any<a name="line.112"></a> -<span class="sourceLineNo">113</span> * </ul><a name="line.113"></a> -<span class="sourceLineNo">114</span> */<a name="line.114"></a> -<span class="sourceLineNo">115</span>@InterfaceAudience.Private<a name="line.115"></a> -<span class="sourceLineNo">116</span>public class RestoreSnapshotHelper {<a name="line.116"></a> -<span class="sourceLineNo">117</span> private static final Logger LOG = LoggerFactory.getLogger(RestoreSnapshotHelper.class);<a name="line.117"></a> -<span class="sourceLineNo">118</span><a name="line.118"></a> -<span class="sourceLineNo">119</span> private final Map<byte[], byte[]> regionsMap = new TreeMap<>(Bytes.BYTES_COMPARATOR);<a name="line.119"></a> -<span class="sourceLineNo">120</span><a name="line.120"></a> -<span class="sourceLineNo">121</span> private final Map<String, Pair<String, String> > parentsMap = new HashMap<>();<a name="line.121"></a> -<span class="sourceLineNo">122</span><a name="line.122"></a> -<span class="sourceLineNo">123</span> private final ForeignExceptionDispatcher monitor;<a name="line.123"></a> -<span class="sourceLineNo">124</span> private final MonitoredTask status;<a name="line.124"></a> -<span class="sourceLineNo">125</span><a name="line.125"></a> -<span class="sourceLineNo">126</span> private final SnapshotManifest snapshotManifest;<a name="line.126"></a> -<span class="sourceLineNo">127</span> private final SnapshotDescription snapshotDesc;<a name="line.127"></a> -<span class="sourceLineNo">128</span> private final TableName snapshotTable;<a name="line.128"></a> -<span class="sourceLineNo">129</span><a name="line.129"></a> -<span class="sourceLineNo">130</span> private final TableDescriptor tableDesc;<a name="line.130"></a> -<span class="sourceLineNo">131</span> private final Path rootDir;<a name="line.131"></a> -<span class="sourceLineNo">132</span> private final Path tableDir;<a name="line.132"></a> -<span class="sourceLineNo">133</span><a name="line.133"></a> -<span class="sourceLineNo">134</span> private final Configuration conf;<a name="line.134"></a> -<span class="sourceLineNo">135</span> private final FileSystem fs;<a name="line.135"></a> -<span class="sourceLineNo">136</span> private final boolean createBackRefs;<a name="line.136"></a> -<span class="sourceLineNo">137</span><a name="line.137"></a> -<span class="sourceLineNo">138</span> public RestoreSnapshotHelper(final Configuration conf,<a name="line.138"></a> -<span class="sourceLineNo">139</span> final FileSystem fs,<a name="line.139"></a> -<span class="sourceLineNo">140</span> final SnapshotManifest manifest,<a name="line.140"></a> -<span class="sourceLineNo">141</span> final TableDescriptor tableDescriptor,<a name="line.141"></a> -<span class="sourceLineNo">142</span> final Path rootDir,<a name="line.142"></a> -<span class="sourceLineNo">143</span> final ForeignExceptionDispatcher monitor,<a name="line.143"></a> -<span class="sourceLineNo">144</span> final MonitoredTask status) {<a name="line.144"></a> -<span class="sourceLineNo">145</span> this(conf, fs, manifest, tableDescriptor, rootDir, monitor, status, true);<a name="line.145"></a> -<span class="sourceLineNo">146</span> }<a name="line.146"></a> -<span class="sourceLineNo">147</span><a name="line.147"></a> -<span class="sourceLineNo">148</span> public RestoreSnapshotHelper(final Configuration conf,<a name="line.148"></a> -<span class="sourceLineNo">149</span> final FileSystem fs,<a name="line.149"></a> -<span class="sourceLineNo">150</span> final SnapshotManifest manifest,<a name="line.150"></a> -<span class="sourceLineNo">151</span> final TableDescriptor tableDescriptor,<a name="line.151"></a> -<span class="sourceLineNo">152</span> final Path rootDir,<a name="line.152"></a> -<span class="sourceLineNo">153</span> final ForeignExceptionDispatcher monitor,<a name="line.153"></a> -<span class="sourceLineNo">154</span> final MonitoredTask status,<a name="line.154"></a> -<span class="sourceLineNo">155</span> final boolean createBackRefs)<a name="line.155"></a> -<span class="sourceLineNo">156</span> {<a name="line.156"></a> -<span class="sourceLineNo">157</span> this.fs = fs;<a name="line.157"></a> -<span class="sourceLineNo">158</span> this.conf = conf;<a name="line.158"></a> -<span class="sourceLineNo">159</span> this.snapshotManifest = manifest;<a name="line.159"></a> -<span class="sourceLineNo">160</span> this.snapshotDesc = manifest.getSnapshotDescription();<a name="line.160"></a> -<span class="sourceLineNo">161</span> this.snapshotTable = TableName.valueOf(snapshotDesc.getTable());<a name="line.161"></a> -<span class="sourceLineNo">162</span> this.tableDesc = tableDescriptor;<a name="line.162"></a> -<span class="sourceLineNo">163</span> this.rootDir = rootDir;<a name="line.163"></a> -<span class="sourceLineNo">164</span> this.tableDir = FSUtils.getTableDir(rootDir, tableDesc.getTableName());<a name="line.164"></a> -<span class="sourceLineNo">165</span> this.monitor = monitor;<a name="line.165"></a> -<span class="sourceLineNo">166</span> this.status = status;<a name="line.166"></a> -<span class="sourceLineNo">167</span> this.createBackRefs = createBackRefs;<a name="line.167"></a> -<span class="sourceLineNo">168</span> }<a name="line.168"></a> -<span class="sourceLineNo">169</span><a name="line.169"></a> -<span class="sourceLineNo">170</span> /**<a name="line.170"></a> -<span class="sourceLineNo">171</span> * Restore the on-disk table to a specified snapshot state.<a name="line.171"></a> -<span class="sourceLineNo">172</span> * @return the set of regions touched by the restore operation<a name="line.172"></a> -<span class="sourceLineNo">173</span> */<a name="line.173"></a> -<span class="sourceLineNo">174</span> public RestoreMetaChanges restoreHdfsRegions() throws IOException {<a name="line.174"></a> -<span class="sourceLineNo">175</span> ThreadPoolExecutor exec = SnapshotManifest.createExecutor(conf, "RestoreSnapshot");<a name="line.175"></a> -<span class="sourceLineNo">176</span> try {<a name="line.176"></a> -<span class="sourceLineNo">177</span> return restoreHdfsRegions(exec);<a name="line.177"></a> -<span class="sourceLineNo">178</span> } finally {<a name="line.178"></a> -<span class="sourceLineNo">179</span> exec.shutdown();<a name="line.179"></a> -<span class="sourceLineNo">180</span> }<a name="line.180"></a> -<span class="sourceLineNo">181</span> }<a name="line.181"></a> -<span class="sourceLineNo">182</span><a name="line.182"></a> -<span class="sourceLineNo">183</span> private RestoreMetaChanges restoreHdfsRegions(final ThreadPoolExecutor exec) throws IOException {<a name="line.183"></a> -<span class="sourceLineNo">184</span> LOG.info("starting restore table regions using snapshot=" + snapshotDesc);<a name="line.184"></a> -<span class="sourceLineNo">185</span><a name="line.185"></a> -<span class="sourceLineNo">186</span> Map<String, SnapshotRegionManifest> regionManifests = snapshotManifest.getRegionManifestsMap();<a name="line.186"></a> -<span class="sourceLineNo">187</span> if (regionManifests == null) {<a name="line.187"></a> -<span class="sourceLineNo">188</span> LOG.warn("Nothing to restore. Snapshot " + snapshotDesc + " looks empty");<a name="line.188"></a> -<span class="sourceLineNo">189</span> return null;<a name="line.189"></a> -<span class="sourceLineNo">190</span> }<a name="line.190"></a> -<span class="sourceLineNo">191</span><a name="line.191"></a> -<span class="sourceLineNo">192</span> RestoreMetaChanges metaChanges = new RestoreMetaChanges(tableDesc, parentsMap);<a name="line.192"></a> -<span class="sourceLineNo">193</span><a name="line.193"></a> -<span class="sourceLineNo">194</span> // Take a copy of the manifest.keySet() since we are going to modify<a name="line.194"></a> -<span class="sourceLineNo">195</span> // this instance, by removing the regions already present in the restore dir.<a name="line.195"></a> -<span class="sourceLineNo">196</span> Set<String> regionNames = new HashSet<>(regionManifests.keySet());<a name="line.196"></a> -<span class="sourceLineNo">197</span><a name="line.197"></a> -<span class="sourceLineNo">198</span> List<RegionInfo> tableRegions = getTableRegions();<a name="line.198"></a> -<span class="sourceLineNo">199</span><a name="line.199"></a> -<span class="sourceLineNo">200</span> RegionInfo mobRegion = MobUtils.getMobRegionInfo(snapshotManifest.getTableDescriptor()<a name="line.200"></a> -<span class="sourceLineNo">201</span> .getTableName());<a name="line.201"></a> -<span class="sourceLineNo">202</span> if (tableRegions != null) {<a name="line.202"></a> -<span class="sourceLineNo">203</span> // restore the mob region in case<a name="line.203"></a> -<span class="sourceLineNo">204</span> if (regionNames.contains(mobRegion.getEncodedName())) {<a name="line.204"></a> -<span class="sourceLineNo">205</span> monitor.rethrowException();<a name="line.205"></a> -<span class="sourceLineNo">206</span> status.setStatus("Restoring mob region...");<a name="line.206"></a> -<span class="sourceLineNo">207</span> List<RegionInfo> mobRegions = new ArrayList<>(1);<a name="line.207"></a> -<span class="sourceLineNo">208</span> mobRegions.add(mobRegion);<a name="line.208"></a> -<span class="sourceLineNo">209</span> restoreHdfsMobRegions(exec, regionManifests, mobRegions);<a name="line.209"></a> -<span class="sourceLineNo">210</span> regionNames.remove(mobRegion.getEncodedName());<a name="line.210"></a> -<span class="sourceLineNo">211</span> status.setStatus("Finished restoring mob region.");<a name="line.211"></a> -<span class="sourceLineNo">212</span> }<a name="line.212"></a> -<span class="sourceLineNo">213</span> }<a name="line.213"></a> -<span class="sourceLineNo">214</span> if (regionNames.contains(mobRegion.getEncodedName())) {<a name="line.214"></a> -<span class="sourceLineNo">215</span> // add the mob region<a name="line.215"></a> -<span class="sourceLineNo">216</span> monitor.rethrowException();<a name="line.216"></a> -<span class="sourceLineNo">217</span> status.setStatus("Cloning mob region...");<a name="line.217"></a> -<span class="sourceLineNo">218</span> cloneHdfsMobRegion(regionManifests, mobRegion);<a name="line.218"></a> -<span class="sourceLineNo">219</span> regionNames.remove(mobRegion.getEncodedName());<a name="line.219"></a> -<span class="sourceLineNo">220</span> status.setStatus("Finished cloning mob region.");<a name="line.220"></a> -<span class="sourceLineNo">221</span> }<a name="line.221"></a> -<span class="sourceLineNo">222</span><a name="line.222"></a> -<span class="sourceLineNo">223</span> // Identify which region are still available and which not.<a name="line.223"></a> -<span class="sourceLineNo">224</span> // NOTE: we rely upon the region name as: "table name, start key, end key"<a name="line.224"></a> -<span class="sourceLineNo">225</span> if (tableRegions != null) {<a name="line.225"></a> -<span class="sourceLineNo">226</span> monitor.rethrowException();<a name="line.226"></a> -<span class="sourceLineNo">227</span> for (RegionInfo regionInfo: tableRegions) {<a name="line.227"></a> -<span class="sourceLineNo">228</span> String regionName = regionInfo.getEncodedName();<a name="line.228"></a> -<span class="sourceLineNo">229</span> if (regionNames.contains(regionName)) {<a name="line.229"></a> -<span class="sourceLineNo">230</span> LOG.info("region to restore: " + regionName);<a name="line.230"></a> -<span class="sourceLineNo">231</span> regionNames.remove(regionName);<a name="line.231"></a> -<span class="sourceLineNo">232</span> metaChanges.addRegionToRestore(ProtobufUtil.toRegionInfo(regionManifests.get(regionName)<a name="line.232"></a> -<span class="sourceLineNo">233</span> .getRegionInfo()));<a name="line.233"></a> -<span class="sourceLineNo">234</span> } else {<a name="line.234"></a> -<span class="sourceLineNo">235</span> LOG.info("region to remove: " + regionName);<a name="line.235"></a> -<span class="sourceLineNo">236</span> metaChanges.addRegionToRemove(regionInfo);<a name="line.236"></a> -<span class="sourceLineNo">237</span> }<a name="line.237"></a> -<span class="sourceLineNo">238</span> }<a name="line.238"></a> -<span class="sourceLineNo">239</span> }<a name="line.239"></a> -<span class="sourceLineNo">240</span><a name="line.240"></a> -<span class="sourceLineNo">241</span> // Regions to Add: present in the snapshot but not in the current table<a name="line.241"></a> -<span class="sourceLineNo">242</span> List<RegionInfo> regionsToAdd = new ArrayList<>(regionNames.size());<a name="line.242"></a> -<span class="sourceLineNo">243</span> if (regionNames.size() > 0) {<a name="line.243"></a> -<span class="sourceLineNo">244</span> monitor.rethrowException();<a name="line.244"></a> -<span class="sourceLineNo">245</span> for (String regionName: regionNames) {<a name="line.245"></a> -<span class="sourceLineNo">246</span> LOG.info("region to add: " + regionName);<a name="line.246"></a> -<span class="sourceLineNo">247</span> regionsToAdd.add(ProtobufUtil.toRegionInfo(regionManifests.get(regionName)<a name="line.247"></a> -<span class="sourceLineNo">248</span> .getRegionInfo()));<a name="line.248"></a> -<span class="sourceLineNo">249</span> }<a name="line.249"></a> -<span class="sourceLineNo">250</span> }<a name="line.250"></a> -<span class="sourceLineNo">251</span><a name="line.251"></a> -<span class="sourceLineNo">252</span> // Create new regions cloning from the snapshot<a name="line.252"></a> -<span class="sourceLineNo">253</span> // HBASE-19980: We need to call cloneHdfsRegions() before restoreHdfsRegions() because<a name="line.253"></a> -<span class="sourceLineNo">254</span> // regionsMap is constructed in cloneHdfsRegions() and it can be used in restoreHdfsRegions().<a name="line.254"></a> -<span class="sourceLineNo">255</span> monitor.rethrowException();<a name="line.255"></a> -<span class="sourceLineNo">256</span> status.setStatus("Cloning regions...");<a name="line.256"></a> -<span class="sourceLineNo">257</span> RegionInfo[] clonedRegions = cloneHdfsRegions(exec, regionManifests, regionsToAdd);<a name="line.257"></a> -<span class="sourceLineNo">258</span> metaChanges.setNewRegions(clonedRegions);<a name="line.258"></a> -<span class="sourceLineNo">259</span> status.setStatus("Finished cloning regions.");<a name="line.259"></a> -<span class="sourceLineNo">260</span><a name="line.260"></a> -<span class="sourceLineNo">261</span> // Restore regions using the snapshot data<a name="line.261"></a> -<span class="sourceLineNo">262</span> monitor.rethrowException();<a name="line.262"></a> -<span class="sourceLineNo">263</span> status.setStatus("Restoring table regions...");<a name="line.263"></a> -<span class="sourceLineNo">264</span> restoreHdfsRegions(exec, regionManifests, metaChanges.getRegionsToRestore());<a name="line.264"></a> -<span class="sourceLineNo">265</span> status.setStatus("Finished restoring all table regions.");<a name="line.265"></a> -<span class="sourceLineNo">266</span><a name="line.266"></a> -<span class="sourceLineNo">267</span> // Remove regions from the current table<a name="line.267"></a> -<span class="sourceLineNo">268</span> monitor.rethrowException();<a name="line.268"></a> -<span class="sourceLineNo">269</span> status.setStatus("Starting to delete excess regions from table");<a name="line.269"></a> -<span class="sourceLineNo">270</span> removeHdfsRegions(exec, metaChanges.getRegionsToRemove());<a name="line.270"></a> -<span class="sourceLineNo">271</span> status.setStatus("Finished deleting excess regions from table.");<a name="line.271"></a> -<span class="sourceLineNo">272</span><a name="line.272"></a> -<span class="sourceLineNo">273</span> LOG.info("finishing restore table regions using snapshot=" + snapshotDesc);<a name="line.273"></a> -<span class="sourceLineNo">274</span><a name="line.274"></a> -<span class="sourceLineNo">275</span> return metaChanges;<a name="line.275"></a> -<span class="sourceLineNo">276</span> }<a name="line.276"></a> -<span class="sourceLineNo">277</span><a name="line.277"></a> -<span class="sourceLineNo">278</span> /**<a name="line.278"></a> -<span class="sourceLineNo">279</span> * Describe the set of operations needed to update hbase:meta after restore.<a name="line.279"></a> -<span class="sourceLineNo">280</span> */<a name="line.280"></a> -<span class="sourceLineNo">281</span> public static class RestoreMetaChanges {<a name="line.281"></a> -<span class="sourceLineNo">282</span> private final Map<String, Pair<String, String> > parentsMap;<a name="line.282"></a> -<span class="sourceLineNo">283</span> private final TableDescriptor htd;<a name="line.283"></a> -<span class="sourceLineNo">284</span><a name="line.284"></a> -<span class="sourceLineNo">285</span> private List<RegionInfo> regionsToRestore = null;<a name="line.285"></a> -<span class="sourceLineNo">286</span> private List<RegionInfo> regionsToRemove = null;<a name="line.286"></a> -<span class="sourceLineNo">287</span> private List<RegionInfo> regionsToAdd = null;<a name="line.287"></a> -<span class="sourceLineNo">288</span><a name="line.288"></a> -<span class="sourceLineNo">289</span> public RestoreMetaChanges(TableDescriptor htd, Map<String, Pair<String, String> > parentsMap) {<a name="line.289"></a> -<span class="sourceLineNo">290</span> this.parentsMap = parentsMap;<a name="line.290"></a> -<span class="sourceLineNo">291</span> this.htd = htd;<a name="line.291"></a> -<span class="sourceLineNo">292</span> }<a name="line.292"></a> -<span class="sourceLineNo">293</span><a name="line.293"></a> -<span class="sourceLineNo">294</span> public TableDescriptor getTableDescriptor() {<a name="line.294"></a> -<span class="sourceLineNo">295</span> return htd;<a name="line.295"></a> -<span class="sourceLineNo">296</span> }<a name="line.296"></a> -<span class="sourceLineNo">297</span><a name="line.297"></a> -<span class="sourceLineNo">298</span> /**<a name="line.298"></a> -<span class="sourceLineNo">299</span> * Returns the map of parent-children_pair.<a name="line.299"></a> -<span class="sourceLineNo">300</span> * @return the map<a name="line.300"></a> -<span class="sourceLineNo">301</span> */<a name="line.301"></a> -<span class="sourceLineNo">302</span> public Map<String, Pair<String, String>> getParentToChildrenPairMap() {<a name="line.302"></a> -<span class="sourceLineNo">303</span> return this.parentsMap;<a name="line.303"></a> -<span class="sourceLineNo">304</span> }<a name="line.304"></a> -<span class="sourceLineNo">305</span><a name="line.305"></a> -<span class="sourceLineNo">306</span> /**<a name="line.306"></a> -<span class="sourceLineNo">307</span> * @return true if there're new regions<a name="line.307"></a> -<span class="sourceLineNo">308</span> */<a name="line.308"></a> -<span class="sourceLineNo">309</span> public boolean hasRegionsToAdd() {<a name="line.309"></a> -<span class="sourceLineNo">310</span> return this.regionsToAdd != null && this.regionsToAdd.size() > 0;<a name="line.310"></a> -<span class="sourceLineNo">311</span> }<a name="line.311"></a> -<span class="sourceLineNo">312</span><a name="line.312"></a> -<span class="sourceLineNo">313</span> /**<a name="line.313"></a> -<span class="sourceLineNo">314</span> * Returns the list of new regions added during the on-disk restore.<a name="line.314"></a> -<span class="sourceLineNo">315</span> * The caller is responsible to add the regions to META.<a name="line.315"></a> -<span class="sourceLineNo">316</span> * e.g MetaTableAccessor.addRegionsToMeta(...)<a name="line.316"></a> -<span class="sourceLineNo">317</span> * @return the list of regions to add to META<a name="line.317"></a> -<span class="sourceLineNo">318</span> */<a name="line.318"></a> -<span class="sourceLineNo">319</span> public List<RegionInfo> getRegionsToAdd() {<a name="line.319"></a> -<span class="sourceLineNo">320</span> return this.regionsToAdd;<a name="line.320"></a> -<span class="sourceLineNo">321</span> }<a name="line.321"></a> -<span class="sourceLineNo">322</span><a name="line.322"></a> -<span class="sourceLineNo">323</span> /**<a name="line.323"></a> -<span class="sourceLineNo">324</span> * @return true if there're regions to restore<a name="line.324"></a> -<span class="sourceLineNo">325</span> */<a name="line.325"></a> -<span class="sourceLineNo">326</span> public boolean hasRegionsToRestore() {<a name="line.326"></a> -<span class="sourceLineNo">327</span> return this.regionsToRestore != null && this.regionsToRestore.size() > 0;<a name="line.327"></a> -<span class="sourceLineNo">328</span> }<a name="line.328"></a> -<span class="sourceLineNo">329</span><a name="line.329"></a> -<span class="sourceLineNo">330</span> /**<a name="line.330"></a> -<span class="sourceLineNo">331</span> * Returns the list of 'restored regions' during the on-disk restore.<a name="line.331"></a> -<span class="sourceLineNo">332</span> * The caller is responsible to add the regions to hbase:meta if not present.<a name="line.332"></a> -<span class="sourceLineNo">333</span> * @return the list of regions restored<a name="line.333"></a> -<span class="sourceLineNo">334</span> */<a name="line.334"></a> -<span class="sourceLineNo">335</span> public List<RegionInfo> getRegionsToRestore() {<a name="line.335"></a> -<span class="sourceLineNo">336</span> return this.regionsToRestore;<a name="line.336"></a> -<span class="sourceLineNo">337</span> }<a name="line.337"></a> -<span class="sourceLineNo">338</span><a name="line.338"></a> -<span class="sourceLineNo">339</span> /**<a name="line.339"></a> -<span class="sourceLineNo">340</span> * @return true if there're regions to remove<a name="line.340"></a> -<span class="sourceLineNo">341</span> */<a name="line.341"></a> -<span class="sourceLineNo">342</span> public boolean hasRegionsToRemove() {<a name="line.342"></a> -<span class="sourceLineNo">343</span> return this.regionsToRemove != null && this.regionsToRemove.size() > 0;<a name="line.343"></a> -<span class="sourceLineNo">344</span> }<a name="line.344"></a> -<span class="sourceLineNo">345</span><a name="line.345"></a> -<span class="sourceLineNo">346</span> /**<a name="line.346"></a> -<span class="sourceLineNo">347</span> * Returns the list of regions removed during the on-disk restore.<a name="line.347"></a> -<span class="sourceLineNo">348</span> * The caller is responsible to remove the regions from META.<a name="line.348"></a> -<span class="sourceLineNo">349</span> * e.g. MetaTableAccessor.deleteRegions(...)<a name="line.349"></a> -<span class="sourceLineNo">350</span> * @return the list of regions to remove from META<a name="line.350"></a> -<span class="sourceLineNo">351</span> */<a name="line.351"></a> -<span class="sourceLineNo">352</span> public List<RegionInfo> getRegionsToRemove() {<a name="line.352"></a> -<span class="sourceLineNo">353</span> return this.regionsToRemove;<a name="line.353"></a> -<span class="sourceLineNo">354</span> }<a name="line.354"></a> -<span class="sourceLineNo">355</span><a name="line.355"></a> -<span class="sourceLineNo">356</span> void setNewRegions(final RegionInfo[] hris) {<a name="line.356"></a> -<span class="sourceLineNo">357</span> if (hris != null) {<a name="line.357"></a> -<span class="sourceLineNo">358</span> regionsToAdd = Arrays.asList(hris);<a name="line.358"></a> -<span class="sourceLineNo">359</span> } else {<a name="line.359"></a> -<span class="sourceLineNo">360</span> regionsToAdd = null;<a name="line.360"></a> -<span class="sourceLineNo">361</span> }<a name="line.361"></a> -<span class="sourceLineNo">362</span> }<a name="line.362"></a> -<span class="sourceLineNo">363</span><a name="line.363"></a> -<span class="sourceLineNo">364</span> void addRegionToRemove(final RegionInfo hri) {<a name="line.364"></a> -<span class="sourceLineNo">365</span> if (regionsToRemove == null) {<a name="line.365"></a> -<span class="sourceLineNo">366</span> regionsToRemove = new LinkedList<>();<a name="line.366"></a> -<span class="sourceLineNo">367</span> }<a name="line.367"></a> -<span class="sourceLineNo">368</span> regionsToRemove.add(hri);<a name="line.368"></a> -<span class="sourceLineNo">369</span> }<a name="line.369"></a> -<span class="sourceLineNo">370</span><a name="line.370"></a> -<span class="sourceLineNo">371</span> void addRegionToRestore(final RegionInfo hri) {<a name="line.371"></a> -<span class="sourceLineNo">372</span> if (regionsToRestore == null) {<a name="line.372"></a> -<span class="sourceLineNo">373</span> regionsToRestore = new LinkedList<>();<a name="line.373"></a> -<span class="sourceLineNo">374</span> }<a name="line.374"></a> -<span class="sourceLineNo">375</span> regionsToRestore.add(hri);<a name="line.375"></a> -<span class="sourceLineNo">376</span> }<a name="line.376"></a> -<span class="sourceLineNo">377</span><a name="line.377"></a> -<span class="sourceLineNo">378</span> public void updateMetaParentRegions(Connection connection,<a name="line.378"></a> -<span class="sourceLineNo">379</span> final List<RegionInfo> regionInfos) throws IOException {<a name="line.379"></a> -<span class="sourceLineNo">380</span> if (regionInfos == null || parentsMap.isEmpty()) return;<a name="line.380"></a> -<span class="sourceLineNo">381</span><a name="line.381"></a> -<span class="sourceLineNo">382</span> // Extract region names and offlined regions<a name="line.382"></a> -<span class="sourceLineNo">383</span> Map<String, RegionInfo> regionsByName = new HashMap<>(regionInfos.size());<a name="line.383"></a> -<span class="sourceLineNo">384</span> List<RegionInfo> parentRegions = new LinkedList<>();<a name="line.384"></a> -<span class="sourceLineNo">385</span> for (RegionInfo regionInfo: regionInfos) {<a name="line.385"></a> -<span class="sourceLineNo">386</span> if (regionInfo.isSplitParent()) {<a name="line.386"></a> -<span class="sourceLineNo">387</span> parentRegions.add(regionInfo);<a name="line.387"></a> -<span class="sourceLineNo">388</span> } else {<a name="line.388"></a> -<span class="sourceLineNo">389</span> regionsByName.put(regionInfo.getEncodedName(), regionInfo);<a name="line.389"></a> -<span class="sourceLineNo">390</span> }<a name="line.390"></a> -<span class="sourceLineNo">391</span> }<a name="line.391"></a> -<span class="sourceLineNo">392</span><a name="line.392"></a> -<span class="sourceLineNo">393</span> // Update Offline parents<a name="line.393"></a> -<span class="sourceLineNo">394</span> for (RegionInfo regionInfo: parentRegions) {<a name="line.394"></a> -<span class="sourceLineNo">395</span> Pair<String, String> daughters = parentsMap.get(regionInfo.getEncodedName());<a name="line.395"></a> -<span class="sourceLineNo">396</span> if (daughters == null) {<a name="line.396"></a> -<span class="sourceLineNo">397</span> // The snapshot contains an unreferenced region.<a name="line.397"></a> -<span class="sourceLineNo">398</span> // It will be removed by the CatalogJanitor.<a name="line.398"></a> -<span class="sourceLineNo">399</span> LOG.warn("Skip update of unreferenced offline parent: " + regionInfo);<a name="line.399"></a> -<span class="sourceLineNo">400</span> continue;<a name="line.400"></a> -<span class="sourceLineNo">401</span> }<a name="line.401"></a> -<span class="sourceLineNo">402</span><a name="line.402"></a> -<span class="sourceLineNo">403</span> // One side of the split is already compacted<a name="line.403"></a> -<span class="sourceLineNo">404</span> if (daughters.getSecond() == null) {<a name="line.404"></a> -<span class="sourceLineNo">405</span> daughters.setSecond(daughters.getFirst());<a name="line.405"></a> -<span class="sourceLineNo">406</span> }<a name="line.406"></a> -<span class="sourceLineNo">407</span><a name="line.407"></a> -<span class="sourceLineNo">408</span> LOG.debug("Update splits parent " + regionInfo.getEncodedName() + " -> " + daughters);<a name="line.408"></a> -<span class="sourceLineNo">409</span> MetaTableAccessor.addSplitsToParent(connection, regionInfo,<a name="line.409"></a> -<span class="sourceLineNo">410</span> regionsByName.get(daughters.getFirst()),<a name="line.410"></a> -<span class="sourceLineNo">411</span> regionsByName.get(daughters.getSecond()));<a name="line.411"></a> -<span class="sourceLineNo">412</span> }<a name="line.412"></a> -<span class="sourceLineNo">413</span> }<a name="line.413"></a> -<span class="sourceLineNo">414</span> }<a name="line.414"></a> -<span class="sourceLineNo">415</span><a name="line.415"></a> -<span class="sourceLineNo">416</span> /**<a name="line.416"></a> -<span class="sourceLineNo">417</span> * Remove specified regions from the file-system, using the archiver.<a name="line.417"></a> -<span class="sourceLineNo">418</span> */<a name="line.418"></a> -<span class="sourceLineNo">419</span> private void removeHdfsRegions(final ThreadPoolExecutor exec, final List<RegionInfo> regions)<a name="line.419"></a> -<span class="sourceLineNo">420</span> throws IOException {<a name="line.420"></a> -<span class="sourceLineNo">421</span> if (regions == null || regions.isEmpty()) return;<a name="line.421"></a> -<span class="sourceLineNo">422</span> ModifyRegionUtils.editRegions(exec, regions, new ModifyRegionUtils.RegionEditTask() {<a name="line.422"></a> -<span class="sourceLineNo">423</span> @Override<a name="line.423"></a> -<span class="sourceLineNo">424</span> public void editRegion(final RegionInfo hri) throws IOException {<a name="line.424"></a> -<span class="sourceLineNo">425</span> HFileArchiver.archiveRegion(conf, fs, hri);<a name="line.425"></a> -<span class="sourceLineNo">426</span> }<a name="line.426"></a> -<span class="sourceLineNo">427</span> });<a name="line.427"></a> -<span class="sourceLineNo">428</span> }<a name="line.428"></a> -<span class="sourceLineNo">429</span><a name="line.429"></a> -<span class="sourceLineNo">430</span> /**<a name="line.430"></a> -<span class="sourceLineNo">431</span> * Restore specified regions by restoring content to the snapshot state.<a name="line.431"></a> -<span class="sourceLineNo">432</span> */<a name="line.432"></a> -<span class="sourceLineNo">433</span> private void restoreHdfsRegions(final ThreadPoolExecutor exec,<a name="line.433"></a> -<span class="sourceLineNo">434</span> final Map<String, SnapshotRegionManifest> regionManifests,<a name="line.434"></a> -<span class="sourceLineNo">435</span> final List<RegionInfo> regions) throws IOException {<a name="line.435"></a> -<span class="sourceLineNo">436</span> if (regions == null || regions.isEmpty()) return;<a name="line.436"></a> -<span class="sourceLineNo">437</span> ModifyRegionUtils.editRegions(exec, regions, new ModifyRegionUtils.RegionEditTask() {<a name="line.437"></a> -<span class="sourceLineNo">438</span> @Override<a name="line.438"></a> -<span class="sourceLineNo">439</span> public void editRegion(final RegionInfo hri) throws IOException {<a name="line.439"></a> -<span class="sourceLineNo">440</span> restoreRegion(hri, regionManifests.get(hri.getEncodedName()));<a name="line.440"></a> -<span class="sourceLineNo">441</span> }<a name="line.441"></a> -<span class="sourceLineNo">442</span> });<a name="line.442"></a> -<span class="sourceLineNo">443</span> }<a name="line.443"></a> -<span class="sourceLineNo">444</span><a name="line.444"></a> -<span class="sourceLineNo">445</span> /**<a name="line.445"></a> -<span class="sourceLineNo">446</span> * Restore specified mob regions by restoring content to the snapshot state.<a name="line.446"></a> -<span class="sourceLineNo">447</span> */<a name="line.447"></a> -<span class="sourceLineNo">448</span> private void restoreHdfsMobRegions(final ThreadPoolExecutor exec,<a name="line.448"></a> -<span class="sourceLineNo">449</span> final Map<String, SnapshotRegionManifest> regionManifests,<a name="line.449"></a> -<span class="sourceLineNo">450</span> final List<RegionInfo> regions) throws IOException {<a name="line.450"></a> -<span class="sourceLineNo">451</span> if (regions == null || regions.isEmpty()) return;<a name="line.451"></a> -<span class="sourceLineNo">452</span> ModifyRegionUtils.editRegions(exec, regions, new ModifyRegionUtils.RegionEditTask() {<a name="line.452"></a> -<span class="sourceLineNo">453</span> @Override<a name="line.453"></a> -<span class="sourceLineNo">454</span> public void editRegion(final RegionInfo hri) throws IOException {<a name="line.454"></a> -<span class="sourceLineNo">455</span> restoreMobRegion(hri, regionManifests.get(hri.getEncodedName()));<a name="line.455"></a> -<span class="sourceLineNo">456</span> }<a name="line.456"></a> -<span class="sourceLineNo">457</span> });<a name="line.457"></a> -<span class="sourceLineNo">458</span> }<a name="line.458"></a> -<span class="sourceLineNo">459</span><a name="line.459"></a> -<span class="sourceLineNo">460</span> private Map<String, List<SnapshotRegionManifest.StoreFile>> getRegionHFileReferences(<a name="line.460"></a> -<span class="sourceLineNo">461</span> final SnapshotRegionManifest manifest) {<a name="line.461"></a> -<span class="sourceLineNo">462</span> Map<String, List<SnapshotRegionManifest.StoreFile>> familyMap =<a name="line.462"></a> -<span class="sourceLineNo">463</span> new HashMap<>(manifest.getFamilyFilesCount());<a name="line.463"></a> -<span class="sourceLineNo">464</span> for (SnapshotRegionManifest.FamilyFiles familyFiles: manifest.getFamilyFilesList()) {<a name="line.464"></a> -<span class="sourceLineNo">465</span> familyMap.put(familyFiles.getFamilyName().toStringUtf8(),<a name="line.465"></a> -<span class="sourceLineNo">466</span> new ArrayList<>(familyFiles.getStoreFilesList()));<a name="line.466"></a> -<span class="sourceLineNo">467</span> }<a name="line.467"></a> -<span class="sourceLineNo">468</span> return familyMap;<a name="line.468"></a> -<span class="sourceLineNo">469</span> }<a name="line.469"></a> -<span class="sourceLineNo">470</span><a name="line.470"></a> -<span class="sourceLineNo">471</span> /**<a name="line.471"></a> -<span class="sourceLineNo">472</span> * Restore region by removing files not in the snapshot<a name="line.472"></a> -<span class="sourceLineNo">473</span> * and adding the missing ones from the snapshot.<a name="line.473"></a> -<span class="sourceLineNo">474</span> */<a name="line.474"></a> -<span class="sourceLineNo">475</span> private void restoreRegion(final RegionInfo regionInfo,<a name="line.475"></a> -<span class="sourceLineNo">476</span> final SnapshotRegionManifest regionManifest) throws IOException {<a name="line.476"></a> -<span class="sourceLineNo">477</span> restoreRegion(regionInfo, regionManifest, new Path(tableDir, regionInfo.getEncodedName()));<a name="line.477"></a> -<span class="sourceLineNo">478</span> }<a name="line.478"></a> -<span class="sourceLineNo">479</span><a name="line.479"></a> -<span class="sourceLineNo">480</span> /**<a name="line.480"></a> -<span class="sourceLineNo">481</span> * Restore mob region by removing files not in the snapshot<a name="line.481"></a> -<span class="sourceLineNo">482</span> * and adding the missing ones from the snapshot.<a name="line.482"></a> -<span class="sourceLineNo">483</span> */<a name="line.483"></a> -<span class="sourceLineNo">484</span> private void restoreMobRegion(final RegionInfo regionInfo,<a name="line.484"></a> -<span class="sourceLineNo">485</span> final SnapshotRegionManifest regionManifest) throws IOException {<a name="line.485"></a> -<span class="sourceLineNo">486</span> if (regionManifest == null) {<a name="line.486"></a> -<span class="sourceLineNo">487</span> return;<a name="line.487"></a> -<span class="sourceLineNo">488</span> }<a name="line.488"></a> -<span class="sourceLineNo">489</span> restoreRegion(regionInfo, regionManifest,<a name="line.489"></a> -<span class="sourceLineNo">490</span> MobUtils.getMobRegionPath(conf, tableDesc.getTableName()));<a name="line.490"></a> -<span class="sourceLineNo">491</span> }<a name="line.491"></a> -<span class="sourceLineNo">492</span><a name="line.492"></a> -<span class="sourceLineNo">493</span> /**<a name="line.493"></a> -<span class="sourceLineNo">494</span> * Restore region by removing files not in the snapshot<a name="line.494"></a> -<span class="sourceLineNo">495</span> * and adding the missing ones from the snapshot.<a name="line.495"></a> -<span class="sourceLineNo">496</span> */<a name="line.496"></a> -<span class="sourceLineNo">497</span> private void restoreRegion(final RegionInfo regionInfo,<a name="line.497"></a> -<span class="sourceLineNo">498</span> final SnapshotRegionManifest regionManifest, Path regionDir) throws IOException {<a name="line.498"></a> -<span class="sourceLineNo">499</span> Map<String, List<SnapshotRegionManifest.StoreFile>> snapshotFiles =<a name="line.499"></a> -<span class="sourceLineNo">500</span> getRegionHFileReferences(regionManifest);<a name="line.500"></a> -<span class="sourceLineNo">501</span><a name="line.501"></a> -<span class="sourceLineNo">502</span> String tableName = tableDesc.getTableName().getNameAsString();<a name="line.502"></a> -<span class="sourceLineNo">503</span><a name="line.503"></a> -<span class="sourceLineNo">504</span> // Restore families present in the table<a name="line.504"></a> -<span class="sourceLineNo">505</span> for (Path familyDir: FSUtils.getFamilyDirs(fs, regionDir)) {<a name="line.505"></a> -<span class="sourceLineNo">506</span> byte[] family = Bytes.toBytes(familyDir.getName());<a name="line.506"></a> -<span class="sourceLineNo">507</span> Set<String> familyFiles = getTableRegionFamilyFiles(familyDir);<a name="line.507"></a> -<span class="sourceLineNo">508</span> List<SnapshotRegionManifest.StoreFile> snapshotFamilyFiles =<a name="line.508"></a> -<span class="sourceLineNo">509</span> snapshotFiles.remove(familyDir.getName());<a name="line.509"></a> -<span class="sourceLineNo">510</span> if (snapshotFamilyFiles != null) {<a name="line.510"></a> -<span class="sourceLineNo">511</span> List<SnapshotRegionManifest.StoreFile> hfilesToAdd = new ArrayList<>();<a name="line.511"></a> -<span class="sourceLineNo">512</span> for (SnapshotRegionManifest.StoreFile storeFile: snapshotFamilyFiles) {<a name="line.512"></a> -<span class="sourceLineNo">513</span> if (familyFiles.contains(storeFile.getName())) {<a name="line.513"></a> -<span class="sourceLineNo">514</span> // HFile already present<a name="line.514"></a> -<span class="sourceLineNo">515</span> familyFiles.remove(storeFile.getName());<a name="line.515"></a> -<span class="sourceLineNo">516</span> } else {<a name="line.516"></a> -<span class="sourceLineNo">517</span> // HFile missing<a name="line.517"></a> -<span class="sourceLineNo">518</span> hfilesToAdd.add(storeFile);<a name="line.518"></a> -<span class="sourceLineNo">519</span> }<a name="line.519"></a> -<span class="sourceLineNo">520</span> }<a name="line.520"></a> -<span class="sourceLineNo">521</span><a name="line.521"></a> -<span class="sourceLineNo">522</span> // Remove hfiles not present in the snapshot<a name="line.522"></a> -<span class="sourceLineNo">523</span> for (String hfileName: familyFiles) {<a name="line.523"></a> -<span class="sourceLineNo">524</span> Path hfile = new Path(familyDir, hfileName);<a name="line.524"></a> -<span class="sourceLineNo">525</span> LOG.trace("Removing hfile=" + hfileName +<a name="line.525"></a> -<span class="sourceLineNo">526</span> " from region=" + regionInfo.getEncodedName() + " table=" + tableName);<a name="line.526"></a> -<span class="sourceLineNo">527</span> HFileArchiver.archiveStoreFile(conf, fs, regionInfo, tableDir, family, hfile);<a name="line.527"></a> -<span class="sourceLineNo">528</span> }<a name="line.528"></a> -<span class="sourceLineNo">529</span><a name="line.529"></a> -<span class="sourceLineNo">530</span> // Restore Missing files<a name="line.530"></a> -<span class="sourceLineNo">531</span> for (SnapshotRegionManifest.StoreFile storeFile: hfilesToAdd) {<a name="line.531"></a> -<span class="sourceLineNo">532</span> LOG.debug("Adding HFileLink " + storeFile.getName() +<a name="line.532"></a> -<span class="sourceLineNo">533</span> " to region=" + regionInfo.getEncodedName() + " table=" + tableName);<a name="line.533"></a> -<span class="sourceLineNo">534</span> restoreStoreFile(familyDir, regionInfo, storeFile, createBackRefs);<a name="line.534"></a> -<span class="sourceLineNo">535</span> }<a name="line.535"></a> -<span class="sourceLineNo">536</span> } else {<a name="line.536"></a> -<span class="sourceLineNo">537</span> // Family doesn't exists in the snapshot<a name="line.537"></a> -<span class="sourceLineNo">538</span> LOG.trace("Removing family=" + Bytes.toString(family) +<a name="line.538"></a> -<span class="sourceLineNo">539</span> " from region=" + regionInfo.getEncodedName() + " table=" + tableName);<a name="line.539"></a> -<span class="sourceLineNo">540</span> HFileArchiver.archiveFamilyByFamilyDir(fs, conf, regionInfo, familyDir, family);<a name="line.540"></a> -<span class="sourceLineNo">541</span> fs.delete(familyDir, true);<a name="line.541"></a> -<span class="sourceLineNo">542</span> }<a name="line.542"></a> -<span class="sourceLineNo">543</span> }<a name="line.543"></a> -<span class="sourceLineNo">544</span><a name="line.544"></a> -<span class="sourceLineNo">545</span> // Add families not present in the table<a name="line.545"></a> -<span class="sourceLineNo">546</span> for (Map.Entry<String, List<SnapshotRegionManifest.StoreFile>> familyEntry:<a name="line.546"></a> -<span class="sourceLineNo">547</span> snapshotFiles.entrySet()) {<a name="line.547"></a> -<span class="sourceLineNo">548</span> Path familyDir = new Path(regionDir, familyEntry.getKey());<a name="line.548"></a> -<span class="sourceLineNo">549</span> if (!fs.mkdirs(familyDir)) {<a name="line.549"></a> -<span class="sourceLineNo">550</span> throw new IOException("Unable to create familyDir=" + familyDir);<a name="line.550"></a> -<span class="sourceLineNo">551</span> }<a name="line.551"></a> -<span class="sourceLineNo">552</span><a name="line.552"></a> -<span class="sourceLineNo">553</span> for (SnapshotRegionManifest.StoreFile storeFile: familyEntry.getValue()) {<a name="line.553"></a> -<span class="sourceLineNo">554</span> LOG.trace("Adding HFileLink " + storeFile.getName() + " to table=" + tableName);<a name="line.554"></a> -<span class="sourceLineNo">555</span> restoreStoreFile(familyDir, regionInfo, storeFile, createBackRefs);<a name="line.555"></a> -<span class="sourceLineNo">556</span> }<a name="line.556"></a> -<span class="sourceLineNo">557</span> }<a name="line.557"></a> -<span class="sourceLineNo">558</span> }<a name="line.558"></a> -<span class="sourceLineNo">559</span><a name="line.559"></a> -<span class="sourceLineNo">560</span> /**<a name="line.560"></a> -<span class="sourceLineNo">561</span> * @return The set of files in the specified family directory.<a name="line.561"></a> -<span class="sourceLineNo">562</span> */<a name="line.562"></a> -<span class="sourceLineNo">563</span> private Set<String> getTableRegionFamilyFiles(final Path familyDir) throws IOException {<a name="line.563"></a> -<span class="sourceLineNo">564</span> FileStatus[] hfiles = FSUtils.listStatus(fs, familyDir);<a name="line.564"></a> -<span class="sourceLineNo">565</span> if (hfiles == null) return Collections.emptySet();<a name="line.565"></a> -<span class="sourceLineNo">566</span><a name="line.566"></a> -<span class="sourceLineNo">567</span> Set<String> familyFiles = new HashSet<>(hfiles.length);<a name="line.567"></a> -<span class="sourceLineNo">568</span> for (int i = 0; i < hfiles.length; ++i) {<a name="line.568"></a> -<span class="sourceLineNo">569</span> String hfileName = hfiles[i].getPath().getName();<a name="line.569"></a> -<span class="sourceLineNo">570</span> familyFiles.add(hfileName);<a name="line.570"></a> -<span class="sourceLineNo">571</span> }<a name="line.571"></a> -<span class="sourceLineNo">572</span><a name="line.572"></a> -<span class="sourceLineNo">573</span> return familyFiles;<a name="line.573"></a> -<span class="sourceLineNo">574</span> }<a name="line.574"></a> -<span class="sourceLineNo">575</span><a name="line.575"></a> -<span class="sourceLineNo">576</span> /**<a name="line.576"></a> -<span class="sourceLineNo">577</span> * Clone specified regions. For each region create a new region<a name="line.577"></a> -<span class="sourceLineNo">578</span> * and create a HFileLink for each hfile.<a name="line.578"></a> -<span class="sourceLineNo">579</span> */<a name="line.579"></a> -<span class="sourceLineNo">580</span> private RegionInfo[] cloneHdfsRegions(final ThreadPoolExecutor exec,<a name="line.580"></a> -<span class="sourceLineNo">581</span> final Map<String, SnapshotRegionManifest> regionManifests,<a name="line.581"></a> -<span class="sourceLineNo">582</span> final List<RegionInfo> regions) throws IOException {<a name="line.582"></a> -<span class="sourceLineNo">583</span> if (regions == null || regions.isEmpty()) return null;<a name="line.583"></a> -<span class="sourceLineNo">584</span><a name="line.584"></a> -<span class="sourceLineNo">585</span> final Map<String, RegionInfo> snapshotRegions = new HashMap<>(regions.size());<a name="line.585"></a> -<span class="sourceLineNo">586</span><a name="line.586"></a> -<span class="sourceLineNo">587</span> // clone region info (change embedded tableName with the new one)<a name="line.587"></a> -<span class="sourceLineNo">588</span> RegionInfo[] clonedRegionsInfo = new RegionInfo[regions.size()];<a name="line.588"></a> -<span class="sourceLineNo">589</span> for (int i = 0; i < clonedRegionsInfo.length; ++i) {<a name="line.589"></a> -<span class="sourceLineNo">590</span> // clone the region info from the snapshot region info<a name="line.590"></a> -<span class="sourceLineNo">591</span> RegionInfo snapshotRegionInfo = regions.get(i);<a name="line.591"></a> -<span class="sourceLineNo">592</span> clonedRegionsInfo[i] = cloneRegionInfo(snapshotRegionInfo);<a name="line.592"></a> -<span class="sourceLineNo">593</span><a name="line.593"></a> -<span class="sourceLineNo">594</span> // add the region name mapping between snapshot and cloned<a name="line.594"></a> -<span class="sourceLineNo">595</span> String snapshotRegionName = snapshotRegionInfo.getEncodedName();<a name="line.595"></a> -<span class="sourceLineNo">596</span> String clonedRegionName = clonedRegionsInfo[i].getEncodedName();<a name="line.596"></a> -<span class="sourceLineNo">597</span> regionsMap.put(Bytes.toBytes(snapshotRegionName), Bytes.toBytes(clonedRegionName));<a name="line.597"></a> -<span class="sourceLineNo">598</span> LOG.info("clone region=" + snapshotRegionName + " as " + clonedRegionName);<a name="line.598"></a> -<span class="sourceLineNo">599</span><a name="line.599"></a> -<span class="sourceLineNo">600</span> // Add mapping between cloned region name and snapshot region info<a name="line.600"></a> -<span class="sourceLineNo">601</span> snapshotRegions.put(clonedRegionName, snapshotRegionInfo);<a name="line.601"></a> -<span class="sourceLineNo">602</span> }<a name="line.602"></a> -<span class="sourceLineNo">603</span><a name="line.603"></a> -<span class="sourceLineNo">604</span> // create the regions on disk<a name="line.604"></a> -<span class="sourceLineNo">605</span> ModifyRegionUtils.createRegions(exec, conf, rootDir,<a name="line.605"></a> -<span class="sourceLineNo">606</span> tableDesc, clonedRegionsInfo, new ModifyRegionUtils.RegionFillTask() {<a name="line.606"></a> -<span class="sourceLineNo">607</span> @Override<a name="line.607"></a> -<span class="sourceLineNo">608</span> public void fillRegion(final HRegion region) throws IOException {<a name="line.608"></a> -<span class="sourceLineNo">609</span> RegionInfo snapshotHri = snapshotRegions.get(region.getRegionInfo().getEncodedName());<a name="line.609"></a> -<span class="sourceLineNo">610</span> cloneRegion(region, snapshotHri, regionManifests.get(snapshotHri.getEncodedName()));<a name="line.610"></a> -<span class="sourceLineNo">611</span> }<a name="line.611"></a> -<span class="sourceLineNo">612</span> });<a name="line.612"></a> -<span class="sourceLineNo">613</span><a name="line.613"></a> -<span class="sourceLineNo">614</span> return clonedRegionsInfo;<a name="line.614"></a> -<span class="sourceLineNo">615</span> }<a name="line.615"></a> -<span class="sourceLineNo">616</span><a name="line.616"></a> -<span class="sourceLineNo">617</span> /**<a name="line.617"></a> -<span class="sourceLineNo">618</span> * Clone the mob region. For the region create a new region<a name="line.618"></a> -<span class="sourceLineNo">619</span> * and create a HFileLink for each hfile.<a name="line.619"></a> -<span class="sourceLineNo">620</span> */<a name="line.620"></a> -<span class="sourceLineNo">621</span> private void cloneHdfsMobRegion(final Map<String, SnapshotRegionManifest> regionManifests,<a name="line.621"></a> -<span class="sourceLineNo">622</span> final RegionInfo region) throws IOException {<a name="line.622"></a> -<span class="sourceLineNo">623</span> // clone region info (change embedded tableName with the new one)<a name="line.623"></a> -<span class="sourceLineNo">624</span> Path clonedRegionPath = MobUtils.getMobRegionPath(conf, tableDesc.getTableName());<a name="line.624"></a> -<span class="sourceLineNo">625</span> cloneRegion(clonedRegionPath, region, regionManifests.get(region.getEncodedName()));<a name="line.625"></a> -<span class="sourceLineNo">626</span> }<a name="line.626"></a> -<span class="sourceLineNo">627</span><a name="line.627"></a> -<span class="sourceLineNo">628</span> /**<a name="line.628"></a> -<span class="sourceLineNo">629</span> * Clone region directory content from the snapshot info.<a name="line.629"></a> -<span class="sourceLineNo">630</span> *<a name="line.630"></a> -<span class="sourceLineNo">631</span> * Each region is encoded with the table name, so the cloned region will have<a name="line.631"></a> -<span class="sourceLineNo">632</span> * a different region name.<a name="line.632"></a> -<span class="sourceLineNo">633</span> *<a name="line.633"></a> -<span class="sourceLineNo">634</span> * Instead of copying the hfiles a HFileLink is created.<a name="line.634"></a> -<span class="sourceLineNo">635</span> *<a name="line.635"></a> -<span class="sourceLineNo">636</span> * @param regionDir {@link Path} cloned dir<a name="line.636"></a> -<span class="sourceLineNo">637</span> * @param snapshotRegionInfo<a name="line.637"></a> -<span class="sourceLineNo">638</span> */<a name="line.638"></a> -<span class="sourceLineNo">639</span> private void cloneRegion(final Path regionDir, final RegionInfo snapshotRegionInfo,<a name="line.639"></a> -<span class="sourceLineNo">640</span> final SnapshotRegionManifest manifest) throws IOException {<a name="line.640"></a> -<span class="sourceLineNo">641</span> final String tableName = tableDesc.getTableName().getNameAsString();<a name="line.641"></a> -<span class="sourceLineNo">642</span> for (SnapshotRegionManifest.FamilyFiles familyFiles: manifest.getFamilyFilesList()) {<a name="line.642"></a> -<span class="sourceLineNo">643</span> Path familyDir = new Path(regionDir, familyFiles.getFamilyName().toStringUtf8());<a name="line.643"></a> -<span class="sourceLineNo">644</span> for (SnapshotRegionManifest.StoreFile storeFile: familyFiles.getStoreFilesList()) {<a name="line.644"></a> -<span class="sourceLineNo">645</span> LOG.info("Adding HFileLink " + storeFile.getName() + " to table=" + tableName);<a name="line.645"></a> -<span class="sourceLineNo">646</span> restoreStoreFile(familyDir, snapshotRegionInfo, storeFile, createBackRefs);<a name="line.646"></a> -<span class="sourceLineNo">647</span> }<a name="line.647"></a> -<span class="sourceLineNo">648</span> }<a name="line.648"></a> -<span class="sourceLineNo">649</span> }<a name="line.649"></a> -<span class="sourceLineNo">650</span><a name="line.650"></a> -<span class="sourceLineNo">651</span> /**<a name="line.651"></a> -<span class="sourceLineNo">652</span> * Clone region directory content from the snapshot info.<a name="line.652"></a> -<span class="sourceLineNo">653</span> *<a name="line.653"></a> -<span class="sourceLineNo">654</span> * Each region is encoded with the table name, so the cloned region will have<a name="line.654"></a> -<span class="sourceLineNo">655</span> * a different region name.<a name="line.655"></a> -<span class="sourceLineNo">656</span> *<a name="line.656"></a> -<span class="sourceLineNo">657</span> * Instead of copying the hfiles a HFileLink is created.<a name="line.657"></a> -<span class="sourceLineNo">658</span> *<a name="line.658"></a> -<span class="sourceLineNo">659</span> * @param region {@link HRegion} cloned<a name="line.659"></a> -<span class="sourceLineNo">660</span> * @param snapshotRegionInfo<a name="line.660"></a> -<span class="sourceLineNo">661</span> */<a name="line.661"></a> -<span class="sourceLineNo">662</span> private void cloneRegion(final HRegion region, final RegionInfo snapshotRegionInfo,<a name="line.662"></a> -<span class="sourceLineNo">663</span> final SnapshotRegionManifest manifest) throws IOException {<a name="line.663"></a> -<span class="sourceLineNo">664</span> cloneRegion(new Path(tableDir, region.getRegionInfo().getEncodedName()), snapshotRegionInfo,<a name="line.664"></a> -<span class="sourceLineNo">665</span> manifest);<a name="line.665"></a> -<span class="sourceLineNo">666</span> }<a name="line.666"></a> -<span class="sourceLineNo">667</span><a name="line.667"></a> -<span class="sourceLineNo">668</span> /**<a name="line.668"></a> -<span class="sourceLineNo">669</span> * Create a new {@link HFileLink} to reference the store file.<a name="line.669"></a> -<span class="sourceLineNo">670</span> * <p>The store file in the snapshot can be a simple hfile, an HFileLink or a reference.<a name="line.670"></a> -<span class="sourceLineNo">671</span> * <ul><a name="line.671"></a> -<span class="sourceLineNo">672</span> * <li>hfile: abc -> table=region-abc<a name="line.672"></a> -<span class="sourceLineNo">673</span> * <li>reference: abc.1234 -> table=region-abc.1234<a name="line.673"></a> -<span class="sourceLineNo">674</span> * <li>hfilelink: table=region-hfile -> table=region-hfile<a name="line.674"></a> -<span class="sourceLineNo">675</span> * </ul><a name="line.675"></a> -<span class="sourceLineNo">676</span> * @param familyDir destination directory for the store file<a name="line.676"></a> -<span class="sourceLineNo">677</span> * @param regionInfo destination region info for the table<a name="line.677"></a> -<span class="sourceLineNo">678</span> * @param createBackRef - Whether back reference should be created. Defaults to true.<a name="line.678"></a> -<span class="sourceLineNo">679</span> * @param storeFile store file name (can be a Reference, HFileLink or simple HFile)<a name="line.679"></a> -<span class="sourceLineNo">680</span> */<a name="line.680"></a> -<span class="sourceLineNo">681</span> private void restoreStoreFile(final Path familyDir, final RegionInfo regionInfo,<a name="line.681"></a> -<span class="sourceLineNo">682</span> final SnapshotRegionManifest.StoreFile storeFile, final boolean createBackRef)<a name="line.682"></a> -<span class="sourceLineNo">683</span> throws IOException {<a name="line.683"></a> -<span class="sourceLineNo">684</span> String hfileName = storeFile.getName();<a name="line.684"></a> -<span class="sourceLineNo">685</span> if (HFileLink.isHFileLink(hfileName)) {<a name="line.685"></a> -<span class="sourceLineNo">686</span> HFileLink.createFromHFileLink(conf, fs, familyDir, hfileName, createBackRef);<a name="line.686"></a> -<span class="sourceLineNo">687</span> } else if (StoreFileInfo.isReference(hfileName)) {<a name="line.687"></a> -<span class="sourceLineNo">688</span> restoreReferenceFile(familyDir, regionInfo, storeFile);<a name="line.688"></a> -<span class="sourceLineNo">689</span> } else {<a name="line.689"></a> -<span class="sourceLineNo">690</span> HFileLink.create(conf, fs, familyDir, regionInfo, hfileName, createBackRef);<a name="line.690"></a> -<span class="sourceLineNo">691</span> }<a name="line.691"></a> -<span class="sourceLineNo">692</span> }<a name="line.692"></a> -<span class="sourceLineNo">693</span><a name="line.693"></a> -<span class="sourceLineNo">694</span> /**<a name="line.694"></a> -<span class="sourceLineNo">695</span> * Create a new {@link Reference} as copy of the source one.<a name="line.695"></a> -<span class="sourceLineNo">696</span> * <p><blockquote><pre><a name="line.696"></a> -<span class="sourceLineNo">697</span> * The source table looks like:<a name="line.697"></a> -<span class="sourceLineNo">698</span> * 1234/abc (original file)<a name="line.698"></a> -<span class="sourceLineNo">699</span> * 5678/abc.1234 (reference file)<a name="line.699"></a> -<span class="sourceLineNo">700</span> *<a name="line.700"></a> -<span class="sourceLineNo">701</span> * After the clone operation looks like:<a name="line.701"></a> -<span class="sourceLineNo">702</span> * wxyz/table=1234-abc<a name="line.702"></a> -<span class="sourceLineNo">703</span> * stuv/table=1234-abc.wxyz<a name="line.703"></a> -<span class="sourceLineNo">704</span> *<a name="line.704"></a> -<span class="sourceLineNo">705</span> * NOTE that the region name in the clone changes (md5 of regioninfo)<a name="line.705"></a> -<span class="sourceLineNo">706</span> * and the reference should reflect that change.<a name="line.706"></a> -<span class="sourceLineNo">707</span> * </pre></blockquote><a name="line.707"></a> -<span class="sourceLineNo">708</span> * @param familyDir destination directory for the store file<a name="line.708"></a> -<span class="sourceLineNo">709</span> * @param regionInfo destination region info for the table<a name="line.709"></a> -<span class="sourceLineNo">710</span> * @param storeFile reference file name<a name="line.710"></a> -<span class="sourceLineNo">711</span> */<a name="line.711"></a> -<span class="sourceLineNo">712</span> private void restoreReferenceFile(final Path familyDir, final RegionInfo regionInfo,<a name="line.712"></a> -<span class="sourceLineNo">713</span> final SnapshotRegionManifest.StoreFile storeFile) throws IOException {<a name="line.713"></a> -<span class="sourceLineNo">714</span> String hfileName = storeFile.getName();<a name="line.714"></a> -<span class="sourceLineNo">715</span><a name="line.715"></a> -<span class="sourceLineNo">716</span> // Extract the referred information (hfile name and parent region)<a name="line.716"></a> -<span class="sourceLineNo">717</span> Path refPath =<a name="line.717"></a> -<span class="sourceLineNo">718</span> StoreFileInfo.getReferredToFile(new Path(new Path(new Path(new Path(snapshotTable<a name="line.718"></a> -<span class="sourceLineNo">719</span> .getNamespaceAsString(), snapshotTable.getQualifierAsString()), regionInfo<a name="line.719"></a> -<span class="sourceLineNo">720</span> .getEncodedName()), familyDir.getName()), hfileName));<a name="line.720"></a> -<span class="sourceLineNo">721</span> String snapshotRegionName = refPath.getParent().getParent().getName();<a name="line.721"></a> -<span class="sourceLineNo">722</span> String fileName = refPath.getName();<a name="line.722"></a> -<span class="sourceLineNo">723</span><a name="line.723"></a> -<span class="sourceLineNo">724</span> // The new reference should have the cloned region name as parent, if it is a clone.<a name="line.724"></a> -<span class="sourceLineNo">725</span> String clonedRegionName = Bytes.toString(regionsMap.get(Bytes.toBytes(snapshotRegionName)));<a name="line.725"></a> -<span class="sourceLineNo">726</span> if (clonedRegionName == null) clonedRegionName = snapshotRegionName;<a name="line.726"></a> -<span class="sourceLineNo">727</span><a name="line.727"></a> -<span class="sourceLineNo">728</span> // The output file should be a reference link table=snapshotRegion-fileName.clonedRegionName<a name="line.728"></a> -<span class="sourceLineNo">729</span> Path linkPath = null;<a name="line.729"></a> -<span class="sourceLineNo">730</span> String refLink = fileName;<a name="line.730"></a> -<span class="sourceLineNo">731</span> if (!HFileLink.isHFileLink(fileName)) {<a name="line.731"></a> -<span class="sourceLineNo">732</span> refLink = HFileLink.createHFileLinkName(snapshotTable, snapshotRegionName, fileName);<a name="line.732"></a> -<span class="sourceLineNo">733</span> linkPath = new Path(familyDir,<a name="line.733"></a> -<span class="sourceLineNo">734</span> HFileLink.createHFileLinkName(snapshotTable, regionInfo.getEncodedName(), hfileName));<a name="line.734"></a> -<span class="sourceLineNo">735</span> }<a name="line.735"></a> -<span class="sourceLineNo">736</span><a name="line.736"></a> -<span class="sourceLineNo">737</span> Path outPath = new Path(familyDir, refLink + '.' + clonedRegionName);<a name="line.737"></a> -<span class="sourceLineNo">738</span><a name="line.738"></a> -<span class="sourceLineNo">739</span> // Create the new reference<a name="line.739"></a> -<span class="sourceLineNo">740</span> if (storeFile.hasReference()) {<a name="line.740"></a> -<span class="sourceLineNo">741</span> Reference reference = Reference.convert(storeFile.getReference());<a name="line.741"></a> -<span class="sourceLineNo">742</span> reference.write(fs, outPath);<a name="line.742"></a> -<span class="sourceLineNo">743</span> } else {<a name="line.743"></a> -<span class="sourceLineNo">744</span> InputStream in;<a name="line.744"></a> -<span class="sourceLineNo">745</span> if (linkPath != null) {<a name="line.745"></a> -<span class="sourceLineNo">746</span> in = HFileLink.buildFromHFileLinkPattern(conf, linkPath).open(fs);<a name="line.746"></a> -<span class="sourceLineNo">747</span> } else {<a name="line.747"></a> -<span class="sourceLineNo">748</span> linkPath = new Path(new Path(HRegion.getRegionDir(snapshotManifest.getSnapshotDir(),<a name="line.748"></a> -<span class="sourceLineNo">749</span> regionInfo.getEncodedName()), familyDir.getName()), hfileName);<a name="line.749"></a> -<span class="sourceLineNo">750</span> in = fs.open(linkPath);<a name="line.750"></a> -<span class="sourceLineNo">751</span> }<a name="line.751"></a> -<span class="sourceLineNo">752</span> OutputStream out = fs.create(outPath);<a name="line.752"></a> -<span class="sourceLineNo">753</span> IOUtils.copyBytes(in, out, conf);<a name="line.753"></a> -<span class="sourceLineNo">754</span> }<a name="line.754"></a> -<span class="sourceLineNo">755</span><a name="line.755"></a> -<span class="sourceLineNo">756</span> // Add the daughter region to the map<a name="line.756"></a> -<span class="sourceLineNo">757</span> String regionName = Bytes.toString(regionsMap.get(regionInfo.getEncodedNameAsBytes()));<a name="line.757"></a> -<span class="sourceLineNo">758</span> if (regionName == null) {<a name="line.758"></a> -<span class="sourceLineNo">759</span> regionName = regionInfo.getEncodedName();<a name="line.759"></a> -<span class="sourceLineNo">760</span> }<a name="line.760"></a> -<span class="sourceLineNo">761</span> LOG.debug("Restore reference " + regionName + " to " + clonedRegionName);<a name="line.761"></a> -<span class="sourceLineNo">762</span> synchronized (parentsMap) {<a name="line.762"></a> -<span class="sourceLineNo">763</span> Pair<String, String> daughters = parentsMap.get(clonedRegionName);<a name="line.763"></a> -<span class="sourceLineNo">764</span> if (daughters == null) {<a name="line.764"></a> -<span class="sourceLineNo">765</span> // In case one side of the split is already compacted, regionName is put as both first and<a name="line.765"></a> -<span class="sourceLineNo">766</span> // second of Pair<a name="line.766"></a> -<span class="sourceLineNo">767</span> daughters = new Pair<>(regionName, regionName);<a name="line.767"></a> -<span class="sourceLineNo">768</span> parentsMap.put(clonedRegionName, daughters);<a name="line.768"></a> -<span class="sourceLineNo">769</span> } else if (!regionName.equals(daughters.getFirst())) {<a name="line.769"></a> -<span class="sourceLineNo">770</span> daughters.setSecond(regionName);<a name="line.770"></a> -<span class="sourceLineNo">771</span> }<a name="line.771"></a> -<span class="sourceLineNo">772</span> }<a name="line.772"></a> -<span class="sourceLineNo">773</span> }<a name="line.773"></a> -<span class="sourceLineNo">774</span><a name="line.774"></a> -<span class="sourceLineNo">775</span> /**<a name="line.775"></a> -<span class="sourceLineNo">776</span> * Create a new {@link RegionInfo} from the snapshot region info.<a name="line.776"></a> -<span class="sourceLineNo">777</span> * Keep the same startKey, endKey, regionId and split information but change<a name="line.777"></a> -<span class="sourceLineNo">778</span> * the table name.<a name="line.778"></a> -<span class="sourceLineNo">779</span> *<a name="line.779"></a> -<span class="sourceLineNo">780</span> * @param snapshotRegionInfo Info for region to clone.<a name="line.780"></a> -<span class="sourceLineNo">781</span> * @return the new HRegion instance<a name="line.781"></a> -<span class="sourceLineNo">782</span> */<a name="line.782"></a> -<span class="sourceLineNo">783</span> public RegionInfo cloneRegionInfo(final RegionInfo snapshotRegionInfo) {<a name="line.783"></a> -<span class="sourceLineNo">784</span> return cloneRegionInfo(tableDesc.getTableName(), snapshotRegionInfo);<a name="line.784"></a> -<span class="sourceLineNo">785</span> }<a name="line.785"></a> -<span class="sourceLineNo">786</span><a name="line.786"></a> -<span class="sourceLineNo">787</span> public static RegionInfo cloneRegionInfo(TableName tableName, RegionInfo snapshotRegionInfo) {<a name="line.787"></a> -<span class="sourceLineNo">788</span> return RegionInfoBuilder.newBuilder(tableName)<a name="line.788"></a> -<span class="sourceLineNo">789</span> .setStartKey(snapshotRegionInfo.getStartKey())<a name="line.789"></a> -<span class="sourceLineNo">790</span> .setEndKey(snapshotRegionInfo.getEndKey())<a name="line.790"></a> -<span class="sourceLineNo">791</span> .setSplit(snapshotRegionInfo.isSplit())<a name="line.791"></a> -<span class="sourceLineNo">792</span> .setRegionId(snapshotRegionInfo.getRegionId())<a name="line.792"></a> -<span class="sourceLineNo">793</span> .setOffline(snapshotRegionInfo.isOffline())<a name="line.793"></a> -<span class="sourceLineNo">794</span> .build();<a name="line.794"></a> -<span class="sourceLineNo">795</span> }<a name="line.795"></a> -<span class="sourceLineNo">796</span><a name="line.796"></a> -<span class="sourceLineNo">797</span> /**<a name="line.797"></a> -<span class="sourceLineNo">798</span> * @return the set of the regions contained in the table<a name="line.798"></a> -<span class="sourceLineNo">799</span> */<a name="line.799"></a> -<span class="sourceLineNo">800</span> private List<RegionInfo> getTableRegions() throws IOException {<a name="line.800"></a> -<span class="sourceLineNo">801</span> LOG.debug("get table regions: " + tableDir);<a name="line.801"></a> -<span class="sourceLineNo">802</span> FileStatus[] regionDirs = FSUtils.listStatus(fs, tableDir, new FSUtils.RegionDirFilter(fs));<a name="line.802"></a> -<span class="sourceLineNo">803</span> if (regionDirs == null) return null;<a name="line.803"></a> -<span class="sourceLineNo">804</span><a name="line.804"></a> -<span class="sourceLineNo">805</span> List<RegionInfo> regions = new ArrayList<>(regionDirs.length);<a name="line.805"></a> -<span class="sourceLineNo">806</span> for (int i = 0; i < regionDirs.length; ++i) {<a name="line.806"></a> -<span class="sourceLineNo">807</span> RegionInfo hri = HRegionFileSystem.loadRegionInfoFileContent(fs, regionDirs[i].getPath());<a name="line.807"></a> -<span class="sourceLineNo">808</span> regions.add(hri);<a name="line.808"></a> -<span class="sourceLineNo">809</span> }<a name="line.809"></a> -<span class="sourceLineNo">810</span> LOG.debug("found " + regions.size() + " regions for table=" +<a name="line.810"></a> -<span class="sourceLineNo">811</span> tableDesc.getTableName().getNameAsString());<a name="line.811"></a> -<span class="sourceLineNo">812</span> return regions;<a name="line.812"></a> -<span class="sourceLineNo">813</span> }<a name="line.813"></a> -<span class="sourceLineNo">814</span><a name="line.814"></a> -<span class="sourceLineNo">815</span> /**<a name="line.815"></a> -<span class="sourceLineNo">816</span> * Copy the snapshot files for a snapshot scanner, discards meta changes.<a name="line.816"></a> -<span class="sourceLineNo">817</span> * @param conf<a name="line.817"></a> -<span class="sourceLineNo">818</span> * @param fs<a name="line.818"></a> -<span class="sourceLineNo">819</span> * @param rootDir<a name="line.819"></a> -<span class="sourceLineNo">820</span> * @param restoreDir<a name="line.820"></a> -<span class="sourceLineNo">821</span> * @param snapshotName<a name="line.821"></a> -<span class="sourceLineNo">822</span> * @throws IOException<a name="line.822"></a> -<span class="sourceLineNo">823</span> */<a name="line.823"></a> -<span class="sourceLineNo">824</span> public static RestoreMetaChanges copySnapshotForScanner(Configuration conf, FileSystem fs,<a name="line.824"></a> -<span class="sourceLineNo">825</span> Path rootDir, Path restoreDir, String snapshotName) throws IOException {<a name="line.825"></a> -<span class="sourceLineNo">826</span> // ensure that restore dir is not under root dir<a name="line.826"></a> -<span class="sourceLineNo">827</span> if (!restoreDir.getFileSystem(conf).getUri().equals(rootDir.getFileSystem(conf).getUri())) {<a name="line.827"></a> -<span class="sourceLineNo">828</span> throw new IllegalArgumentException("Filesystems for restore directory and HBase root " +<a name="line.828"></a> -<span class="sourceLineNo">829</span> "directory should be the same");<a name="line.829"></a> -<span class="sourceLineNo">830</span> }<a name="line.830"></a> -<span class="sourceLineNo">831</span> if (restoreDir.toUri().getPath().startsWith(rootDir.toUri().getPath())) {<a name="line.831"></a> -<span class="sourceLineNo">832</span> throw new IllegalArgumentException("Restore directory cannot be a sub directory of HBase " +<a name="line.832"></a> -<span class="sourceLineNo">833</span> "root directory. RootDir: " + rootDir + ", restoreDir: " + restoreDir);<a name="line.833"></a> -<span class="sourceLineNo">834</span> }<a name="line.834"></a> -<span class="sourceLineNo">835</span><a name="line.835"></a> -<span class="sourceLineNo">836</span> Path snapshotDir = SnapshotDescriptionUtils.getCompletedSnapshotDir(snapshotName, rootDir);<a name="line.836"></a> -<span class="sourceLineNo">837</span> SnapshotDescription snapshotDesc = SnapshotDescriptionUtils.readSnapshotInfo(fs, snapshotDir);<a name="line.837"></a> -<span class="sourceLineNo">838</span> SnapshotManifest manifest = SnapshotManifest.open(conf, fs, snapshotDir, snapshotDesc);<a name="line.838"></a> -<span class="sourceLineNo">839</span><a name="line.839"></a> -<span class="sourceLineNo">840</span> MonitoredTask status = TaskMonitor.get().createStatus(<a name="line.840"></a> -<span class="sourceLineNo">841</span> "Restoring snapshot '" + snapshotName + "' to directory " + restoreDir);<a name="line.841"></a> -<span class="sourceLineNo">842</span> ForeignExceptionDispatcher monitor = new ForeignExceptionDispatcher();<a name="line.842"></a> -<span class="sourceLineNo">843</span><a name="line.843"></a> -<span class="sourceLineNo">844</span> // we send createBackRefs=false so that restored hfiles do not create back reference links<a name="line.844"></a> -<span class="sourceLineNo">845</span> // in the base hbase root dir.<a name="line.845"></a> -<span class="sourceLineNo">846</span> RestoreSnapshotHelper helper = new RestoreSnapshotHelper(conf, fs,<a name="line.846"></a> -<span class="sourceLineNo">847</span> manifest, manifest.getTableDescriptor(), restoreDir, monitor, status, false);<a name="line.847"></a> -<span class="sourceLineNo">848</span> RestoreMetaChanges metaChanges = helper.restoreHdfsRegions(); // TODO: parallelize.<a name="line.848"></a> -<span class="sourceLineNo">849</span><a name="line.849"></a> -<span class="sourceLineNo">850</span> if (LOG.isDebugEnabled()) {<a name="line.850"></a> -<span class="sourceLineNo">851</span> LOG.debug("Restored table dir:" + restoreDir);<a name="line.851"></a> -<span class="sourceLineNo">852</span> FSUtils.logFileSystemState(fs, restoreDir, LOG);<a name="line.852"></a> -<span class="sourceLineNo">853</span> }<a name="line.853"></a> -<span class="sourceLineNo">854</span> return metaChanges;<a name="line.854"></a> -<span class="sourceLineNo">855</span> }<a name="line.855"></a> -<span class="sourceLineNo">856</span><a name="line.856"></a> -<span class="sourceLineNo">857</span> public static void restoreSnapshotAcl(SnapshotDescription snapshot, TableName newTableName,<a name="line.857"></a> -<span class="sourceLineNo">858</span> Configuration conf) throws IOException {<a name="line.858"></a> -<span class="sourceLineNo">859</span> if (snapshot.hasUsersAndPermissions() && snapshot.getUsersAndPermissions() != null) {<a name="line.859"></a> -<span class="sourceLineNo">860</span> LOG.info("Restore snapshot acl to table. snapshot: " + snapshot + ", table: " + newTableName);<a name="line.860"></a> -<span class="sourceLineNo">861</span> ListMultimap<String, TablePermission> perms =<a name="line.861"></a> -<span class="sourceLineNo">862</span> ShadedAccessControlUtil.toUserTablePermissions(snapshot.getUsersAndPermissions());<a name="line.862"></a> -<span class="sourceLineNo">863</span> try (Connection conn = ConnectionFactory.createConnection(conf)) {<a name="line.863"></a> -<span class="sourceLineNo">864</span> for (Entry<String, TablePermission> e : perms.entries()) {<a name="line.864"></a> -<span class="sourceLineNo">865</span> String user = e.getKey();<a name="line.865"></a> -<span class="sourceLineNo">866</span> TablePermission perm = e.getValue();<a name="line.866"></a> -<span class="sourceLineNo">867</span> perm.setTableName(newTableName);<a name="line.867"></a> -<span class="sourceLineNo">868</span> AccessControlClient.grant(conn, perm.getTableName(), user, perm.getFamily(),<a name="line.868"></a> -<span class="sourceLineNo">869</span> perm.getQualifier(), perm.getActions());<a name="line.869"></a> -<span class="sourceLineNo">870</span> }<a name="line.870"></a> -<span class="sourceLineNo">871</span> } catch (Throwable e) {<a name="line.871"></a> -<span class="sourceLineNo">872</span> throw new IOException("Grant acl into newly creatd table failed. snapshot: " + snapshot<a name="line.872"></a> -<span class="sourceLineNo">873</span> + ", table: " + newTableName, e);<a name="line.873"></a> -<span class="sourceLineNo">874</span> }<a name="line.874"></a> -<span class="sourceLineNo">875</span> }<a name="line.875"></a> -<span class="sourceLineNo">876</span> }<a name="line.876"></a> -<span class="sourceLineNo">877</span>}<a name="line.877"></a> +<span class="sourceLineNo">059</span>import org.apache.hadoop.hbase.security.access.Permission;<a name="line.59"></a> +<span class="sourceLineNo">060</span>import org.apache.hadoop.hbase.security.access.ShadedAccessControlUtil;<a name="line.60"></a> +<span class="sourceLineNo">061</span>import org.apache.hadoop.hbase.security.access.TablePermission;<a name="line.61"></a> +<span class="sourceLineNo">062</span>import org.apache.hadoop.hbase.util.Bytes;<a name="line.62"></a> +<span class="sourceLineNo">063</span>import org.apache.hadoop.hbase.util.FSUtils;<a name="line.63"></a> +<span class="sourceLineNo">064</span>import org.apache.hadoop.hbase.util.ModifyRegionUtils;<a name="line.64"></a> +<span class="sourceLineNo">065</span>import org.apache.hadoop.hbase.util.Pair;<a name="line.65"></a> +<span class="sourceLineNo">066</span>import org.apache.hadoop.io.IOUtils;<a name="line.66"></a> +<span class="sourceLineNo">067</span>import org.apache.yetus.audience.InterfaceAudience;<a name="line.67"></a> +<span class="sourceLineNo">068</span>import org.slf4j.Logger;<a name="line.68"></a> +<span class="sourceLineNo">069</span>import org.slf4j.LoggerFactory;<a name="line.69"></a> +<span class="sourceLineNo">070</span>import org.apache.hbase.thirdparty.com.google.common.collect.ListMultimap;<a name="line.70"></a> +<span class="sourceLineNo">071</span>import org.apache.hadoop.hbase.shaded.protobuf.ProtobufUtil;<a name="line.71"></a> +<span class="sourceLineNo">072</span>import org.apache.hadoop.hbase.shaded.protobuf.generated.SnapshotProtos.SnapshotDescription;<a name="line.72"></a> +<span class="sourceLineNo">073</span>import org.apache.hadoop.hbase.shaded.protobuf.generated.SnapshotProtos.SnapshotRegionManifest;<a name="line.73"></a> +<span class="sourceLineNo">074</span><a name="line.74"></a> +<span class="sourceLineNo">075</span>/**<a name="line.75"></a> +<span class="sourceLineNo">076</span> * Helper to Restore/Clone a Snapshot<a name="line.76"></a> +<span class="sourceLineNo">077</span> *<a name="line.77"></a> +<span class="sourceLineNo">078</span> * <p>The helper assumes that a table is already created, and by calling restore()<a name="line.78"></a> +<span class="sourceLineNo">079</span> * the content present in the snapshot will be restored as the new content of the table.<a name="line.79"></a> +<span class="sourceLineNo">080</span> *<a name="line.80"></a> +<span class="sourceLineNo">081</span> * <p>Clone from Snapshot: If the target table is empty, the restore operation<a name="line.81"></a> +<span class="sourceLineNo">082</span> * is just a "clone operation", where the only operations are:<a name="line.82"></a> +<span class="sourceLineNo">083</span> * <ul><a name="line.83"></a> +<span class="sourceLineNo">084</span> * <li>for each region in the snapshot create a new region<a name="line.84"></a> +<span class="sourceLineNo">085</span> * (note that the region will have a different name, since the encoding contains the table name)<a name="line.85"></a> +<span class="sourceLineNo">086</span> * <li>for each file in the region create a new HFileLink to point to the original file.<a name="line.86"></a> +<span class="sourceLineNo">087</span> * <li>restore the logs, if any<a name="line.87"></a> +<span class="sourceLineNo">088</span> * </ul><a name="line.88"></a> +<span class="sourceLineNo">089</span> *<a name="line.89"></a> +<span class="sourceLineNo">090</span> * <p>Restore from Snapshot:<a name="line.90"></a> +<span class="sourceLineNo">091</span> * <ul><a name="line.91"></a> +<span class="sourceLineNo">092</span> * <li>for each region in the table verify which are available in the snapshot and which are not<a name="line.92"></a> +<span class="sourceLineNo">093</span> * <ul><a name="line.93"></a> +<span class="sourceLineNo">094</span> * <li>if the region is not present in the snapshot, remove it.<a name="line.94"></a> +<span class="sourceLineNo">095</span> * <li>if the region is present in the snapshot<a name="line.95"></a> +<span class="sourceLineNo">096</span> * <ul><a name="line.96"></a> +<span class="sourceLineNo">097</span> * <li>for each file in the table region verify which are available in the snapshot<a name="line.97"></a> +<span class="sourceLineNo">098</span> * <ul><a name="line.98"></a> +<span class="sourceLineNo">099</span> * <li>if the hfile is not
<TRUNCATED>