Changeset 4b1c93d

Timestamp:

01/08/21 15:05:58 (4 years ago)

Author:

Vzdra <vladko.zdravkovski@…>

Branches:

master

Children:

277b400

Parents:

336d09e

Message:

added response types to team and company

Location:

src/main/java/it/finki/tinki

Files:

• model/dto/CompanyResponseDTO.java (modified) (1 diff)
• model/dto/TeamResponseDTO.java (modified) (2 diffs)
• web/controller/LoginController.java (modified) (3 diffs)

src/main/java/it/finki/tinki/model/dto/CompanyResponseDTO.java

@@ -11 +11 @@
 public class CompanyResponseDTO extends LoginResponseDTO {
     private Address address;
-    private List<Job> jobs;
-    private List<Internship> internships;
+    private List<JobResponseDTO> jobs;
+    private List<InternshipResponseDTO> internships;
 }

src/main/java/it/finki/tinki/model/dto/TeamResponseDTO.java

@@ -1 +1 @@
 package it.finki.tinki.model.dto;
 
-import it.finki.tinki.model.Jobs.Job;
-import it.finki.tinki.model.Jobs.Project;
 import lombok.Data;
 
@@ -10 +8 @@
 public class TeamResponseDTO extends LoginResponseDTO{
     private int members;
-    private List<Job> jobs;
-    private List<Project> projects;
+    private List<JobResponseDTO> jobs;
+    private List<ProjectResponseDTO> projects;
 }

src/main/java/it/finki/tinki/web/controller/LoginController.java

@@ -32 +32 @@
         this.workService = workService;
     }
-
-    //TODO: ADD JOB/INTERNSHIP RESPONSE TYPE TO COMPANY AND TEAM ---------------------v
 
     @PostMapping(path = "/login")
@@ -90 +88 @@
                 tDto.setMembers(((Team) a1).getMembers());
 
-                tDto.setJobs(this.workService.getAllJobsByAccount(a1.getId()));
-                tDto.setProjects(this.workService.getAllProjectsByAccount(a1.getId()));
+                List<Job> jobs = this.workService.getAllJobsByAccount(a1.getId());
+                List<Project> projects = this.workService.getAllProjectsByAccount(a1.getId());
+
+                jobs.forEach(job -> {
+                    JobResponseDTO dto = new JobResponseDTO(job);
+                    tDto.getJobs().add(dto);
+                });
+
+                projects.forEach(project -> {
+                    ProjectResponseDTO dto = new ProjectResponseDTO(project);
+                    tDto.getProjects().add(dto);
+                });
 
                 return tDto;
@@ -107 +115 @@
                 cDto.setAddress(((Company) a1).getAddress());
 
-                cDto.setJobs(this.workService.getAllJobsByAccount(a1.getId()));
-                cDto.setInternships(this.workService.getAllInternshipsByAccount(a1.getId()));
+                List<Job> jobs = this.workService.getAllJobsByAccount(a1.getId());
+                List<Internship> internships = this.workService.getAllInternshipsByAccount(a1.getId());
+
+                jobs.forEach(job -> {
+                    JobResponseDTO dto = new JobResponseDTO(job);
+                    cDto.getJobs().add(dto);
+                });
+
+                internships.forEach(internship -> {
+                    InternshipResponseDTO dto = new InternshipResponseDTO(internship);
+                    cDto.getInternships().add(dto);
+                });
 
                 return cDto;