Igor Kulikov
9 years ago
4 changed files with 192 additions and 2 deletions
@ -0,0 +1,161 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2018 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.controller; |
||||
|
|
||||
|
import org.springframework.http.HttpStatus; |
||||
|
import org.springframework.security.access.prepost.PreAuthorize; |
||||
|
import org.springframework.web.bind.annotation.*; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.audit.ActionType; |
||||
|
import org.thingsboard.server.common.data.id.PluginId; |
||||
|
import org.thingsboard.server.common.data.id.RuleChainId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.page.TextPageData; |
||||
|
import org.thingsboard.server.common.data.page.TextPageLink; |
||||
|
import org.thingsboard.server.common.data.plugin.ComponentLifecycleEvent; |
||||
|
import org.thingsboard.server.common.data.plugin.PluginMetaData; |
||||
|
import org.thingsboard.server.common.data.rule.RuleChain; |
||||
|
import org.thingsboard.server.common.data.security.Authority; |
||||
|
import org.thingsboard.server.dao.model.ModelConstants; |
||||
|
import org.thingsboard.server.exception.ThingsboardException; |
||||
|
|
||||
|
import java.util.List; |
||||
|
|
||||
|
@RestController |
||||
|
@RequestMapping("/api") |
||||
|
public class RuleChainController extends BaseController { |
||||
|
|
||||
|
public static final String RULE_CHAIN_ID = "ruleChainId"; |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('SYS_ADMIN', 'TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/ruleChain/{ruleChainId}", method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public RuleChain getRuleChainById(@PathVariable(RULE_CHAIN_ID) String strRuleChainId) throws ThingsboardException { |
||||
|
checkParameter(RULE_CHAIN_ID, strRuleChainId); |
||||
|
try { |
||||
|
RuleChainId ruleChainId = new RuleChainId(toUUID(strRuleChainId)); |
||||
|
return checkRuleChain(ruleChainId); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('SYS_ADMIN', 'TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/ruleChain", method = RequestMethod.POST) |
||||
|
@ResponseBody |
||||
|
public RuleChain saveRuleChain(@RequestBody RuleChain ruleChain) throws ThingsboardException { |
||||
|
try { |
||||
|
boolean created = ruleChain.getId() == null; |
||||
|
ruleChain.setTenantId(getCurrentUser().getTenantId()); |
||||
|
RuleChain savedRuleChain = checkNotNull(ruleChainService.saveRuleChain(ruleChain)); |
||||
|
|
||||
|
logEntityAction(savedRuleChain.getId(), savedRuleChain, |
||||
|
null, |
||||
|
created ? ActionType.ADDED : ActionType.UPDATED, null); |
||||
|
|
||||
|
return savedRuleChain; |
||||
|
} catch (Exception e) { |
||||
|
|
||||
|
logEntityAction(emptyId(EntityType.RULE_CHAIN), ruleChain, |
||||
|
null, ruleChain.getId() == null ? ActionType.ADDED : ActionType.UPDATED, e); |
||||
|
|
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('SYS_ADMIN', 'TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/ruleChains", params = {"limit"}, method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public TextPageData<RuleChain> getRuleChains( |
||||
|
@RequestParam int limit, |
||||
|
@RequestParam(required = false) String textSearch, |
||||
|
@RequestParam(required = false) String idOffset, |
||||
|
@RequestParam(required = false) String textOffset) throws ThingsboardException { |
||||
|
try { |
||||
|
TextPageLink pageLink = createPageLink(limit, textSearch, idOffset, textOffset); |
||||
|
if (getCurrentUser().getAuthority() == Authority.SYS_ADMIN) { |
||||
|
return checkNotNull(ruleChainService.findSystemRuleChains(pageLink)); |
||||
|
} else { |
||||
|
TenantId tenantId = getCurrentUser().getTenantId(); |
||||
|
TextPageData<RuleChain> ruleChainsData = checkNotNull(ruleChainService.findAllTenantRuleChainsByTenantIdAndPageLink(tenantId, pageLink)); |
||||
|
List<RuleChain> ruleChains = ruleChainsData.getData(); |
||||
|
ruleChains.stream() |
||||
|
.filter(ruleChain -> ruleChain.getTenantId().getId().equals(ModelConstants.NULL_UUID)) |
||||
|
.forEach(ruleChain -> ruleChain.setConfiguration(null)); |
||||
|
return ruleChainsData; |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAuthority('SYS_ADMIN')") |
||||
|
@RequestMapping(value = "/system/ruleChains", params = {"limit"}, method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public TextPageData<RuleChain> getSystemRuleChains( |
||||
|
@RequestParam int limit, |
||||
|
@RequestParam(required = false) String textSearch, |
||||
|
@RequestParam(required = false) String idOffset, |
||||
|
@RequestParam(required = false) String textOffset) throws ThingsboardException { |
||||
|
try { |
||||
|
TextPageLink pageLink = createPageLink(limit, textSearch, idOffset, textOffset); |
||||
|
return checkNotNull(ruleChainService.findSystemRuleChains(pageLink)); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAuthority('TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/tenant/ruleChains", params = {"limit"}, method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public TextPageData<RuleChain> getTenantRuleChains( |
||||
|
@RequestParam int limit, |
||||
|
@RequestParam(required = false) String textSearch, |
||||
|
@RequestParam(required = false) String idOffset, |
||||
|
@RequestParam(required = false) String textOffset) throws ThingsboardException { |
||||
|
try { |
||||
|
TenantId tenantId = getCurrentUser().getTenantId(); |
||||
|
TextPageLink pageLink = createPageLink(limit, textSearch, idOffset, textOffset); |
||||
|
return checkNotNull(ruleChainService.findTenantRuleChains(tenantId, pageLink)); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('SYS_ADMIN', 'TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/ruleChain/{ruleChainId}", method = RequestMethod.DELETE) |
||||
|
@ResponseStatus(value = HttpStatus.OK) |
||||
|
public void deleteRuleChain(@PathVariable(RULE_CHAIN_ID) String strRuleChainId) throws ThingsboardException { |
||||
|
checkParameter(RULE_CHAIN_ID, strRuleChainId); |
||||
|
try { |
||||
|
RuleChainId ruleChainId = new RuleChainId(toUUID(strRuleChainId)); |
||||
|
RuleChain ruleChain = checkRuleChain(ruleChainId); |
||||
|
ruleChainService.deleteRuleChainById(ruleChainId); |
||||
|
|
||||
|
logEntityAction(ruleChainId, ruleChain, |
||||
|
null, |
||||
|
ActionType.DELETED, null, strRuleChainId); |
||||
|
|
||||
|
} catch (Exception e) { |
||||
|
logEntityAction(emptyId(EntityType.RULE_CHAIN), |
||||
|
null, |
||||
|
null, |
||||
|
ActionType.DELETED, e, strRuleChainId); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue