dendi-wang/thingsboard
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactoring

Browse Source
This commit is contained in:
Yevhenii 2025-09-16 18:19:45 +03:00
parent d6d9413082
commit caec9699d4
2 changed files with 10 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
application/src/main/java/org/thingsboard/server/service/edge/rpc/processor/ai/AiModelEdgeProcessor.java
View File

@ -63,12 +63,7 @@ public class AiModelEdgeProcessor extends BaseAiModelProcessor implements AiMode
return handleUnsupportedMsgType(aiModelUpdateMsg.getMsgType());
}
} catch (DataValidationException e) {
if (e.getMessage().contains("limit reached")) {
log.warn("[{}] Number of allowed aiModel violated {}", tenantId, aiModelUpdateMsg, e);
return Futures.immediateFuture(null);
} else {
return Futures.immediateFailedFuture(e);
}
return Futures.immediateFailedFuture(e);
} finally {
edgeSynchronizationManager.getEdgeId().remove();
}

22
dao/src/main/java/org/thingsboard/server/dao/ai/AiModelServiceImpl.java
View File

@ -37,6 +37,7 @@ import org.thingsboard.server.dao.sql.JpaExecutorService;
import java.util.Optional;
import java.util.Set;
import java.util.UUID;
import static org.thingsboard.server.dao.service.Validator.validatePageLink;
@ -125,11 +126,7 @@ class AiModelServiceImpl extends CachedVersionedEntityService<AiModelCacheKey, A
@Override
@Transactional
public boolean deleteByTenantIdAndId(TenantId tenantId, AiModelId modelId) {
AiModel aiModel = aiModelDao.findById(tenantId, modelId.getId());
if (aiModel == null) {
return true;
}
return deleteByTenantIdAndIdInternal(tenantId, aiModel);
return deleteByTenantIdAndIdInternal(tenantId, modelId.getId());
}
@Override
@ -146,17 +143,16 @@ class AiModelServiceImpl extends CachedVersionedEntityService<AiModelCacheKey, A
@Override
@Transactional
public void deleteEntity(TenantId tenantId, EntityId id, boolean force) {
AiModel aiModel = aiModelDao.findById(tenantId, id.getId());
if (aiModel == null) {
return;
}
deleteByTenantIdAndIdInternal(tenantId, aiModel);
deleteByTenantIdAndIdInternal(tenantId, id.getId());
}
private boolean deleteByTenantIdAndIdInternal(TenantId tenantId, AiModel aiModel) {
boolean deleted = aiModelDao.deleteByTenantIdAndId(tenantId, aiModel.getId());
private boolean deleteByTenantIdAndIdInternal(TenantId tenantId, UUID aiModelId) {
AiModel aiModel = aiModelDao.findById(tenantId, aiModelId);
if (aiModel == null) {
return false;
}
boolean deleted = aiModelDao.deleteByTenantIdAndId(tenantId, aiModel.getId());
if (deleted) {
publishEvictEvent(new AiModelCacheEvictEvent.Deleted(AiModelCacheKey.of(tenantId, aiModel.getId())));
eventPublisher.publishEvent(DeleteEntityEvent.builder().tenantId(tenantId).entityId(aiModel.getId()).entity(aiModel).build());