SpringMVC整合Shiro

 

SpringMVC-3.2.4和Shiro-1.2.2

首先是web.xml

<?xml version="1.0" encoding="UTF-8"?>
<web-app version="2.5"
	xmlns="http://java.sun.com/xml/ns/javaee"
	xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
	xsi:schemaLocation="http://java.sun.com/xml/ns/javaee
	http://java.sun.com/xml/ns/javaee/web-app_2_5.xsd">
	<!-- 指定Spring的配置文件 -->
	<!-- 否則Spring會默認從WEB-INF下尋找配置文件,contextConfigLocation屬性是Spring內部固定的 -->
	<!-- 通過ContextLoaderListener的父類ContextLoader的第120行發現CONFIG_LOCATION_PARAM固定爲contextConfigLocation -->
	<context-param>
		<param-name>contextConfigLocation</param-name>
		<param-value>classpath:applicationContext.xml</param-value>
	</context-param>

	<!-- 防止發生java.beans.Introspector內存泄露,應將它配置在ContextLoaderListener的前面 -->
	<!-- 詳細描述見http://blog.csdn.net/jadyer/article/details/11991457 -->
	<listener>
		<listener-class>org.springframework.web.util.IntrospectorCleanupListener</listener-class>
	</listener>
	
	<!-- 實例化Spring容器 -->
	<!-- 應用啓動時,該監聽器被執行,它會讀取Spring相關配置文件,其默認會到WEB-INF中查找applicationContext.xml -->
	<listener>
		<listener-class>org.springframework.web.context.ContextLoaderListener</listener-class>
	</listener>

	<!-- 解決亂碼問題 -->
	<filter>
		<filter-name>SpringEncodingFilter</filter-name>
		<filter-class>org.springframework.web.filter.CharacterEncodingFilter</filter-class>
		<init-param>
			<param-name>encoding</param-name>
			<param-value>UTF-8</param-value>
		</init-param>
		<init-param>
			<param-name>forceEncoding</param-name>
			<param-value>true</param-value>
		</init-param>
	</filter>
	<filter-mapping>
		<filter-name>SpringEncodingFilter</filter-name>
		<url-pattern>/*</url-pattern>
	</filter-mapping>
	
	<!-- 配置Shiro過濾器,先讓Shiro過濾系統接收到的請求 -->
	<!-- 這裏filter-name必須對應applicationContext.xml中定義的<bean id="shiroFilter"/> -->
	<!-- 使用[/*]匹配所有請求,保證所有的可控請求都經過Shiro的過濾 -->
	<!-- 通常會將此filter-mapping放置到最前面(即其他filter-mapping前面),以保證它是過濾器鏈中第一個起作用的 -->
	<filter>
		<filter-name>shiroFilter</filter-name>
		<filter-class>org.springframework.web.filter.DelegatingFilterProxy</filter-class>
		<init-param>
			<!-- 該值缺省爲false,表示生命週期由SpringApplicationContext管理,設置爲true則表示由ServletContainer管理 -->
			<param-name>targetFilterLifecycle</param-name>
			<param-value>true</param-value>
		</init-param>
	</filter>
	<filter-mapping>
		<filter-name>shiroFilter</filter-name>
		<url-pattern>/*</url-pattern>
	</filter-mapping>

	<!-- SpringMVC核心分發器 -->
	<servlet>
		<servlet-name>SpringMVC</servlet-name>
		<servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>
		<init-param>
			<param-name>contextConfigLocation</param-name>
			<param-value>classpath:applicationContext.xml</param-value>
		</init-param>
		<load-on-startup>1</load-on-startup>
	</servlet>
	<servlet-mapping>
		<servlet-name>SpringMVC</servlet-name>
		<url-pattern>/</url-pattern>
	</servlet-mapping>

	<!-- 默認歡迎頁 -->
	<!-- Servlet2.5中可直接在此處執行Servlet應用,如<welcome-file>servlet/InitSystemParamServlet</welcome-file> -->
	<!-- 這裏使用了SpringMVC提供的<mvc:view-controller>標籤,實現了首頁隱藏的目的,詳見applicationContext.xml -->
	<!-- 
	<welcome-file-list>
		<welcome-file>login.jsp</welcome-file>
	</welcome-file-list>
	 -->
	
	<error-page>
		<error-code>405</error-code>
		<location>/WEB-INF/405.html</location>
	</error-page>
	<error-page>
		<error-code>404</error-code>
		<location>/WEB-INF/404.jsp</location>
	</error-page>
	<error-page>
		<error-code>500</error-code>
		<location>/WEB-INF/500.jsp</location>
	</error-page>
	<error-page>
		<error-code>java.lang.Throwable</error-code>
		<location>/WEB-INF/500.jsp</location>
	</error-page>
</web-app>

下面是用於顯示Request method 'GET' not supported的//WebRoot//WEB-INF//405.html

<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN">
<html>
	<head>
		<title>405.html</title>
		<meta http-equiv="content-type" content="text/html; charset=UTF-8">
	</head>
	<body>
		<font color="blue">
			Request method 'GET' not supported
			<br/><br/>
			The specified HTTP method is not allowed for the requested resource.
		</font>
	</body>
</html>


下面是允許匿名用戶訪問的//WebRoot//login.jsp

<%@ page language="java" pageEncoding="UTF-8"%>

<script type="text/javascript">
<!--
function reloadVerifyCode(){
	document.getElementById('verifyCodeImage').setAttribute('src', '${pageContext.request.contextPath}/mydemo/getVerifyCodeImage');
}
//-->
</script>

<div style="color:red; font-size:22px;">${message_login}</div>

<form action="<%=request.getContextPath()%>/mydemo/login" method="POST">
	姓名:<input type="text" name="username"/><br/>
	密碼:<input type="text" name="password"/><br/>
	驗證:<input type="text" name="verifyCode"/>
		   
		 <img id="verifyCodeImage" onclick="reloadVerifyCode()" src="<%=request.getContextPath()%>/mydemo/getVerifyCodeImage"/><br/>
	<input type="submit" value="確認"/>
</form>


下面是用戶登錄後顯示的//WebRoot//main.jsp

<span style="font-size:14px;"><%@ page language="java" pageEncoding="UTF-8"%>
</span><span style="font-size:12px;">普通用戶可訪問<a href="<%=request.getContextPath()%>/mydemo/getUserInfo" target="_blank">用戶信息頁面</a>
<br/>
<br/>
管理員可訪問<a href="<%=request.getContextPath()%>/admin/listUser.jsp" target="_blank">用戶列表頁面</a>
<br/>
<br/>
<a href="<%=request.getContextPath()%>/mydemo/logout" target="_blank">Logout</a></span>


下面是隻有管理員才允許訪問的//WebRoot//admin//listUser.jsp

<pre class="plain" name="code"><span style="font-size:12px;"><%@ page language="java" pageEncoding="UTF-8"%></span>
<span style="font-size:12px;">This is listUser.jsp
<br/>
<br/>
<a href="<%=request.getContextPath()%>/mydemo/logout" target="_blank">Logout</a></span>


下面是普通的登錄用戶所允許訪問的//WebRoot//user//info.jsp

<span style="font-size:12px;"><%@ page language="java" pageEncoding="UTF-8"%>
當前登錄的用戶爲${currUser}
<br/>
<br/>
<a href="<%=request.getContextPath()%>/mydemo/logout" target="_blank">Logout</a></span>

下面是//src//log4j.properties

<span style="font-size:12px;">#use Root for GobalConfig
log4j.rootLogger=DEBUG,CONSOLE

log4j.logger.java.sql=DEBUG
log4j.logger.org.apache.shiro=DEBUG
log4j.logger.org.apache.commons=DEBUG
log4j.logger.org.springframework=DEBUG

#use ConsoleAppender for ConsoleOut
log4j.appender.CONSOLE=org.apache.log4j.ConsoleAppender
log4j.appender.CONSOLE.Threshold=DEBUG
log4j.appender.CONSOLE.Target=System.out
log4j.appender.CONSOLE.layout=org.apache.log4j.PatternLayout
log4j.appender.CONSOLE.layout.ConversionPattern=[%d{yyyyMMdd HH:mm:ss}][%t][%C{1}.%M]%m%n</span>


下面是//src//applicationContext.xml

<span style="font-size:12px;"><?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
	xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
	xmlns:mvc="http://www.springframework.org/schema/mvc"
	xmlns:context="http://www.springframework.org/schema/context"
	xsi:schemaLocation="http://www.springframework.org/schema/beans
						http://www.springframework.org/schema/beans/spring-beans-3.2.xsd
						http://www.springframework.org/schema/mvc
						http://www.springframework.org/schema/mvc/spring-mvc-3.2.xsd
						http://www.springframework.org/schema/context
						http://www.springframework.org/schema/context/spring-context-3.2.xsd">
	<!-- 它背後註冊了很多用於解析註解的處理器,其中就包括<context:annotation-config/>配置的註解所使用的處理器 -->
	<!-- 所以配置了<context:component-scan base-package="">之後,便無需再配置<context:annotation-config> -->
	<context:component-scan base-package="com.jadyer"/>
	
	<!-- 啓用SpringMVC的註解功能,它會自動註冊HandlerMapping、HandlerAdapter、ExceptionResolver的相關實例 -->
	<mvc:annotation-driven/>

	<!-- 配置SpringMVC的視圖解析器 -->
	<!-- 其viewClass屬性的默認值就是org.springframework.web.servlet.view.JstlView -->
	<bean class="org.springframework.web.servlet.view.InternalResourceViewResolver">
		<property name="prefix" value="/"/>
		<property name="suffix" value=".jsp"/>
	</bean>

	<!-- 默認訪問跳轉到登錄頁面(即定義無需Controller的url<->view直接映射) -->
	<mvc:view-controller path="/" view-name="forward:/login.jsp"/>

	<!-- 由於web.xml中設置是:由SpringMVC攔截所有請求,於是在讀取靜態資源文件的時候就會受到影響(說白了就是讀不到) -->
	<!-- 經過下面的配置,該標籤的作用就是:所有頁面中引用"/js/**"的資源,都會從"/resources/js/"裏面進行查找 -->
	<!-- 我們可以訪問http://IP:8080/xxx/js/my.css和http://IP:8080/xxx/resources/js/my.css對比出來 -->
	<mvc:resources mapping="/js/**" location="/resources/js/"/>
	<mvc:resources mapping="/css/**" location="/resources/css/"/>
	<mvc:resources mapping="/WEB-INF/**" location="/WEB-INF/"/>

	<!-- SpringMVC在超出上傳文件限制時,會拋出org.springframework.web.multipart.MaxUploadSizeExceededException -->
	<!-- 該異常是SpringMVC在檢查上傳的文件信息時拋出來的,而且此時還沒有進入到Controller方法中 -->
	<bean class="org.springframework.web.servlet.handler.SimpleMappingExceptionResolver">
		<property name="exceptionMappings">
			<props>
				<!-- 遇到MaxUploadSizeExceededException異常時,自動跳轉到/WEB-INF/error_fileupload.jsp頁面 -->
				<prop key="org.springframework.web.multipart.MaxUploadSizeExceededException">WEB-INF/error_fileupload</prop>
				<!-- 處理其它異常(包括Controller拋出的) -->
				<prop key="java.lang.Throwable">WEB-INF/500</prop>
			</props>
		</property>
	</bean>

	<!-- 繼承自AuthorizingRealm的自定義Realm,即指定Shiro驗證用戶登錄的類爲自定義的ShiroDbRealm.java -->
	<bean id="myRealm" class="com.jadyer.realm.MyRealm"/>

	<!-- Shiro默認會使用Servlet容器的Session,可通過sessionMode屬性來指定使用Shiro原生Session -->
	<!-- 即<property name="sessionMode" value="native"/>,詳細說明見官方文檔 -->
	<!-- 這裏主要是設置自定義的單Realm應用,若有多個Realm,可使用'realms'屬性代替 -->
	<bean id="securityManager" class="org.apache.shiro.web.mgt.DefaultWebSecurityManager">
		<property name="realm" ref="myRealm"/>
	</bean>

	<!-- Shiro主過濾器本身功能十分強大,其強大之處就在於它支持任何基於URL路徑表達式的、自定義的過濾器的執行 -->
	<!-- Web應用中,Shiro可控制的Web請求必須經過Shiro主過濾器的攔截,Shiro對基於Spring的Web應用提供了完美的支持 -->
	<bean id="shiroFilter" class="org.apache.shiro.spring.web.ShiroFilterFactoryBean">
		<!-- Shiro的核心安全接口,這個屬性是必須的 -->
		<property name="securityManager" ref="securityManager"/>
		<!-- 要求登錄時的鏈接(可根據項目的URL進行替換),非必須的屬性,默認會自動尋找Web工程根目錄下的"/login.jsp"頁面 -->
		<property name="loginUrl" value="/"/>
		<!-- 登錄成功後要跳轉的連接(本例中此屬性用不到,因爲登錄成功後的處理邏輯在LoginController裏硬編碼爲main.jsp了) -->
		<!-- <property name="successUrl" value="/system/main"/> -->
		<!-- 用戶訪問未對其授權的資源時,所顯示的連接 -->
		<!-- 若想更明顯的測試此屬性可以修改它的值,如unauthor.jsp,然後用[玄玉]登錄後訪問/admin/listUser.jsp就看見瀏覽器會顯示unauthor.jsp -->
		<property name="unauthorizedUrl" value="/"/>
		<!-- Shiro連接約束配置,即過濾鏈的定義 -->
		<!-- 此處可配合我的這篇文章來理解各個過濾連的作用http://blog.csdn.net/jadyer/article/details/12172839 -->
		<!-- 下面value值的第一個'/'代表的路徑是相對於HttpServletRequest.getContextPath()的值來的 -->
		<!-- anon:它對應的過濾器裏面是空的,什麼都沒做,這裏.do和.jsp後面的*表示參數,比方說login.jsp?main這種 -->
		<!-- authc:該過濾器下的頁面必須驗證後才能訪問,它是Shiro內置的一個攔截器org.apache.shiro.web.filter.authc.FormAuthenticationFilter -->
		<property name="filterChainDefinitions">
			<value>
				/mydemo/login=anon
				/mydemo/getVerifyCodeImage=anon
				/main**=authc
				/user/info**=authc
				/admin/listUser**=authc,perms[admin:manage]
			</value>
		</property>
	</bean>

	<!-- 保證實現了Shiro內部lifecycle函數的bean執行 -->
	<bean id="lifecycleBeanPostProcessor" class="org.apache.shiro.spring.LifecycleBeanPostProcessor"/>

	<!-- 開啓Shiro的註解(如@RequiresRoles,@RequiresPermissions),需藉助SpringAOP掃描使用Shiro註解的類,並在必要時進行安全邏輯驗證 -->
	<!-- 配置以下兩個bean即可實現此功能 -->
	<!-- Enable Shiro Annotations for Spring-configured beans. Only run after the lifecycleBeanProcessor has run -->
	<!-- 由於本例中並未使用Shiro註解,故註釋掉這兩個bean(個人覺得將權限通過註解的方式硬編碼在程序中,查看起來不是很方便,沒必要使用) -->
	<!-- 
	<bean class="org.springframework.aop.framework.autoproxy.DefaultAdvisorAutoProxyCreator" depends-on="lifecycleBeanPostProcessor"/>
	<bean class="org.apache.shiro.spring.security.interceptor.AuthorizationAttributeSourceAdvisor">
		<property name="securityManager" ref="securityManager"/>
	</bean>
	 -->
</beans></span>


下面是自定義的Realm類----MyRealm.java

<span style="font-size:12px;">package com.jadyer.realm;

import org.apache.commons.lang3.builder.ReflectionToStringBuilder;
import org.apache.commons.lang3.builder.ToStringStyle;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.AuthenticationException;
import org.apache.shiro.authc.AuthenticationInfo;
import org.apache.shiro.authc.AuthenticationToken;
import org.apache.shiro.authc.SimpleAuthenticationInfo;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.session.Session;
import org.apache.shiro.subject.PrincipalCollection;
import org.apache.shiro.subject.Subject;

/**
 * 自定義的指定Shiro驗證用戶登錄的類
 * @see 在本例中定義了2個用戶:jadyer和玄玉,jadyer具有admin角色和admin:manage權限,玄玉不具有任何角色和權限
 * @create Sep 29, 2013 3:15:31 PM
 * @author 玄玉<http://blog.csdn.net/jadyer>
 */
