jpa~為欄位新增insert的預設值

张占岭發表於2024-09-27

前言

如果多個實體類都有 isDelete 欄位，並且你希望在插入時為它們統一設定預設值，可以採取以下幾種方法來減少程式碼重複：

1. 使用基類（抽象類）

建立一個基類，其中包含 isDelete 欄位和 @PrePersist 方法。然後讓所有需要這個欄位的實體類繼承這個基類。

示例程式碼：

import javax.persistence.MappedSuperclass;
import javax.persistence.PrePersist;

@MappedSuperclass
public abstract class BaseEntity {

    protected Integer isDelete;

    @PrePersist
    public void prePersist() {
        if (isDelete == null) {
            isDelete = 0; // 設定預設值為0
        }
    }

    // Getter 和 Setter
    public Integer getIsDelete() {
        return isDelete;
    }

    public void setIsDelete(Integer isDelete) {
        this.isDelete = isDelete;
    }
}

然後在其他實體類中繼承 BaseEntity：

import javax.persistence.Entity;
import javax.persistence.Id;

@Entity
public class MyEntity extends BaseEntity {

    @Id
    private Long id;

    // 其他欄位、getter 和 setter
}

2. 使用 AOP（面向切面程式設計）

透過 Spring AOP 建立一個切面，在插入操作時檢查並設定 isDelete 的預設值。這種方式不需要修改每個實體類，適合大規模應用。

示例程式碼：

import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.stereotype.Component;

import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
import java.lang.reflect.Field;

@Aspect
@Component
public class DefaultValueAspect {

    @PersistenceContext
    private EntityManager entityManager;

    @Before("execution(* com.example.repository.*.save(..))") // 根據你的倉庫路徑調整
    public void setDefaultValues(Object entity) throws IllegalAccessException {
        Field[] fields = entity.getClass().getDeclaredFields();
        for (Field field : fields) {
            if ("isDelete".equals(field.getName())) { // 檢查欄位名
                field.setAccessible(true);
                if (field.get(entity) == null) {
                    field.set(entity, 0); // 設定預設值為0
                }
            }
        }
    }
}

3. 使用 JPA 審計功能

使用 Spring Data JPA 的審計功能，透過實現 AuditorAware 介面來統一處理審計欄位，包括 建立時間，更新時間，建立人，刪除人等，這個功能比較侷限

示例程式碼：

@EntityListeners(AuditingEntityListener.class)
public abstract class EntityBase implements Serializable {

	/**
	 * 建立人
	 */
	@Column(name = "created_by")
	@CreatedBy
	private String createdBy;

	/**
	 * 修改人
	 */
	@Column(name = "updated_by")
	@LastModifiedBy
	private String updatedBy;

	/**
	 * 建立時間
	 */
	@Column(name = "created_time")
	@CreatedDate
	private java.time.LocalDateTime createdTime;

	/**
	 * 更新時間
	 */
	@Column(name = "update_time")
	@LastModifiedDate
	private java.time.LocalDateTime updateTime;

}

// 基於dubbo下的netty中獲取當前請求頭中的使用者名稱
@EnableJpaAuditing
@Component
public class AuditorAwareImpl implements AuditorAware<String> {

	@Override
	public Optional<String> getCurrentAuditor() {
		Object request = RpcContext.getServiceContext().getRequest();
		if (request != null) {
			NettyRequestFacade requestFacade = (NettyRequestFacade) request;

			if (requestFacade != null && requestFacade.getHeader("username") != null) {
				return Optional.of(requestFacade.getHeader("username"));
			}
		}
		return Optional.of("none");
	}

}

4. 使用事件監聽@EntityListeners

JPA 提供了事件監聽器的功能，你可以定義一個事件監聽器來處理所有需要設定預設值的實體類。

示例程式碼：

import javax.persistence.PostLoad;
import javax.persistence.PrePersist;
import javax.persistence.EntityListeners;

public interface DeletedField {

  	Integer getDeletedFlag();

	void setDeletedFlag(Integer deletedFlag);
}

public class DeleteDefaultValueListener {

	@PrePersist
	public void setDefaultValues(DeletedFlagField deletedFlagField) {
		if (deletedFlagField.getDeletedFlag() == null) {
			deletedFlagField.setDeletedFlag(0); // 設定預設值為0
		}
	}

}

@EntityListeners(DefaultValueListener.class)
@Entity
public class TableUserAccount extends EntityBase implements DeletedFlagField {

  	/**
	 * 刪除標識（邏輯刪除），1刪除 0未刪除
	 */
	@Column(name = "deleted_flag")
	private Integer deletedFlag;
}

相關文章