Browse Source

refactoring due to comments (removed serviceId duplications in ToUsageStatsServiceMsgPack)

pull/12191/head
YevhenBondarenko 2 years ago
parent
commit
933d1447d2
  1. 7
      application/src/main/java/org/thingsboard/server/service/apiusage/DefaultTbApiUsageStateService.java
  2. 2
      common/proto/src/main/proto/queue.proto
  3. 3
      common/queue/src/main/java/org/thingsboard/server/queue/usagestats/DefaultTbApiUsageReportClient.java

7
application/src/main/java/org/thingsboard/server/service/apiusage/DefaultTbApiUsageStateService.java

@ -157,11 +157,12 @@ public class DefaultTbApiUsageStateService extends AbstractPartitionBasedService
@Override
public void process(TbProtoQueueMsg<ToUsageStatsServiceMsgPack> msgPack, TbCallback callback) {
msgPack.getValue().getMsgsList().forEach(this::process);
String serviceId = msgPack.getValue().getServiceId();
msgPack.getValue().getMsgsList().forEach(msg -> process(msg, serviceId));
callback.onSuccess();
}
private void process(ToUsageStatsServiceMsg statsMsg) {
private void process(ToUsageStatsServiceMsg statsMsg, String serviceId) {
TenantId tenantId = TenantId.fromUUID(new UUID(statsMsg.getTenantIdMSB(), statsMsg.getTenantIdLSB()));
EntityId ownerId;
if (statsMsg.getCustomerIdMSB() != 0 && statsMsg.getCustomerIdLSB() != 0) {
@ -170,7 +171,7 @@ public class DefaultTbApiUsageStateService extends AbstractPartitionBasedService
ownerId = tenantId;
}
processEntityUsageStats(tenantId, ownerId, statsMsg.getValuesList(), statsMsg.getServiceId());
processEntityUsageStats(tenantId, ownerId, statsMsg.getValuesList(), serviceId);
}
private void processEntityUsageStats(TenantId tenantId, EntityId ownerId, List<UsageStatsKVProto> values, String serviceId) {

2
common/proto/src/main/proto/queue.proto

@ -1615,11 +1615,11 @@ message ToUsageStatsServiceMsg {
repeated UsageStatsKVProto values = 5;
int64 customerIdMSB = 6;
int64 customerIdLSB = 7;
string serviceId = 8;
}
message ToUsageStatsServiceMsgPack {
repeated ToUsageStatsServiceMsg msgs = 1;
string serviceId = 2;
}
message ToOtaPackageStateServiceMsg {

3
common/queue/src/main/java/org/thingsboard/server/queue/usagestats/DefaultTbApiUsageReportClient.java

@ -115,7 +115,6 @@ public class DefaultTbApiUsageReportClient implements TbApiUsageReportClient {
newStatsMsg.setCustomerIdLSB(customerId.getId().getLeastSignificantBits());
}
newStatsMsg.setServiceId(serviceInfoProvider.getServiceId());
return newStatsMsg;
});
@ -124,7 +123,6 @@ public class DefaultTbApiUsageReportClient implements TbApiUsageReportClient {
.setValue(value);
statsMsg.addValues(statsItem.build());
});
statsForKey.clear();
}
Map<TopicPartitionInfo, List<ToUsageStatsServiceMsg>> reportStatsPerTpi = new HashMap<>();
@ -160,6 +158,7 @@ public class DefaultTbApiUsageReportClient implements TbApiUsageReportClient {
.map(partition ->
ToUsageStatsServiceMsgPack.newBuilder()
.addAllMsgs(partition)
.setServiceId(serviceInfoProvider.getServiceId())
.build())
.toList();
}

Loading…
Cancel
Save