added restclient to tools
This commit is contained in:
parent
1a66a840b8
commit
4dfe05865d
@ -35,6 +35,30 @@
|
|||||||
<main.dir>${basedir}/..</main.dir>
|
<main.dir>${basedir}/..</main.dir>
|
||||||
</properties>
|
</properties>
|
||||||
|
|
||||||
|
<dependencies>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.thingsboard.common</groupId>
|
||||||
|
<artifactId>data</artifactId>
|
||||||
|
<version>${project.version}</version>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.springframework.boot</groupId>
|
||||||
|
<artifactId>spring-boot-starter-web</artifactId>
|
||||||
|
<version>${spring-boot.version}</version>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.eclipse.paho</groupId>
|
||||||
|
<artifactId>org.eclipse.paho.client.mqttv3</artifactId>
|
||||||
|
<version>${paho.client.version}</version>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.projectlombok</groupId>
|
||||||
|
<artifactId>lombok</artifactId>
|
||||||
|
<version>${lombok.version}</version>
|
||||||
|
<scope>provided</scope>
|
||||||
|
</dependency>
|
||||||
|
</dependencies>
|
||||||
|
|
||||||
<build>
|
<build>
|
||||||
<plugins>
|
<plugins>
|
||||||
<plugin>
|
<plugin>
|
||||||
|
|||||||
@ -0,0 +1,76 @@
|
|||||||
|
/**
|
||||||
|
* 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.client.tools;
|
||||||
|
|
||||||
|
import com.fasterxml.jackson.databind.JsonNode;
|
||||||
|
import lombok.RequiredArgsConstructor;
|
||||||
|
import org.springframework.http.HttpRequest;
|
||||||
|
import org.springframework.http.ResponseEntity;
|
||||||
|
import org.springframework.http.client.ClientHttpRequestExecution;
|
||||||
|
import org.springframework.http.client.ClientHttpRequestInterceptor;
|
||||||
|
import org.springframework.http.client.ClientHttpResponse;
|
||||||
|
import org.springframework.http.client.support.HttpRequestWrapper;
|
||||||
|
import org.springframework.web.client.RestTemplate;
|
||||||
|
import org.thingsboard.server.common.data.Device;
|
||||||
|
import org.thingsboard.server.common.data.id.DeviceId;
|
||||||
|
import org.thingsboard.server.common.data.security.DeviceCredentials;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.Collections;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Andrew Shvayka
|
||||||
|
*/
|
||||||
|
@RequiredArgsConstructor
|
||||||
|
public class RestClient implements ClientHttpRequestInterceptor {
|
||||||
|
private static final String JWT_TOKEN_HEADER_PARAM = "X-Authorization";
|
||||||
|
private final RestTemplate restTemplate = new RestTemplate();
|
||||||
|
private String token;
|
||||||
|
private final String baseURL;
|
||||||
|
|
||||||
|
public void login(String username, String password) {
|
||||||
|
Map<String, String> loginRequest = new HashMap<>();
|
||||||
|
loginRequest.put("username", username);
|
||||||
|
loginRequest.put("password", password);
|
||||||
|
ResponseEntity<JsonNode> tokenInfo = restTemplate.postForEntity(baseURL + "/api/auth/login", loginRequest, JsonNode.class);
|
||||||
|
this.token = tokenInfo.getBody().get("token").asText();
|
||||||
|
restTemplate.setInterceptors(Collections.singletonList(this));
|
||||||
|
}
|
||||||
|
|
||||||
|
public Device createDevice(String name) {
|
||||||
|
Device device = new Device();
|
||||||
|
device.setName(name);
|
||||||
|
return restTemplate.postForEntity(baseURL + "/api/device", device, Device.class).getBody();
|
||||||
|
}
|
||||||
|
|
||||||
|
public DeviceCredentials getCredentials(DeviceId id) {
|
||||||
|
return restTemplate.getForEntity(baseURL + "/api/device/" + id.getId().toString() + "/credentials", DeviceCredentials.class).getBody();
|
||||||
|
}
|
||||||
|
|
||||||
|
public RestTemplate getRestTemplate() {
|
||||||
|
return restTemplate;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ClientHttpResponse intercept(HttpRequest request, byte[] bytes, ClientHttpRequestExecution execution) throws IOException {
|
||||||
|
HttpRequest wrapper = new HttpRequestWrapper(request);
|
||||||
|
wrapper.getHeaders().set(JWT_TOKEN_HEADER_PARAM, "Bearer " + token);
|
||||||
|
return execution.execute(wrapper, bytes);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user