Hibernate - SortedSet 映射

SortedSet 是一个不包含任何重复元素的 java 集合,并且元素使用其自然顺序或提供的比较器进行排序。

SortedSet 使用映射表中的 <set> 元素进行映射,并使用 java.util.TreeSet 进行初始化。 sort 属性可以设置为比较器或自然排序。 如果我们使用 natural 排序,那么它的迭代器将按元素升序遍历集合。


定义 RDBMS 表

考虑一种情况,我们需要将员工记录存储在 EMPLOYEE 表中,该表将具有以下结构 −

create table EMPLOYEE (
   id INT NOT NULL auto_increment,
   first_name VARCHAR(20) default NULL,
   last_name  VARCHAR(20) default NULL,
   salary     INT  default NULL,
   PRIMARY KEY (id)
);

此外,假设每个员工都可以拥有一个或多个与其相关联的证书。 因此,我们将证书相关信息存储在具有以下结构的单独表中 −

create table CERTIFICATE (
   id INT NOT NULL auto_increment,
   certificate_name VARCHAR(30) default NULL,
   employee_id INT default NULL,
   PRIMARY KEY (id)
);

EMPLOYEE 和 CERTIFICATE 对象之间会有一对多的关系 −


定义 POJO 类

让我们实现我们的 POJO 类 Employee,它将用于持久化与 EMPLOYEE 表相关的对象,并在 SortedSet 变量中拥有一组证书。

import java.util.*;

public class Employee {
   private int id;
   private String firstName; 
   private String lastName;   
   private int salary;
   private SortedSet certificates;

   public Employee() {}
   
   public Employee(String fname, String lname, int salary) {
      this.firstName = fname;
      this.lastName = lname;
      this.salary = salary;
   }
   
   public int getId() {
      return id;
   }
   
   public void setId( int id ) {
      this.id = id;
   }
   
   public String getFirstName() {
      return firstName;
   }
   
   public void setFirstName( String first_name ) {
      this.firstName = first_name;
   }
   
   public String getLastName() {
      return lastName;
   }
   
   public void setLastName( String last_name ) {
      this.lastName = last_name;
   }
   
   public int getSalary() {
      return salary;
   }
   
   public void setSalary( int salary ) {
      this.salary = salary;
   }

   public SortedSet getCertificates() {
      return certificates;
   }
   
   public void setCertificates( SortedSet certificates ) {
      this.certificates = certificates;
   }
}

现在让我们定义另一个与 CERTIFICATE 表对应的 POJO 类,以便可以将证书对象存储和检索到 CERTIFICATE 表中。该类还应实现 Comparable 接口和 compareTo 方法,如果您在映射文件中设置 sort="natural",该方法将用于对元素进行排序(参见下面的映射文件) −

public class Certificate implements Comparable <Certificate>{
   private int id;
   private String name; 

   public Certificate() {}
   
   public Certificate(String name) {
      this.name = name;
   }
   
   public int getId() {
      return id;
   }
   
   public void setId( int id ) {
      this.id = id;
   }
   
   public String getName() {
      return name;
   }
   
   public void setName( String name ) {
      this.name = name;
   }
   
   public int compareTo(Certificate that){
      final int BEFORE = -1;
      final int AFTER = 1;

      if (that == null) {
         return BEFORE;
      }

      Comparable thisCertificate = this.getName();
      Comparable thatCertificate = that.getName();

      if(thisCertificate == null) {
         return AFTER;
      } else if(thatCertificate == null) {
         return BEFORE;
      } else {
         return thisCertificate.compareTo(thatCertificate);
      }
   }
}

定义 Hibernate 映射文件

让我们开发我们的映射文件,它指示 Hibernate 如何将定义的类映射到数据库表。 <set> 元素将用于定义所使用的 SortedSet 集合的规则。

<?xml version = "1.0" encoding = "utf-8"?>
<!DOCTYPE hibernate-mapping PUBLIC 
"-//Hibernate/Hibernate Mapping DTD//EN"
"http://www.hibernate.org/dtd/hibernate-mapping-3.0.dtd"> 

<hibernate-mapping>
   <class name = "Employee" table = "EMPLOYEE">
      <meta attribute = "class-description">
         This class contains the employee detail. 
      </meta>
      <id name = "id" type = "int" column = "id">
         <generator class="native"/>
      </id>
      <set name = "certificates" cascade="all" sort="MyClass">
         <key column = "employee_id"/>
         <one-to-many class="Certificate"/>
      </set>
      <property name = "firstName" column = "first_name" type = "string"/>
      <property name = "lastName" column = "last_name" type = "string"/>
      <property name = "salary" column = "salary" type = "int"/>
   </class>

   <class name = "Certificate" table = "CERTIFICATE">
      <meta attribute = "class-description">
         This class contains the certificate records. 
      </meta>
      <id name = "id" type = "int" column = "id">
         <generator class="native"/>
      </id>
      <property name = "name" column = "certificate_name" type = "string"/>
   </class>