public class MyRealm extends AuthorizingRealm {
	/**
	 * 爲當前登錄的Subject授予角色和權限
	 * @see 經測試:本例中該方法的調用時機爲需授權資源被訪問時
	 * @see 經測試:並且每次訪問需授權資源時都會執行該方法中的邏輯,這表明本例中默認並未啓用AuthorizationCache
	 * @see 個人感覺若使用了Spring3.1開始提供的ConcurrentMapCache支持,則可靈活決定是否啓用AuthorizationCache
	 * @see 比如說這裏從數據庫獲取權限信息時,先去訪問Spring3.1提供的緩存,而不使用Shior提供的AuthorizationCache
	 */
	@Override
	protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals){
		//獲取當前登錄的用戶名,等價於(String)principals.fromRealm(this.getName()).iterator().next()
		String currentUsername = (String)super.getAvailablePrincipal(principals);
//		List<String> roleList = new ArrayList<String>();
//		List<String> permissionList = new ArrayList<String>();
//		//從數據庫中獲取當前登錄用戶的詳細信息
//		User user = userService.getByUsername(currentUsername);
//		if(null != user){
//			//實體類User中包含有用戶角色的實體類信息
//			if(null!=user.getRoles() && user.getRoles().size()>0){
//				//獲取當前登錄用戶的角色
//				for(Role role : user.getRoles()){
//					roleList.add(role.getName());
//					//實體類Role中包含有角色權限的實體類信息
//					if(null!=role.getPermissions() && role.getPermissions().size()>0){
//						//獲取權限
//						for(Permission pmss : role.getPermissions()){
//							if(!StringUtils.isEmpty(pmss.getPermission())){
//								permissionList.add(pmss.getPermission());
//							}
//						}
//					}
//				}
//			}
//		}else{
//			throw new AuthorizationException();
//		}
//		//爲當前用戶設置角色和權限
//		SimpleAuthorizationInfo simpleAuthorInfo = new SimpleAuthorizationInfo();
//		simpleAuthorInfo.addRoles(roleList);
//		simpleAuthorInfo.addStringPermissions(permissionList);
		SimpleAuthorizationInfo simpleAuthorInfo = new SimpleAuthorizationInfo();
		//實際中可能會像上面註釋的那樣從數據庫取得
		if(null!=currentUsername && "jadyer".equals(currentUsername)){
			//添加一個角色,不是配置意義上的添加,而是證明該用戶擁有admin角色  
			simpleAuthorInfo.addRole("admin");
			//添加權限
			simpleAuthorInfo.addStringPermission("admin:manage");
			System.out.println("已爲用戶[jadyer]賦予了[admin]角色和[admin:manage]權限");
			return simpleAuthorInfo;
		}else if(null!=currentUsername && "玄玉".equals(currentUsername)){
			System.out.println("當前用戶[玄玉]無授權");
			return simpleAuthorInfo;
		}
		//若該方法什麼都不做直接返回null的話,就會導致任何用戶訪問/admin/listUser.jsp時都會自動跳轉到unauthorizedUrl指定的地址
		//詳見applicationContext.xml中的<bean id="shiroFilter">的配置
		return null;
	}

	
	/**
	 * 驗證當前登錄的Subject
	 * @see 經測試:本例中該方法的調用時機爲LoginController.login()方法中執行Subject.login()時
	 */
	@Override
	protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken authcToken) throws AuthenticationException {
		//獲取基於用戶名和密碼的令牌
		//實際上這個authcToken是從LoginController裏面currentUser.login(token)傳過來的
		//兩個token的引用都是一樣的,本例中是org.apache.shiro.authc.UsernamePasswordToken@33799a1e
		UsernamePasswordToken token = (UsernamePasswordToken)authcToken;
		System.out.println("驗證當前Subject時獲取到token爲" + ReflectionToStringBuilder.toString(token, ToStringStyle.MULTI_LINE_STYLE));
//		User user = userService.getByUsername(token.getUsername());
//		if(null != user){
//			AuthenticationInfo authcInfo = new SimpleAuthenticationInfo(user.getUsername(), user.getPassword(), user.getNickname());
//			this.setSession("currentUser", user);
//			return authcInfo;
//		}else{
//			return null;
//		}
		//此處無需比對,比對的邏輯Shiro會做,我們只需返回一個和令牌相關的正確的驗證信息
		//說白了就是第一個參數填登錄用戶名,第二個參數填合法的登錄密碼(可以是從數據庫中取到的,本例中爲了演示就硬編碼了)
		//這樣一來,在隨後的登錄頁面上就只有這裏指定的用戶和密碼才能通過驗證
		if("jadyer".equals(token.getUsername())){
			AuthenticationInfo authcInfo = new SimpleAuthenticationInfo("jadyer", "jadyer", this.getName());
			this.setSession("currentUser", "jadyer");
			return authcInfo;
		}else if("玄玉".equals(token.getUsername())){
			AuthenticationInfo authcInfo = new SimpleAuthenticationInfo("玄玉", "xuanyu", this.getName());
			this.setSession("currentUser", "玄玉");
			return authcInfo;
		}
		//沒有返回登錄用戶名對應的SimpleAuthenticationInfo對象時,就會在LoginController中拋出UnknownAccountException異常
		return null;
	}
	
	
	/**
	 * 將一些數據放到ShiroSession中,以便於其它地方使用
	 * @see 比如Controller,使用時直接用HttpSession.getAttribute(key)就可以取到
	 */
	private void setSession(Object key, Object value){
		Subject currentUser = SecurityUtils.getSubject();
		if(null != currentUser){
			Session session = currentUser.getSession();
			System.out.println("Session默認超時時間爲[" + session.getTimeout() + "]毫秒");
			if(null != session){
				session.setAttribute(key, value);
			}
		}
	}
}</span>


