Advertisement
Nik_Perepelov

ыаыа жаба

Dec 27th, 2021
1,103
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.38 KB | None | 0 0
  1. package com.example.web;
  2.  
  3. import com.example.db.DatabaseService;
  4. import org.springframework.boot.Banner;
  5. import org.springframework.stereotype.Controller;
  6. import org.springframework.ui.Model;
  7. import org.springframework.web.bind.annotation.*;
  8. import org.springframework.web.multipart.MultipartFile;
  9.  
  10. import java.io.IOException;
  11. import java.io.InputStream;
  12. import java.sql.SQLException;
  13. import java.util.ArrayList;
  14. import java.util.List;
  15. import java.util.Objects;
  16. import java.util.Scanner;
  17.  
  18. @Controller
  19. public class WebController {
  20.  
  21.     @GetMapping("/csv_upload")
  22.     public String csvUpload() {
  23.  
  24.         return "csv_form";
  25.     }
  26.  
  27.     @PostMapping("/csv_upload")
  28.     public String csvUploadHandle(Model model, @RequestParam("file") MultipartFile file) throws IOException, SQLException {
  29.  
  30.         InputStream inputStream = file.getInputStream();
  31.         Scanner s = new Scanner(inputStream).useDelimiter("\n");
  32.         List<String[]> result = new ArrayList<>();
  33.         String file_name = file.getOriginalFilename();
  34.  
  35.  
  36.  
  37.         if (Objects.equals(file_name, "tr_mcc_codes.csv")
  38.                 || Objects.equals(file_name, "tr_types.csv")) {
  39.             while (s.hasNext()) {
  40.                 String[] curr_string = s.next().split(";", -1);
  41.                 result.add(curr_string);
  42.             }
  43.         } else {
  44.             while (s.hasNext()) {
  45.                 String[] curr_string = s.next().split(",", -1);
  46.                 result.add(curr_string);
  47.             }
  48.         }
  49.  
  50.         DatabaseService databaseService = DatabaseService.getDatabaseService();
  51.         databaseService.insertCSV(result, file_name.substring(0, file_name.length() - 4));
  52.  
  53.         model.addAttribute("message", "Загрузка успешна!");
  54.         return "csv_form";
  55.     }
  56.  
  57.     @GetMapping("/request")
  58.     public String requestResults() {
  59.         return "request_results";
  60.     }
  61.  
  62.     @PostMapping("/request")
  63.     public String returnResults(@ModelAttribute("string_to_find") String string_to_find, Model model) {
  64.         System.out.println(string_to_find);
  65.  
  66.         DatabaseService databaseService = DatabaseService.getDatabaseService();
  67.         var result = databaseService.findTransactions(string_to_find);
  68.         model.addAttribute("result", result);
  69.         if (!result.isEmpty()){
  70.             model.addAttribute("flag", true);
  71.         }
  72.  
  73.         return "request_results";
  74.     }
  75. }
  76.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement