1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.hbase.master.snapshot;
19
20 import java.io.IOException;
21 import java.util.HashSet;
22 import java.util.List;
23 import java.util.Set;
24 import java.util.concurrent.ThreadPoolExecutor;
25
26 import org.apache.commons.logging.Log;
27 import org.apache.commons.logging.LogFactory;
28 import org.apache.hadoop.conf.Configuration;
29 import org.apache.hadoop.hbase.HConstants;
30 import org.apache.hadoop.hbase.classification.InterfaceAudience;
31 import org.apache.hadoop.hbase.classification.InterfaceStability;
32 import org.apache.hadoop.hbase.HRegionInfo;
33 import org.apache.hadoop.hbase.ServerName;
34 import org.apache.hadoop.hbase.errorhandling.ForeignException;
35 import org.apache.hadoop.hbase.master.MasterServices;
36 import org.apache.hadoop.hbase.protobuf.generated.HBaseProtos.SnapshotDescription;
37 import org.apache.hadoop.hbase.snapshot.ClientSnapshotDescriptionUtils;
38 import org.apache.hadoop.hbase.snapshot.SnapshotManifest;
39 import org.apache.hadoop.hbase.util.FSUtils;
40 import org.apache.hadoop.hbase.util.ModifyRegionUtils;
41 import org.apache.hadoop.hbase.util.Pair;
42 import org.apache.zookeeper.KeeperException;
43
44
45
46
47
48
49 @InterfaceAudience.Private
50 @InterfaceStability.Evolving
51 public class DisabledTableSnapshotHandler extends TakeSnapshotHandler {
52 private static final Log LOG = LogFactory.getLog(DisabledTableSnapshotHandler.class);
53
54
55
56
57
58 public DisabledTableSnapshotHandler(SnapshotDescription snapshot,
59 final MasterServices masterServices) {
60 super(snapshot, masterServices);
61 }
62
63 @Override
64 public DisabledTableSnapshotHandler prepare() throws Exception {
65 return (DisabledTableSnapshotHandler) super.prepare();
66 }
67
68
69
70 @Override
71 public void snapshotRegions(List<Pair<HRegionInfo, ServerName>> regionsAndLocations)
72 throws IOException, KeeperException {
73 try {
74
75
76
77 Set<HRegionInfo> regions = new HashSet<HRegionInfo>();
78 for (Pair<HRegionInfo, ServerName> p : regionsAndLocations) {
79 regions.add(p.getFirst());
80 }
81
82
83 String msg = "Starting to write region info and WALs for regions for offline snapshot:"
84 + ClientSnapshotDescriptionUtils.toString(snapshot);
85 LOG.info(msg);
86 status.setStatus(msg);
87
88 ThreadPoolExecutor exec = SnapshotManifest.createExecutor(conf, "DisabledTableSnapshot");
89 try {
90 ModifyRegionUtils.editRegions(exec, regions, new ModifyRegionUtils.RegionEditTask() {
91 @Override
92 public void editRegion(final HRegionInfo regionInfo) throws IOException {
93 snapshotManifest.addRegion(FSUtils.getTableDir(rootDir, snapshotTable), regionInfo);
94 }
95 });
96 } finally {
97 exec.shutdown();
98 }
99 } catch (Exception e) {
100
101 String reason = "Failed snapshot " + ClientSnapshotDescriptionUtils.toString(snapshot)
102 + " due to exception:" + e.getMessage();
103 ForeignException ee = new ForeignException(reason, e);
104 monitor.receive(ee);
105 status.abort("Snapshot of table: "+ snapshotTable + " failed because " + e.getMessage());
106 } finally {
107 LOG.debug("Marking snapshot" + ClientSnapshotDescriptionUtils.toString(snapshot)
108 + " as finished.");
109 }
110 }
111 }