io.irain.reactor.security.service.ClientInfoService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of shore-security Show documentation

shore security starter

There is a newer version: 1.0.3

package io.irain.reactor.security.service;

import io.irain.reactor.commons.exception.ServerException;
import io.irain.reactor.commons.exception.TokenException;
import io.irain.reactor.security.domain.ClientInfo;
import io.irain.reactor.security.util.ClientUtil;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

/**
 * @author youta
 **/
public interface ClientInfoService {


    /**
     * 获取客户端id
     *
     * @param exchange 请求
     * @return {@link ClientInfo}
     */
    default Mono client(ServerWebExchange exchange) {
        var client = ClientUtil.getClient(exchange);
        return this.loadClientById(client.getClientId())
                .doOnError(throwable -> {
                    throw new ServerException("You must implement the loadClientById method");
                })
                .switchIfEmpty(Mono.error(new TokenException("Invalid ClientId")))
                .handle((clientInfo, sink) -> {
                    if (!clientInfo.getClientSecret().equals(client.getClientSecret())) {
                        sink.error(new TokenException("Invalid ClientSecret"));
                        return;
                    }
                    sink.next(clientInfo);
                });
    }

    /**
     * 根据clientId获取客户端信息
     *
     * @param clientId 客户端id
     * @return ClientInfo 客户端信息
     */
    default Mono loadClientById(String clientId) {
        return Mono.justOrEmpty(ClientInfo.base(clientId, ""));
    }

}