Andrew Shvayka
9 years ago
7 changed files with 202 additions and 45 deletions
@ -0,0 +1,129 @@ |
|||
/** |
|||
* Copyright © 2016-2017 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 com.google.common.util.concurrent.ListenableFuture; |
|||
import org.apache.commons.lang3.StringUtils; |
|||
import org.springframework.http.HttpStatus; |
|||
import org.springframework.security.access.prepost.PreAuthorize; |
|||
import org.springframework.web.bind.annotation.*; |
|||
import org.thingsboard.server.common.data.Customer; |
|||
import org.thingsboard.server.common.data.Event; |
|||
import org.thingsboard.server.common.data.alarm.Alarm; |
|||
import org.thingsboard.server.common.data.alarm.AlarmId; |
|||
import org.thingsboard.server.common.data.alarm.AlarmQuery; |
|||
import org.thingsboard.server.common.data.alarm.AlarmStatus; |
|||
import org.thingsboard.server.common.data.asset.Asset; |
|||
import org.thingsboard.server.common.data.id.*; |
|||
import org.thingsboard.server.common.data.page.TextPageData; |
|||
import org.thingsboard.server.common.data.page.TextPageLink; |
|||
import org.thingsboard.server.common.data.page.TimePageData; |
|||
import org.thingsboard.server.common.data.page.TimePageLink; |
|||
import org.thingsboard.server.dao.asset.AssetSearchQuery; |
|||
import org.thingsboard.server.dao.exception.IncorrectParameterException; |
|||
import org.thingsboard.server.dao.model.ModelConstants; |
|||
import org.thingsboard.server.exception.ThingsboardErrorCode; |
|||
import org.thingsboard.server.exception.ThingsboardException; |
|||
import org.thingsboard.server.service.security.model.SecurityUser; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.List; |
|||
import java.util.stream.Collectors; |
|||
|
|||
@RestController |
|||
@RequestMapping("/api") |
|||
public class AlarmController extends BaseController { |
|||
|
|||
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
|||
@RequestMapping(value = "/alarm/{alarmId}", method = RequestMethod.GET) |
|||
@ResponseBody |
|||
public Alarm getAlarmById(@PathVariable("alarmId") String strAlarmId) throws ThingsboardException { |
|||
checkParameter("alarmId", strAlarmId); |
|||
try { |
|||
AlarmId alarmId = new AlarmId(toUUID(strAlarmId)); |
|||
return checkAlarmId(alarmId); |
|||
} catch (Exception e) { |
|||
throw handleException(e); |
|||
} |
|||
} |
|||
|
|||
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
|||
@RequestMapping(value = "/alarm", method = RequestMethod.POST) |
|||
@ResponseBody |
|||
public Alarm saveAlarm(@RequestBody Alarm alarm) throws ThingsboardException { |
|||
try { |
|||
alarm.setTenantId(getCurrentUser().getTenantId()); |
|||
return checkNotNull(alarmService.createOrUpdateAlarm(alarm)); |
|||
} catch (Exception e) { |
|||
throw handleException(e); |
|||
} |
|||
} |
|||
|
|||
@PreAuthorize("hasAuthority('TENANT_ADMIN')") |
|||
@RequestMapping(value = "/alarm/{alarmId}/ack", method = RequestMethod.POST) |
|||
@ResponseStatus(value = HttpStatus.OK) |
|||
public void ackAlarm(@PathVariable("alarmId") String strAlarmId) throws ThingsboardException { |
|||
checkParameter("alarmId", strAlarmId); |
|||
try { |
|||
AlarmId alarmId = new AlarmId(toUUID(strAlarmId)); |
|||
checkAlarmId(alarmId); |
|||
alarmService.ackAlarm(alarmId, System.currentTimeMillis()).get(); |
|||
} catch (Exception e) { |
|||
throw handleException(e); |
|||
} |
|||
} |
|||
|
|||
@PreAuthorize("hasAuthority('TENANT_ADMIN')") |
|||
@RequestMapping(value = "/alarm/{alarmId}/clear", method = RequestMethod.POST) |
|||
@ResponseStatus(value = HttpStatus.OK) |
|||
public void clearAlarm(@PathVariable("alarmId") String strAlarmId) throws ThingsboardException { |
|||
checkParameter("alarmId", strAlarmId); |
|||
try { |
|||
AlarmId alarmId = new AlarmId(toUUID(strAlarmId)); |
|||
checkAlarmId(alarmId); |
|||
alarmService.clearAlarm(alarmId, System.currentTimeMillis()).get(); |
|||
} catch (Exception e) { |
|||
throw handleException(e); |
|||
} |
|||
} |
|||
|
|||
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
|||
@RequestMapping(value = "/alarm/{entityType}/{entityId}", method = RequestMethod.GET) |
|||
@ResponseBody |
|||
public TimePageData<Alarm> getAlarms( |
|||
@PathVariable("entityType") String strEntityType, |
|||
@PathVariable("entityId") String strEntityId, |
|||
@RequestParam(required = false) String status, |
|||
@RequestParam int limit, |
|||
@RequestParam(required = false) Long startTime, |
|||
@RequestParam(required = false) Long endTime, |
|||
@RequestParam(required = false, defaultValue = "false") boolean ascOrder, |
|||
@RequestParam(required = false) String offset |
|||
) throws ThingsboardException { |
|||
checkParameter("EntityId", strEntityId); |
|||
checkParameter("EntityType", strEntityType); |
|||
EntityId entityId = EntityIdFactory.getByTypeAndId(strEntityType, strEntityId); |
|||
AlarmStatus alarmStatus = StringUtils.isEmpty(status) ? null : AlarmStatus.valueOf(status); |
|||
checkEntityId(entityId); |
|||
try { |
|||
TimePageLink pageLink = createPageLink(limit, startTime, endTime, ascOrder, offset); |
|||
return checkNotNull(alarmService.findAlarms(new AlarmQuery(entityId, pageLink, alarmStatus)).get()); |
|||
} catch (Exception e) { |
|||
throw handleException(e); |
|||
} |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue