src/main/java/database/Database1.java

package database;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.Persistence;

import com.querydsl.jpa.impl.JPAQuery;

import entity.BegInfo;
import entity.Notice;
import entity.Post;
import entity.Profile;
import entity.QNotice;
import entity.QSeed;
import entity.QUser;
import entity.QUserPT;
import entity.QUserStar;
import entity.Seed;
import entity.User;
import entity.UserPT;
import entity.UserStar;

public class Database1 implements DataManagerInterface {

    private EntityManagerFactory emf;

    public Database1() {
        entity.config cfg = new entity.config();
        Map<String, Object> props = new HashMap<>();
        props.put("javax.persistence.jdbc.url", "jdbc:mysql://" + cfg.SqlURL + "/" + cfg.Database);
        props.put("javax.persistence.jdbc.user", cfg.SqlUsername);
        props.put("javax.persistence.jdbc.password", cfg.SqlPassword);
        this.emf = Persistence.createEntityManagerFactory("myPersistenceUnit", props);
    }

    // 方便测试时注入
    public Database1(EntityManagerFactory emf) {
        this.emf = emf;
    }

    private EntityManager getEntityManager() {
        return emf.createEntityManager();
    }

    // 返回状态：0 success，1 邮箱重复，2其他原因
    @Override
    public int RegisterUser(User userinfo) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<String> query = new JPAQuery<>(em);
            QUser u = QUser.user;
            List<String> allEmails = query.select(u.email).from(u).fetch();

