基于HTTP的REST软件架构实例

package net.bjstd.rest.cookie; import java.io.IOException; import java.util.HashSet; import java.util.List; import java.util.Random; import java.util.Set; import org.dom4j.Document; import org.dom4j.DocumentException; import org.dom4j.DocumentHelper; import org.restlet.Request; import org.restlet.Response; import org.restlet.data.Cookie; import org.restlet.data.CookieSetting; import org.restlet.data.MediaType; import org.restlet.representation.Representation; import org.restlet.representation.StringRepresentation; import org.restlet.routing.Filter; import org.restlet.util.Series; public class AuthFilter extends Filter { public static final String SESSION_ID = "sessionid"; private Set<String> sessions = new HashSet<String>(); String sessionid = null; @Override public int beforeHandle(Request request, Response response){ Series<Cookie> cookies = request.getCookies(); sessionid = cookies.getFirstValue(SESSION_ID, true); System.out.println("Auth====>"); if(sessionid == null){ System.out.println("start to auth!...1"); //开始认证 //... try { String content = request.getEntity().getText(); System.out.println("entity====>"+content); Document document = DocumentHelper.parseText(content); List list = document.selectNodes("/login"); if(list != null && list.size() != 0){ System.out.println("auth pass!..."); sessionid = this.createSessionID(); sessions.add(sessionid); return SKIP; }else{ System.out.println("auth failure!..."); return SKIP; } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } catch (DocumentException e) { // TODO Auto-generated catch block e.printStackTrace(); } }else{ if(!sessions.contains(sessionid)){ System.out.println("start to auth!...2"); try { String content = request.getEntity().getText(); System.out.println("entity====>"+content); Document document = DocumentHelper.parseText(content); List list = document.selectNodes("/login"); if(list != null && list.size() != 0){ System.out.println("auth pass!..."); sessionid = this.createSessionID(); sessions.add(sessionid); return SKIP; }else{ System.out.println("auth failure!..."); return SKIP; } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } catch (DocumentException e) { // TODO Auto-generated catch block e.printStackTrace(); } }else{ System.out.println("session has exist!..."); } } return CONTINUE; } public String createSessionID(){ Random random = new Random(); random.setSeed(10000); return String.valueOf(random.nextLong()); } @Override public void afterHandle(Request request, Response response){ System.out.println("set sessionid==>"+sessionid); if(sessionid != null){ //设置返回cookie Series<CookieSetting> cookies = response.getCookieSettings(); CookieSetting cookieOne = new CookieSetting(); cookieOne.setName(SESSION_ID); cookieOne.setValue(sessionid); cookies.add(cookieOne); response.setCookieSettings(cookies); }else{ Representation rep = new StringRepresentation("You should login first!...",MediaType.TEXT_PLAIN); response.setEntity(rep); } } }