ViacheslavKlimov
2 years ago
14 changed files with 259 additions and 170 deletions
@ -0,0 +1,26 @@ |
|||
/** |
|||
* Copyright © 2016-2024 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.server.dao.config; |
|||
|
|||
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
|||
|
|||
import java.lang.annotation.Retention; |
|||
import java.lang.annotation.RetentionPolicy; |
|||
|
|||
@Retention(RetentionPolicy.RUNTIME) |
|||
@ConditionalOnProperty(value = "spring.datasource.dedicated.enabled", havingValue = "false", matchIfMissing = true) |
|||
public @interface DefaultDataSource { |
|||
} |
|||
@ -0,0 +1,87 @@ |
|||
/** |
|||
* Copyright © 2016-2024 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.server.dao.sql.audit; |
|||
|
|||
import jakarta.persistence.EntityManager; |
|||
import jakarta.persistence.PersistenceContext; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.beans.factory.annotation.Qualifier; |
|||
import org.springframework.jdbc.core.JdbcTemplate; |
|||
import org.springframework.stereotype.Component; |
|||
import org.springframework.transaction.annotation.Transactional; |
|||
import org.thingsboard.server.common.data.audit.AuditLog; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.dao.config.DedicatedDataSource; |
|||
import org.thingsboard.server.dao.sqlts.insert.sql.DedicatedSqlPartitioningRepository; |
|||
import org.thingsboard.server.dao.util.SqlDao; |
|||
|
|||
import java.util.Collection; |
|||
import java.util.UUID; |
|||
|
|||
import static org.thingsboard.server.dao.config.DedicatedJpaDaoConfig.DEDICATED_JDBC_TEMPLATE; |
|||
import static org.thingsboard.server.dao.config.DedicatedJpaDaoConfig.DEDICATED_PERSISTENCE_UNIT; |
|||
import static org.thingsboard.server.dao.config.DedicatedJpaDaoConfig.DEDICATED_TRANSACTION_MANAGER; |
|||
|
|||
@DedicatedDataSource |
|||
@Component |
|||
@SqlDao |
|||
public class DedicatedJpaAuditLogDao extends JpaAuditLogDao { |
|||
|
|||
@Autowired |
|||
@Qualifier(DEDICATED_JDBC_TEMPLATE) |
|||
private JdbcTemplate jdbcTemplate; |
|||
@PersistenceContext(unitName = DEDICATED_PERSISTENCE_UNIT) |
|||
private EntityManager entityManager; |
|||
|
|||
public DedicatedJpaAuditLogDao(AuditLogRepository auditLogRepository, DedicatedSqlPartitioningRepository partitioningRepository) { |
|||
super(auditLogRepository, partitioningRepository); |
|||
} |
|||
|
|||
@Transactional(transactionManager = DEDICATED_TRANSACTION_MANAGER) |
|||
@Override |
|||
public AuditLog save(TenantId tenantId, AuditLog domain) { |
|||
return super.save(tenantId, domain); |
|||
} |
|||
|
|||
@Transactional(transactionManager = DEDICATED_TRANSACTION_MANAGER) |
|||
@Override |
|||
public AuditLog saveAndFlush(TenantId tenantId, AuditLog domain) { |
|||
return super.saveAndFlush(tenantId, domain); |
|||
} |
|||
|
|||
@Transactional(transactionManager = DEDICATED_TRANSACTION_MANAGER) |
|||
@Override |
|||
public boolean removeById(TenantId tenantId, UUID id) { |
|||
return super.removeById(tenantId, id); |
|||
} |
|||
|
|||
@Transactional(transactionManager = DEDICATED_TRANSACTION_MANAGER) |
|||
@Override |
|||
public void removeAllByIds(Collection<UUID> ids) { |
|||
super.removeAllByIds(ids); |
|||
} |
|||
|
|||
@Override |
|||
protected EntityManager getEntityManager() { |
|||
return entityManager; |
|||
} |
|||
|
|||
@Override |
|||
protected JdbcTemplate getJdbcTemplate() { |
|||
return jdbcTemplate; |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,36 @@ |
|||
/** |
|||
* Copyright © 2016-2024 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.server.dao.sql.event; |
|||
|
|||
import org.springframework.beans.factory.annotation.Qualifier; |
|||
import org.springframework.jdbc.core.JdbcTemplate; |
|||
import org.springframework.stereotype.Repository; |
|||
import org.springframework.transaction.support.TransactionTemplate; |
|||
import org.thingsboard.server.dao.config.DedicatedDataSource; |
|||
|
|||
import static org.thingsboard.server.dao.config.DedicatedJpaDaoConfig.DEDICATED_JDBC_TEMPLATE; |
|||
import static org.thingsboard.server.dao.config.DedicatedJpaDaoConfig.DEDICATED_TRANSACTION_TEMPLATE; |
|||
|
|||
@DedicatedDataSource |
|||
@Repository |
|||
public class DedicatedEventInsertRepository extends EventInsertRepository { |
|||
|
|||
public DedicatedEventInsertRepository(@Qualifier(DEDICATED_JDBC_TEMPLATE) JdbcTemplate jdbcTemplate, |
|||
@Qualifier(DEDICATED_TRANSACTION_TEMPLATE) TransactionTemplate transactionTemplate) { |
|||
super(jdbcTemplate, transactionTemplate); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,45 @@ |
|||
/** |
|||
* Copyright © 2016-2024 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.server.dao.sql.event; |
|||
|
|||
import org.springframework.stereotype.Component; |
|||
import org.thingsboard.server.common.stats.StatsFactory; |
|||
import org.thingsboard.server.dao.config.DedicatedDataSource; |
|||
import org.thingsboard.server.dao.sql.ScheduledLogExecutorComponent; |
|||
import org.thingsboard.server.dao.sqlts.insert.sql.DedicatedSqlPartitioningRepository; |
|||
import org.thingsboard.server.dao.util.SqlDao; |
|||
|
|||
@DedicatedDataSource |
|||
@Component |
|||
@SqlDao |
|||
public class DedicatedJpaEventDao extends JpaBaseEventDao { |
|||
|
|||
public DedicatedJpaEventDao(EventPartitionConfiguration partitionConfiguration, |
|||
DedicatedSqlPartitioningRepository partitioningRepository, |
|||
LifecycleEventRepository lcEventRepository, |
|||
StatisticsEventRepository statsEventRepository, |
|||
ErrorEventRepository errorEventRepository, |
|||
DedicatedEventInsertRepository eventInsertRepository, |
|||
RuleNodeDebugEventRepository ruleNodeDebugEventRepository, |
|||
RuleChainDebugEventRepository ruleChainDebugEventRepository, |
|||
ScheduledLogExecutorComponent logExecutor, |
|||
StatsFactory statsFactory) { |
|||
super(partitionConfiguration, partitioningRepository, lcEventRepository, statsEventRepository, |
|||
errorEventRepository, eventInsertRepository, ruleNodeDebugEventRepository, |
|||
ruleChainDebugEventRepository, logExecutor, statsFactory); |
|||
} |
|||
|
|||
} |
|||
@ -1,47 +0,0 @@ |
|||
/** |
|||
* Copyright © 2016-2024 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.server.dao.sql.event; |
|||
|
|||
import lombok.RequiredArgsConstructor; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.stereotype.Repository; |
|||
import org.thingsboard.server.common.data.event.EventType; |
|||
import org.thingsboard.server.dao.sql.JpaAbstractDaoListeningExecutorService; |
|||
import org.thingsboard.server.dao.sqlts.insert.sql.DedicatedSqlPartitioningRepository; |
|||
|
|||
|
|||
@Slf4j |
|||
@Repository |
|||
@RequiredArgsConstructor |
|||
public class SqlEventCleanupRepository extends JpaAbstractDaoListeningExecutorService implements EventCleanupRepository { |
|||
|
|||
private final EventPartitionConfiguration partitionConfiguration; |
|||
private final DedicatedSqlPartitioningRepository partitioningRepository; |
|||
|
|||
@Override |
|||
public void cleanupEvents(long eventExpTime, boolean debug) { |
|||
for (EventType eventType : EventType.values()) { |
|||
if (eventType.isDebug() == debug) { |
|||
cleanupEvents(eventType, eventExpTime); |
|||
} |
|||
} |
|||
} |
|||
|
|||
private void cleanupEvents(EventType eventType, long eventExpTime) { |
|||
partitioningRepository.dropPartitionsBefore(eventType.getTable(), eventExpTime, partitionConfiguration.getPartitionSizeInMs(eventType)); |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue