com.nebula.boxes.mould.fetcher.loader.FirmStaffDataLoader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-recruit Show documentation

The newest version!

package com.nebula.boxes.mould.fetcher.loader;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;
import java.util.function.Function;
import java.util.stream.Collectors;

import com.nebula.boxes.mould.entity.FirmStaff;
import com.nebula.boxes.mould.service.IFirmStaffService;
import org.apache.commons.collections4.CollectionUtils;
import org.dataloader.BatchLoaderEnvironment;
import org.dataloader.MappedBatchLoaderWithContext;
import org.springframework.beans.factory.annotation.Autowired;
import com.spring.boxes.dollar.support.ContextAwarePoolExecutor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;

import com.netflix.graphql.dgs.DgsDataLoader;
import com.spring.boxes.dollar.support.MoreStream;

/**
* 企业基础信息 DataLoader
*/
@Lazy
@Slf4j
@Service
@DgsDataLoader(name = "firmStaffDataLoader")
public class FirmStaffDataLoader implements MappedBatchLoaderWithContext {

    @Autowired
    private IFirmStaffService firmStaffService;

    @Autowired
    private ContextAwarePoolExecutor contextAwarePoolExecutor;

    @Override
    public CompletionStage> load(Set keys, BatchLoaderEnvironment batchLoaderEnvironment) {
        return CompletableFuture.supplyAsync(() -> {
            List ids = CollectionUtils.emptyIfNull(keys)
                .stream()
                .filter(e -> e > 0)
                .collect(Collectors.toList());
            List list = firmStaffService.listByIds(ids);
            return MoreStream.toMap(list, FirmStaff::getId, Function.identity());
        }, contextAwarePoolExecutor);
    }
}