Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 1 | package org.onlab.onos.store.service.impl; |
| 2 | |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 3 | import static com.google.common.base.Preconditions.checkNotNull; |
| 4 | |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 5 | import java.util.List; |
Madan Jampani | f5d263b | 2014-11-13 10:04:40 -0800 | [diff] [blame^] | 6 | import java.util.Set; |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 7 | import java.util.concurrent.CompletableFuture; |
| 8 | import java.util.concurrent.ExecutionException; |
| 9 | |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 10 | import net.kuujo.copycat.Copycat; |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 11 | |
Madan Jampani | 12390c1 | 2014-11-12 00:35:56 -0800 | [diff] [blame] | 12 | import org.onlab.onos.store.service.BatchReadRequest; |
| 13 | import org.onlab.onos.store.service.BatchWriteRequest; |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 14 | import org.onlab.onos.store.service.DatabaseException; |
Madan Jampani | 12390c1 | 2014-11-12 00:35:56 -0800 | [diff] [blame] | 15 | import org.onlab.onos.store.service.ReadResult; |
| 16 | import org.onlab.onos.store.service.WriteResult; |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 17 | |
Madan Jampani | 686fa18 | 2014-11-04 23:16:27 -0800 | [diff] [blame] | 18 | /** |
| 19 | * Client for interacting with the Copycat Raft cluster. |
| 20 | */ |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 21 | public class DatabaseClient { |
| 22 | |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 23 | private final Copycat copycat; |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 24 | |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 25 | public DatabaseClient(Copycat copycat) { |
| 26 | this.copycat = checkNotNull(copycat); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 27 | } |
| 28 | |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 29 | public boolean createTable(String tableName) { |
| 30 | |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 31 | CompletableFuture<Boolean> future = copycat.submit("createTable", tableName); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 32 | try { |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 33 | return future.get(); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 34 | } catch (InterruptedException | ExecutionException e) { |
| 35 | throw new DatabaseException(e); |
| 36 | } |
| 37 | } |
| 38 | |
Madan Jampani | def2c65 | 2014-11-12 13:50:10 -0800 | [diff] [blame] | 39 | public boolean createTable(String tableName, int ttlMillis) { |
| 40 | |
Madan Jampani | f5d263b | 2014-11-13 10:04:40 -0800 | [diff] [blame^] | 41 | CompletableFuture<Boolean> future = copycat.submit("createTableWithExpiration", tableName); |
Madan Jampani | def2c65 | 2014-11-12 13:50:10 -0800 | [diff] [blame] | 42 | try { |
| 43 | return future.get(); |
| 44 | } catch (InterruptedException | ExecutionException e) { |
| 45 | throw new DatabaseException(e); |
| 46 | } |
| 47 | } |
| 48 | |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 49 | public void dropTable(String tableName) { |
| 50 | |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 51 | CompletableFuture<Void> future = copycat.submit("dropTable", tableName); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 52 | try { |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 53 | future.get(); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 54 | } catch (InterruptedException | ExecutionException e) { |
| 55 | throw new DatabaseException(e); |
| 56 | } |
| 57 | } |
| 58 | |
| 59 | public void dropAllTables() { |
| 60 | |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 61 | CompletableFuture<Void> future = copycat.submit("dropAllTables"); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 62 | try { |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 63 | future.get(); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 64 | } catch (InterruptedException | ExecutionException e) { |
| 65 | throw new DatabaseException(e); |
| 66 | } |
| 67 | } |
| 68 | |
Madan Jampani | f5d263b | 2014-11-13 10:04:40 -0800 | [diff] [blame^] | 69 | public Set<String> listTables() { |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 70 | |
Madan Jampani | f5d263b | 2014-11-13 10:04:40 -0800 | [diff] [blame^] | 71 | CompletableFuture<Set<String>> future = copycat.submit("listTables"); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 72 | try { |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 73 | return future.get(); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 74 | } catch (InterruptedException | ExecutionException e) { |
| 75 | throw new DatabaseException(e); |
| 76 | } |
| 77 | } |
| 78 | |
Madan Jampani | 12390c1 | 2014-11-12 00:35:56 -0800 | [diff] [blame] | 79 | public List<ReadResult> batchRead(BatchReadRequest batchRequest) { |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 80 | |
Madan Jampani | 12390c1 | 2014-11-12 00:35:56 -0800 | [diff] [blame] | 81 | CompletableFuture<List<ReadResult>> future = copycat.submit("read", batchRequest); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 82 | try { |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 83 | return future.get(); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 84 | } catch (InterruptedException | ExecutionException e) { |
| 85 | throw new DatabaseException(e); |
| 86 | } |
| 87 | } |
| 88 | |
Madan Jampani | 12390c1 | 2014-11-12 00:35:56 -0800 | [diff] [blame] | 89 | public List<WriteResult> batchWrite(BatchWriteRequest batchRequest) { |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 90 | |
Madan Jampani | 12390c1 | 2014-11-12 00:35:56 -0800 | [diff] [blame] | 91 | CompletableFuture<List<WriteResult>> future = copycat.submit("write", batchRequest); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 92 | try { |
Yuta HIGUCHI | f846844 | 2014-11-11 10:09:20 -0800 | [diff] [blame] | 93 | return future.get(); |
Madan Jampani | 08822c4 | 2014-11-04 17:17:46 -0800 | [diff] [blame] | 94 | } catch (InterruptedException | ExecutionException e) { |
| 95 | throw new DatabaseException(e); |
| 96 | } |
| 97 | } |
| 98 | } |