用户删除功能

黑马训练营

 

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Scanner;

import cn.model.User;
import cn.service.UserService;

public class Test01 {

 public User deInput(){
  User us=new User();
  while(true){
  System.out.println("1.ID删除     2.名字删除    3.号码删除     4.退出");
      Scanner st = new Scanner(System.in);
      int a = st.nextInt();
     
      if(a==1){
       String ss = st.next();
       us.setId(ss);
       us.setName(null);
       us.setTel(null);
       break;
      }
      else if(a==2){
       String ss = st.next();
       us.setId(null);
       us.setName(ss);
       us.setTel(null);
       break;
      }
      else if(a==3){
       String ss = st.next();
       us.setId(null);
       us.setName(null);
       us.setTel(ss);
       break;
       
      }
      else if(a==4){
       
       break;
       
      }
      else{
       System.out.print("请重新输入：");
       continue;
      }
  }
  return us;

 }
 public User adInput(){
  User us=new User();
  Scanner st = new Scanner(System.in);
  System.out.print("ID:");
  us.setId(st.next());
  System.out.print("name:");
  us.setName(st.next());
  System.out.print("sex:");
  us.setSex(st.next());
  System.out.print("tel:");
  us.setTel(st.next());
  return us;

 }
 public static void main(String[] args) {
  
     UserService us = new UserService();
     
    while(true){
      
      
      System.out.println("1.添加     2.查询     3.删除     4.退出");
       Scanner sc = new Scanner(System.in);
       int i = sc.nextInt();
      
      
      if(i==1){
       User user = new User();
       Test01 tc = new Test01();
       us.addUser(tc.adInput());
       System.out.println("添加成功！");
       continue;
      }
      else if(i==2){
       ArrayList<User> userList =us.getUsers();
       for(User u:userList)
       {
        
        System.out.println(u.getId()+"    "+u.getName());
       }
       System.out.println("退出请按 0");
          Scanner sb = new Scanner(System.in);
          int j = sb.nextInt();
          while(j==0);
          continue;
      }
      else if(i==3){
       Test01 tt = new Test01();
       User user = tt.deInput();
       us.delete(user);
      continue;
       
      }
      else if(i==4){
       break;
       
      }
      else{
       System.out.print("请重新输入：");
       continue;
      }
     }

 }

}