Andrew Shvayka
4 years ago
committed by
GitHub
GitHub
59 changed files with 302 additions and 181 deletions
@ -1,43 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2022 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.entity; |
|
||||
|
|
||||
import org.apache.commons.text.CaseUtils; |
|
||||
import org.springframework.beans.factory.BeanFactory; |
|
||||
import org.springframework.stereotype.Service; |
|
||||
import org.thingsboard.server.common.data.EntityType; |
|
||||
|
|
||||
@Service |
|
||||
public class EntityServiceBeanFactory { |
|
||||
|
|
||||
private static final String SERVICE_SUFFIX = "DaoService"; |
|
||||
|
|
||||
private final BeanFactory beanFactory; |
|
||||
|
|
||||
public EntityServiceBeanFactory(BeanFactory beanFactory) { |
|
||||
this.beanFactory = beanFactory; |
|
||||
} |
|
||||
|
|
||||
public TbEntityService getServiceByEntityType(EntityType entityType) { |
|
||||
String beanName = EntityType.RULE_NODE.equals(entityType) ? getBeanName(EntityType.RULE_CHAIN) : getBeanName(entityType); |
|
||||
return beanFactory.getBean(beanName, TbEntityService.class); |
|
||||
} |
|
||||
|
|
||||
private String getBeanName(EntityType entityType) { |
|
||||
return CaseUtils.toCamelCase(entityType.name(), true, '_') + SERVICE_SUFFIX; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -0,0 +1,24 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2022 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.entity; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
|
||||
|
public interface EntityServiceRegistry { |
||||
|
|
||||
|
EntityDaoService getServiceByEntityType(EntityType entityType); |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,53 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2022 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.entity; |
||||
|
|
||||
|
import org.springframework.context.ApplicationContext; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
|
||||
|
import javax.annotation.PostConstruct; |
||||
|
import java.util.HashMap; |
||||
|
import java.util.Map; |
||||
|
|
||||
|
@Service |
||||
|
public class DefaultEntityServiceRegistry implements EntityServiceRegistry { |
||||
|
|
||||
|
private final ApplicationContext applicationContext; |
||||
|
private final Map<EntityType, EntityDaoService> entityDaoServicesMap; |
||||
|
|
||||
|
public DefaultEntityServiceRegistry(ApplicationContext applicationContext) { |
||||
|
this.applicationContext = applicationContext; |
||||
|
this.entityDaoServicesMap = new HashMap<>(); |
||||
|
} |
||||
|
|
||||
|
@PostConstruct |
||||
|
public void init() { |
||||
|
applicationContext.getBeansOfType(EntityDaoService.class).values().forEach(entityDaoService -> { |
||||
|
EntityType entityType = entityDaoService.getEntityType(); |
||||
|
entityDaoServicesMap.put(entityType, entityDaoService); |
||||
|
if (EntityType.RULE_CHAIN.equals(entityType)) { |
||||
|
entityDaoServicesMap.put(EntityType.RULE_NODE, entityDaoService); |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public EntityDaoService getServiceByEntityType(EntityType entityType) { |
||||
|
return entityDaoServicesMap.get(entityType); |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue