/*
 * Copyright 2016 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package de.vorb.platon.web.rest;


import de.vorb.platon.persistence.CommentRepository;
import de.vorb.platon.web.rest.json.CommentCountsJson;

import org.springframework.stereotype.Component;

import javax.inject.Inject;
import javax.validation.constraints.NotNull;
import javax.ws.rs.Consumes;
import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.MediaType;
import java.util.Map;
import java.util.Set;


@Component
@Path("comment-count")
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
public class CommentCountResource {

    private final CommentRepository commentRepository;

    @Inject
    public CommentCountResource(CommentRepository commentRepository) {
        this.commentRepository = commentRepository;
    }

    @GET
    public CommentCountsJson getCommentCounts(@NotNull @QueryParam("threadUrl[]") Set<String> threadUrls) {

        final CommentCountsJson commentCounts = new CommentCountsJson();

        Map<String, Long> counts = commentRepository.countByThreadUrls(threadUrls);

        counts.forEach(commentCounts::setCommentCount);

        return commentCounts;
    }
}

@@ -1725,21 +1725,25 @@
         
-count
+threadUrl
 s.forEac
@@ -1744,16 +1744,45 @@
 forEach(
+threadUrl -%3E%0A                
 commentC
@@ -1790,10 +1790,9 @@
 unts
-::
+.
 setC
@@ -1802,16 +1802,63 @@
 entCount
+(threadUrl, counts.getOrDefault(threadUrl, 0L))
 );%0A%0A    

Improve comment counts for non-existing threads