利用thrift API访问Cassandra 第二天

忘记说Cassandra的版本信息了，apache-cassandra-0.7.2 今天写几个基本的操作，插入，删除，查询 import java.io.UnsupportedEncodingException; import java.nio.ByteBuffer; import java.util.ArrayList; import java.util.List; import org.apache.cassandra.thrift.CfDef; import org.apache.cassandra.thrift.Column; import org.apache.cassandra.thrift.ColumnOrSuperColumn; import org.apache.cassandra.thrift.ColumnParent; import org.apache.cassandra.thrift.ColumnPath; import org.apache.cassandra.thrift.ConsistencyLevel; import org.apache.cassandra.thrift.InvalidRequestException; import org.apache.cassandra.thrift.KeyRange; import org.apache.cassandra.thrift.KeySlice; import org.apache.cassandra.thrift.KsDef; import org.apache.cassandra.thrift.NotFoundException; import org.apache.cassandra.thrift.SlicePredicate; import org.apache.cassandra.thrift.SliceRange; import org.apache.cassandra.thrift.TimedOutException; import org.apache.cassandra.thrift.UnavailableException; import org.apache.cassandra.thrift.Cassandra.Client; import org.apache.cassandra.utils.ByteBufferUtil; import org.apache.thrift.TException; import org.apache.thrift.protocol.TBinaryProtocol; import org.apache.thrift.transport.TFramedTransport; import org.apache.thrift.transport.TSocket; /* * Cassandra操作サンプル * @author ryu * @date 2011-03-07 */ public class JCassandraClient { public static void main(String[] args) throws InvalidRequestException, NotFoundException, UnavailableException, TimedOutException, TException, UnsupportedEncodingException { // TSocket socket = new TSocket("192.168.1.113", 9160); TSocket socket = new TSocket("localhost", 9160); TBinaryProtocol binaryProtocol = new TBinaryProtocol(new TFramedTransport(socket)); Client client = new Client(binaryProtocol); socket.open(); // ※データの最小単位はcolumnです。columnの構造は[name,value,timestamp]です。 // 指定しているrow keyにcolumnを追加する insertColumnSample(client); // 指定しているcolumnを取得する getOneColumnSample(client); // 指定しているcolumnを削除する deleteColumnSample(client); // column family内にすべてのcolumnを取得する //getRowsColumnSample(client); // 一行のcolumnを取得する //getOneRowColumnSample(client); // データベース切断 socket.close(); } /** * 登録データ * * @param client */ private static void insertColumnSample(Client client){ try{ // Keyspace String keyspace = "Ryu"; // Row Key String key = "1000000001"; // Column Family String columnFamily = "T_ERPBASE_USER"; // 指定Keyspace client.set_keyspace(keyspace); ByteBuffer keyBuffer = ByteBuffer.wrap(key.getBytes("UTF-8")); ColumnParent parent = new ColumnParent(columnFamily).setSuper_column((ByteBuffer) null); // 新規しているColumn:[name,value,timestamp] Column newCol = new Column(ByteBufferUtil.bytes("USER_NAME"),ByteBufferUtil.bytes("テスト太郎"), System.currentTimeMillis()); // ①新規または更新 client.insert(keyBuffer, parent, newCol,ConsistencyLevel.ONE); } catch(UnsupportedEncodingException e){ e.printStackTrace(); } catch(TException e){ e.printStackTrace(); } catch(InvalidRequestException e){ e.printStackTrace(); } catch(TimedOutException e){ e.printStackTrace(); } catch(UnavailableException e){ e.printStackTrace(); } } /** * 指定しているcolumnを取得する * * @param client */ private static void getOneColumnSample(Client client){ try{ // Keyspace String keyspace = "Ryu"; // Row Key String key = "1000000001"; // Column Family String columnFamily = "T_ERPBASE_USER"; ByteBuffer keyBuffer = ByteBuffer.wrap(key.getBytes("UTF-8")); // 指定Keyspace client.set_keyspace(keyspace); // ②指定しているcolumnデータを取得 // 検索対象column;検索条件name = USER_NAME ColumnPath col = new ColumnPath(columnFamily).setSuper_column((ByteBuffer) null).setColumn("USER_NAME".getBytes()); Column column = client.get(keyBuffer, col, ConsistencyLevel.ONE).column; System.out.println("read row " + new String(keyBuffer.array(), "UTF-8") + " " + new String(column.getName(), "UTF-8") + ":" + new String(column.getValue(), "UTF-8") + ":" + column.timestamp); } catch(UnsupportedEncodingException e){ e.printStackTrace(); } catch(TException e){ e.printStackTrace(); } catch(InvalidRequestException e){ e.printStackTrace(); } catch(TimedOutException e){ e.printStackTrace(); } catch(UnavailableException e){ e.printStackTrace(); } catch(NotFoundException e){ e.printStackTrace(); } } /** * 指定しているcolumnを削除する(物理削除) * * @param client */ private static void deleteColumnSample(Client client){ try{ // Keyspace String keyspace = "Ryu"; // Row Key String key = "1000000001"; // Column Family String columnFamily = "T_ERPBASE_USER"; ByteBuffer keyBuffer = ByteBuffer.wrap(key.getBytes("UTF-8")); // 指定Keyspace client.set_keyspace(keyspace); ColumnPath col = new ColumnPath(columnFamily).setSuper_column((ByteBuffer) null).setColumn("USER_NAME".getBytes()); // ③削除 client.remove(keyBuffer, col, System.currentTimeMillis(), ConsistencyLevel.ONE); } catch(UnsupportedEncodingException e){ e.printStackTrace(); } catch(TException e){ e.printStackTrace(); } catch(InvalidRequestException e){ e.printStackTrace(); } catch(TimedOutException e){ e.printStackTrace(); } catch(UnavailableException e){ e.printStackTrace(); } } /** * column familyの内にすべてのcolumnを * * @param client */ private static void getRowsColumnSample(Client client){ try{ // Keyspace String keyspace = "Ryu"; // Row Key String key = "rowKey_1000000001"; // Column Family String columnFamily = "T_ERPBASE_USER"; ByteBuffer keyBuffer = ByteBuffer.wrap(key.getBytes("UTF-8")); ColumnParent parent = new ColumnParent(columnFamily).setSuper_column((ByteBuffer) null); // 指定Keyspace client.set_keyspace(keyspace); // 複数行データを取得 KsDef ksDef = client.describe_keyspace(keyspace); if(ksDef != null){ // (列設定) SlicePredicate predicate = new SlicePredicate(); // 指定しているcolumn名により抽出 // predicate.addToColumn_names(ByteBuffer.wrap("USER_ID".getBytes("UTF-8"))); // predicate.addToColumn_names(ByteBuffer.wrap("USER_PASS".getBytes("UTF-8"))); // predicate.addToColumn_names(ByteBuffer.wrap("USER_NAME".getBytes("UTF-8"))); // 指定しているcolumn範囲により抽出 SliceRange sliceRange = new SliceRange(); // すべて抽出 // nameにより開始箇所未限定 sliceRange.setStart(new byte[0]); // nameにより終了箇所未限定 sliceRange.setFinish(new byte[0]); // データ件数(column列数) //sliceRange.setCount(3); // 順次優先で　データ件数設定により、データを取得。true:降順；false:昇順 sliceRange.setReversed(false); predicate.setSlice_range(sliceRange); List cf_defs = ksDef.getCf_defs(); for(int i = 0; i < cf_defs.size(); i ++){ CfDef cfDef = cf_defs.get(i); if(cfDef.getName().equals(columnFamily)){ // すべてのキーに検索(行設定) KeyRange keyRange = new KeyRange(); keyRange.setStart_key(ByteBuffer.wrap("".getBytes("UTF-8"))); keyRange.setEnd_key(ByteBuffer.wrap("".getBytes("UTF-8"))); int length = client.get_count(keyBuffer ,parent, predicate, ConsistencyLevel.ONE); System.out.println("length==========================================>" + length); List lstRow = client.get_range_slices(parent, predicate, keyRange, ConsistencyLevel.ONE); System.out.println("size==========================================>" + lstRow.size()); for (KeySlice slice : lstRow) { String rowKey = new String(slice.getKey(),"UTF-8"); List cols = slice.getColumns(); for (ColumnOrSuperColumn cs : cols) { Column col = cs.getColumn(); System.out.println("read row " + rowKey + " " + new String(col.getName(), "UTF-8") + ":" + new String(col.getValue(), "UTF-8")); } } } } } } catch(UnsupportedEncodingException e){ e.printStackTrace(); } catch(TException e){ e.printStackTrace(); } catch(InvalidRequestException e){ e.printStackTrace(); } catch(TimedOutException e){ e.printStackTrace(); } catch(UnavailableException e){ e.printStackTrace(); } catch(NotFoundException e){ e.printStackTrace(); } } /** * 指定しているrow keyにより、一行のcolumnを取得する * * @param client */ private static void getOneRowColumnSample(Client client){ try{ // Keyspace String keyspace = "Ryu"; // Row Key String key = "rowKey_1000000001"; // Column Family String columnFamily = "T_ERPBASE_USER"; ByteBuffer keyBuffer = ByteBuffer.wrap(key.getBytes("UTF-8")); ColumnParent parent = new ColumnParent(columnFamily).setSuper_column((ByteBuffer) null); // 指定Keyspace client.set_keyspace(keyspace); // read entire row SlicePredicate predicate = new SlicePredicate(); SliceRange sliceRange = new SliceRange(); // 抽出したいcolumn範囲 // BIRTHDAY ～ NICK_NAME sliceRange.setStart("BIRTHDAY".getBytes()); sliceRange.setFinish("NICK_NAME".getBytes()); // データ件数 //sliceRange.setCount(3); // 順次優先で　データ件数設定により、データを取得。true:降順；false:昇順 sliceRange.setReversed(false); predicate.setSlice_range(sliceRange); // 指定しているcolumn名 List lstCols = new ArrayList(); ByteBuffer column_names1 = ByteBuffer.wrap("BIRTHDAY".getBytes("UTF-8")); lstCols.add(column_names1); ByteBuffer column_names2 = ByteBuffer.wrap("NICK_NAME".getBytes("UTF-8")); lstCols.add(column_names2); // 抽出したいcolumn名リスト //predicate.setColumn_names(lstCols); // 一行データを取得 List results = client.get_slice(keyBuffer, parent, predicate, ConsistencyLevel.ONE); for (ColumnOrSuperColumn result : results) { Column columnShow = result.column; System.out.println(new String(columnShow.getName(), "UTF8") + " -> " + new String(columnShow.getValue(), "UTF8")); } } catch(UnsupportedEncodingException e){ e.printStackTrace(); } catch(TException e){ e.printStackTrace(); } catch(InvalidRequestException e){ e.printStackTrace(); } catch(TimedOutException e){ e.printStackTrace(); } catch(UnavailableException e){ e.printStackTrace(); } } }