Hbase Coprocessor(协处理器)的使用

本博客记录初次使用hbase coprocessor的过程。协处理器分两种类型,系统协处理器可以全局导入region server上的所有数据表,表协处理器即是用户可以指定一张表使用协处理器。协处理器框架为了更好支持其行为的灵活性,提供了两个不同方面的插件。一个是观察者(observer),类似于关系数据库的触发器。另一个是终端(endpoint),动态的终端有点像存储过程。
本次主要使用EndPoint完成计数和求和的功能。终端是动态RPC插件的接口,它的实现代码被安装在服务器端,从而能够通过HBase RPC唤醒。客户端类库提供了非常方便的方法来调用这些动态接口,它们可以在任意时候调用一个终端,它们的实现代码会被目标region远程执行,结果会返回到终端。用户可以结合使用这些强大的插件接口,为HBase添加全新的特性。

准备工作

1、开发endpoint需要用到google protobuf,protobuf用于生成RPC框架代码,protpbuf版本需要和hbase对应,版本跨度太大可能导致未知问题,我开始就是踩了这个坑,具体版本可查看hbase安装目录下的lib中protobuf-java-[version].jar。本次使用的hbase版本是1.2.6,对应protobuf是2.5.0,从网上下载protoc-2.5.0-win32.zip,解压后可得到protoc.exe,将protoc.exe配置到环境变量中备用,protobuf的详细使用方法可参考网上其他教程。
protobuf下载链接

2、创建一个maven工程,pom.xml添加如下依赖:

<properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <jdk.version>1.7</jdk.version>
        <hbase.version>1.2.5</hbase.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>com.google.protobuf</groupId>
            <artifactId>protobuf-java</artifactId>
            <version>2.5.0</version>
        </dependency>
        <dependency>
            <groupId>org.apache.hbase</groupId>
            <artifactId>hbase-client</artifactId>
            <version>${hbase.version}</version>
        </dependency>
        <dependency>
            <groupId>org.apache.hbase</groupId>
            <artifactId>hbase-server</artifactId>
            <version>${hbase.version}</version>
        </dependency>
    </dependencies>

创建一个Endpoint的基本流程可以归纳为:
(1)创建一个通信协议:准备一个proto文件,然后使用protoc工具来生成协议类文件。这个文件需要在服务端及客户端存 在。
(2)创建一个Service类,实现具体的业务逻辑
(3)创建表时指定使用这个EndPoint,或者是全局配置。
(4)创建一个Client类,调用这个RPC方法。

(一)创建测试表

HBase表中有一个family命名为0, 一个column命名为c,rowkey为某个id,使用hbase shell创建表并添加测试数据。

create 'test','0'
put 'test','id1','0:c',100
put 'test','id2','0:c',200
put 'test','id3','0:c',300
put 'test','id4','0:c',400
put 'test','id5','0:c',500

接下来我们需要实现数据计数并计算“0:c”列的和。

(二)准备proto文件

新建文件,命名为count_sum.proto,添加如下内容:

syntax = "proto2";
option java_package = "com.hny.hbase.coprocessor";
option java_outer_classname = "CountAndSumProtocol";
option java_generic_services = true;
option java_generate_equals_and_hash = true;
option optimize_for = SPEED;

message CountAndSumRequest {
    required string family = 1;
    required string column = 2;
}

message CountAndSumResponse {
    required int64 count = 1 [default = 0];
    required double sum = 2 [default = 0];
}

service RowCountAndSumService {
  rpc getCountAndSum(CountAndSumRequest)
    returns (CountAndSumResponse);
}

(二)使用protoc生成类文件

windows下使用cmd进入上一步创建的proto文件的目录下,执行如下命令(由于已经将protoc.exe加入了环境变量,所以可以直接执行,如果提示protoc命令不存在可将protoc.exe复制到当前目录下也可以)

protoc --java_out=./ count_sum.proto

命令执行完成后会在当前目录下生产一个名称为CountAndSumProtocol的类,将这个类复制到IDE中,这个类文件有几个地方需要注意:
1、生成了一个CountAndSumRequest 内部类,表示请求信息
2、生成了一个CountAndSumResponse 内部类,表示返回信息
3、生成了一个 RowCountAndSumService 内部类,表示所提供的服务,这个类还有一个内部接口,这个接口定义了 getCountAndSum()这个方法。
我们下面需要做的就是实现这个接口的这个方法,提供真正的服务。

