CardSetGrpcClient.java

package flipnote.reaction.common.grpc;

import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

import org.springframework.stereotype.Component;

import cardset.Cardset.CardSetSummary;
import cardset.Cardset.GetCardSetsByIdsRequest;
import cardset.Cardset.GetCardSetsByIdsResponse;
import cardset.Cardset.IsCardSetViewableRequest;
import cardset.Cardset.IsCardSetViewableResponse;
import cardset.CardsetServiceGrpc;
import flipnote.reaction.common.exception.BizException;
import flipnote.reaction.common.exception.CommonErrorCode;
import io.grpc.ManagedChannel;
import io.grpc.StatusRuntimeException;
import lombok.extern.slf4j.Slf4j;

@Slf4j
@Component
public class CardSetGrpcClient {

	private final CardsetServiceGrpc.CardsetServiceBlockingStub stub;

	public CardSetGrpcClient(ManagedChannel cardSetChannel) {
		this.stub = CardsetServiceGrpc.newBlockingStub(cardSetChannel);
	}

	public boolean isCardSetViewable(Long cardSetId, Long userId) {
		try {
			IsCardSetViewableRequest request = IsCardSetViewableRequest.newBuilder()
				.setCardSetId(cardSetId.intValue())
				.setUserId(userId.intValue())
				.build();

			IsCardSetViewableResponse response = stub.isCardSetViewable(request);
			return response.getViewable();
		} catch (StatusRuntimeException e) {
			log.error("gRPC call failed: IsCardSetViewable, cardSetId={}, userId={}", cardSetId, userId, e);
			throw new BizException(CommonErrorCode.GRPC_CALL_FAILED);
		}
	}

	public Map<Long, CardSetSummary> getCardSetsByIds(List<Long> cardSetIds, Long userId) {
		try {
			GetCardSetsByIdsRequest request = GetCardSetsByIdsRequest.newBuilder()
				.addAllCardSetIds(cardSetIds)
				.setUserId(userId.intValue())
				.build();

			GetCardSetsByIdsResponse response = stub.getCardSetsByIds(request);
			return response.getCardSetsList().stream()
				.collect(Collectors.toMap(
					cs -> (long) cs.getId(),
					Function.identity()
				));
		} catch (StatusRuntimeException e) {
			log.error("gRPC call failed: GetCardSetsByIds, cardSetIds={}, userId={}", cardSetIds, userId, e);
			throw new BizException(CommonErrorCode.GRPC_CALL_FAILED);
		}
	}
}