下面是處理用戶登錄的LoginController.java

package com.jadyer.controller;

import javax.servlet.http.HttpServletRequest;

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;

@Controller
@RequestMapping("mydemo")
public class UserController {
	@RequestMapping(value="/getUserInfo")
	public String getUserInfo(HttpServletRequest request){
		String currentUser = (String)request.getSession().getAttribute("currentUser");
		System.out.println("當前登錄的用戶爲[" + currentUser + "]");
		request.setAttribute("currUser", currentUser);
		return "/user/info";
	}
}


最後是用於生成登錄驗證碼的VerifyCodeUtil.java

package com.jadyer.util;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.util.Random;

/**
 * 驗證碼生成器
 * @see --------------------------------------------------------------------------------------------------------------
 * @see 可生成數字、大寫、小寫字母及三者混合類型的驗證碼
 * @see 支持自定義驗證碼字符數量,支持自定義驗證碼圖片的大小,支持自定義需排除的特殊字符,支持自定義干擾線的數量,支持自定義驗證碼圖文顏色
 * @see --------------------------------------------------------------------------------------------------------------
 * @see 另外,給Shiro加入驗證碼有多種方式,也可以通過繼承修改FormAuthenticationFilter類,通過Shiro去驗證驗證碼
 * @see 而這裏既然使用了SpringMVC,也爲了簡化操作,就使用此工具生成驗證碼,並在Controller中處理驗證碼的校驗
 * @see --------------------------------------------------------------------------------------------------------------
 * @create Sep 29, 2013 4:23:13 PM
 * @author 玄玉<http://blog.csdn.net/jadyer>
 */