(三)实现真实的服务

在CountAndSumProtocol同目录下创建类CountAndSum,继承CountAndSumProtocol,同时需要实现Coprocessor和CoprocessorService2个接口:

package com.hny.hbase.coprocessor;

import com.google.protobuf.RpcCallback;
import com.google.protobuf.RpcController;
import com.google.protobuf.Service;
import org.apache.hadoop.hbase.Cell;
import org.apache.hadoop.hbase.CellUtil;
import org.apache.hadoop.hbase.Coprocessor;
import org.apache.hadoop.hbase.CoprocessorEnvironment;
import org.apache.hadoop.hbase.client.Scan;
import org.apache.hadoop.hbase.coprocessor.CoprocessorException;
import org.apache.hadoop.hbase.coprocessor.CoprocessorService;
import org.apache.hadoop.hbase.coprocessor.RegionCoprocessorEnvironment;
import org.apache.hadoop.hbase.protobuf.ResponseConverter;
import org.apache.hadoop.hbase.regionserver.InternalScanner;
import org.apache.hadoop.hbase.util.Bytes;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class CountAndSum extends CountAndSumProtocol.RowCountAndSumService implements Coprocessor, CoprocessorService {

    private RegionCoprocessorEnvironment env;

    @Override
    public void getCountAndSum(RpcController controller, CountAndSumProtocol.CountAndSumRequest request, RpcCallback<CountAndSumProtocol.CountAndSumResponse> done) {
        String family = request.getFamily();
        if (null == family || "".equals(family)) {
            throw new NullPointerException("you need specify the family");
        }
        String column = request.getColumn();
        if (null == column || "".equals(column)) {
            throw new NullPointerException("you need specify the column");
        }
        Scan scan = new Scan();
        scan.addColumn(Bytes.toBytes(family), Bytes.toBytes(column));

        CountAndSumProtocol.CountAndSumResponse response = null;
        InternalScanner scanner = null;
        try {
            // 计数
            long count = 0;
            // 求和
            double sum = 0;

            scanner = env.getRegion().getScanner(scan);
            List<Cell> results = new ArrayList<>();
            boolean hasMore;
            // 切记不要用while(){}的方式,这种方式会丢失最后一条数据
            do {
                hasMore = scanner.next(results);
                if (results.isEmpty()) {
                    continue;
                }
                Cell kv = results.get(0);
                double value = 0;
                try {
                    value = Double.parseDouble(Bytes.toString(CellUtil.cloneValue(kv)));
                } catch (Exception e) {
                }
                count++;
                sum += value;
                results.clear();
            } while (hasMore);

            // 生成response
            response = CountAndSumProtocol.CountAndSumResponse.newBuilder().setCount(count).setSum(sum).build();
        } catch (IOException e) {
            e.printStackTrace();
            ResponseConverter.setControllerException(controller, e);
        } finally {
            if (scanner != null) {
                try {
                    scanner.close();
                } catch (IOException ignored) {
                }
            }
        }
        done.run(response);
    }

    @Override
    public void start(CoprocessorEnvironment env) throws IOException {
        if (env instanceof RegionCoprocessorEnvironment) {
            this.env = (RegionCoprocessorEnvironment) env;
        } else {
            throw new CoprocessorException("Must be loaded on a table region!");
        }
    }

    @Override
    public void stop(CoprocessorEnvironment env) throws IOException {
        // do nothing
    }

    @Override
    public Service getService() {
        return this;
    }
}

它需要实现以下4个方法,下面我们逐一讨论一下:
getService():这个方法直接返回自身即可。
start(CoprocessorEnvironment env):这个方法会在coprocessor启动时调用,这里判断了是否在一个region内被使用,而不是master,WAL等环境下被调用。
stop(CoprocessorEnvironment env):这个方法会在coprocessor完成时被调用,可用于关闭资源等,这里为空。
getCountAndSum(…):这是整个类的核心方法,用于实现真正的业务逻辑。关键的步骤有:
(1)根据request创建一个Scanner,然后使用它创建一个 InternalScanner,可以更高效的进行scan
(2)对扫描出来的行进行分析处理,将结果保存在几个变量中。
(3)调用response的各个set()方法,设置返回的结果。
(4)使用 done.run(response); 返回结果到客户端。

(四)部署coprocessor

