diff --git a/src/main/java/dtos/OptionCount.java b/src/main/java/dtos/OptionCount.java new file mode 100644 index 0000000..b25361e --- /dev/null +++ b/src/main/java/dtos/OptionCount.java @@ -0,0 +1,26 @@ +package dtos; + +public class OptionCount { + private Long optionId; + private int count; + public OptionCount(Long optionId, int count){ + this.optionId = optionId; + this.count = count; + } + + public Long getOptionId() { + return optionId; + } + + public void setOptionId(Long optionId) { + this.optionId = optionId; + } + + public int getCount() { + return count; + } + + public void setCount(int count) { + this.count = count; + } +} diff --git a/src/main/java/dtos/VoteResult.java b/src/main/java/dtos/VoteResult.java new file mode 100644 index 0000000..32bd7b8 --- /dev/null +++ b/src/main/java/dtos/VoteResult.java @@ -0,0 +1,31 @@ +package dtos; + +import io.zipcoder.tc_spring_poll_application.domain.Option; + +import java.util.Collection; + +public class VoteResult { + private int totalVotes; + private Collection results; + + public int getTotalVotes() { + int count = 0; + for(OptionCount optionCount : results){ + count = count + optionCount.getCount(); + } + totalVotes = count; + return totalVotes; + } + + public void setTotalVotes(int totalVotes) { + this.totalVotes = totalVotes; + } + + public Collection getResults() { + return results; + } + + public void setResults(Collection results) { + this.results = results; + } +} diff --git a/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/ComputeResultController.java b/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/ComputeResultController.java new file mode 100644 index 0000000..6ad91fd --- /dev/null +++ b/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/ComputeResultController.java @@ -0,0 +1,60 @@ +package io.zipcoder.tc_spring_poll_application.Controller; + +import dtos.OptionCount; +import dtos.VoteResult; +import io.zipcoder.tc_spring_poll_application.Repositories.VoteRepository; +import io.zipcoder.tc_spring_poll_application.domain.Option; +import io.zipcoder.tc_spring_poll_application.domain.Vote; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.http.HttpStatus; +import org.springframework.http.ResponseEntity; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestMethod; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; + +import java.lang.reflect.Array; +import java.util.*; + +@RestController +public class ComputeResultController { + private VoteRepository voteRepository; + + + + @Autowired + public ComputeResultController(VoteRepository voteRepository) { + this.voteRepository = voteRepository; + } + + @RequestMapping(value = "/computeresult", method = RequestMethod.GET) + public ResponseEntity computeResult(@RequestParam Long pollId) { + VoteResult voteResult = new VoteResult(); + Iterable allVotes = voteRepository.findVotesByPoll(pollId); + //now we are creating a map of option id and total vote count + Map map = new HashMap<>(); + Long optionId; + for (Vote vote : allVotes) { + optionId = vote.getOption().getId(); + int updateCount = 0; + if (!map.containsKey(optionId)) + map.put(optionId, 1); + else { + updateCount = map.get(optionId) + 1; + map.put(optionId, updateCount); + } + } + Set optionCounts = new HashSet<>(); + for (Long id : map.keySet()) { + optionCounts.add(new OptionCount(id, map.get(id))); + } + voteResult.setResults(optionCounts); + voteResult.setTotalVotes(voteResult.getTotalVotes()); + + + + + //TODO: Implement algorithm to count votes + return new ResponseEntity(voteResult, HttpStatus.OK); + } +} \ No newline at end of file diff --git a/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/OptionController.java b/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/OptionController.java new file mode 100644 index 0000000..ed0f413 --- /dev/null +++ b/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/OptionController.java @@ -0,0 +1,7 @@ +package io.zipcoder.tc_spring_poll_application.Controller; + +import org.springframework.web.bind.annotation.RestController; + +@RestController +public class OptionController { +} diff --git a/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/PollController.java b/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/PollController.java new file mode 100644 index 0000000..5575346 --- /dev/null +++ b/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/PollController.java @@ -0,0 +1,75 @@ +package io.zipcoder.tc_spring_poll_application.Controller; + +import io.zipcoder.tc_spring_poll_application.Repositories.PollRepository; +import io.zipcoder.tc_spring_poll_application.domain.Poll; +import io.zipcoder.tc_spring_poll_application.exception.ResourceNotFoundException; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.data.domain.Page; +import org.springframework.data.domain.Pageable; +import org.springframework.http.HttpHeaders; +import org.springframework.http.HttpStatus; +import org.springframework.http.ResponseEntity; +import org.springframework.web.bind.annotation.*; +import org.springframework.web.servlet.support.ServletUriComponentsBuilder; + +import javax.validation.Valid; + +import java.net.URI; + +@RestController +public class PollController { + @Autowired + PollRepository pollRepository; + //Poll poll; + + @Autowired + public PollController(PollRepository pollRepository) { + this.pollRepository = pollRepository; + } + @Valid + @RequestMapping(value = "/polls", method = RequestMethod.GET) + public ResponseEntity> getAllPolls(Pageable pageable) { + Page allPolls = pollRepository.findAll(pageable); + return new ResponseEntity>(allPolls, HttpStatus.OK); + } + @Valid + @RequestMapping(value = "/polls", method = RequestMethod.POST) + public ResponseEntity createPoll(@RequestBody Poll poll) { + poll = pollRepository.save(poll); + URI newPollUri = ServletUriComponentsBuilder + .fromCurrentRequest() + .path("/{id}") + .buildAndExpand(poll.getId()) + .toUri(); + HttpHeaders httpHeaders = new HttpHeaders(); + httpHeaders.setLocation(newPollUri); + return new ResponseEntity<>(newPollUri, HttpStatus.CREATED); + } + @Valid + @RequestMapping(value = "/polls/{pollId}", method = RequestMethod.GET) + public ResponseEntity getPoll(@PathVariable Long pollId) { + Poll p = pollRepository.findOne(pollId); + return new ResponseEntity<>(p, HttpStatus.OK); + } + @Valid + @RequestMapping(value = "/polls/{pollId}", method = RequestMethod.PUT) + public ResponseEntity updatePoll(@RequestBody Poll poll, @PathVariable Long pollId) { + // Save the entity + Poll p = pollRepository.save(poll); + return new ResponseEntity<>(HttpStatus.OK); + } + @Valid + @RequestMapping(value = "/polls/{pollId}", method = RequestMethod.DELETE) + public ResponseEntity deletePoll(@PathVariable Long pollId) { + pollRepository.delete(pollId); + return new ResponseEntity<>(HttpStatus.OK); + } + public void verifyPoll(@PathVariable Long pollId) { + if(!pollRepository.exists(pollId)) + throw new ResourceNotFoundException(); + } + + + + +} diff --git a/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/VoteController.java b/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/VoteController.java new file mode 100644 index 0000000..2bfff21 --- /dev/null +++ b/src/main/java/io/zipcoder/tc_spring_poll_application/Controller/VoteController.java @@ -0,0 +1,43 @@ +package io.zipcoder.tc_spring_poll_application.Controller; + +import io.zipcoder.tc_spring_poll_application.Repositories.VoteRepository; +import io.zipcoder.tc_spring_poll_application.domain.Vote; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.data.jpa.repository.Query; +import org.springframework.data.repository.CrudRepository; +import org.springframework.http.HttpHeaders; +import org.springframework.http.HttpStatus; +import org.springframework.http.ResponseEntity; +import org.springframework.web.bind.annotation.*; +import org.springframework.web.servlet.support.ServletUriComponentsBuilder; + +@RestController +public class VoteController { + private VoteRepository voteRepository; + + @Autowired + public VoteController(VoteRepository voteRepository) { + this.voteRepository = voteRepository; + } + + @RequestMapping(value = "/polls/{pollId}/votes", method = RequestMethod.POST) + public ResponseEntity createVote(@PathVariable Long pollId, @RequestBody Vote + vote) { + vote = voteRepository.save(vote); + // Set the headers for the newly created resource + HttpHeaders responseHeaders = new HttpHeaders(); + responseHeaders.setLocation(ServletUriComponentsBuilder. + fromCurrentRequest().path("/{id}").buildAndExpand(vote.getId()).toUri()); + return new ResponseEntity<>(null, responseHeaders, HttpStatus.CREATED); + } + @RequestMapping(value="/polls/votes", method=RequestMethod.GET) + public Iterable getAllVotes() { + return voteRepository.findAll(); + } + @RequestMapping(value="/polls/{pollId}/votes", method=RequestMethod.GET) + public Iterable getVote(@PathVariable Long pollId) { + return voteRepository.findById(pollId); + } + + +} diff --git a/src/main/java/io/zipcoder/tc_spring_poll_application/Repositories/OptionRepository.java b/src/main/java/io/zipcoder/tc_spring_poll_application/Repositories/OptionRepository.java new file mode 100644 index 0000000..7e2e2cb --- /dev/null +++ b/src/main/java/io/zipcoder/tc_spring_poll_application/Repositories/OptionRepository.java @@ -0,0 +1,9 @@ +package io.zipcoder.tc_spring_poll_application.Repositories; + +import io.zipcoder.tc_spring_poll_application.domain.Option; +import org.springframework.data.repository.CrudRepository; +import org.springframework.stereotype.Repository; + +@Repository +public interface OptionRepository extends CrudRepository { +} diff --git a/src/main/java/io/zipcoder/tc_spring_poll_application/Repositories/PollRepository.java b/src/main/java/io/zipcoder/tc_spring_poll_application/Repositories/PollRepository.java new file mode 100644 index 0000000..759be5f --- /dev/null +++ b/src/main/java/io/zipcoder/tc_spring_poll_application/Repositories/PollRepository.java @@ -0,0 +1,11 @@ +package io.zipcoder.tc_spring_poll_application.Repositories; + + +import io.zipcoder.tc_spring_poll_application.domain.Poll; +import org.springframework.data.repository.CrudRepository; +import org.springframework.data.repository.PagingAndSortingRepository; +import org.springframework.stereotype.Repository; + +@Repository +public interface PollRepository extends PagingAndSortingRepository { +} diff --git a/src/main/java/io/zipcoder/tc_spring_poll_application/Repositories/VoteRepository.java b/src/main/java/io/zipcoder/tc_spring_poll_application/Repositories/VoteRepository.java new file mode 100644 index 0000000..ecfd92d --- /dev/null +++ b/src/main/java/io/zipcoder/tc_spring_poll_application/Repositories/VoteRepository.java @@ -0,0 +1,18 @@ +package io.zipcoder.tc_spring_poll_application.Repositories; + +import io.zipcoder.tc_spring_poll_application.domain.Vote; +import org.springframework.data.jpa.repository.Query; +import org.springframework.data.repository.CrudRepository; +import org.springframework.stereotype.Repository; + +@Repository +public interface VoteRepository extends CrudRepository { + @Query(value = "SELECT v.* " + + "FROM Option o, Vote v " + + "WHERE o.POLL_ID = ?1 " + + "AND v.OPTION_ID = o.OPTION_ID", nativeQuery = true) + public Iterable findVotesByPoll(Long pollId); + + Iterable findById(Long pollId); +} + diff --git a/src/main/java/io/zipcoder/tc_spring_poll_application/domain/Option.java b/src/main/java/io/zipcoder/tc_spring_poll_application/domain/Option.java new file mode 100644 index 0000000..82728ca --- /dev/null +++ b/src/main/java/io/zipcoder/tc_spring_poll_application/domain/Option.java @@ -0,0 +1,32 @@ +package io.zipcoder.tc_spring_poll_application.domain; + +import javax.persistence.Column; +import javax.persistence.Entity; +import javax.persistence.GeneratedValue; +import javax.persistence.Id; + +@Entity +public class Option { + @Id + @GeneratedValue + @Column(name = "OPTION_ID") + private Long id; + @Column(name = "OPTION_VALUE") + private String value; + + public Long getId() { + return id; + } + + public String getValue() { + return value; + } + + public void setId(Long id) { + this.id = id; + } + + public void setValue(String value) { + this.value = value; + } +} diff --git a/src/main/java/io/zipcoder/tc_spring_poll_application/domain/Poll.java b/src/main/java/io/zipcoder/tc_spring_poll_application/domain/Poll.java new file mode 100644 index 0000000..5800e2c --- /dev/null +++ b/src/main/java/io/zipcoder/tc_spring_poll_application/domain/Poll.java @@ -0,0 +1,49 @@ +package io.zipcoder.tc_spring_poll_application.domain; + +import org.hibernate.validator.constraints.NotEmpty; + +import javax.persistence.*; +import javax.validation.Valid; +import javax.validation.constraints.Size; +import java.util.Set; + + +@Entity +public class Poll { + @Id + @GeneratedValue + @Column(name = "POLL_ID") +private Long id; + @NotEmpty + @Column(name = "QUESTION") + private String question; + @OneToMany(cascade = CascadeType.ALL) + @JoinColumn(name = "POLL_ID") + @OrderBy + @Size(min=2, max = 6) + Set