public class VerifyCodeUtil {
	/**
	 * 驗證碼類型爲僅數字,即0~9
	 */
	public static final int TYPE_NUM_ONLY = 0;

	/**
	 * 驗證碼類型爲僅字母,即大小寫字母混合
	 */
	public static final int TYPE_LETTER_ONLY = 1;

	/**
	 * 驗證碼類型爲數字和大小寫字母混合
	 */
	public static final int TYPE_ALL_MIXED = 2;

	/**
	 * 驗證碼類型爲數字和大寫字母混合
	 */
	public static final int TYPE_NUM_UPPER = 3;

	/**
	 * 驗證碼類型爲數字和小寫字母混合
	 */
	public static final int TYPE_NUM_LOWER = 4;

	/**
	 * 驗證碼類型爲僅大寫字母
	 */
	public static final int TYPE_UPPER_ONLY = 5;

	/**
	 * 驗證碼類型爲僅小寫字母
	 */
	public static final int TYPE_LOWER_ONLY = 6;

	private VerifyCodeUtil(){}
	
	/**
	 * 生成隨機顏色
	 */
	private static Color generateRandomColor() {
		Random random = new Random();
		return new Color(random.nextInt(255), random.nextInt(255), random.nextInt(255));
	}
	
	
	/**
	 * 生成圖片驗證碼
	 * @param type           驗證碼類型,參見本類的靜態屬性
	 * @param length         驗證碼字符長度,要求大於0的整數
	 * @param excludeString  需排除的特殊字符
	 * @param width          圖片寬度(注意此寬度若過小,容易造成驗證碼文本顯示不全,如4個字符的文本可使用85到90的寬度)
	 * @param height         圖片高度
	 * @param interLine      圖片中干擾線的條數
	 * @param randomLocation 每個字符的高低位置是否隨機
	 * @param backColor      圖片顏色,若爲null則表示採用隨機顏色
	 * @param foreColor      字體顏色,若爲null則表示採用隨機顏色
	 * @param lineColor      干擾線顏色,若爲null則表示採用隨機顏色
	 * @return 圖片緩存對象
	 */
	public static BufferedImage generateImageCode(int type, int length, String excludeString, int width, int height, int interLine, boolean randomLocation, Color backColor, Color foreColor, Color lineColor){
		String textCode = generateTextCode(type, length, excludeString);
		return generateImageCode(textCode, width, height, interLine, randomLocation, backColor, foreColor, lineColor);
	}
	