将上述2个类进行打包,打包时不用包含protobuf和hbase相关的依赖。本示例暂时使用静态部署的方式,将jar复制到每个regionserver节点的hbase/lib目录下,然后修改hbase-site.xml,添加如下属性:

<property>
     <name>hbase.coprocessor.region.classes</name>
     <value>com.hny.hbase.coprocessor.CountAndSum</value>
</property>

重启hbase。
建议在hbase-site.xml中再加入以下配置,防止协处理器出现错误时导致regionServer挂掉。

<property>
      <name>hbase.coprocessor.abortonerror</name>
      <value>false</value>
</property>

(五)编写调用端

客户端的作用是将各个region的结果再次进行合并,客户端需要依赖CountAndSumProtocol类,代码如下:

package com.hny.hbase.coprocessor.client;

import com.hny.hbase.coprocessor.CountAndSumProtocol;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.HBaseConfiguration;
import org.apache.hadoop.hbase.TableName;
import org.apache.hadoop.hbase.client.Connection;
import org.apache.hadoop.hbase.client.ConnectionFactory;
import org.apache.hadoop.hbase.client.Table;
import org.apache.hadoop.hbase.client.coprocessor.Batch;
import org.apache.hadoop.hbase.ipc.BlockingRpcCallback;
import org.apache.hadoop.hbase.util.Bytes;

import java.io.IOException;
import java.util.Map;

public class CountAndSumClient {
    public static class CountAndSumResult {
        public long count;
        public double sum;

    }

    private Connection connection;

    public CountAndSumClient(Connection connection) {
        this.connection = connection;
    }

    public CountAndSumResult call(String tableName, String family, String column, String
            startRow, String endRow) throws Throwable {
        Table table = connection.getTable(TableName.valueOf(Bytes.toBytes(tableName)));
        final CountAndSumProtocol.CountAndSumRequest request = CountAndSumProtocol.CountAndSumRequest
                .newBuilder()
                .setFamily(family)
                .setColumn(column)
                .build();

        byte[] startKey = (null != startRow) ? Bytes.toBytes(startRow) : null;
        byte[] endKey = (null != endRow) ? Bytes.toBytes(endRow) : null;
        // coprocessorService方法的第二、三个参数是定位region的,是不是范围查询,在startKey和endKey之间的region上的数据都会参与计算
        Map<byte[], CountAndSumResult> map = table.coprocessorService(CountAndSumProtocol.RowCountAndSumService.class,
                startKey, endKey, new Batch.Call<CountAndSumProtocol.RowCountAndSumService,
                        CountAndSumResult>() {
                    @Override
                    public CountAndSumResult call(CountAndSumProtocol.RowCountAndSumService service) throws IOException {
                        BlockingRpcCallback<CountAndSumProtocol.CountAndSumResponse> rpcCallback = new BlockingRpcCallback<>();
                        service.getCountAndSum(null, request, rpcCallback);
                        CountAndSumProtocol.CountAndSumResponse response = rpcCallback.get();
                        //直接返回response也行。
                        CountAndSumResult responseInfo = new CountAndSumResult();
                        responseInfo.count = response.getCount();
                        responseInfo.sum = response.getSum();
                        return responseInfo;
                    }
                });

        CountAndSumResult result = new CountAndSumResult();
        for (CountAndSumResult ri : map.values()) {
            result.count += ri.count;
            result.sum += ri.sum;
        }

        return result;
    }
}

测试代码:

public class Test{

    public static void main(String[] args) throws Throwable {
        // 使用该方式需要将hbase-site.xml复制到resources目录下
        Configuration conf = HBaseConfiguration.create();
        // hbase-site.xml不在resources目录下时使用如下方式指定
        // conf.addResource(new Path("/home/hadoop/conf/hbase", "hbase-site.xml"));
        Connection connection = ConnectionFactory.createConnection(conf);

        String tableName = "test";
        CountAndSumClient client = new CountAndSumClient(connection);
        CountAndSumResult result = client.call(tableName, "0", "c", null, null);

        System.out.println("count: " + result.count + ", sum: " + result.sum);
    }
} 

运行测试代码输出如下:

count: 5, sum: 1500.0

注意:部署到集群的jar包包括Service类和protocol类,而运行任务的jar包包括client类与protocol类。

参考文章:https://blog.csdn.net/jediael_lu/article/details/76577072

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章