rankin/rankin-resource-service/src/main/java/cn/rankin/resourceservice/service/ResourceService.java

package cn.rankin.resourceservice.service;

import cn.rankin.common.utils.api.model.APICode;
import cn.rankin.common.utils.api.model.APIResult;
import cn.rankin.common.utils.api.page.Page;
import cn.rankin.common.utils.constant.ResourceType;
import cn.rankin.common.utils.dto.resource.ResourceSearchDTO;
import cn.rankin.common.utils.enums.BaseOrderEnum;
import cn.rankin.common.utils.exception.DuplicateValueException;
import cn.rankin.common.utils.exception.NotFoundException;
import cn.rankin.common.utils.exception.UnsupportedOperationException;
import cn.rankin.data.api.resource.entity.AudioImg;
import cn.rankin.data.api.resource.entity.Resource;
import cn.rankin.resourceservice.dto.ResourceDetail;
import cn.rankin.resourceservice.dto.ResourceRemote;
import cn.rankin.resourceservice.dto.ResourceVideo;
import cn.rankin.resourceservice.proxy.RemoteResourceProxy;
import cn.rankin.resourceservice.repository.AudioImgRepository;
import cn.rankin.resourceservice.repository.ResourceRepository;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.domain.Sort;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.*;


@Slf4j
@Service
public class ResourceService {

    @Autowired
    private AudioImgRepository audioImgRepository;

    @Autowired
    private ResourceRepository resourceRepository;

    @Autowired
    private RemoteResourceProxy resourceProxy;

    @Value("${resource.img.domain}")
    private String imgDomain;

    @Value("${resource.audio.domain}")
    private String audioDomain;

    @Value("${resource.video.domain}")
    private String videoDomain;

    public boolean exists(String code) {
        Long count = resourceRepository.countByCode(code);
        return count > 0L;
    }

    public Resource get(String id) {
        return resourceRepository.find(id);
    }

    public Resource save(Resource resource) {
        return resourceRepository.save(resource);
    }

    public APIResult<Page<Resource>> search(Resource resource, Integer pageNo, Integer pageSize) {
        Long count = resourceRepository.count(resource);
        Page<Resource> page = new Page(count, pageNo, pageSize);

        if (count == 0) {
            return APIResult.ok(page);
        }

        List<Resource> resourceList = resourceRepository.find(resource, page.getStart(), pageSize, new Sort(Sort.Direction.DESC, "gmtModified"));
        resourceList.forEach(r -> {
            initResourceUrl(r);
        });

        page.setList(resourceList);

        return APIResult.ok(page);
    }

    public APIResult<Map<String, Resource>> findMixByIds(List<String> ids) {
        Map<String, Resource> resourceMap = new HashMap<>();

        // 取本地
        List<Resource> nativeResourceList = resourceRepository.findByIds(ids);
        nativeResourceList.forEach(resource -> {
            initResourceUrl(resource);
            resourceMap.put(resource.getId(), resource);
        });

        // 本地基础资源取够了不取了
        if (nativeResourceList.size() == ids.size() && nativeResourceList.size() > 0) {
            return APIResult.ok(resourceMap);
        }

        // 取本地,有声读物资源
        List<AudioImg> audioImgList = audioImgRepository.findByIds(ids);
        audioImgList.forEach(ai -> {
            Resource resource = audioImgToResource(ai);
            resourceMap.put(resource.getId(), resource);
        });


        // 本地取够了不取了
        if (nativeResourceList.size() == ids.size() && nativeResourceList.size() > 0) {
            return APIResult.ok(resourceMap);
        }

        // 取远程
        APIResult<List<ResourceDetail>> remoteResult = resourceProxy.getBatch(ids);
        if (remoteResult.getSuccess() && remoteResult.getData() != null) {
            List<ResourceDetail> remoteList = remoteResult.getData();
            remoteList.forEach(remote -> {
                Resource resource = remote.toNativeResource();
                resourceMap.put(resource.getId(), resource);
            });
        } else {
            log.error("remote resource empty or error!");
        }

        return APIResult.ok(resourceMap);
    }

