visitstatdao.java

个人Blog java编写的Blog可以直接使用！

/*
 *  VisitStatDAO.java
 *  
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Library General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *  
 *  Author: Winter Lau ([e-mail])
 *  http://dlog4j.sourceforge.net
 */
package com.liusoft.dlog4j.dao;

import java.util.Calendar;
import java.util.Date;

import org.hibernate.Query;
import org.hibernate.Session;

import com.liusoft.dlog4j.beans.SiteBean;
import com.liusoft.dlog4j.beans.SiteStatBean;

/**
 * 访问统计
 * @author Winter Lau
 */
public class VisitStatDAO extends DAO {

	/**
	 * 今日访问人次
	 * @param site
	 * @return
	 */
	public static long getUVToday(SiteBean site){		
		if(site==null) return -1;
		Calendar cal = Calendar.getInstance();
		int statDate = cal.get(Calendar.YEAR)*10000 + cal.get(Calendar.MONTH)*100 + cal.get(Calendar.DATE);
		return toLong(executeNamedStat("VISIT_STAT", site.getId(), statDate), 1);
	}

	/**
	 * 本周访问人次
	 * @param site
	 * @return
	 */
	public static long getUVThisWeek(SiteBean site){	
		if(site==null) return -1;	
		Calendar cal = Calendar.getInstance();
		int week = cal.get(Calendar.DAY_OF_WEEK);
		if(week > 1)
			cal.add(Calendar.DATE, 1-week);
		int statDate = cal.get(Calendar.YEAR)*10000 + cal.get(Calendar.MONTH)*100 + cal.get(Calendar.DATE);
		return toLong(executeNamedStat("VISIT_STAT", site.getId(), statDate), 1);
	}

	/**
	 * 本月访问人次
	 * @param site
	 * @return
	 */
	public static long getUVThisMonth(SiteBean site){
		if(site==null) return -1;	
		Calendar cal = Calendar.getInstance();
		cal.set(Calendar.DATE, 1);
		int statDate = cal.get(Calendar.YEAR)*10000 + cal.get(Calendar.MONTH)*100 + cal.get(Calendar.DATE);
		return toLong(executeNamedStat("VISIT_STAT", site.getId(), statDate), 1);
	}

	/**
	 * 今年访问人次
	 * @param site
	 * @return
	 */
	public static long getUVThisYear(SiteBean site){
		if(site==null) return -1;	
		Calendar cal = Calendar.getInstance();
		cal.set(Calendar.MONTH, 1);
		cal.set(Calendar.DATE, 1);
		int statDate = cal.get(Calendar.YEAR)*10000 + cal.get(Calendar.MONTH)*100 + cal.get(Calendar.DATE);
		return toLong(executeNamedStat("VISIT_STAT", site.getId(), statDate), 1);
	}

	/**
	 * 素有访问人次
	 * @param site
	 * @return
	 */
	public static long getUVTotal(SiteBean site){
		if(site==null) return -1;	
		return toLong(executeNamedStat("VISIT_STAT_2", site.getId()), 1);
	}
	
	private static long toLong(Number v, int defaultValue){
		return (v!=null)?v.longValue():defaultValue;
	}
	
	/**
	 * 写统计数据到数据库
	 * @param ssb
	 * @throws Exception 
	 */
	public static void writeStatData(int siteid, int uvCount, int source) throws Exception{
		Calendar cal = Calendar.getInstance();
		int statDate = cal.get(Calendar.YEAR)*10000 + cal.get(Calendar.MONTH)*100 + cal.get(Calendar.DATE);
		Session ssn = getSession();
		try{
			beginTransaction();
			Query update_q = ssn.getNamedQuery((siteid>0)?"UPDATE_SITE_STAT_1":"UPDATE_SITE_STAT_2");
			update_q.setInteger(0, uvCount);
			update_q.setInteger(1, statDate);
			update_q.setInteger(2, source);
			if(siteid>0)
				update_q.setInteger(3, siteid);
			if(update_q.executeUpdate()<1){
				SiteStatBean ssb = new SiteStatBean();
				ssb.setSiteId(siteid);
				ssb.setUvCount(uvCount);
				ssb.setUpdateTime(new Date());
				ssb.setSource(source);
				ssb.setStatDate(statDate);
				ssn.save(ssb);
			}
			commit();
		}catch(Exception e){
			rollback();
			throw e;
		}
	}
	
}