</hibernate-mapping>

您应该将映射文档保存在格式为 <classname>.hbm.xml 的文件中。我们将映射文档保存在文件 Employee.hbm.xml 中。 您已经熟悉了大部分映射细节,但让我们再次查看映射文件的所有元素 −

  • 映射文档是以 <hibernate-mapping> 为根元素的XML文档,每个类包含两个 <class> 元素。

  • <class> 元素用于定义从 Java 类到数据库表的特定映射。Java 类名使用类元素的 name 属性指定,数据库表名使用table 属性指定。

  • <meta> 元素是可选元素,可用于创建类描述。

  • <id> 元素将类中的唯一 ID 属性映射到数据库表的主键。id 元素的name 属性引用类中的属性,column 属性引用数据库表中的列。type 属性保存hibernate 映射类型,这种映射类型将从Java 转换为SQL 数据类型。

  • id 元素中的 <generator> 元素用于自动生成主键值。生成器元素的 class 属性设置为 native 以让 hibernate 选择任一 identity, sequencehilo 算法来根据底层数据库的功能创建主键。

  • <property> 元素用于将 Java 类属性映射到数据库表中的列。元素的name 属性引用类中的属性,column 属性引用数据库表中的列。type 属性保存hibernate 映射类型,这种映射类型将从Java 转换为SQL 数据类型。

  • <set> 元素用于设置 Certificate 和 Employee 类之间的关系。 我们使用 <set> 元素中的 cascade 属性来告诉 Hibernate 将 Certificate 对象与 Employee 对象同时保存。name 属性设置为父类中定义的 SortedSet 变量,在我们的例子中是 certificatessort 属性可以设置为 natural 以进行自然排序,也可以设置为实现 java.util.Comparator 的自定义类。我们使用了实现 java.util.Comparator 的类 MyClass 来反转 Certificate 类中实现的排序顺序。

  • <key> 元素是 CERTIFICATE 表中的列,它保存父对象的外键,即表 EMPLOYEE。

  • <one-to-many> 元素表示一个 Employee 对象与许多 Certificate 对象相关,因此,Certificate 对象必须有一个与之关联的 Employee 父对象。 您可以根据需要使用任一 <one-to-one>, <many-to-one><many-to-many> 元素。

如果我们使用 sort="natural" 设置,那么我们不需要创建一个单独的类,因为 Certificate 类已经实现了 Comparable 接口,hibernate 将使用 Certificate 类中定义的 compareTo() 方法来比较证书名称。但是我们在映射文件中使用了自定义比较器类MyClass,因此我们必须根据我们的排序算法创建这个类。 让我们使用这个类在这个类中进行降序排序。

import java.util.Comparator;
我们使用 a1 元素中的 cascade 属性来告诉 Hibernate 将 Certificate 对象与 Employee 对象同时保存。
public class MyClass implements Comparator<Certificate>{
   public int compare(Certificate o1, Certificate o2) {
      final int BEFORE = -1;
      final int AFTER = 1;

      /* To reverse the sorting order, multiple by -1 */
      if (o2 == null) {
         return BEFORE * -1;
      }

      Comparable thisCertificate = o1.getName();
      Comparable thatCertificate = o2.getName();

      if(thisCertificate == null) {
         return AFTER * 1;
      } else if(thatCertificate == null) {
         return BEFORE * -1;
      } else {
         return thisCertificate.compareTo(thatCertificate) * -1;
      }
   }
}

创建应用程序类

最后,我们将使用 main() 方法创建我们的应用程序类来运行应用程序。 我们将使用此应用程序保存少数员工的记录及其证书,然后我们将对这些记录应用 CRUD 操作。

import java.util.*;
 
import org.hibernate.HibernateException; 
import org.hibernate.Session; 
import org.hibernate.Transaction;
import org.hibernate.SessionFactory;
import org.hibernate.cfg.Configuration;

public class ManageEmployee {
   private static SessionFactory factory; 
   public static void main(String[] args) {
      try{
         factory = new Configuration().configure().buildSessionFactory();
      }catch (Throwable ex) { 
         System.err.println("Failed to create sessionFactory object." + ex);
         throw new ExceptionInInitializerError(ex); 
      }
		
      ManageEmployee ME = new ManageEmployee();
      /* Let us have a set of certificates for the first employee  */
      TreeSet set1 = new TreeSet();
      set1.add(new Certificate("MCA"));
      set1.add(new Certificate("MBA"));
      set1.add(new Certificate("PMP"));
     
      /* Add employee records in the database */
      Integer empID1 = ME.addEmployee("Manoj", "Kumar", 4000, set1);

      /* Another set of certificates for the second employee  */
      TreeSet set2 = new TreeSet();
      set2.add(new Certificate("BCA"));
      set2.add(new Certificate("BA"));

      /* Add another employee record in the database */
      Integer empID2 = ME.addEmployee("Dilip", "Kumar", 3000, set2);

      /* List down all the employees */
      ME.listEmployees();

      /* Update employee's salary records */
      ME.updateEmployee(empID1, 5000);

      /* Delete an employee from the database */
      ME.deleteEmployee(empID2);

      /* List down all the employees */
      ME.listEmployees();

   }

