Changeset 4b1c93d for src/main/java/it/finki/tinki/web

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

File:

• src/main/java/it/finki/tinki/web/controller/LoginController.java (modified) (3 diffs)

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;