1.1、在CustomDAO中添加查询所有客户的方法
public List<Custom> selectAllCustoms() { String sql = "SELECT c.*, u.nick_name FROM t_custom AS c LEFT JOIN t_user AS u ON c.uid=u.id"; ResultSet rs = this.executeQuery(sql); List<Custom> list = new ArrayList<Custom>(); //用于保存所有客户信息 try { while(rs.next()){ Custom custom = new Custom(); custom.setName(rs.getString("name")); custom.setAge(rs.getInt("age")); custom.setSex(rs.getInt("sex")); custom.setPhone(rs.getString("phone")); custom.setWechat(rs.getString("wechat")); custom.setAddr(rs.getString("addr")); custom.setHoby(rs.getString("hoby")); custom.setEmail(rs.getString("email")); custom.setOccupation(rs.getString("occupation")); custom.setUid(rs.getInt("uid")); custom.setNickName(rs.getString("nick_name")); list.add(custom); } this.closeAll(); } catch (SQLException throwables) { throwables.printStackTrace(); } return list; }
1.2、在CustomDAO中添加根据用户id查询所有客户的方法
public List<Custom> selectCustomsByUid(int uid) { String sql = "select * from t_custom where uid="+uid; ResultSet rs = this.executeQuery(sql); List<Custom> list = new ArrayList<Custom>(); //用于保存所有客户信息 try { while(rs.next()){ Custom custom = new Custom(); custom.setName(rs.getString("name")); custom.setAge(rs.getInt("age")); custom.setSex(rs.getInt("sex")); custom.setPhone(rs.getString("phone")); custom.setWechat(rs.getString("wechat")); custom.setAddr(rs.getString("addr")); custom.setHoby(rs.getString("hoby")); custom.setEmail(rs.getString("email")); custom.setOccupation(rs.getString("occupation")); try { custom.setNickName(rs.getString("nick_name")); } catch (SQLException throwables) { throwables.printStackTrace(); } custom.setUid(uid); list.add(custom); } this.closeAll(); } catch (SQLException throwables) { throwables.printStackTrace(); } return list; }
1.3、添加CustomAddServlet
在servlet中获取session中保存的用户,判断当前登录的用户是否是admin,如果是则显示全部用户,否则显示当前登录用户的客户列表
package controller;
import com.alibaba.fastjson.JSON;
import dao.CustomDAO;
import dao.impl.CustomDAOImpl;
import entity.Custom;
import entity.User;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRes