Andrew Shvayka
2 years ago
committed by
GitHub
GitHub
18 changed files with 209 additions and 265 deletions
@ -0,0 +1,54 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.transport.coap.callback; |
||||
|
|
||||
|
import org.eclipse.californium.core.coap.Response; |
||||
|
import org.eclipse.californium.core.server.resources.CoapExchange; |
||||
|
import org.thingsboard.server.common.transport.TransportServiceCallback; |
||||
|
|
||||
|
public class CoapResponseCallback implements TransportServiceCallback<Void> { |
||||
|
|
||||
|
protected final CoapExchange exchange; |
||||
|
protected final Response onSuccessResponse; |
||||
|
protected final Response onFailureResponse; |
||||
|
|
||||
|
public CoapResponseCallback(CoapExchange exchange, Response onSuccessResponse, Response onFailureResponse) { |
||||
|
this.exchange = exchange; |
||||
|
this.onSuccessResponse = onSuccessResponse; |
||||
|
this.onFailureResponse = onFailureResponse; |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* @param msg |
||||
|
*/ |
||||
|
@Override |
||||
|
public void onSuccess(Void msg) { |
||||
|
this.onSuccessResponse.setConfirmable(isConRequest()); |
||||
|
exchange.respond(this.onSuccessResponse); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* @param e |
||||
|
*/ |
||||
|
@Override |
||||
|
public void onError(Throwable e) { |
||||
|
exchange.respond(onFailureResponse); |
||||
|
} |
||||
|
|
||||
|
protected boolean isConRequest() { |
||||
|
return exchange.advanced().getRequest().isConfirmable(); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,81 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.msa; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.node.ObjectNode; |
||||
|
import org.eclipse.californium.core.CoapClient; |
||||
|
import org.eclipse.californium.core.coap.MediaTypeRegistry; |
||||
|
import org.eclipse.californium.core.config.CoapConfig; |
||||
|
import org.eclipse.californium.elements.config.Configuration; |
||||
|
import org.eclipse.californium.elements.config.Configuration.ModuleDefinitionsProvider; |
||||
|
import org.eclipse.californium.elements.config.IntegerDefinition; |
||||
|
import org.eclipse.californium.elements.config.TcpConfig; |
||||
|
import org.thingsboard.common.util.JacksonUtil; |
||||
|
import org.thingsboard.server.common.msg.session.FeatureType; |
||||
|
|
||||
|
public abstract class AbstractCoapClientTest extends AbstractContainerTest{ |
||||
|
|
||||
|
private static final String COAP_BASE_URL = "coap://localhost:5683/api/v1/"; |
||||
|
private static final long CLIENT_REQUEST_TIMEOUT = 60000L; |
||||
|
|
||||
|
|
||||
|
private static final String COAP_CLIENT_TEST = "COAP_CLIENT_TEST."; |
||||
|
private static final IntegerDefinition COAP_PORT_DEF = CoapConfig.COAP_PORT; |
||||
|
|
||||
|
private static final ModuleDefinitionsProvider MODULE_DEFINITIONS_PROVIDER = new ModuleDefinitionsProvider() { |
||||
|
|
||||
|
@Override |
||||
|
public String getModule() { |
||||
|
return COAP_CLIENT_TEST; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void applyDefinitions(Configuration config) { |
||||
|
TcpConfig.register(); |
||||
|
config.set(COAP_PORT_DEF, 5683); |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
protected CoapClient client; |
||||
|
|
||||
|
protected byte[] createCoapClientAndPublish(String deviceName) throws Exception { |
||||
|
String provisionRequestMsg = createTestProvisionMessage(deviceName); |
||||
|
Configuration.addDefaultModule(MODULE_DEFINITIONS_PROVIDER); |
||||
|
String featureTokenUrl = COAP_BASE_URL + FeatureType.PROVISION.name().toLowerCase(); |
||||
|
client = new CoapClient(featureTokenUrl); |
||||
|
return client.setTimeout(CLIENT_REQUEST_TIMEOUT) |
||||
|
.post(provisionRequestMsg.getBytes(), MediaTypeRegistry.APPLICATION_JSON) |
||||
|
.getPayload(); |
||||
|
} |
||||
|
|
||||
|
protected void disconnect() { |
||||
|
if (client != null) { |
||||
|
client.shutdown(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private String createTestProvisionMessage(String deviceName) { |
||||
|
ObjectNode provisionRequest = JacksonUtil.newObjectNode(); |
||||
|
provisionRequest.put("provisionDeviceKey", TEST_PROVISION_DEVICE_KEY); |
||||
|
provisionRequest.put("provisionDeviceSecret", TEST_PROVISION_DEVICE_SECRET); |
||||
|
if (deviceName != null) { |
||||
|
provisionRequest.put("deviceName", deviceName); |
||||
|
} |
||||
|
return provisionRequest.toString(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
|
||||
@ -1,122 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 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.msa; |
|
||||
|
|
||||
import org.eclipse.californium.core.CoapClient; |
|
||||
import org.eclipse.californium.core.CoapHandler; |
|
||||
import org.eclipse.californium.core.CoapObserveRelation; |
|
||||
import org.eclipse.californium.core.CoapResponse; |
|
||||
import org.eclipse.californium.core.coap.CoAP; |
|
||||
import org.eclipse.californium.core.coap.MediaTypeRegistry; |
|
||||
import org.eclipse.californium.core.coap.Request; |
|
||||
import org.eclipse.californium.elements.exception.ConnectorException; |
|
||||
import org.thingsboard.server.common.msg.session.FeatureType; |
|
||||
|
|
||||
import java.io.IOException; |
|
||||
|
|
||||
public class TestCoapClient { |
|
||||
|
|
||||
private static final String COAP_BASE_URL = "coap://localhost:5683/api/v1/"; |
|
||||
private static final long CLIENT_REQUEST_TIMEOUT = 60000L; |
|
||||
|
|
||||
private final CoapClient client; |
|
||||
|
|
||||
public TestCoapClient(){ |
|
||||
this.client = createClient(); |
|
||||
} |
|
||||
|
|
||||
public TestCoapClient(String accessToken, FeatureType featureType) { |
|
||||
this.client = createClient(getFeatureTokenUrl(accessToken, featureType)); |
|
||||
} |
|
||||
|
|
||||
public TestCoapClient(String featureTokenUrl) { |
|
||||
this.client = createClient(featureTokenUrl); |
|
||||
} |
|
||||
|
|
||||
public void connectToCoap(String accessToken) { |
|
||||
setURI(accessToken, null); |
|
||||
} |
|
||||
|
|
||||
public void connectToCoap(String accessToken, FeatureType featureType) { |
|
||||
setURI(accessToken, featureType); |
|
||||
} |
|
||||
|
|
||||
public void disconnect() { |
|
||||
if (client != null) { |
|
||||
client.shutdown(); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
public CoapResponse postMethod(String requestBody) throws ConnectorException, IOException { |
|
||||
return this.postMethod(requestBody.getBytes()); |
|
||||
} |
|
||||
|
|
||||
public CoapResponse postMethod(byte[] requestBodyBytes) throws ConnectorException, IOException { |
|
||||
return client.setTimeout(CLIENT_REQUEST_TIMEOUT).post(requestBodyBytes, MediaTypeRegistry.APPLICATION_JSON); |
|
||||
} |
|
||||
|
|
||||
public void postMethod(CoapHandler handler, String payload, int format) { |
|
||||
client.post(handler, payload, format); |
|
||||
} |
|
||||
|
|
||||
public void postMethod(CoapHandler handler, byte[] payload, int format) { |
|
||||
client.post(handler, payload, format); |
|
||||
} |
|
||||
|
|
||||
public CoapResponse getMethod() throws ConnectorException, IOException { |
|
||||
return client.setTimeout(CLIENT_REQUEST_TIMEOUT).get(); |
|
||||
} |
|
||||
|
|
||||
public CoapObserveRelation getObserveRelation(TestCoapClientCallback callback){ |
|
||||
Request request = Request.newGet().setObserve(); |
|
||||
request.setType(CoAP.Type.CON); |
|
||||
return client.observe(request, callback); |
|
||||
} |
|
||||
|
|
||||
public void setURI(String featureTokenUrl) { |
|
||||
if (client == null) { |
|
||||
throw new RuntimeException("Failed to connect! CoapClient is not initialized!"); |
|
||||
} |
|
||||
client.setURI(featureTokenUrl); |
|
||||
} |
|
||||
|
|
||||
public void setURI(String accessToken, FeatureType featureType) { |
|
||||
if (featureType == null){ |
|
||||
featureType = FeatureType.ATTRIBUTES; |
|
||||
} |
|
||||
setURI(getFeatureTokenUrl(accessToken, featureType)); |
|
||||
} |
|
||||
|
|
||||
private CoapClient createClient() { |
|
||||
return new CoapClient(); |
|
||||
} |
|
||||
|
|
||||
private CoapClient createClient(String featureTokenUrl) { |
|
||||
return new CoapClient(featureTokenUrl); |
|
||||
} |
|
||||
|
|
||||
public static String getFeatureTokenUrl(FeatureType featureType) { |
|
||||
return COAP_BASE_URL + featureType.name().toLowerCase(); |
|
||||
} |
|
||||
|
|
||||
public static String getFeatureTokenUrl(String token, FeatureType featureType) { |
|
||||
return COAP_BASE_URL + token + "/" + featureType.name().toLowerCase(); |
|
||||
} |
|
||||
|
|
||||
public static String getFeatureTokenUrl(String token, FeatureType featureType, int requestId) { |
|
||||
return COAP_BASE_URL + token + "/" + featureType.name().toLowerCase() + "/" + requestId; |
|
||||
} |
|
||||
} |
|
||||
@ -1,68 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 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.msa; |
|
||||
|
|
||||
import lombok.Data; |
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.eclipse.californium.core.CoapHandler; |
|
||||
import org.eclipse.californium.core.CoapResponse; |
|
||||
import org.eclipse.californium.core.coap.CoAP; |
|
||||
|
|
||||
import java.util.concurrent.CountDownLatch; |
|
||||
|
|
||||
@Slf4j |
|
||||
@Data |
|
||||
public class TestCoapClientCallback implements CoapHandler { |
|
||||
|
|
||||
protected final CountDownLatch latch; |
|
||||
protected Integer observe; |
|
||||
protected byte[] payloadBytes; |
|
||||
protected CoAP.ResponseCode responseCode; |
|
||||
|
|
||||
public TestCoapClientCallback() { |
|
||||
this.latch = new CountDownLatch(1); |
|
||||
} |
|
||||
|
|
||||
public TestCoapClientCallback(int subscribeCount) { |
|
||||
this.latch = new CountDownLatch(subscribeCount); |
|
||||
} |
|
||||
|
|
||||
public Integer getObserve() { |
|
||||
return observe; |
|
||||
} |
|
||||
|
|
||||
public byte[] getPayloadBytes() { |
|
||||
return payloadBytes; |
|
||||
} |
|
||||
|
|
||||
public CoAP.ResponseCode getResponseCode() { |
|
||||
return responseCode; |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void onLoad(CoapResponse response) { |
|
||||
observe = response.getOptions().getObserve(); |
|
||||
payloadBytes = response.getPayload(); |
|
||||
responseCode = response.getCode(); |
|
||||
latch.countDown(); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void onError() { |
|
||||
log.warn("Command Response Ack Error, No connect"); |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
Loading…
Reference in new issue