    public APIResult<Page<Resource>> findRemotePage(ResourceSearchDTO resourceSearchDTO) {
        APIResult<Page<ResourceRemote>> result = resourceProxy.findPage(new HashMap<String, Object>(){
            {
                this.put("no", resourceSearchDTO.getCode());
                this.put("title", resourceSearchDTO.getName());
                this.put("pageNum", resourceSearchDTO.getPageNo());
                this.put("pageSize", resourceSearchDTO.getPageSize());
                this.put("type", resourceSearchDTO.getType());
                this.put("sort", new LinkedHashMap<String,BaseOrderEnum>().put("createTime",BaseOrderEnum.DESC));
            }
        });

        if (!result.getSuccess()) {
            log.error("Remote Server Error: code={}, message={}", result.getCode(), result.getMessage());
            APIResult errResult = APIResult.error(APICode.REMOTE_SERVER_ERROR);
            result.setMessage(result.getMessage());
            return errResult;
        }

        Page<ResourceRemote> remotePage = result.getData();
        Page<Resource> page = new Page<>(remotePage.getTotalSize(), remotePage.getPageNo(), remotePage.getPageSize());

        List<Resource> resourceList = new ArrayList<>();
        for (ResourceRemote resourceRemote : remotePage.getList()) {
            resourceList.addAll(resourceRemote.toNativeResource());
        }

        page.setList(resourceList);

        return APIResult.ok(page);
    }

    public APIResult<Page<Resource>> findRemoteByCode(String code) {
        APIResult<ResourceRemote> result = resourceProxy.findByCode(code);
        if (!result.getSuccess()) {
            log.error("Remote Server Error: code={}, message={}", result.getCode(), result.getMessage());
            APIResult errResult = APIResult.error(APICode.REMOTE_SERVER_ERROR);
            result.setMessage(result.getMessage());
            return errResult;
        }

        ResourceRemote resourceRemote = result.getData();
        if (resourceRemote == null) {
            return APIResult.ok(new Page<>());
        }
        List<Resource> resourceList = result.getData().toNativeResource();
        Page<Resource> page = new Page((long) resourceList.size());
        page.setList(resourceList);

        return APIResult.ok(page);
    }

    public void initResourceUrl(Resource resource) {
        if (resource == null) {
            return;
        }else if (resource.getPath() == null) {
            return;
        }

        String path = resource.getPath();
        if (ResourceType.VIDEO == resource.getType()) {
            resource.setUrl(concatUrl(videoDomain, path));
        }else if (ResourceType.AUDIO == resource.getType()) {
            resource.setUrl(concatUrl(audioDomain, path));
        }else if (ResourceType.IMG == resource.getType()) {
            resource.setUrl(concatUrl(imgDomain, path).replace(" ", ""));
        }
    }

    public static String concatUrl(String domain, String path) {
        if (StringUtils.isEmpty(domain) || StringUtils.isEmpty(path)) {
            return null;
        }
        if (path.startsWith("/") && domain.endsWith("/")) {
            return domain.substring(0, domain.length()) + path;
        }else if(path.startsWith("/") || domain.endsWith("/")){
            return domain + path;
        }else {
            return domain + "/" + path;
        }
    }

    /**
     *
     * @param resource
     * @throws DuplicateValueException
     * @throws NotFoundException
     */
    public Resource update(Resource resource) throws DuplicateValueException, NotFoundException, UnsupportedOperationException {
        if (null == resource){
            log.error("Resource is null");
            return null;
        }

        if (ResourceType.IMG == resource.getType() || ResourceType.AUDIO == resource.getType() ){
            return updateResource(resource);
        }
        else{
            log.error("Not-Supported-Resource-Type, id={}, type={}", resource.getId(), resource.getType());
            throw new UnsupportedOperationException("Not-Supported-Resource-Type");
        }
    }

    /**
     * update Img
     * @param resource
     * @throws DuplicateValueException
     * @throws NotFoundException
     */
    private void updateImg(Resource resource) throws DuplicateValueException, NotFoundException {

        if (ResourceType.IMG != resource.getType()){
            log.error("Not-Img-Resource");
            return;
        }

        //find target resource
        Resource targetResouce = resourceRepository.find(resource.getId());

        if(null == targetResouce){
            log.error("Cannot Find Resource, id={}", resource.getId());
            throw new NotFoundException("Resource-Not-Found");
        }

        //only support code,title,path
        //first, make sure code is not duplicated
        List<Resource> foundedResource = resourceRepository.findByCode(resource.getCode());
        foundedResource.forEach( (each) -> {
            if ( !resource.getId().equals(each.getId()) ){
                log.error("Code-Already-Exist, code={}", resource.getCode());
                throw new DuplicateValueException("Code-Already-Exist");
            }
        });

        //set and save
        targetResouce.setCode(resource.getCode());
        targetResouce.setName(resource.getName());
        targetResouce.setPath(resource.getPath());
        resourceRepository.save(targetResouce);
    }