   /* Method to add an employee record in the database */
   public Integer addEmployee(String fname, String lname, int salary, SortedSet cert){
      Session session = factory.openSession();
      Transaction tx = null;
      Integer employeeID = null;
      try{
         tx = session.beginTransaction();
         Employee employee = new Employee(fname, lname, salary);
         employee.setCertificates(cert);
         employeeID = (Integer) session.save(employee); 
         tx.commit();
      }catch (HibernateException e) {
         if (tx!=null) tx.rollback();
         e.printStackTrace(); 
      }finally {
         session.close(); 
      }
      return employeeID;
   }

   /* Method to list all the employees detail */
   public void listEmployees( ){
      Session session = factory.openSession();
      Transaction tx = null;
      try{
         tx = session.beginTransaction();
         List employees = session.createQuery("FROM Employee").list(); 
         for (Iterator iterator1 = employees.iterator(); iterator1.hasNext();){
            Employee employee = (Employee) iterator1.next(); 
            System.out.print("First Name: " + employee.getFirstName()); 
            System.out.print("  Last Name: " + employee.getLastName()); 
            System.out.println("  Salary: " + employee.getSalary());
            SortedSet certificates = employee.getCertificates();
            for (Iterator iterator2 = certificates.iterator(); iterator2.hasNext();){
               Certificate certName = (Certificate) iterator2.next(); 
               System.out.println("Certificate: " + certName.getName()); 
            }
         }
         tx.commit();
      }catch (HibernateException e) {
         if (tx!=null) tx.rollback();
         e.printStackTrace(); 
      }finally {
         session.close(); 
      }
   }
   /* Method to update salary for an employee */
   public void updateEmployee(Integer EmployeeID, int salary ){
      Session session = factory.openSession();
      Transaction tx = null;
      try{
         tx = session.beginTransaction();
         Employee employee = (Employee)session.get(Employee.class, EmployeeID); 
         employee.setSalary( salary );
         session.update(employee);
         tx.commit();
      }catch (HibernateException e) {
         if (tx!=null) tx.rollback();
         e.printStackTrace(); 
      }finally {
         session.close(); 
      }
   }
   /* Method to delete an employee from the records */
   public void deleteEmployee(Integer EmployeeID){
      Session session = factory.openSession();
      Transaction tx = null;
      try{
         tx = session.beginTransaction();
         Employee employee = (Employee)session.get(Employee.class, EmployeeID); 
         session.delete(employee); 
         tx.commit();
      }catch (HibernateException e) {
         if (tx!=null) tx.rollback();
         e.printStackTrace(); 
      }finally {
         session.close(); 
      }
   }
}

编译与执行

以下是编译和运行上述应用程序的步骤。 在继续编译和执行之前,请确保您已正确设置 PATH 和 CLASSPATH。

  • 按照配置章节中的说明创建 hibernate.cfg.xml 配置文件。

  • 如上所示创建 Employee.hbm.xml 映射文件。

  • 如上所示创建 Employee.java 源文件并编译它。

  • 如上图创建 Certificate.java源文件并编译。

  • Create MyClass.java source file as shown above and compile it.

  • 如上图创建 ManageEmployee.java源文件并编译。

  • 执行 ManageEmployee 二进制文件以运行程序。

您将在屏幕上获得以下结果,同时将在 EMPLOYEE 和 CERTIFICATE 表中创建记录。您可以看到证书已按相反顺序排序。 您可以尝试更改映射文件,只需设置 sort="natural" 并执行您的程序并比较结果。

$java ManageEmployee
.......VARIOUS LOG MESSAGES WILL DISPLAY HERE........

First Name: Manoj  Last Name: Kumar  Salary: 4000
Certificate: PMP
Certificate: MCA
Certificate: MBA
First Name: Dilip  Last Name: Kumar  Salary: 3000
Certificate: BCA
Certificate: BA
First Name: Manoj  Last Name: Kumar  Salary: 5000
Certificate: PMP
Certificate: MCA
Certificate: MBA

如果您检查您的 EMPLOYEE 和 CERTIFICATE 表,它们应该有以下记录 −

mysql> select * from employee;
+----+------------+-----------+--------+
| id | first_name | last_name | salary |
+----+------------+-----------+--------+
| 1  | Manoj      | Kumar     |   5000 |
+----+------------+-----------+--------+
1 row in set (0.00 sec)

mysql> select * from certificate;
+----+------------------+-------------+
| id | certificate_name | employee_id |
+----+------------------+-------------+
| 1  | MBA              |          1  |
| 2  | PMP              |          1  |
| 3  | MCA              |          1  |
+----+------------------+-------------+
3 rows in set (0.00 sec)

mysql>

❮ Hibernate - O/R 映射