package com.dji.sample.component.mqtt.config;
|
|
import com.auth0.jwt.algorithms.Algorithm;
|
import com.dji.sample.common.util.JwtUtil;
|
import com.dji.sample.component.mqtt.model.MqttClientOptions;
|
import com.dji.sample.component.mqtt.model.MqttProtocolEnum;
|
import com.dji.sample.component.mqtt.model.MqttUseEnum;
|
import com.dji.sample.control.model.dto.MqttBrokerDTO;
|
import lombok.Data;
|
import org.eclipse.paho.client.mqttv3.MqttConnectOptions;
|
import org.springframework.boot.context.properties.ConfigurationProperties;
|
import org.springframework.context.annotation.Bean;
|
import org.springframework.context.annotation.Configuration;
|
import org.springframework.integration.mqtt.core.DefaultMqttPahoClientFactory;
|
import org.springframework.integration.mqtt.core.MqttPahoClientFactory;
|
import org.springframework.util.StringUtils;
|
|
import java.util.Map;
|
|
/**
|
*
|
* @author sean.zhou
|
* @date 2021/11/10
|
* @version 0.1
|
*/
|
@Configuration
|
@Data
|
@ConfigurationProperties
|
public class MqttConfiguration {
|
|
private static Map<MqttUseEnum, MqttClientOptions> mqtt;
|
|
public void setMqtt(Map<MqttUseEnum, MqttClientOptions> mqtt) {
|
MqttConfiguration.mqtt = mqtt;
|
}
|
|
/**
|
* Get the configuration options of the basic link of the mqtt client.
|
* @return
|
*/
|
static MqttClientOptions getBasicClientOptions() {
|
if (!mqtt.containsKey(MqttUseEnum.BASIC)) {
|
throw new Error("请先配置基本的mqtt连接参数，否则应用程序无法启动。");
|
}
|
return mqtt.get(MqttUseEnum.BASIC);
|
}
|
|
/**
|
* Get the mqtt address of the basic link.
|
* @return
|
*/
|
public static String getBasicMqttAddress() {
|
return getMqttAddress(getBasicClientOptions());
|
}
|
|
/**
|
* Splice the mqtt address according to the parameters of different clients.
|
* @param options
|
* @return
|
*/
|
private static String getMqttAddress(MqttClientOptions options) {
|
StringBuilder addr = new StringBuilder()
|
.append(options.getProtocol().getProtocolAddr())
|
.append(options.getHost().trim())
|
.append(":")
|
.append(options.getPort());
|
if ((options.getProtocol() == MqttProtocolEnum.WS || options.getProtocol() == MqttProtocolEnum.WSS)
|
&& StringUtils.hasText(options.getPath())) {
|
addr.append(options.getPath());
|
}
|
return addr.toString();
|
}
|
|
/**
|
* Get the connection parameters of the mqtt client of the drc link.
|
* @param clientId
|
* @param username
|
* @param age The validity period of the token. unit: s
|
* @param map Custom data added in token.
|
* @return
|
*/
|
public static MqttBrokerDTO getMqttBrokerWithDrc(String clientId, String username, Long age, Map<String, ?> map) {
|
if (!mqtt.containsKey(MqttUseEnum.DRC)) {
|
throw new RuntimeException("请先在后端配置文件中配置mqtt的drc链接参数。");
|
}
|
Algorithm algorithm = JwtUtil.algorithm;
|
|
String token = JwtUtil.createToken(map, age, algorithm, null, null);
|
|
return MqttBrokerDTO.builder()
|
.address(getMqttAddress(mqtt.get(MqttUseEnum.DRC)))
|
.username("root")
|
.clientId(clientId)
|
.expireTime(System.currentTimeMillis() / 1000 + age)
|
.password("root")
|
.build();
|
}
|
|
|
@Bean
|
public MqttConnectOptions mqttConnectOptions() {
|
MqttClientOptions customizeOptions = getBasicClientOptions();
|
MqttConnectOptions mqttConnectOptions = new MqttConnectOptions();
|
mqttConnectOptions.setServerURIs(new String[]{ getBasicMqttAddress() });
|
mqttConnectOptions.setUserName(customizeOptions.getUsername());
|
mqttConnectOptions.setPassword(StringUtils.hasText(customizeOptions.getPassword()) ?
|
customizeOptions.getPassword().toCharArray() : new char[0]);
|
mqttConnectOptions.setAutomaticReconnect(true);
|
mqttConnectOptions.setKeepAliveInterval(10);
|
return mqttConnectOptions;
|
}
|
|
@Bean
|
public MqttPahoClientFactory mqttClientFactory() {
|
DefaultMqttPahoClientFactory factory = new DefaultMqttPahoClientFactory();
|
factory.setConnectionOptions(mqttConnectOptions());
|
return factory;
|
}
|
}
|
