@Override
public long getNumRepos(String username) {
try {
THttpClient transport = new THttpClient("http://datahub.csail.mit.edu/service");
TBinaryProtocol protocol = new TBinaryProtocol(transport);
DataHub.Client client = new DataHub.Client(protocol);
ConnectionParams params = new ConnectionParams();
params.setApp_id(DatahubDataModel.getKibitzAppName());
params.setApp_token(DatahubDataModel.getKibitzAppId());
params.setRepo_base(username);
Connection connection = client.open_connection(params);
ResultSet res =
client.execute_sql(
connection,
"select count(distinct schemaname) from pg_tables where tableowner = '"
+ username
+ "';",
null);
HashMap<String, Integer> colToIndex = DatahubDataModel.getFieldNames(res);
for (Tuple t : res.getTuples()) {
List<ByteBuffer> cells = t.getCells();
return Long.parseLong(new String(cells.get(colToIndex.get("count")).array()));
}
} catch (DBException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (TException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return 0;
}
private void assertThatCountDownLatchEquals(
final CountDownLatch countDownLatch, final long expectedCountDownValue) {
assertThat(
Long.valueOf(countDownLatch.getCount()), equalTo(Long.valueOf(expectedCountDownValue)));
}