    /**
     * update Img,audio
     * @param resource
     * @throws DuplicateValueException
     * @throws NotFoundException
     */
    public Resource updateResource(Resource resource) throws DuplicateValueException, NotFoundException {

        if (ResourceType.IMG != resource.getType() && ResourceType.AUDIO!= resource.getType()){
            log.error("Not-Img/audio -Resource");
            return null;
        }

        //find target resource
        Resource targetResouce = resourceRepository.find(resource.getId());

        if(null == targetResouce){
            log.error("Cannot Find Resource, id={}", resource.getId());
            throw new NotFoundException("Resource-Not-Found");
        }

        //only support code,title,path
        //first, make sure code is not duplicated
        List<Resource> foundedResource = resourceRepository.findByCode(resource.getCode());
        foundedResource.forEach( (each) -> {
            if ( !resource.getId().equals(each.getId()) ){
                log.error("Code-Already-Exist, code={}", resource.getCode());
                throw new DuplicateValueException("Code-Already-Exist");
            }
        });

        //set and save
        targetResouce.setCode(resource.getCode());
        targetResouce.setName(resource.getName());
        targetResouce.setPath(resource.getPath());
        return resourceRepository.save(targetResouce);
    }

    /**
     * saveRemoteResource
     * type 0 video , 1 audio
     * @param resource
     * @return
     */
    public APIResult<Resource> saveRemoteResource(Resource resource) {

        Set<ResourceVideo> videos = new HashSet<>();
        ResourceVideo video = new ResourceVideo();
        video.setBucket("efunbox");
        video.setPath(resource.getPath());
        video.setQuality(resource.getQuality());
        video.setFormat(resource.getFormat());
        video.setSize(resource.getSize());
        videos.add(video);


        ResourceRemote resourceRemote = new ResourceRemote();
        resourceRemote.setNo(resource.getCode());
        resourceRemote.setTitle(resource.getName());
        resourceRemote.setType(resource.getType());
        resourceRemote.setVideos(videos);


        APIResult<ResourceDetail> apiResult = resourceProxy.saveResources(resourceRemote);
        if (!apiResult.getSuccess()) {
            return errorResourceAPIResult(apiResult);
        }

        log.info("Remote Server return: code={}, message={}", apiResult.getCode(), apiResult.getMessage());
//        Resource resourceVO = apiResult.getData().toNativeResource();
        return APIResult.ok();
    }

    /**
     * update  RemoteResource
     * @param resource
     * @return
     */
    public APIResult<Resource> updateRemoteResource(Resource resource) {

        if (org.apache.commons.lang.StringUtils.isEmpty(resource.getId())) {
            return APIResult.error(APICode.PARAMETER_ERROR);
        }

        Set<ResourceVideo> videos = new HashSet<>();
        ResourceVideo video = new ResourceVideo();
        video.setId(resource.getId());
        video.setBucket("efunbox");
        video.setPath(resource.getPath());
        video.setQuality(resource.getQuality());
        video.setFormat(resource.getFormat());
        video.setSize(resource.getSize());
        videos.add(video);


        ResourceRemote resourceRemote = new ResourceRemote();
        resourceRemote.setId(resource.getRid());
        resourceRemote.setNo(resource.getCode());
        resourceRemote.setTitle(resource.getName());
        resourceRemote.setType(resource.getType());
        resourceRemote.setVideos(videos);

        APIResult apiResult = resourceProxy.updateResources(resourceRemote);
        if (!apiResult.getSuccess()) {
            return errorResourceAPIResult(apiResult);
        }

        log.info("Remote Server info: code={}, message={}", apiResult.getCode(), apiResult.getMessage());
        return APIResult.ok();
    }

    private APIResult<Resource> errorResourceAPIResult(APIResult<ResourceDetail> apiResult) {
        log.error("Remote Server Error: code={}, message={}", apiResult.getCode(), apiResult.getMessage());
        APIResult errResult = APIResult.error(APICode.REMOTE_SERVER_ERROR);
        errResult.setMessage(apiResult.getMessage());
        return errResult;
    }

    /**
     * 有声读物 转  资源
     * @param ai
     * @return
     */
    public Resource audioImgToResource(AudioImg ai) {
        String audioId = ai.getAudioId();
        Resource audio = resourceRepository.find(audioId);
        initResourceUrl(audio);
        String imgId = ai.getImgId();
        Resource img = resourceRepository.find(imgId);
        initResourceUrl(img);

        Resource resource = new Resource();
        //有声读物info
        resource.setType(ai.getType());
        resource.setId(ai.getId());
        resource.setCode(ai.getCode());
        resource.setName(ai.getName());
        resource.setStatus(ai.getStatus());
        resource.setGmtCreated(ai.getGmtCreated());
        resource.setGmtModified(ai.getGmtModified());

        //图片info
        resource.setImg(img);
        //音频info
        resource.setAudio(audio);
        return resource;
    }

}