使用者模組
要登陸後才能購買,因此我們先寫購買模組.
設計實體
private String id;
private String username;
private String password;
private String email;
private String cellphone;
private String address;
//各種setter、getter
複製程式碼
設計資料庫表
CREATE TABLE user (
id VARCHAR(40) PRIMARY KEY,
username VARCHAR(20) NOT NULL,
cellphone VARCHAR(20) NOT NULL,
address VARCHAR(40) NOT NULL,
email VARCHAR(30),
password VARCHAR(30) NOT NULL
);
複製程式碼
##編寫DAO ##
/**
* 使用者的登入註冊模組
* 1:登陸
* 2:註冊
* 3:根據id查詢具體的使用者
*/
public class UserDaoImpl {
public void register(User user) {
QueryRunner queryRunner = new QueryRunner(Utils2DB.getDataSource());
String sql = "INSERT INTO user (id,username,cellphone,address,email,password) VALUES(?,?,?,?,?,?)";
try {
queryRunner.update(sql, new Object[]{user.getId(),user.getUsername(), user.getCellphone(), user.getAddress(), user.getEmail(), user.getPassword()});
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
public User login(String username, String password) {
QueryRunner queryRunner = new QueryRunner(Utils2DB.getDataSource());
String sql = "SELECT * FROM user WHERE username = ? AND password=?";
try {
return (User) queryRunner.query(sql, new Object[]{username, password}, new BeanHandler(User.class));
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
public User find(String id) {
QueryRunner queryRunner = new QueryRunner(Utils2DB.getDataSource());
String sql = "SELECT * FROM user WHERE id=?";
try {
return (User) queryRunner.query(sql, id, new BeanHandler(User.class));
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
}
複製程式碼
測試DAO
public class UserDemo {
UserDaoImpl userDao = new UserDaoImpl();
@Test
public void add() {
User user = new User();
user.setId("1");
user.setUsername("zhong");
user.setPassword("123");
user.setCellphone("10085");
user.setAddress("廣州蘿崗");
user.setEmail("40368324234234@QQ.com");
userDao.register(user);
}
@Test
public void find() {
String id = "1";
User user = userDao.find(id);
System.out.println(user.getEmail());
}
@Test
public void login() {
String username = "zhong";
String password = "123";
User user = userDao.login(username, password);
System.out.println(user.getAddress());
}
}
複製程式碼
抽取DAO
public interface UserDao {
void register(User user);
User login(String username, String password);
User find(String id);
}
複製程式碼
編寫Service
private UserDao userDao = DaoFactory.getInstance().createDao("zhongfucheng.dao.impl.UserDaoImpl", UserDao.class);
public void registerUser(User user) {
userDao.register(user);
}
public User loginUser(String username,String password) {
return userDao.login(username, password);
}
public User findUser(String id) {
return userDao.find(id);
}
複製程式碼
前臺樣式
- head.jsp
<div id="User">
使用者名稱:<input type="text" name="username">
密碼:<input type="password" name="password">
<button name="login">登陸</button>
<button name="register">註冊</button>
</div>
複製程式碼
- head.css
#body {
position: relative;
}
#user {
position: absolute;
margin-top: 130px;
margin-left: 1364px;
}
複製程式碼
- 效果:
實現登陸註冊功能
當點選登陸按鈕的時候,把資料帶過去給Servlet,讓Servlet呼叫BusinessService方法,實現登陸。註冊同理.....因此,我們需要用到JavaScript程式碼
- head.jsp
<c:if test="${user==null}" >
<div id="User">
使用者名稱:<input type="text" id="username">
密碼:<input type="password" id="password">
<button name="login" onclick="login()">登陸</button>
<button name="register" onclick="register()">註冊</button>
</div>
</c:if>
<c:if test="${user!=null}" >
<div id="User">
歡迎您:${user.username} <a href="${pageContext.request.contextPath}/UserServlet?method=Logout">登出</a>
</div>
</c:if>
複製程式碼
- javaScript程式碼
<script type="text/javascript">
function login() {
//得到輸入框的資料
var username = document.getElementById("username").value;
var password = document.getElementById("password").value;
//跳轉到相對應的Servlet上
window.location.href = "${pageContext.request.contextPath}/UserServlet?method=login&username=" + username + "&password=" + password;
}
function register() {
//跳轉到註冊頁面
window.location.href = "${pageContext.request.contextPath}/client/register.jsp";
}
</script>
複製程式碼
- UserServlet
String method = request.getParameter("method");
BussinessServiceImpl service = new BussinessServiceImpl();
if (method.equals("login")) {
try {
//得到頁面傳遞過來的資料
String username = request.getParameter("username");
String password = request.getParameter("password");
User user = service.loginUser(username, password);
request.getSession().setAttribute("user",user);
request.getRequestDispatcher("/client/head.jsp").forward(request, response);
} catch (Exception e) {
request.setAttribute("message", "登陸失敗了!");
request.getRequestDispatcher("/message.jsp").forward(request, response);
}
} else if (method.equals("register")) {
try {
//得到JSP傳遞過來的資料,封裝成Bean物件
User user = WebUtils.request2Bean(request, User.class);
user.setId(WebUtils.makeId());
service.registerUser(user);
request.setAttribute("message", "註冊成功了!");
} catch (Exception e) {
e.printStackTrace();
request.setAttribute("message", "註冊失敗了!");
}
request.getRequestDispatcher("/message.jsp").forward(request, response);
} else if (method.equals("Logout")) {
//銷燬session
request.getSession().invalidate();
//回到首頁
request.getRequestDispatcher("/client/head.jsp").forward(request, response);
}
複製程式碼
購買模組
在顯示圖書的時候,順便新增購買的超連結
<li><a href="#">購買</a></li>
複製程式碼
設計購物車實體
如果不清楚為什麼這樣設計,可參考我之前的博文:blog.csdn.net/hon_3y/arti…
- Cart實體
public class Cart {
private Map<String, CartItem> map = new HashMap<>();
private double price;
//提供把商品新增到購物的功能
public void addBook2Cart(Book book) {
//得到對應的購物項
CartItem cartItem = map.get(book.getId());
//如果是null,說明購物車還沒有該購物項
if (cartItem == null) {
cartItem = new CartItem();
cartItem.setQuantity(1);
cartItem.setBook(book);
cartItem.setPrice(book.getPrice());
//把購物項加到購物車中
map.put(book.getId(), cartItem);
} else {
//如果購物車有該購物項了,那麼將購物項的數量+1
cartItem.setQuantity(cartItem.getQuantity() + 1);
}
}
//購物車的價錢是購物項價錢的總和
public double getPrice() {
double totalPrice = 0;
for (Map.Entry<String, CartItem> me : map.entrySet()) {
CartItem cartItem = me.getValue();
totalPrice += cartItem.getPrice();
}
return totalPrice;
}
public Map<String, CartItem> getMap() {
return map;
}
public void setMap(Map<String, CartItem> map) {
this.map = map;
}
public void setPrice(double price) {
this.price = price;
}
}
複製程式碼
設計購物項實體
public class CartItem {
private Book book;
private double price;
private int quantity;
public double getPrice() {
return this.book.getPrice() * this.quantity;
}
public void setPrice(double price) {
this.price = price;
}
public Book getBook() {
return book;
}
public void setBook(Book book) {
this.book = book;
}
public int getQuantity() {
return quantity;
}
public void setQuantity(int quantity) {
this.quantity = quantity;
}
}
複製程式碼
處理使用者想要買的書籍Servlet
<li><a href="${pageContext.request
.contextPath}/BuyServlet?book_id=${book.id}">購買</a></li>
複製程式碼
- BuyServlet
BussinessServiceImpl service = new BussinessServiceImpl();
//先檢查該使用者是否登陸了。
User user = (User) request.getSession().getAttribute("user");
if (user == null) {
request.setAttribute("message", "您還沒登陸,請登陸了再來購買");
request.getRequestDispatcher("/message.jsp").forward(request, response);
return ;
}
//如果登陸了...
//得到該使用者的購物車
Cart cart = (Cart) request.getSession().getAttribute("cart");
if (cart == null) {
cart = new Cart();
request.getSession().setAttribute("cart", cart);
}
//得到使用者想買的書籍
String book_id = request.getParameter("book_id");
Book book = service.findBook(book_id);
//把書籍新增到購物車中
service.buyBook(cart, book);
request.setAttribute("message", "該商品已新增到購物車中");
request.getRequestDispatcher("/message.jsp").forward(request,response);
複製程式碼
提供顯示購物車商品的Servlet
//先判斷該使用者是否登陸了
User user = (User) request.getSession().getAttribute("user");
if (user == null) {
request.setAttribute("message", "您還沒有登陸呢!");
request.getRequestDispatcher("/message.jsp").forward(request, response);
return;
}
//如果登陸了.....
Cart cart = (Cart) request.getSession().getAttribute("cart");
//把該使用者的購物車給JSP頁面顯示
request.setAttribute("cart", cart);
request.getRequestDispatcher("/client/listCart.jsp").forward(request, response);
複製程式碼
顯示購物車的JSP頁面
<c:if test="${empty(cart.map)}">
您還沒有購買過任何商品哦!!!
</c:if>
<table border="1px">
<c:if test="${!empty(cart.map)}">
<h1>您購物車下有如下的商品:</h1><br>
<tr>
<td>書名:</td>
<td>作者:</td>
<td>數量:</td>
<td>價錢:</td>
</tr>
<c:forEach items="${cart.map}" var="cartItme">
<tr>
<td>${cartItme.value.book.name}</td>
<td>${cartItme.value.book.author}</td>
<td>${cartItme.value.quantity}</td>
<td>${cartItme.value.price}</td>
</tr>
</c:forEach>
</c:if>
</table>
複製程式碼
效果:
訂單模組
在前臺使用者介面中,當使用者要把購物車付款時,應該提供生成訂單的超連結....
設計訂單實體
訂單應該包含id,收貨人資訊,下單的時間,訂單的總價,訂單的狀態【有無發貨】..而不應該包含商品的資訊的。商品的資訊用一個專門的”訂單項“來表示
一個訂單對應多個訂單項,這是一對多的關係!
private String id;
//下單的時間、日期、狀態
private Date date;
private double price;
private boolean state;
//一個使用者可以有多個訂單,把使用者記住
private String user_id;
//一個訂單中有多個訂單項
private Set<OrderItem> items = new HashSet<>();
//各種的setter和getter
複製程式碼
設計訂單項實體
private String id;
//一本書對應多個訂單項,訂單項一定是由書組成,記住書
private String book_id;
private double price;
private int quantity;
//各種的setter和getter
複製程式碼
設計資料庫表
- 訂單表
mysql不能建立名為”order”的表,後邊加個s就可以
CREATE TABLE orders (
id VARCHAR(40) PRIMARY KEY,
date DATE NOT NULL,
user_id VARCHAR(40) NOT NULL,
state BOOLEAN,
price DOUBLE,
CONSTRAINT user_id_FK FOREIGN KEY (user_id) REFERENCES user (id)
);
複製程式碼
- 訂單項表:
CREATE TABLE orderItem (
id VARCHAR(40) PRIMARY KEY,
price DOUBLE,
quantity INT,
order_id VARCHAR(40) ,
book_id VARCHAR(40) ,
CONSTRAINT order_id_FK FOREIGN KEY (order_id) REFERENCES orders (id),
CONSTRAINT book_id_FK FOREIGN KEY (book_id) REFERENCES book (id)
);
複製程式碼
- 表之間的結構:
設計Dao
public class OrderDaoImpl implements zhongfucheng.dao.OrderDao {
@Override
public void addOrder(Order order) {
QueryRunner queryRunner = new QueryRunner(Utils2DB.getDataSource());
String sql1 = "INSERT INTO orders(id,ordertime,user_id,state,price) VALUES(?,?,?,?,?)";
try {
//訂單的基本資訊
queryRunner.update(sql1, new Object[]{order.getId(), order.getOrdertime(), order.getUser_id(), order.isState(), order.getPrice()});
//訂單項的資訊
String sql2 = "INSERT INTO orderItem(id,price,quantity,order_id,book_id) VALUES(?,?,?,?,?)";
Set<OrderItem> items = order.getItems();
for (OrderItem item : items) {
queryRunner.update(sql2, new Object[]{item.getId(), item.getPrice(), item.getQuantity(), item.getOrder_id(), item.getBook_id()});
}
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
@Override
public Order findOrder(String id) {
QueryRunner queryRunner = new QueryRunner(Utils2DB.getDataSource());
Order order;
try {
//找出訂單的基本資訊
String sql = "SELECT * FROM orders WHERE id=?";
order = (Order) queryRunner.query(sql, new BeanHandler(Order.class), new Object[]{id});
//找出訂單的所有訂單項
String sql2 = "SELECT * FROM orderItem WHERE order_id=?";
List<OrderItem> list = (List<OrderItem>) queryRunner.query(sql2, new BeanListHandler(OrderItem.class), new Object[]{order.getId()});
System.out.println("這是資料庫拿到的list集合:"+list.size());
//將所有訂單項裝到訂單裡邊
order.getItems().addAll(list);
System.out.println("這是資料庫拿到的"+order.getItems().size());
//找出該訂單是屬於哪一個使用者的
String sql3 = "SELECT * FROM orders o,user u WHERE o.user_id=u.id AND o.id=? ";
User user = (User) queryRunner.query(sql3, new BeanHandler(User.class), new Object[]{order.getId()});
order.setUser_id(user.getId());
return order;
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
//更新訂單的狀態
public void updateState(String id) {
QueryRunner queryRunner = new QueryRunner(Utils2DB.getDataSource());
String sql = "UPDATE orders SET state=? WHERE id=?";
try {
queryRunner.update(sql, new Object[]{true, id});
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
//檢視已經發貨或沒發貨的訂單資訊
public List<Order> getAllOrder(boolean state) {
QueryRunner queryRunner = new QueryRunner(Utils2DB.getDataSource());
String sql = "SELECT * FROM orders WHERE state=? ";
try {
return (List<Order>) queryRunner.query(sql, new BeanListHandler(Order.class), new Object[]{state});
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
//通過使用者的id查詢使用者的訂單,可能不止一個
public List<Order> findUserOrder(String user_id) {
QueryRunner queryRunner = new QueryRunner(Utils2DB.getDataSource());
String sql = "SELECT * FROM orders WHERE user_id=? ";
try {
return List<Order> queryRunner.query(sql, new BeanHandler(Order.class), new Object[]{user_id});
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
}
複製程式碼
二次更新
在編寫dao的時候,尤其是Add方法。它是將所有資料都封裝到Order物件上,然後取出資料,把資料插入到資料表中
- 其實,我們的Order和OrderItem的操作可以分開。OrderItem也可以另外編寫一個Dao,那麼我們在插入完Order物件之後,得到Order物件返回的主鍵,再呼叫OrderItemDao的方法來插入OrderItem的資料,這樣我覺得會讓程式碼清晰一些。
- 在OrderItemDao中接收的是一個List,因為我們一個訂單會對應多個訂單項。
抽取成DAO介面
public interface OrderDao {
void addOrder(Order order);
Order findOrder(String id);
List<Order> getAllOrder(boolean state);
void updateState(String user_id);
List<Order> findUserOrder(String user_id);
}
複製程式碼
BussinessService
private OrderDao orderDao = DaoFactory.getInstance().createDao("zhongfucheng.dao.impl.OrderDaoImpl", OrderDao.class);
public void createOrder(Cart cart, User user) {
//訂單的基本資訊
String order_id = WebUtils.makeId();
Order order = new Order();
order.setId(order_id);
order.setPrice(cart.getPrice());
order.setOrdertime(new Date());
order.setState(false);
order.setUser_id(user.getId());
//訂單項的基本資訊
//得到每個購物項,購物項就作為訂單項
for (Map.Entry<String, CartItem> me : cart.getMap().entrySet()) {
OrderItem orderItem = new OrderItem();
CartItem cartItem = me.getValue();
orderItem.setId(WebUtils.makeId());
orderItem.setPrice(cartItem.getPrice());
orderItem.setBook_id(cartItem.getBook().getId());
orderItem.setQuantity(cartItem.getQuantity());
orderItem.setOrder_id(order_id);
order.getItems().add(orderItem);
}
orderDao.addOrder(order);
}
public Order findOrder(String user_id) {
return orderDao.findOrder(user_id);
}
public List<Order> getAllOrder(boolean state) {
return orderDao.getAllOrder(state);
}
public void sendOutOrder(String id) {
orderDao.updateState(id);
}
public List<Order> findUserOrder(String user_id) {
return orderDao.findUserOrder(user_id);
}
複製程式碼
生成訂單的Servlet
BussinessServiceImpl service = new BussinessServiceImpl();
//檢查該使用者的購物車是否有商品
Cart cart = (Cart) request.getSession().getAttribute("cart");
if (cart == null) {
request.setAttribute("message", "您購物車沒有商品,無法生成訂單");
request.getRequestDispatcher("/message.jsp").forward(request, response);
return;
}
//如果有商品,得到當前使用者
User user = (User) request.getSession().getAttribute("user");
service.createOrder(cart, user);
request.setAttribute("message", "訂單已經生成了,準備好錢來收貨把");
request.getRequestDispatcher("/message.jsp").forward(request, response);
return;
複製程式碼
使用者查詢自己的訂單Servlet
<a href="${pageContext.request.contextPath}/LookOrder" target="body">檢視訂單</a>
複製程式碼
BussinessServiceImpl service = new BussinessServiceImpl();
//檢查該使用者是否登陸了
User user = (User) request.getSession().getAttribute("user");
if (user == null) {
request.setAttribute("message", "您還沒登陸,等您登陸了再來看把");
request.getRequestDispatcher("/message.jsp").forward(request, response);
return;
}
//使用者登陸了!
Order order = service.findUserOrder(user.getId());
//交給相對應的JSP 顯示
request.setAttribute("order", order);
request.setAttribute("user",user);
request.getRequestDispatcher("/client/listOrder.jsp").forward(request, response);
return ;
複製程式碼
顯示訂單資料的JSP
<c:if test="${order==null}">
您還沒有下過任何訂單!!
</c:if>
<c:if test="${order!=null}">
<table border="1px">
<tr>
<td>下單人:</td>
<td>訂單時間</td>
<td>訂單狀態</td>
<td>訂單價錢</td>
</tr>
<tr>
<td>${user.username}</td>
<td>${order.ordertime}</td>
<td>${order.state==false?"未發貨":"已發貨"}</td>
<td>${order.price}</td>
</tr>
</table>
</c:if>
複製程式碼
效果:
後臺查詢訂單的狀況Servlet
<a href="${pageContext.request.contextPath}/OrderServlet?state=false" target="body">待處理訂單</a><br>
<a href="${pageContext.request.contextPath}/OrderServlet?state=true" target="body">已發貨訂單</a><br>
複製程式碼
BussinessServiceImpl service = new BussinessServiceImpl();
String state = request.getParameter("state");
if (state.equals("true")) {
List<Order> list = service.getAllOrder(true);
request.setAttribute("list",list);
} else if (state.equals("false")) {
List<Order> list = service.getAllOrder(false);
request.setAttribute("list", list);
}
request.getRequestDispatcher("/background/listOrder.jsp").forward(request, response);
複製程式碼
顯示訂單狀況的JSP
<c:if test="${empty(list)}">
還沒有任何訂單哦!
</c:if>
<c:if test="${!empty(list)}">
<table border="1px">
<tr>
<td>下單人:</td>
<td>訂單時間</td>
<td>訂單狀態</td>
<td>訂單價錢</td>
<td>操作</td>
</tr>
<c:forEach items="${list}" var="order">
<tr>
<td>${order.user_id}</td>
<td>${order.ordertime}</td>
<td>${order.state==false?"未發貨":"已發貨"}</td>
<td>${order.price}</td>
<td>
<a href="${pageContext.request.contextPath}/orderItemServlet?order_id=${order.id}">檢視詳細資訊</a>
<a href="#">刪除</a>
</td>
</tr>
</c:forEach>
</table>
</c:if>
複製程式碼
檢視具體訂單的詳細資訊Servlet
BussinessServiceImpl service = new BussinessServiceImpl();
//得到使用者想要檢視詳細資訊的表單
String order_id = request.getParameter("order_id");
Order order = service.findOrder(order_id);
//將該order物件給相對應的JSP顯示
request.setAttribute("order", order);
request.getRequestDispatcher("/background/listDetail.jsp").forward(request, response);
複製程式碼
檢視具體訂單的詳細資訊JSP
<table border="1px">
<tr>
<td>書籍的編號</td>
<td>價錢</td>
<td>數量</td>
<td>操作</td>
</tr>
<c:forEach items="${order.items}" var="item">
<tr>
<td>${item.book_id}</td>
<td>${item.price}</td>
<td>${item.quantity}</td>
<td><a href="${pageContext.request.contextPath}/SendOutServlet?id=${order.id}">發貨</a></td>
</tr>
</c:forEach>
</table>
複製程式碼
處理髮貨的Servlet##
BussinessServiceImpl service = new BussinessServiceImpl();
String id = request.getParameter("id");
service.sendOutOrder(id);
request.setAttribute("message", "已發貨!");
request.getRequestDispatcher("/message.jsp").forward(request, response);
return;
複製程式碼
效果:
新增許可權控制
目前為止,我們已經學習了動態代理技術和註解技術了。於是我們想要為之前的bookStore專案新增許可權控制.....
只有使用者有許可權的時候,後臺管理才可以進行相對應的操作.....
實現思路
之前我們做許可權管理系統的時候,是根據使用者請求的URI來判斷該連結是否需要許可權的。這次我們使用動態代理的技術和註解來判斷:使用者呼叫該方法時,檢查該方法是否需要許可權...
根據MVC模式,我們在web層都是呼叫service層來實現功能的。那麼我們具體的思路是這樣的:
- web層呼叫service層的時候,得到的並不是ServiceDao物件,而是我們的代理物件
- 在service層中的方法新增註解,如果方法上有註解,那麼說明呼叫該方法需要許可權...
- 當web層呼叫代理物件方法的時候,代理物件會判斷該方法是否需要許可權,再給出相對應的提示....
設計實體、資料庫表
上次我們做的許可權管理系統是引入了角色這個概念的,這次主要為了練習動態代理和註解技術,就以簡單為主,不引入角色這個實體。直接是使用者和許可權之間的關係了。
Privilege實體
public class Privilege {
private String id ;
private String name;
public String getId() {
return id;
}
public void setId(String id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
}
複製程式碼
資料庫表
- privilege表
CREATE TABLE privilege (
id VARCHAR(40) PRIMARY KEY,
name VARCHAR(40)
);
複製程式碼
privilege和user是多對多的關係,於是使用第三方表來維護他們的關係
- user_privilege表
CREATE TABLE user_privilege (
privilege_id VARCHAR(40),
user_id VARCHAR(40),
PRIMARY KEY (privilege_id, user_id),
CONSTRAINT privilege_id_FK FOREIGN KEY (privilege_id) REFERENCES privilege(id),
CONSTRAINT user_id_FK1 FOREIGN KEY (user_id) REFERENCES user(id)
);
複製程式碼
新增測試資料
為了方便,直接新增資料了。就不寫詳細的DAO了。
- 在資料庫中新增了兩個許可權
- 為id為1的user新增了兩個許可權
編寫DAO
後面在動態代理中,我們需要檢查該使用者是否有許可權...那麼就必須查詢出該使用者擁有的哪些許可權。再看看使用者有沒有相對應的許可權
//查詢使用者的所有許可權
public List<Privilege> findUserPrivilege(String user_id) {
QueryRunner queryRunner = new QueryRunner(Utils2DB.getDataSource());
String sql = "SELECT p.* FROM privilege p, user_privilege up WHERE p.id = up.privilege_id AND up.user_id = ?";
try {
return (List<Privilege>) queryRunner.query(sql, new Object[]{user_id}, new BeanListHandler(Privilege.class));
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
複製程式碼
抽取到介面上
List<Privilege> findUserPrivilege(String user_id);
複製程式碼
註解模組
- 編寫註解
@Retention(RetentionPolicy.RUNTIME)
public @interface permission {
String value();
}
複製程式碼
- 在Service層方法中需要許可權的地方新增註解CategoryServiceImpl
@permission("新增分類")
/*新增分類*/
public void addCategory(Category category) {
categoryDao.addCategory(category);
}
/*查詢分類*/
public void findCategory(String id) {
categoryDao.findCategory(id);
}
@permission("查詢分類")
/*檢視分類*/
public List<Category> getAllCategory() {
return categoryDao.getAllCategory();
}
複製程式碼
抽取Service
把Service的方法抽取成ServiceDao。在Servlet中,也是通過ServiceFactory來得到Service的物件【和DaoFactory是類似的】
CategoryService
@permission("新增分類")
/*新增分類*/ void addCategory(Category category);
/*查詢分類*/
void findCategory(String id);
@permission("查詢分類")
/*檢視分類*/ List<Category> getAllCategory();
複製程式碼
ServiceFactory
public class ServiceDaoFactory {
private static final ServiceDaoFactory factory = new ServiceDaoFactory();
private ServiceDaoFactory() {
}
public static ServiceDaoFactory getInstance() {
return factory;
}
//需要判斷該使用者是否有許可權
public <T> T createDao(String className, Class<T> clazz, final User user) {
System.out.println("新增分類進來了!");
try {
//得到該類的型別
final T t = (T) Class.forName(className).newInstance();
//返回一個動態代理物件出去
return (T) Proxy.newProxyInstance(ServiceDaoFactory.class.getClassLoader(), t.getClass().getInterfaces(), new InvocationHandler() {
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws NoSuchMethodException, InvocationTargetException, IllegalAccessException, PrivilegeException {
//判斷使用者呼叫的是什麼方法
String methodName = method.getName();
System.out.println(methodName);
//得到使用者呼叫的真實方法,注意引數!!!
Method method1 = t.getClass().getMethod(methodName,method.getParameterTypes());
//檢視方法上有沒有註解
permission permis = method1.getAnnotation(permission.class);
//如果註解為空,那麼表示該方法並不需要許可權,直接呼叫方法即可
if (permis == null) {
return method.invoke(t, args);
}
//如果註解不為空,得到註解上的許可權
String privilege = permis.value();
//設定許可權【後面通過它來判斷使用者的許可權有沒有自己】
Privilege p = new Privilege();
p.setName(privilege);
//到這裡的時候,已經是需要許可權了,那麼判斷使用者是否登陸了
if (user == null) {
//這裡丟擲的異常是代理物件丟擲的,sun公司會自動轉換成執行期異常丟擲,於是在Servlet上我們根據getCause()來判斷是不是該異常,從而做出相對應的提示。
throw new PrivilegeException("對不起請先登陸");
}
//執行到這裡使用者已經登陸了,判斷使用者有沒有許可權
Method m = t.getClass().getMethod("findUserPrivilege", String.class);
List<Privilege> list = (List<Privilege>) m.invoke(t, user.getId());
//看下許可權集合中有沒有包含方法需要的許可權。使用contains方法,在Privilege物件中需要重寫hashCode和equals()
if (!list.contains(p)) {
//這裡丟擲的異常是代理物件丟擲的,sun公司會自動轉換成執行期異常丟擲,於是在Servlet上我們根據getCause()來判斷是不是該異常,從而做出相對應的提示。
throw new PrivilegeException("您沒有許可權,請聯絡管理員!");
}
//執行到這裡的時候,已經有許可權了,所以可以放行了
return method.invoke(t, args);
}
});
} catch (Exception e) {
new RuntimeException(e);
}
return null;
}
}
複製程式碼
PrivilegeExcetption
當使用者沒有登陸或者沒有許可權的時候,我們應該給使用者一些友好的提示....於是我們自定義了PrivilegeException
public class PrivilegeException extends Exception {
public PrivilegeException() {
super();
}
public PrivilegeException(String message) {
super(message);
}
public PrivilegeException(String message, Throwable cause) {
super(message, cause);
}
public PrivilegeException(Throwable cause) {
super(cause);
}
}
複製程式碼
我們繼承的是Exception,通過方法名丟擲去。但是我們是通過代理物件呼叫方法的,於是sun公司的策略就是把它們轉換成執行期異常丟擲去。
因此,我們就在Servlet上得到異常,再給出友好的提示。。
效果:
- 沒有登陸的時候:
- 登陸了,但是沒有相對應的許可權的時候
- 登陸了,並且有許可權
要點總結
該許可權控制是十分優雅的,只要我在Service層中新增一個註解...那麼當web層呼叫該方法的時候就需要判斷使用者有沒有該許可權....
- 外界呼叫Service層的方法是代理呼叫invoke()方法,我們在invoke()方法可以對其進行增強!
- invoke()方法內部就是在查詢呼叫該方法上有沒有註解,如果沒有註解,就可以直接呼叫。如果有註解,那麼就得到註解的資訊,判斷該使用者有沒有許可權來訪問這個方法
- 在反射具體方法的時候,必須記得要給出相對應的引數!
- 在invoke()方法丟擲的編譯時期異常,java會自動轉換成執行期異常進行丟擲...
- 使用contains()方法時,就要重寫該物件的hashCode()和equals()