mysqloperator.java

基于Jpcap开发的流量监控软件

package joel.ethernettools;
/**
 * THIS IS THE MYSQL DB OPERATOR CLASS.
 * FOR NOW , IT ONLY SUPPORT ADDDATA(INSERT).
 * MORE FEATURE WILL BE DEVELOPPED SOON.
 */
import java.io.File;

import javax.swing.JOptionPane;

import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;
import org.hibernate.exception.JDBCConnectionException;

public class MySQLOperator {
	private Session session;
	private Transaction transaction;
	private boolean isOnline=false;
	public MySQLOperator() {
		try {
			
			Configuration cfg = new Configuration().configure(new File(
					System.getProperty("user.dir")+System.getProperty("file.separator")+"XML/hibernate.cfg.xml"));
			SessionFactory factory = cfg.buildSessionFactory();
			session = factory.openSession();
			transaction = session.beginTransaction();
			isOnline=true;
		} catch (JDBCConnectionException e) {
			JOptionPane.showMessageDialog(null, e.getLocalizedMessage()
					+ "\nplease check the DB & restart the program.");
			isOnline=false;
		}
	}

	protected boolean add(ProcessedPacket p) {
		if(isOnline==false)
			return false;
		session.save(p);
		transaction.commit();
		p=null;
		return true;
	}

	protected boolean add(ProcessedPacket[] p) {
		if(isOnline==false)
			return false;
		for (int i = 0; i < p.length; i++)
			session.save(p[i]);
		transaction.commit();
		transaction.begin();
		p=null;
		return true;
	}

	protected boolean add(OriginalPacket p) {
		if(isOnline==false)
			return false;
		session.save(p);
		transaction.commit();
		transaction.begin();
		p=null;
		return true;
	}

	protected boolean add(OriginalPacket[] p) {
		if(isOnline==false)
			return false;
		for (int i = 0; i < p.length; i++)
			session.save(p[i]);
		transaction.commit();
		transaction.begin();
		p=null;
		return true;
	}

	/**
	 * @param args
	 */
	public static void main(String[] args) {
		new MySQLOperator();
	}

	protected boolean isOnline() {
		return isOnline;
	}

}