1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.hbase.constraint;
19
20 import java.io.IOException;
21 import java.util.ArrayList;
22 import java.util.List;
23
24 import org.apache.commons.logging.Log;
25 import org.apache.commons.logging.LogFactory;
26 import org.apache.hadoop.hbase.CoprocessorEnvironment;
27 import org.apache.hadoop.hbase.HTableDescriptor;
28 import org.apache.hadoop.hbase.client.Put;
29 import org.apache.hadoop.hbase.coprocessor.BaseRegionObserver;
30 import org.apache.hadoop.hbase.coprocessor.ObserverContext;
31 import org.apache.hadoop.hbase.coprocessor.RegionCoprocessorEnvironment;
32 import org.apache.hadoop.hbase.regionserver.wal.WALEdit;
33
34
35
36
37
38
39
40 public class ConstraintProcessor extends BaseRegionObserver {
41
42 private static final Log LOG = LogFactory.getLog(ConstraintProcessor.class);
43
44 private final ClassLoader classloader;
45
46 private List<? extends Constraint> constraints = new ArrayList<Constraint>();
47
48
49
50
51
52
53 public ConstraintProcessor() {
54 classloader = this.getClass().getClassLoader();
55 }
56
57 @Override
58 public void start(CoprocessorEnvironment environment) {
59
60 if (!(environment instanceof RegionCoprocessorEnvironment)) {
61 throw new IllegalArgumentException(
62 "Constraints only act on regions - started in an environment that was not a region");
63 }
64 RegionCoprocessorEnvironment env = (RegionCoprocessorEnvironment) environment;
65 HTableDescriptor desc = env.getRegion().getTableDesc();
66
67 try {
68 this.constraints = Constraints.getConstraints(desc, classloader);
69 } catch (IOException e) {
70 throw new IllegalArgumentException(e);
71 }
72
73 if (LOG.isInfoEnabled()) {
74 LOG.info("Finished loading " + constraints.size()
75 + " user Constraints on table: " + new String(desc.getName()));
76 }
77
78 }
79
80 @Override
81 public void prePut(ObserverContext<RegionCoprocessorEnvironment> e, Put put,
82 WALEdit edit, boolean writeToWAL) throws IOException {
83
84 for (Constraint c : constraints) {
85 c.check(put);
86 }
87
88 }
89 }