wg-gateway/src/main/java/ink/wgink/gateway/handler/route/RouteHandler.java

package ink.wgink.gateway.handler.route;

import ink.wgink.gateway.dao.route.IRouteDao;
import ink.wgink.gateway.handler.BaseHandler;
import ink.wgink.gateway.pojo.ListSearch;
import ink.wgink.gateway.pojo.result.SuccessResultList;
import ink.wgink.gateway.pojo.route.Route;
import org.springframework.data.domain.Example;
import org.springframework.data.domain.ExampleMatcher;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Component;
import org.springframework.web.reactive.function.server.RouterFunction;
import org.springframework.web.reactive.function.server.ServerRequest;
import org.springframework.web.reactive.function.server.ServerResponse;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.util.Arrays;
import java.util.List;
import java.util.Optional;

/**
 * When you feel like quitting. Think about why you started
 * 当你想要放弃的时候，想想当初你为何开始
 *
 * @ClassName: RouteHandler
 * @Description: 路由
 * @Author: WangGeng
 * @Date: 2021/4/19 21:40
 * @Version: 1.0
 **/
@Component
public class RouteHandler extends BaseHandler {

    private IRouteDao routeDao;

    public RouteHandler(IRouteDao routeDao) {
        this.routeDao = routeDao;
    }

    /**
     * 新增
     *
     * @param serverRequest
     * @return
     */
    public Mono<ServerResponse> save(ServerRequest serverRequest) {
        Mono<Route> routeMono = serverRequest.bodyToMono(Route.class);
        return routeMono.flatMap(route -> {
            setSave(route);
            return routeDao.save(route);
        }).then(ServerResponse.ok().build());
    }

    /**
     * 删除
     *
     * @param serverRequest
     * @return
     */
    public Mono<ServerResponse> delete(ServerRequest serverRequest) {
        String ids = serverRequest.pathVariable("ids");
        List<String> idList = Arrays.asList(ids.split(","));
        return routeDao.findAllById(idList).flatMap(
                route -> routeDao.delete(route)
        ).then(ServerResponse.ok().contentType(MediaType.APPLICATION_JSON).build());
    }

    /**
     * 修改
     *
     * @param serverRequest
     * @return
     */
    public Mono<ServerResponse> update(ServerRequest serverRequest) {
        String id = serverRequest.pathVariable("id");
        Mono<Route> routeMono = serverRequest.bodyToMono(Route.class);
        return routeDao.findById(id).flatMap(
                route -> routeMono.flatMap(r -> {
                    route.setSystem(r.getSystem());
                    route.setSummary(r.getSummary());
                    route.setPath(r.getPath());
                    route.setUri(r.getUri());
                    setUpdate(route);
                    return routeDao.save(route);
                }).then(ServerResponse.ok().contentType(MediaType.APPLICATION_JSON).build())
        ).switchIfEmpty(ServerResponse.notFound().build());
    }

    /**
     * 列表
     *
     * @param serverRequest
     * @return
     */
    public Mono<ServerResponse> list(ServerRequest serverRequest) {
        return ServerResponse.ok().contentType(MediaType.APPLICATION_JSON).body(routeDao.findAll(), Route.class);
    }

    /**
     * 列表分页
     *
     * @param serverRequest
     * @return
     */
    public Mono<ServerResponse> listPage(ServerRequest serverRequest) {
        Optional<String> page = serverRequest.queryParam("page");
        int currentPage = 1;
        if (page.isPresent()) {
            currentPage = Integer.parseInt(page.get());
        }
        int totalSize = 5;
        Optional<String> size = serverRequest.queryParam("size");
        if (size.isPresent()) {
            totalSize = Integer.parseInt(size.get());
        }

        Optional<String> keyword = serverRequest.queryParam("keyword");
        Route route = new Route();
        ExampleMatcher exampleMatcher = ExampleMatcher.matchingAny()
                .withMatcher("system", ExampleMatcher.GenericPropertyMatcher::contains)
                .withMatcher("summary", ExampleMatcher.GenericPropertyMatcher::contains)
                .withIgnoreCase("id");
        Example example = Example.of(route, exampleMatcher);
        if (keyword.isPresent()) {
            route.setSystem(keyword.get());
            route.setSummary(keyword.get());
        }
        return ServerResponse.ok().contentType(MediaType.APPLICATION_JSON).body(listPage(example, currentPage - 1, totalSize), Route.class);
        //return routeDao.findAll(example).skip((currentPage - 1) * totalSize).take(totalSize);
    }

    public Mono<SuccessResultList> listPage(Example example, int skip, long totalSize) {
        Flux<Route> routeFlux = routeDao.findAll(example).skip(skip * totalSize).take(totalSize);
        Mono<Long> mono = routeDao.count(example);
        return Mono.just(new SuccessResultList());
    }

    /**
     * 列表
     *
     * @param serverRequest
     * @return
     */
    public Mono<ServerResponse> get(ServerRequest serverRequest) {
        String id = serverRequest.pathVariable("id");
        return routeDao.findById(id).flatMap(
                route -> ServerResponse.ok().contentType(MediaType.APPLICATION_JSON).body(Mono.just(route), Route.class)
        ).switchIfEmpty(ServerResponse.notFound().build());
    }

}