	/**
	 * 生成驗證碼字符串
	 * @param type          驗證碼類型,參見本類的靜態屬性
	 * @param length        驗證碼長度,要求大於0的整數
	 * @param excludeString 需排除的特殊字符(無需排除則爲null)
	 * @return 驗證碼字符串
	 */
	public static String generateTextCode(int type, int length, String excludeString){
		if(length <= 0){
			return "";
		}
		StringBuffer verifyCode = new StringBuffer();
		int i = 0;
		Random random = new Random();
		switch(type){
			case TYPE_NUM_ONLY:
				while(i < length){
					int t = random.nextInt(10);
					//排除特殊字符
					if(null==excludeString || excludeString.indexOf(t+"")<0) {
						verifyCode.append(t);
						i++;
					}
				}
			break;
			case TYPE_LETTER_ONLY:
				while(i < length){
					int t = random.nextInt(123);
					if((t>=97 || (t>=65&&t<=90)) && (null==excludeString||excludeString.indexOf((char)t)<0)){
						verifyCode.append((char)t);
						i++;
					}
				}
			break;
			case TYPE_ALL_MIXED:
				while(i < length){
					int t = random.nextInt(123);
					if((t>=97 || (t>=65&&t<=90) || (t>=48&&t<=57)) && (null==excludeString||excludeString.indexOf((char)t)<0)){
						verifyCode.append((char)t);
						i++;
					}
				}
			break;
			case TYPE_NUM_UPPER:
				while(i < length){
					int t = random.nextInt(91);
					if((t>=65 || (t>=48&&t<=57)) && (null==excludeString || excludeString.indexOf((char)t)<0)){
						verifyCode.append((char)t);
						i++;
					}
				}
			break;
			case TYPE_NUM_LOWER:
				while(i < length){
					int t = random.nextInt(123);
					if((t>=97 || (t>=48&&t<=57)) && (null==excludeString || excludeString.indexOf((char)t)<0)){
						verifyCode.append((char)t);
						i++;
					}
				}
			break;
			case TYPE_UPPER_ONLY:
				while(i < length){
					int t = random.nextInt(91);
					if((t >= 65) && (null==excludeString||excludeString.indexOf((char)t)<0)){
						verifyCode.append((char)t);
						i++;
					}
				}
			break;
			case TYPE_LOWER_ONLY:
				while(i < length){
					int t = random.nextInt(123);
					if((t>=97) && (null==excludeString||excludeString.indexOf((char)t)<0)){
						verifyCode.append((char)t);
						i++;
					}
				}
			break;
		}
		return verifyCode.toString();
	}