            if (allEmails.contains(userinfo.email)) {
                return 1;
            }
            em.getTransaction().begin();
            em.persist(userinfo);
            em.getTransaction().commit();
            return 0;

        } catch (Exception e) {
            e.printStackTrace();
            return 2;
        } finally {
            em.close();
        }

    }

    // 返回状态：0 success，1 不存在，2其他原因
    @Override
    public int UpdateInformation(User userinfo) {
        EntityManager em = getEntityManager();
        try {
            if (userinfo.userid == null) {
                return 2; // userid为null直接返回错误
            }
            JPAQuery<User> query = new JPAQuery<>(em);
            QUser u = QUser.user;
            User updateUser = query.select(u).from(u).where(u.userid.eq(userinfo.userid)).fetchOne();

            if (updateUser == null) {
                return 1;
            }
            // 只更新需要的字段，避免破坏持久化状态和关联关系
            updateUser.email = userinfo.email;
            updateUser.username = userinfo.username;
            updateUser.password = userinfo.password;
            updateUser.sex = userinfo.sex;
            updateUser.school = userinfo.school;
            updateUser.pictureurl = userinfo.pictureurl;
            updateUser.profile = userinfo.profile;
            updateUser.accountstate = userinfo.accountstate;
            updateUser.invitetimes = userinfo.invitetimes;
            // 如有其他字段也一并赋值
            em.getTransaction().begin();
            em.merge(updateUser);
            em.getTransaction().commit();
            return 0;
        } catch (Exception e) {
            e.printStackTrace();
            return 2;
        } finally {
            em.close();
        }

    }

    // 返回用户的全部基本信息
    @Override
    public User GetInformation(String userid) {
        EntityManager em = getEntityManager();
        try {
            User user = em.find(User.class, userid);
            return user;
        } finally {
            em.close();
        }
    }

    //返回用户的全部pt站信息
    @Override
    public UserPT GetInformationPT(String userid) {
        EntityManager em = getEntityManager();
        try {
            UserPT userPT = em.find(UserPT.class, userid);
            return userPT;
        } finally {
            em.close();
        }
    }

    //返回状态：0 success，1 邮箱重复，2其他原因
    @Override
    public int UpdateInformationPT(UserPT userinfo) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<UserPT> query = new JPAQuery<>(em);
            QUserPT u = QUserPT.userPT;
            UserPT userPT = query.select(u).from(u).where(u.userid.eq(userinfo.userid)).fetchOne();

            if (userPT == null) {
                return 1;
            }
            userPT = userinfo;
            em.getTransaction().begin();
            em.merge(userPT);
            em.getTransaction().commit();
            return 0;

        } catch (Exception e) {
            e.printStackTrace();
            return 2;
        } finally {
            em.close();
        }
    }

    //返回状态：0 success，1 id重复，2其他原因
    @Override
    public int RegisterUserPT(UserPT userinfo) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<UserPT> query = new JPAQuery<>(em);
            QUserPT u = QUserPT.userPT;
            UserPT checkUserPT = query.select(u).from(u).where(u.userid.eq(userinfo.userid)).fetchOne();
            if (checkUserPT != null) {
                return 1;
            }
            em.getTransaction().begin();
            em.persist(userinfo);
            em.getTransaction().commit();
            return 0;
        } catch (Exception e) {
            e.printStackTrace();
            return 2;
        } finally {
            em.close();
        }
    }

    //返回种子的全部信息
    @Override
    public Seed GetSeedInformation(String seedid) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<Seed> query = new JPAQuery<>(em);
            QSeed s = QSeed.seed;
            Seed seed = query.select(s).from(s).where(s.seedid.eq(seedid)).fetchOne();
            return seed;
        } finally {
            em.close();
        }
    }

    //添加一个新的种子，0成功，其他失败信息待定;
    @Override
    public int RegisterSeed(Seed seedinfo) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<Seed> query = new JPAQuery<>(em);
            QSeed s = QSeed.seed;
            Seed seed = query.select(s).from(s).where(s.seedid.eq(seedinfo.seedid)).fetchOne();
            if (seed != null) {
                return 1;
            }
            em.getTransaction().begin();
            em.persist(seedinfo);
            em.getTransaction().commit();
            return 0;
        } catch (Exception e) {
            e.printStackTrace();
            return 2;
        } finally {
            em.close();
        }
    }

    //接收新的种子然后更新其全部属性
    @Override
    public int UpdateSeed(Seed seedinfo) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<Seed> query = new JPAQuery<>(em);
            QSeed s = QSeed.seed;
            Seed seed = query.select(s).from(s).where(s.seedid.eq(seedinfo.seedid)).fetchOne();
            if (seed == null) {
                return 1;
            }
            seed = seedinfo;
            em.getTransaction().begin();
            em.merge(seed);
            em.getTransaction().commit();
            return 0;
        } catch (Exception e) {
            e.printStackTrace();
            return 2;
        } finally {
            em.close();
        }
    }

    //传入搜索的关键词或句子，返回搜索到的种子信息（按照公共字符数量排序）
    @Override
    public Seed[] SearchSeed(String userQ) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<Seed> query = new JPAQuery<>(em);
            QSeed s = QSeed.seed;
            List<Seed> seeds = query.select(s).from(s).fetch();

            if (seeds == null || userQ == null || userQ.trim().isEmpty()) {
                return seeds.toArray(new Seed[0]);
            }

            String processedQuery = userQ.toLowerCase().trim();
            Map<Seed, Integer> seedCountMap = new HashMap<>();
            for (Seed seed : seeds) {
                String title = seed.title.toLowerCase().trim();
                int count = countCommonCharacter(processedQuery, title);
                seedCountMap.put(seed, count);
            }
            seeds.sort((s1, s2) -> {
                int count1 = seedCountMap.getOrDefault(s1, 0);
                int count2 = seedCountMap.getOrDefault(s2, 0);
                return Integer.compare(count2, count1);
            });

            return seeds.toArray(new Seed[0]);
        } finally {
            em.close();
        }
    }

    //计算字符串公共字符数量
    private int countCommonCharacter(String str1, String str2) {
        Map<Character, Integer> map1 = new HashMap<>();
        Map<Character, Integer> map2 = new HashMap<>();

        for (char c : str1.toCharArray()) {
            if (!Character.isWhitespace(c)) {
                map1.put(c, map1.getOrDefault(c, 0) + 1);
            }
        }

        for (char c : str2.toCharArray()) {
            if (!Character.isWhitespace(c)) {
                map2.put(c, map2.getOrDefault(c, 0) + 1);
            }
        }

        int res = 0;
        for (char c : map1.keySet()) {
            if (map2.containsKey(c)) {
                res += Math.min(map1.get(c), map2.get(c));
            }

        }
        return res;
    }

    //返回状态：0 success，1 重复，2其他原因
    @Override
    public int AddNotice(Notice notice) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<Notice> query = new JPAQuery<>(em);
            QNotice n = QNotice.notice;
            Notice checkNotice = query.select(n).from(n).where(n.noticeid.eq(notice.noticeid)).fetchOne();
            if (checkNotice != null) {
                return 1;
            }

            em.getTransaction().begin();
            em.persist(notice);
            em.getTransaction().commit();
            return 0;
        } catch (Exception e) {
            e.printStackTrace();
            return 2;
        } finally {
            em.close();
        }

    }

    //返回状态：0 success，1 不存在，2其他原因
    @Override
    public boolean UpdateNotice(Notice notice) {
        EntityManager em = getEntityManager();
        try {
            Notice oldNotice = em.find(Notice.class, notice.noticeid);
            if (oldNotice == null) {
                return false;
            }
            oldNotice = notice;
            em.getTransaction().begin();
            em.merge(oldNotice);
            em.getTransaction().commit();
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        } finally {
            em.close();
        }
    }

    //删除公告，返回状态：0 success，1 不存在，2其他原因
    @Override
    public boolean DeleteNotice(String noticeid) {
        EntityManager em = getEntityManager();
        try {
            Notice notice = em.find(Notice.class, noticeid);
            if (notice == null) {
                return false;
            }
            em.getTransaction().begin();
            em.remove(notice);
            em.getTransaction().commit();
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        } finally {
            em.close();
        }

    }

    //获取用户的剩余邀请次数
    public int GetUserAvailableInviteTimes(String userid) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<Integer> query = new JPAQuery<>(em);
            QUser u = QUser.user;
            int invite_left = query.select(u.invitetimes).from(u).where(u.userid.eq(userid)).fetchOne();

            return invite_left;
        } catch (Exception e) {
            e.printStackTrace();
            return -1;
        } finally {
            em.close();
        }

    }

    //邀请用户，返回状态：0 success，1 剩余次数不足，2,3其他原因
    @Override
    public int InviteUser(String inviterid, String inviteemail) {
        EntityManager em = getEntityManager();
        try {
            User user = em.find(User.class, inviterid);
            if (user == null || !user.email.equals(inviteemail)) {
                return 3;
            }
            if (user.invitetimes <= 0) {
                return 1;
            }
            user.invitetimes -= 1;
            em.getTransaction().begin();
            em.merge(user);
            em.getTransaction().commit();
            return 0;
        } catch (Exception e) {
            e.printStackTrace();
            return 2;
        } finally {
            em.close();
        }

    }

    //添加一个收藏，返回状态：0 success，1 不存在,2其他原因
    @Override
    public boolean AddCollect(String userid, String seedid) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<User> query2 = new JPAQuery<>(em);
            QUser u2 = QUser.user;
            User user = query2.select(u2).from(u2).where(u2.userid.eq(userid)).fetchOne();
            if (user == null) {
                return false;
            }
            JPAQuery<Seed> query3 = new JPAQuery<>(em);
            QSeed p = QSeed.seed;
            Seed seed = query3.select(p).from(p).where(p.seedid.eq(seedid)).fetchOne();
            if (seed == null) {
                return false;
            }
            JPAQuery<String> query = new JPAQuery<>(em);
            QUserStar u = QUserStar.userStar;
            List<String> allSeedId = query.select(u.seedid).from(u).where(u.userid.eq(userid)).fetch();

            if (allSeedId.contains(seedid)) {
                return false;
            }
            UserStar userStar = new UserStar();
            userStar.userid = userid;
            userStar.seedid = seedid;
            em.getTransaction().begin();
            em.persist(userStar);
            em.getTransaction().commit();
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        } finally {
            em.close();
        }
    }

    //删除一个收藏，返回状态：0 success，1 不存在,2其他原因
    @Override
    public boolean DeleteCollect(String userid, String seedid) {
        EntityManager em = getEntityManager();
        try {
            JPAQuery<UserStar> query = new JPAQuery<>(em);
            QUserStar u = QUserStar.userStar;
            UserStar userStar = query.select(u).from(u).where(u.userid.eq(userid).and(u.seedid.eq(seedid))).fetchOne();
            if (userStar == null) {
                return true; // 收藏不存在
            }
            em.getTransaction().begin();
            em.remove(userStar);
            em.getTransaction().commit();
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        } finally {
            em.close();
        }

    }

    @Override
    public int AddBegSeed(BegInfo info) {
        return 0;
    }

    @Override
    public int UpdateBegSeed(BegInfo info) {
        return 0;
    }

    @Override
    public int DeleteBegSeed(String begid) {
        return 0;
    }

    @Override
    public int VoteSeed(String begId, String seedId, String userId) {
        return 0;
    }

    @Override
    public int SubmitSeed(String begid, Seed seed) {
        return 0;
    }

    @Override
    public void SettleBeg() {

    }

    @Override
    public int AddPost(Post post) {
        return 0;
    }

    @Override
    public int UpdatePost(Post post) {
        return 0;
    }

    @Override
    public int DeletePost(String postid) {
        return 0;
    }

    @Override
    public int AddComment(String postid, String userid, String comment) {
        return 0;
    }

    @Override
    public int DeleteComment(String postid, String commentid) {
        return 0;
    }

    @Override
    public boolean ExchangeMagicToUpload(String userid, int magic)//将魔力值兑换为上传量，返回状态：0 success，1 不存在,2其他原因
    {
        return true;
    }

    @Override
    public boolean ExchangeMagicToDownload(String userid, int magic) {
        return true;
    }//将魔力值兑换为下载量，返回状态：0 success，1 不存在,2其他原因

    @Override
    public boolean ExchangeMagicToVip(String userid, int magic) {
        return true;
    }
    //将魔力值兑换为VIP次数，返回状态：0 success，1 不存在,2其他原因

    @Override
    public boolean UploadTransmitProfile(Profile profile) {
        return true;
    }

    @Override
    public Profile GetTransmitProfile(String profileid) {
        Profile profile = new Profile();
        return profile;
    }
    //获取迁移信息

    @Override
    public boolean ExamTransmitProfile(String profileid, boolean result) {
        return true;
    }

    //审核迁移信息,0成功，1失败
    @Override
    public Profile[] GetTransmitProfileList() {
        return new Profile[0];
    }
    //获取所有迁移信息

}