friend.java

来自「一个用java写的一种聊天工具」· Java 代码 · 共 42 行

package business;
import java.sql.ResultSet;
import java.sql.SQLException;

import DataLayer.SqlHelper;
import business.*;

public class Friend {
	private int userNum;
	private int friendNum;
	private SqlHelper sh;
	public Friend()
	{
		
	}
	public Friend(int user,int friend)
	{
		sh = new SqlHelper();
		sh.setStatement(sh.getConnection());
		this.userNum = user;
		this.friendNum = friend;
	}
	public int addFriend()
	{
		ResultSet rs = sh.getQuery("select count(*) from tbl_Friends where userNum="+userNum+" and friendNum="+friendNum);
		try {
			if(rs.next())
			{
				if(rs.getInt(1)>0)
				{
					return 0;
				}
			}
		} catch (SQLException e) {
			// TODO 自动生成 catch 块
			e.printStackTrace();
		}
		sh.getUpdate("insert into tbl_Friends values("+userNum+","+friendNum+")");
		return 1;
	}
}