	/**
	 * 已有驗證碼,生成驗證碼圖片
	 * @param textCode       文本驗證碼
	 * @param width          圖片寬度(注意此寬度若過小,容易造成驗證碼文本顯示不全,如4個字符的文本可使用85到90的寬度)
	 * @param height         圖片高度
	 * @param interLine      圖片中干擾線的條數
	 * @param randomLocation 每個字符的高低位置是否隨機
	 * @param backColor      圖片顏色,若爲null則表示採用隨機顏色
	 * @param foreColor      字體顏色,若爲null則表示採用隨機顏色
	 * @param lineColor      干擾線顏色,若爲null則表示採用隨機顏色
	 * @return 圖片緩存對象
	 */
	public static BufferedImage generateImageCode(String textCode, int width, int height, int interLine, boolean randomLocation, Color backColor, Color foreColor, Color lineColor){
		//創建內存圖像
		BufferedImage bufferedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
		//獲取圖形上下文
		Graphics graphics = bufferedImage.getGraphics();
		//畫背景圖
		graphics.setColor(null==backColor ? generateRandomColor() : backColor);
		graphics.fillRect(0, 0, width, height);
		//畫干擾線
		Random random = new Random();
		if(interLine > 0){
			int x = 0, y = 0, x1 = width, y1 = 0;
			for(int i=0; i<interLine; i++){
				graphics.setColor(null==lineColor ? generateRandomColor() : lineColor);
				y = random.nextInt(height);
				y1 = random.nextInt(height);
				graphics.drawLine(x, y, x1, y1);
			}
		}
		//字體大小爲圖片高度的80%
		int fsize = (int)(height * 0.8);
		int fx = height - fsize;
		int fy = fsize;
		//設定字體
		graphics.setFont(new Font("Default", Font.PLAIN, fsize));
		//寫驗證碼字符
		for(int i=0; i<textCode.length(); i++){
			fy = randomLocation ? (int)((Math.random()*0.3+0.6)*height) : fy;
			graphics.setColor(null==foreColor ? generateRandomColor() : foreColor);
			//將驗證碼字符顯示到圖象中
			graphics.drawString(textCode.charAt(i)+"", fx, fy);
			fx += fsize * 0.9;
		}
		graphics.dispose();
		return bufferedImage;
	}
}


 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章