38 lines
1.2 KiB
Java
38 lines
1.2 KiB
Java
package com.muyu.gateway.handler;
|
|
|
|
import com.muyu.common.core.exception.CaptchaException;
|
|
import com.muyu.common.core.domain.Result;
|
|
import com.muyu.gateway.service.ValidateCodeService;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.http.HttpStatus;
|
|
import org.springframework.stereotype.Component;
|
|
import org.springframework.web.reactive.function.BodyInserters;
|
|
import org.springframework.web.reactive.function.server.HandlerFunction;
|
|
import org.springframework.web.reactive.function.server.ServerRequest;
|
|
import org.springframework.web.reactive.function.server.ServerResponse;
|
|
import reactor.core.publisher.Mono;
|
|
|
|
import java.io.IOException;
|
|
|
|
/**
|
|
* 验证码获取
|
|
*
|
|
* @author muyu
|
|
*/
|
|
@Component
|
|
public class ValidateCodeHandler implements HandlerFunction<ServerResponse> {
|
|
@Autowired
|
|
private ValidateCodeService validateCodeService;
|
|
|
|
@Override
|
|
public Mono<ServerResponse> handle (ServerRequest serverRequest) {
|
|
Result ajax;
|
|
try {
|
|
ajax = validateCodeService.createCaptcha();
|
|
} catch (CaptchaException | IOException e) {
|
|
return Mono.error(e);
|
|
}
|
|
return ServerResponse.status(HttpStatus.OK).body(BodyInserters.fromValue(ajax));
|
|
}
|
|
}
|