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 @Override
public void process(TbProtoQueueMsg<ToUsageStatsServiceMsgPack> msgPack, TbCallback callback) { 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(); callback.onSuccess();
} }
private void process(ToUsageStatsServiceMsg statsMsg) { private void process(ToUsageStatsServiceMsg statsMsg, String serviceId) {
TenantId tenantId = TenantId.fromUUID(new UUID(statsMsg.getTenantIdMSB(), statsMsg.getTenantIdLSB())); TenantId tenantId = TenantId.fromUUID(new UUID(statsMsg.getTenantIdMSB(), statsMsg.getTenantIdLSB()));
EntityId ownerId; EntityId ownerId;
if (statsMsg.getCustomerIdMSB() != 0 && statsMsg.getCustomerIdLSB() != 0) { if (statsMsg.getCustomerIdMSB() != 0 && statsMsg.getCustomerIdLSB() != 0) {
@ -170,7 +171,7 @@ public class DefaultTbApiUsageStateService extends AbstractPartitionBasedService
ownerId = tenantId; 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) { 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; repeated UsageStatsKVProto values = 5;
int64 customerIdMSB = 6; int64 customerIdMSB = 6;
int64 customerIdLSB = 7; int64 customerIdLSB = 7;
string serviceId = 8;
} }
message ToUsageStatsServiceMsgPack { message ToUsageStatsServiceMsgPack {
repeated ToUsageStatsServiceMsg msgs = 1; repeated ToUsageStatsServiceMsg msgs = 1;
string serviceId = 2;
} }
message ToOtaPackageStateServiceMsg { 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.setCustomerIdLSB(customerId.getId().getLeastSignificantBits());
} }
newStatsMsg.setServiceId(serviceInfoProvider.getServiceId());
return newStatsMsg; return newStatsMsg;
}); });
@ -124,7 +123,6 @@ public class DefaultTbApiUsageReportClient implements TbApiUsageReportClient {
.setValue(value); .setValue(value);
statsMsg.addValues(statsItem.build()); statsMsg.addValues(statsItem.build());
}); });
statsForKey.clear();
} }
Map<TopicPartitionInfo, List<ToUsageStatsServiceMsg>> reportStatsPerTpi = new HashMap<>(); Map<TopicPartitionInfo, List<ToUsageStatsServiceMsg>> reportStatsPerTpi = new HashMap<>();
@ -160,6 +158,7 @@ public class DefaultTbApiUsageReportClient implements TbApiUsageReportClient {
.map(partition -> .map(partition ->
ToUsageStatsServiceMsgPack.newBuilder() ToUsageStatsServiceMsgPack.newBuilder()
.addAllMsgs(partition) .addAllMsgs(partition)
.setServiceId(serviceInfoProvider.getServiceId())
.build()) .build())
.toList(); .toList();
} }

Loading…
Cancel
Save