Skip to content

Commit

Permalink
mvn spotless
Browse files Browse the repository at this point in the history
  • Loading branch information
@SzyWilliam
SzyWilliam committed Nov 28, 2024
1 parent 90cce52 commit 99627e0
Show file tree
Hide file tree
Showing 2 changed files with 128 additions and 107 deletions.
231 changes: 126 additions & 105 deletions iotdb-client/session/src/main/java/org/apache/iotdb/session/SessionConnection.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -284,11 +284,12 @@ protected IClientRPCService.Iface getClient() {

protected void setTimeZone(String zoneId)
throws StatementExecutionException, IoTDBConnectionException {
doOperation(() -> {
TSSetTimeZoneReq req = new TSSetTimeZoneReq(sessionId, zoneId);
RpcUtils.verifySuccess(client.setTimeZone(req));
return null;
});
doOperation(
() -> {
TSSetTimeZoneReq req = new TSSetTimeZoneReq(sessionId, zoneId);
RpcUtils.verifySuccess(client.setTimeZone(req));
return null;
});
setTimeZoneOfSession(zoneId);
}

Expand All @@ -305,45 +306,50 @@ protected String getTimeZone() {

protected void setStorageGroup(String storageGroup)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
RpcUtils.verifySuccess(client.setStorageGroup(sessionId, storageGroup));
return null;
});
doOperation(
() -> {
RpcUtils.verifySuccess(client.setStorageGroup(sessionId, storageGroup));
return null;
});
}

protected void deleteStorageGroups(List<String> storageGroups)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
RpcUtils.verifySuccess(client.deleteStorageGroups(sessionId, storageGroups));
return null;
});
doOperation(
() -> {
RpcUtils.verifySuccess(client.deleteStorageGroups(sessionId, storageGroups));
return null;
});
}

protected void createTimescreateTimeserieseries(TSCreateTimeseriesReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.createTimeseries(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.createTimeseries(request));
return null;
});
}

protected void createAlignedTimeseries(TSCreateAlignedTimeseriesReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.createAlignedTimeseries(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.createAlignedTimeseries(request));
return null;
});
}

protected void createMultiTimeseries(TSCreateMultiTimeseriesReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.createMultiTimeseries(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.createMultiTimeseries(request));
return null;
});
}

protected boolean checkTimeseriesExists(String path, long timeout)
Expand Down Expand Up @@ -1256,56 +1262,62 @@ private TSStatus deleteDataInternal(TSDeleteDataReq request) throws TException {

protected void testInsertRecord(TSInsertStringRecordReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertStringRecord(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertStringRecord(request));
return null;
});
}

protected void testInsertRecord(TSInsertRecordReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertRecord(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertRecord(request));
return null;
});
}

public void testInsertRecords(TSInsertStringRecordsReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertStringRecords(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertStringRecords(request));
return null;
});
}

public void testInsertRecords(TSInsertRecordsReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertRecords(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertRecords(request));
return null;
});
}

protected void testInsertTablet(TSInsertTabletReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertTablet(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertTablet(request));
return null;
});
}

protected void testInsertTablets(TSInsertTabletsReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertTablets(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.testInsertTablets(request));
return null;
});
}

@SuppressWarnings({
Expand Down Expand Up @@ -1358,89 +1370,98 @@ private boolean reconnect() {

protected void createSchemaTemplate(TSCreateSchemaTemplateReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.createSchemaTemplate(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.createSchemaTemplate(request));
return null;
});
}

protected void appendSchemaTemplate(TSAppendSchemaTemplateReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.appendSchemaTemplate(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.appendSchemaTemplate(request));
return null;
});
}

protected void pruneSchemaTemplate(TSPruneSchemaTemplateReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.pruneSchemaTemplate(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.pruneSchemaTemplate(request));
return null;
});
}

protected TSQueryTemplateResp querySchemaTemplate(TSQueryTemplateReq req)
throws StatementExecutionException, IoTDBConnectionException {
return doOperation(() -> {
req.setSessionId(sessionId);
TSQueryTemplateResp execResp = client.querySchemaTemplate(req);
RpcUtils.verifySuccess(execResp.getStatus());
return execResp;
});
return doOperation(
() -> {
req.setSessionId(sessionId);
TSQueryTemplateResp execResp = client.querySchemaTemplate(req);
RpcUtils.verifySuccess(execResp.getStatus());
return execResp;
});
}

protected void setSchemaTemplate(TSSetSchemaTemplateReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.setSchemaTemplate(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.setSchemaTemplate(request));
return null;
});
}

protected void unsetSchemaTemplate(TSUnsetSchemaTemplateReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.unsetSchemaTemplate(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.unsetSchemaTemplate(request));
return null;
});
}

protected void dropSchemaTemplate(TSDropSchemaTemplateReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.dropSchemaTemplate(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.dropSchemaTemplate(request));
return null;
});
}

protected void createTimeseriesUsingSchemaTemplate(
TCreateTimeseriesUsingSchemaTemplateReq request)
throws IoTDBConnectionException, StatementExecutionException {
doOperation(() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.createTimeseriesUsingSchemaTemplate(request));
return null;
});
doOperation(
() -> {
request.setSessionId(sessionId);
RpcUtils.verifySuccess(client.createTimeseriesUsingSchemaTemplate(request));
return null;
});
}

protected TSBackupConfigurationResp getBackupConfiguration()
throws IoTDBConnectionException, StatementExecutionException {
return doOperation(() -> {
TSBackupConfigurationResp execResp = client.getBackupConfiguration();
RpcUtils.verifySuccess(execResp.getStatus());
return execResp;
});
return doOperation(
() -> {
TSBackupConfigurationResp execResp = client.getBackupConfiguration();
RpcUtils.verifySuccess(execResp.getStatus());
return execResp;
});
}

private <RETURN> RETURN doOperation(CheckedSupplier<RETURN, TException> supplier)
throws IoTDBConnectionException, StatementExecutionException {
throws IoTDBConnectionException, StatementExecutionException {
RETURN ret;
try {
ret = supplier.get();
Expand Down
4 changes: 2 additions & 2 deletions iotdb-client/session/src/main/java/org/apache/iotdb/session/util/CheckedSupplier.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,8 +25,8 @@
@FunctionalInterface
public interface CheckedSupplier<OUTPUT, THROWABLE extends Throwable> {
/**
* The same as {@link java.util.function.Supplier#get()}
* except that this method is declared with a throws-clause.
* The same as {@link java.util.function.Supplier#get()} except that this method is declared with
* a throws-clause.
*/
OUTPUT get() throws THROWABLE, StatementExecutionException;
}

0 comments on commit 99627e0

Please sign in to comment.