feat():获取获取token
parent
a3968b7b3f
commit
05d11caa6b
|
|
@ -131,7 +131,7 @@
|
|||
<dependency>
|
||||
<groupId>com.alibaba</groupId>
|
||||
<artifactId>fastjson</artifactId>
|
||||
<version>1.2.73</version>
|
||||
<version>1.2.75</version>
|
||||
</dependency>
|
||||
|
||||
<!--常用工具类 -->
|
||||
|
|
|
|||
|
|
@ -18,6 +18,11 @@
|
|||
</properties>
|
||||
<dependencies>
|
||||
|
||||
<!--mq-->
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-amqp</artifactId>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>com.aliyun</groupId>
|
||||
<artifactId>ecs20140526</artifactId>
|
||||
|
|
|
|||
|
|
@ -0,0 +1,26 @@
|
|||
package com.zhiLian.common.req;
|
||||
|
||||
import lombok.AllArgsConstructor;
|
||||
import lombok.Data;
|
||||
import lombok.NoArgsConstructor;
|
||||
|
||||
|
||||
/**
|
||||
* 连接权重 车子服务器
|
||||
* @author YunFei.Du
|
||||
* @date 22:31 2024/5/28
|
||||
*/
|
||||
@Data
|
||||
@AllArgsConstructor
|
||||
@NoArgsConstructor
|
||||
public class ConnectWeight {
|
||||
|
||||
/**
|
||||
* 服务器ip
|
||||
*/
|
||||
private String carServerIp;
|
||||
/**
|
||||
* 权重值
|
||||
*/
|
||||
private Integer weightValue;
|
||||
}
|
||||
|
|
@ -0,0 +1,15 @@
|
|||
package com.zhiLian.common.req;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
/**
|
||||
* @ClassName UsersReq
|
||||
* @Description 描述
|
||||
* @Author YunFei.Du
|
||||
* @Date 2024/5/28 22:10
|
||||
*/
|
||||
@Data
|
||||
public class UsersReq {
|
||||
private String username;
|
||||
private String password;
|
||||
}
|
||||
|
|
@ -0,0 +1,54 @@
|
|||
package com.zhiLian.config;
|
||||
|
||||
import org.springframework.amqp.core.*;
|
||||
import org.springframework.beans.factory.annotation.Qualifier;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
/**
|
||||
* rabbitMq配置类
|
||||
*
|
||||
* @author YunFei Du
|
||||
* @ClassName: RabbitmqConfig
|
||||
* @Description: rabbitMq配置类
|
||||
* @CreateTime: 2024/5/27 16:56
|
||||
*/
|
||||
@Configuration
|
||||
public class RabbitmqConfig {
|
||||
public static final String QUEUE_INFORM_EMAIL = "queue_inform_email";
|
||||
public static final String QUEUE_INFORM_SMS = "disconnect_connect";
|
||||
public static final String EXCHANGE_TOPICS_INFORM="exchange_topics_inform";
|
||||
public static final String ROUTINGKEY_EMAIL="inform.#.email.#";
|
||||
public static final String ROUTINGKEY_SMS="inform.#.sms.#";
|
||||
|
||||
|
||||
@Bean(EXCHANGE_TOPICS_INFORM)
|
||||
public Exchange EXCHANGE_TOPICS_INFORM(){
|
||||
//durable(true) 持久化，mq重启之后交换机还在
|
||||
return ExchangeBuilder.topicExchange(EXCHANGE_TOPICS_INFORM).durable(true).build();
|
||||
}
|
||||
|
||||
//声明QUEUE_INFORM_EMAIL队列
|
||||
@Bean(QUEUE_INFORM_EMAIL)
|
||||
public Queue QUEUE_INFORM_EMAIL(){
|
||||
return new Queue(QUEUE_INFORM_EMAIL);
|
||||
}
|
||||
//声明QUEUE_INFORM_SMS队列
|
||||
@Bean(QUEUE_INFORM_SMS)
|
||||
public Queue QUEUE_INFORM_SMS(){
|
||||
return new Queue(QUEUE_INFORM_SMS);
|
||||
}
|
||||
|
||||
//ROUTINGKEY_EMAIL队列绑定交换机，指定routingKey
|
||||
@Bean
|
||||
public Binding BINDING_QUEUE_INFORM_EMAIL(@Qualifier(QUEUE_INFORM_EMAIL) Queue queue,
|
||||
@Qualifier(EXCHANGE_TOPICS_INFORM) Exchange exchange){
|
||||
return BindingBuilder.bind(queue).to(exchange).with(ROUTINGKEY_EMAIL).noargs();
|
||||
}
|
||||
//ROUTINGKEY_SMS队列绑定交换机，指定routingKey
|
||||
@Bean
|
||||
public Binding BINDING_ROUTINGKEY_SMS(@Qualifier(QUEUE_INFORM_SMS) Queue queue,
|
||||
@Qualifier(EXCHANGE_TOPICS_INFORM) Exchange exchange){
|
||||
return BindingBuilder.bind(queue).to(exchange).with(ROUTINGKEY_SMS).noargs();
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,33 @@
|
|||
package com.zhiLian.config;
|
||||
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.data.redis.connection.RedisConnectionFactory;
|
||||
import org.springframework.data.redis.core.RedisTemplate;
|
||||
import org.springframework.data.redis.serializer.Jackson2JsonRedisSerializer;
|
||||
import org.springframework.data.redis.serializer.StringRedisSerializer;
|
||||
|
||||
/**
|
||||
* redsi配置类
|
||||
*
|
||||
* @author YunFei Du
|
||||
* @ClassName: RedisConfoig
|
||||
* @Description: redsi配置类
|
||||
* @CreateTime: 2024/5/27 14:16
|
||||
*/
|
||||
@Configuration
|
||||
public class RedisConfig {
|
||||
@Bean
|
||||
public RedisTemplate<String,String> redisTemplate(RedisConnectionFactory redisConnectionFactory){
|
||||
RedisTemplate<String, String> redisTemplate = new RedisTemplate<>();
|
||||
redisTemplate.setConnectionFactory(redisConnectionFactory);
|
||||
|
||||
redisTemplate.setKeySerializer(new StringRedisSerializer());
|
||||
redisTemplate.setValueSerializer(new Jackson2JsonRedisSerializer<Object>(Object.class));
|
||||
|
||||
redisTemplate.setHashKeySerializer(new StringRedisSerializer());
|
||||
redisTemplate.setHashValueSerializer(new StringRedisSerializer());
|
||||
return redisTemplate;
|
||||
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,33 @@
|
|||
package com.zhiLian.config;
|
||||
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.http.client.ClientHttpRequestFactory;
|
||||
import org.springframework.http.client.SimpleClientHttpRequestFactory;
|
||||
import org.springframework.web.client.RestTemplate;
|
||||
|
||||
/**
|
||||
* rest配置类
|
||||
*
|
||||
* @author YunFei Du
|
||||
* @ClassName: RestTemplateConfig
|
||||
* @Description: rest配置类
|
||||
* @CreateTime: 2024/5/27 10:01
|
||||
*/
|
||||
@Configuration
|
||||
public class RestTemplateConfig {
|
||||
@Bean
|
||||
public RestTemplate restTemplate(ClientHttpRequestFactory factory) {
|
||||
return new RestTemplate(factory);
|
||||
}
|
||||
|
||||
@Bean
|
||||
public ClientHttpRequestFactory simpleClientHttpRequestFactory() {
|
||||
SimpleClientHttpRequestFactory factory = new SimpleClientHttpRequestFactory();
|
||||
//超时设置
|
||||
factory.setReadTimeout(5000);//ms
|
||||
factory.setConnectTimeout(15000);//ms
|
||||
return factory;
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -1,16 +1,22 @@
|
|||
package com.zhiLian.controller;
|
||||
|
||||
import com.alibaba.fastjson2.JSON;
|
||||
import com.alibaba.fastjson2.JSONArray;
|
||||
import com.alibaba.fastjson2.JSONObject;
|
||||
import com.zhiLian.common.Result;
|
||||
import com.zhiLian.common.req.ConnectWeight;
|
||||
import com.zhiLian.service.LoadBalanceService;
|
||||
import lombok.extern.log4j.Log4j2;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.web.bind.annotation.GetMapping;
|
||||
import org.springframework.web.bind.annotation.RequestMapping;
|
||||
import org.springframework.web.bind.annotation.RequestParam;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
import org.springframework.data.redis.core.RedisTemplate;
|
||||
import org.springframework.http.*;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
import org.springframework.web.client.RestTemplate;
|
||||
|
||||
|
||||
import java.util.List;
|
||||
import java.math.BigDecimal;
|
||||
import java.math.RoundingMode;
|
||||
import java.util.*;
|
||||
|
||||
/**
|
||||
* @ClassName LoadBalanceController
|
||||
|
|
@ -24,6 +30,14 @@ import java.util.List;
|
|||
public class LoadBalanceController {
|
||||
|
||||
|
||||
private static List< String > ipListThread = new ArrayList<> ( );
|
||||
|
||||
@Autowired
|
||||
private RestTemplate restTemplate;
|
||||
|
||||
@Autowired
|
||||
private RedisTemplate redisTemplate;
|
||||
|
||||
@Autowired
|
||||
private LoadBalanceService loadBalanceService;
|
||||
|
||||
|
|
@ -61,7 +75,87 @@ public class LoadBalanceController {
|
|||
@GetMapping("/getIpList")
|
||||
public Result< List< String > > getIpList() {
|
||||
List< String > ipList = loadBalanceService.getIpList ( );
|
||||
// ipListThread = ipList;
|
||||
ipListThread=ipList;
|
||||
return Result.success ( ipList );
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取token
|
||||
*/
|
||||
@PostMapping("/getToken")
|
||||
public void getToken() {
|
||||
|
||||
List< ConnectWeight > connectWeightList = new ArrayList<> ( );
|
||||
|
||||
for (String ip : ipListThread) {
|
||||
// 向服务器发送登录请求
|
||||
String url = "http://" + ip + ":8080/public/login";
|
||||
Map< String, Object > request = new HashMap<> ( );
|
||||
request.put ( "username", "fluxmq" );
|
||||
request.put ( "password", "fluxmq" );
|
||||
HttpHeaders httpHeaders = new HttpHeaders ( );
|
||||
httpHeaders.setContentType ( MediaType.APPLICATION_JSON );
|
||||
HttpEntity< Map< String, Object > > r = new HttpEntity< Map< String, Object > > ( request, httpHeaders );
|
||||
String result = restTemplate.postForObject ( url, r, String.class );
|
||||
// 添加随机参数以避免缓存
|
||||
int nextInt = new Random ( ).nextInt ( 1000 );
|
||||
String getInfoUrl = "http://" + ip + ":8080/public/cluster?random=" + nextInt;
|
||||
// 使用获取到的token向服务器查询当前连接数
|
||||
HttpHeaders httpHeadersGetInfo = new HttpHeaders ( );
|
||||
httpHeadersGetInfo.setContentType ( MediaType.APPLICATION_JSON );
|
||||
httpHeadersGetInfo.setAccept ( Collections.singletonList ( MediaType.APPLICATION_JSON ) );
|
||||
httpHeadersGetInfo.set ( "Cookie", result );
|
||||
HttpEntity getInfoRequest = new HttpEntity ( httpHeadersGetInfo );
|
||||
ResponseEntity< String > responseInfo = restTemplate.exchange ( getInfoUrl, HttpMethod.GET, getInfoRequest, String.class, 1 );
|
||||
|
||||
// 将字符串转换为有效的JSON格式
|
||||
|
||||
JSONArray jsonArray = JSON.parseArray ( responseInfo.getBody ( ) );
|
||||
log.info ( "响应是" + jsonArray );
|
||||
|
||||
if (jsonArray.size ( ) > 0) {
|
||||
|
||||
JSONObject jsonObject = jsonArray.getJSONObject ( 0 );
|
||||
Integer connectSize = Integer.valueOf ( jsonObject.getJSONObject ( "mqttInfo" ).getString ( "connectSize" ) );
|
||||
connectWeightList.add ( new ConnectWeight ( ip, 100 - connectSize ) );
|
||||
log.error ( "链接数量:{}", connectSize );
|
||||
} else {
|
||||
log.error ( "得到的相应数据为null" );
|
||||
}
|
||||
}
|
||||
Integer sum = 0;
|
||||
for (ConnectWeight connectWeight : connectWeightList) {
|
||||
sum = sum + connectWeight.getWeightValue ( );
|
||||
}
|
||||
|
||||
int max = 0;
|
||||
for (ConnectWeight connectWeight : connectWeightList) {
|
||||
log.error ( "权重值:{}", connectWeight.getWeightValue ( ) );
|
||||
Integer result = BigDecimal.valueOf ( connectWeight.getWeightValue ( ) * 100 ).divide ( BigDecimal.valueOf ( sum ), 0, RoundingMode.DOWN ).intValue ( );
|
||||
if (result > max) {
|
||||
max = result;
|
||||
}
|
||||
connectWeight.setWeightValue ( result );
|
||||
log.error ( "100次轮询次数:{}", result );
|
||||
}
|
||||
|
||||
ArrayList< String > weightIpList = new ArrayList<> ( );
|
||||
|
||||
//轮询出现次数
|
||||
for (int i = 0; i <= max; i++) {
|
||||
for (ConnectWeight connectWeight : connectWeightList) {
|
||||
if (connectWeight.getWeightValue ( ) > i) {
|
||||
weightIpList.add ( connectWeight.getCarServerIp ( ) );
|
||||
} else if (connectWeight.getWeightValue ( ) == max) {
|
||||
weightIpList.add ( connectWeight.getCarServerIp ( ) );
|
||||
}
|
||||
}
|
||||
}
|
||||
//存入redis
|
||||
redisTemplate.delete ( "ipList" );
|
||||
for (String ip : weightIpList) {
|
||||
redisTemplate.opsForList ( ).rightPush ( "ipList", ip );
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
|||
|
|
@ -49,7 +49,7 @@ public class LoadBalanceServiceImpl implements LoadBalanceService {
|
|||
@SneakyThrows
|
||||
@Override
|
||||
public List< String > getIpList() {
|
||||
FindInstance ( "cn-zhangjiakou" );
|
||||
return null;
|
||||
List< String > ipList = FindInstance ( "cn-zhangjiakou" );
|
||||
return ipList;
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,5 +1,6 @@
|
|||
package com.zhiLian.utils;
|
||||
|
||||
import com.alibaba.fastjson2.JSON;
|
||||
import com.alibaba.fastjson2.JSONArray;
|
||||
import com.alibaba.fastjson2.JSONObject;
|
||||
import com.aliyun.ecs20140526.Client;
|
||||
|
|
@ -9,10 +10,13 @@ import com.aliyun.tea.TeaModel;
|
|||
import com.aliyun.teaopenapi.models.Config;
|
||||
import com.aliyun.teautil.Common;
|
||||
import com.aliyun.teautil.models.RuntimeOptions;
|
||||
import com.zhiLian.common.req.ConnectWeight;
|
||||
import lombok.extern.log4j.Log4j2;
|
||||
import org.springframework.http.*;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.math.BigDecimal;
|
||||
import java.math.RoundingMode;
|
||||
import java.util.*;
|
||||
import java.util.stream.Collectors;
|
||||
import java.util.stream.IntStream;
|
||||
|
||||
|
|
@ -24,6 +28,8 @@ import java.util.stream.IntStream;
|
|||
*/
|
||||
@Log4j2
|
||||
public class ECSTool {
|
||||
|
||||
|
||||
/**
|
||||
* 创建ECS客户端
|
||||
* @return ECS客户端实例
|
||||
|
|
@ -135,7 +141,7 @@ public class ECSTool {
|
|||
|
||||
|
||||
|
||||
public static void FindInstance( String regionId) throws Exception {
|
||||
public static List<String> FindInstance( String regionId) throws Exception {
|
||||
// 创建ECS客户端
|
||||
Client client = ECSTool.createClient ( );
|
||||
DescribeInstancesRequest describeInstancesRequest = new DescribeInstancesRequest ( )
|
||||
|
|
@ -153,7 +159,8 @@ public class ECSTool {
|
|||
}
|
||||
System.out.println ( "------------------------" );
|
||||
}
|
||||
log.info ( "ipList: " + ipList );
|
||||
log.info ( "ipList: " + ipList ); // [39.100.89.218, 39.100.87.192]
|
||||
return ipList;
|
||||
} catch (TeaException error) {
|
||||
// 此处仅做打印展示，请谨慎对待异常处理，在工程项目中切勿直接忽略异常。
|
||||
// 错误 message
|
||||
|
|
@ -170,6 +177,7 @@ public class ECSTool {
|
|||
System.out.println ( error.getData ( ).get ( "Recommend" ) );
|
||||
com.aliyun.teautil.Common.assertAsString ( error.message );
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,2 +1,13 @@
|
|||
server:
|
||||
port: 84
|
||||
spring:
|
||||
rabbitmq:
|
||||
host: 111.229.102.61
|
||||
port: 5672
|
||||
username: guest
|
||||
password: guest
|
||||
virtualHost: /
|
||||
redis:
|
||||
host: 127.0.0.1
|
||||
port: 6379
|
||||
password: dyf@123
|
||||
|
|
|
|||
Loading…
Reference in New Issue