userfrienddb.java

cwbbs 云网论坛源码

package com.redmoon.forum.person;

import java.sql.*;
import cn.js.fan.db.*;
import cn.js.fan.web.*;
import org.apache.log4j.*;
import cn.js.fan.util.ErrMsgException;
import cn.js.fan.base.ObjectDb;
import cn.js.fan.util.DateUtil;
import com.redmoon.forum.SequenceMgr;
import cn.js.fan.util.ResKeyException;

public class UserFriendDb extends ObjectDb {
    String connname;
    Logger logger = Logger.getLogger(UserFriendDb.class.getName());

    public UserFriendDb() {
        connname = Global.defaultDB;
        if (connname.equals(""))
            logger.info("UserFriendDb:connname is empty.");
        init();
    }

    public UserFriendDb(int id){
        this.id = id;
        connname = Global.defaultDB;
        if (connname.equals(""))
            logger.info("UserFriendDb:connname is empty.");
        load();
        init();
    }

    public void initDB() {
        objectCache = new UserFriendCache(this);
        tableName = "sq_friend";
        primaryKey = new PrimaryKey("id", PrimaryKey.TYPE_INT);

        QUERY_LOAD =
            "SELECT name,friend,rq FROM " + tableName + " WHERE id=?";
        QUERY_SAVE =
            "?";
        QUERY_DEL = "delete from " + tableName + " where id=?";
        QUERY_CREATE = "insert into " + tableName + " (name,friend,rq,id) values (?,?,?,?)";
        QUERY_LIST = "select id from " + tableName + " order by rq asc";
        isInitFromConfigDB = false;
    }

    public boolean create() throws ResKeyException {
        Conn conn = null;
        boolean re = false;
        try {
            conn = new Conn(connname);
            String sql = "select friend from sq_friend where name=? and friend=?";
            ResultSet rs = null;
            PreparedStatement pstmt = conn.prepareStatement(sql);
            pstmt.setString(1, name);
            pstmt.setString(2, friend);
            rs = conn.executePreQuery();
            if (rs != null && rs.next()) {
                throw new ResKeyException("res.forum.person.UserFriendDb", "err_add_repeat");
                
            }

            if (rs!=null) {
                rs.close();
                rs = null;
            }

            if (pstmt!=null) {
                pstmt.close();
                pstmt = null;
            }

            pstmt = conn.prepareStatement(this.QUERY_CREATE);
            pstmt.setString(1, name);
            pstmt.setString(2, friend);
            pstmt.setString(3, "" + System.currentTimeMillis());
            int id = (int)SequenceMgr.nextID(SequenceMgr.SQ_FRIEND);
            pstmt.setInt(4, id);
            re = conn.executePreUpdate() == 1 ? true : false;
            if (re) {
                UserFriendCache mc = new UserFriendCache(this);
                mc.refreshCreate();
            }
        } catch (SQLException e) {
            logger.error("create:" + e.getMessage());
            throw new ResKeyException(SkinUtil.ERR_DB);
        } finally {
            if (conn != null) {
                conn.close();
                conn = null;
            }
        }
        return re;
    }

    public ObjectDb getObjectRaw(PrimaryKey pk) {
        return new UserFriendDb(pk.getIntValue());
    }

    public int delFriendsOfUser(String userName) {
        
        PreparedStatement pstmt = null;
        ResultSet rs = null;
        String sql = "select id from sq_friend where name=?";
        int count = 0;
        Conn conn = new Conn(connname);
        try {
            pstmt = conn.prepareStatement(sql);
            pstmt.setString(1, userName);
            rs = conn.executePreQuery();
            if (rs != null) {
                while (rs.next()) {
                    getUserFriendDb(rs.getInt(1)).del();
                    count++;
                }
            }
        } catch (SQLException e) {
            logger.error("delFriendsOfUser:" + e.getMessage());
        } finally {
            if (conn != null) {
                conn.close();
                conn = null;
            }
        }
        return count;
    }

    public boolean del() {
        Conn conn = new Conn(connname);
        PreparedStatement pstmt = null;
        boolean re = false;
        try {
            pstmt = conn.prepareStatement(QUERY_DEL);
            pstmt.setInt(1, id);
            re = conn.executePreUpdate() > 0 ? true : false;
            if (re) {
                UserFriendCache bc = new UserFriendCache(this);
                primaryKey.setValue(new Integer(id));
                bc.refreshDel(primaryKey);
            }
        } catch (SQLException e) {
            logger.error("del:" + e.getMessage());
        } finally {
            if (conn != null) {
                conn.close();
                conn = null;
            }
        }
        return re;
    }

    public boolean save() {
        return false;
    }

    public void load() {
        
        PreparedStatement pstmt = null;
        ResultSet rs = null;
        Conn conn = new Conn(connname);
        try {
            
            pstmt = conn.prepareStatement(QUERY_LOAD);
            pstmt.setInt(1, id);
            rs = conn.executePreQuery();
            if (!rs.next()) {
                logger.error("load: " + id +
                             " is not found in DB.");
            } else {
                name = rs.getString(1);
                friend = rs.getString(2);
                rq = DateUtil.parse(rs.getString(3));
                loaded = true;
            }
        } catch (SQLException e) {
            logger.error("load:" + e.getMessage());
        } finally {
            if (rs != null) {
                try {
                    rs.close();
                } catch (Exception e) {}
                rs = null;
            }
            if (conn != null) {
                conn.close();
                conn = null;
            }
        }
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getName() {
        return name;
    }

    public UserFriendDb getUserFriendDb(int id) {
        return (UserFriendDb)getObjectDb(new Integer(id));
    }

    public int getId() {
        return id;
    }

    public String getFriend() {
        return friend;
    }

    public java.util.Date getRq() {
        return rq;
    }

    public void setId(int id) {
        this.id = id;
    }

    public void setFriend(String friend) {
        this.friend = friend;
    }

    public void setRq(Date rq) {
        this.rq = rq;
    }

    private String name;
    private int id;
    private String friend;
    private java.util.Date rq;

}