Andrew Shvayka
9 years ago
29 changed files with 752 additions and 36 deletions
@ -0,0 +1,123 @@ |
|||
/** |
|||
* 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.system; |
|||
|
|||
import com.fasterxml.jackson.core.type.TypeReference; |
|||
import com.google.common.collect.ImmutableMap; |
|||
import org.junit.Before; |
|||
import org.junit.Test; |
|||
import org.thingsboard.server.common.data.Device; |
|||
import org.thingsboard.server.common.data.security.DeviceCredentials; |
|||
import org.thingsboard.server.controller.AbstractControllerTest; |
|||
|
|||
import java.util.List; |
|||
import java.util.UUID; |
|||
import java.util.concurrent.TimeUnit; |
|||
|
|||
import static org.junit.Assert.assertEquals; |
|||
|
|||
public class BaseDeviceOfflineTest extends AbstractControllerTest { |
|||
|
|||
private Device deviceA; |
|||
private Device deviceB; |
|||
private DeviceCredentials credA; |
|||
private DeviceCredentials credB; |
|||
|
|||
@Before |
|||
public void before() throws Exception { |
|||
loginTenantAdmin(); |
|||
deviceA = createDevice("DevA", "VMS"); |
|||
credA = getCredentials(deviceA.getUuidId()); |
|||
deviceB = createDevice("DevB", "SOLAR"); |
|||
credB = getCredentials(deviceB.getUuidId()); |
|||
} |
|||
|
|||
@Test |
|||
public void offlineDevicesCanBeFoundByLastConnectField() throws Exception { |
|||
makeDeviceContact(credA); |
|||
Thread.sleep(1000); |
|||
makeDeviceContact(credB); |
|||
Thread.sleep(100); |
|||
List<Device> devices = doGetTyped("/api/device/offline?contactType=CONNECT&threshold=700", new TypeReference<List<Device>>() { |
|||
}); |
|||
|
|||
assertEquals(devices.toString(),1, devices.size()); |
|||
assertEquals("DevA", devices.get(0).getName()); |
|||
} |
|||
|
|||
@Test |
|||
public void offlineDevicesCanBeFoundByLastUpdateField() throws Exception { |
|||
makeDeviceUpdate(credA); |
|||
Thread.sleep(1000); |
|||
makeDeviceUpdate(credB); |
|||
makeDeviceContact(credA); |
|||
Thread.sleep(100); |
|||
List<Device> devices = doGetTyped("/api/device/offline?contactType=UPLOAD&threshold=700", new TypeReference<List<Device>>() { |
|||
}); |
|||
|
|||
assertEquals(devices.toString(),1, devices.size()); |
|||
assertEquals("DevA", devices.get(0).getName()); |
|||
} |
|||
|
|||
@Test |
|||
public void onlineDevicesCanBeFoundByLastConnectField() throws Exception { |
|||
makeDeviceContact(credB); |
|||
Thread.sleep(1000); |
|||
makeDeviceContact(credA); |
|||
Thread.sleep(100); |
|||
List<Device> devices = doGetTyped("/api/device/online?contactType=CONNECT&threshold=700", new TypeReference<List<Device>>() { |
|||
}); |
|||
|
|||
assertEquals(devices.toString(),1, devices.size()); |
|||
assertEquals("DevA", devices.get(0).getName()); |
|||
} |
|||
|
|||
@Test |
|||
public void onlineDevicesCanBeFoundByLastUpdateField() throws Exception { |
|||
makeDeviceUpdate(credB); |
|||
Thread.sleep(1000); |
|||
makeDeviceUpdate(credA); |
|||
makeDeviceContact(credB); |
|||
Thread.sleep(100); |
|||
List<Device> devices = doGetTyped("/api/device/online?contactType=UPLOAD&threshold=700", new TypeReference<List<Device>>() { |
|||
}); |
|||
|
|||
assertEquals(devices.toString(),1, devices.size()); |
|||
assertEquals("DevA", devices.get(0).getName()); |
|||
} |
|||
|
|||
private Device createDevice(String name, String type) throws Exception { |
|||
Device device = new Device(); |
|||
device.setName(name); |
|||
device.setType(type); |
|||
long currentTime = System.currentTimeMillis() - TimeUnit.SECONDS.toMillis(10); |
|||
device.setLastConnectTs(currentTime); |
|||
device.setLastUpdateTs(currentTime); |
|||
return doPost("/api/device", device, Device.class); |
|||
} |
|||
|
|||
private DeviceCredentials getCredentials(UUID deviceId) throws Exception { |
|||
return doGet("/api/device/" + deviceId.toString() + "/credentials", DeviceCredentials.class); |
|||
} |
|||
|
|||
private void makeDeviceUpdate(DeviceCredentials credentials) throws Exception { |
|||
doPost("/api/v1/" + credentials.getCredentialsId() + "/attributes", ImmutableMap.of("keyA", "valueA"), new String[]{}); |
|||
} |
|||
|
|||
private void makeDeviceContact(DeviceCredentials credentials) throws Exception { |
|||
doGet("/api/v1/" + credentials.getCredentialsId() + "/attributes?clientKeys=keyA,keyB,keyC"); |
|||
} |
|||
} |
|||
@ -0,0 +1,23 @@ |
|||
/** |
|||
* 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.system.nosql; |
|||
|
|||
import org.thingsboard.server.dao.service.DaoNoSqlTest; |
|||
import org.thingsboard.server.system.BaseDeviceOfflineTest; |
|||
|
|||
@DaoNoSqlTest |
|||
public class DeviceOfflineNoSqlTest extends BaseDeviceOfflineTest { |
|||
} |
|||
@ -0,0 +1,23 @@ |
|||
/** |
|||
* 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.system.sql; |
|||
|
|||
import org.thingsboard.server.dao.service.DaoSqlTest; |
|||
import org.thingsboard.server.system.BaseDeviceOfflineTest; |
|||
|
|||
@DaoSqlTest |
|||
public class DeviceOfflineSqlTest extends BaseDeviceOfflineTest { |
|||
} |
|||
@ -0,0 +1,40 @@ |
|||
/** |
|||
* 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.common.data.device; |
|||
|
|||
import lombok.AllArgsConstructor; |
|||
import lombok.Data; |
|||
import lombok.ToString; |
|||
|
|||
@Data |
|||
@AllArgsConstructor |
|||
@ToString |
|||
public class DeviceStatusQuery { |
|||
|
|||
private Status status; |
|||
private ContactType contactType; |
|||
private long threshold; |
|||
|
|||
|
|||
public enum Status { |
|||
ONLINE, OFFLINE |
|||
} |
|||
|
|||
public enum ContactType { |
|||
CONNECT, UPLOAD |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,34 @@ |
|||
/** |
|||
* 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.dao.device; |
|||
|
|||
import com.google.common.util.concurrent.ListenableFuture; |
|||
import org.thingsboard.server.common.data.Device; |
|||
import org.thingsboard.server.common.data.device.DeviceStatusQuery; |
|||
|
|||
import java.util.List; |
|||
import java.util.UUID; |
|||
|
|||
public interface DeviceOfflineService { |
|||
|
|||
void online(Device device, boolean isUpdate); |
|||
|
|||
void offline(Device device); |
|||
|
|||
ListenableFuture<List<Device>> findOfflineDevices(UUID tenantId, DeviceStatusQuery.ContactType contactType, long offlineThreshold); |
|||
|
|||
ListenableFuture<List<Device>> findOnlineDevices(UUID tenantId, DeviceStatusQuery.ContactType contactType, long offlineThreshold); |
|||
} |
|||
@ -0,0 +1,62 @@ |
|||
/** |
|||
* 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.dao.device; |
|||
|
|||
import com.google.common.util.concurrent.ListenableFuture; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.stereotype.Service; |
|||
import org.thingsboard.server.common.data.Device; |
|||
import org.thingsboard.server.common.data.device.DeviceStatusQuery; |
|||
|
|||
import java.util.List; |
|||
import java.util.UUID; |
|||
|
|||
import static org.thingsboard.server.common.data.device.DeviceStatusQuery.Status.OFFLINE; |
|||
import static org.thingsboard.server.common.data.device.DeviceStatusQuery.Status.ONLINE; |
|||
|
|||
@Service |
|||
public class DeviceOfflineServiceImpl implements DeviceOfflineService { |
|||
|
|||
@Autowired |
|||
private DeviceDao deviceDao; |
|||
|
|||
@Override |
|||
public void online(Device device, boolean isUpdate) { |
|||
long current = System.currentTimeMillis(); |
|||
device.setLastConnectTs(current); |
|||
if(isUpdate) { |
|||
device.setLastUpdateTs(current); |
|||
} |
|||
deviceDao.saveDeviceStatus(device); |
|||
} |
|||
|
|||
@Override |
|||
public void offline(Device device) { |
|||
online(device, false); |
|||
} |
|||
|
|||
@Override |
|||
public ListenableFuture<List<Device>> findOfflineDevices(UUID tenantId, DeviceStatusQuery.ContactType contactType, long offlineThreshold) { |
|||
DeviceStatusQuery statusQuery = new DeviceStatusQuery(OFFLINE, contactType, offlineThreshold); |
|||
return deviceDao.findDevicesByTenantIdAndStatus(tenantId, statusQuery); |
|||
} |
|||
|
|||
@Override |
|||
public ListenableFuture<List<Device>> findOnlineDevices(UUID tenantId, DeviceStatusQuery.ContactType contactType, long offlineThreshold) { |
|||
DeviceStatusQuery statusQuery = new DeviceStatusQuery(ONLINE, contactType, offlineThreshold); |
|||
return deviceDao.findDevicesByTenantIdAndStatus(tenantId, statusQuery); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue