如何輕鬆搞定CRUD的創建人、修改人、時間等字段的賦值

作者 | 一起隨緣

來源 | https://juejin.cn/post/6984555714752561183

背景

數據庫設計過程中,我們往往會給數據庫表添加一些通用字段,比如創建人、創建時間、修改人、修改時間,在一些公司的設計過程中有時會強制要求每個表都要包含這些基礎信息,以便記錄數據操作時的一些基本日誌記錄。按照平常的操作來說,通用做法是輸寫sql時,將這些信息和對象的基本屬性信息一起寫入數據庫,當然,這也是大家習以爲常的操作,這種寫法無可厚非,但是對於一個高級開發人員來說,如果所有的表都進行如此操作,未免顯得有點囉嗦,而且數據表多的話,這樣寫就有點得不償失了。其實還有一種更簡便的做法,spring框架大家應該是比較熟悉的,幾乎每個公司都會用到,其中aop思想(切面編程)的經典應用場景之一就是日誌記錄,本文結合aop思想,着重介紹下springboot框架下如何利用切面編程思想實現將創建人、創建時間、更新人、更新時間等基礎信息寫入數據庫。

核心代碼

@Aspect
@Component
@Configuration
public class CommonDaoAspect {

    private static final String creater = "creater";
    private static final String createTime = "createTime";
    private static final String updater = "updater";
    private static final String updateTime = "updateTime";

    @Pointcut("execution(* com.xx.xxxx.*.dao.*.update*(..))")
    public void daoUpdate() {
    }

    @Pointcut("execution(* com.xx.xxxx.*.dao.*.insert*(..))")
    public void daoCreate() {
    }

    @Around("daoUpdate()")
    public Object doDaoUpdate(ProceedingJoinPoint pjp) throws Throwable {
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        if (attributes == null) {
            return pjp.proceed();
        }
        HttpServletRequest request = attributes.getRequest();
        String token = request.getHeader("token");
        String username = getUserName();
        if (token != null && username != null) {
            Object[] objects = pjp.getArgs();
            if (objects != null && objects.length > 0) {
                for (Object arg : objects) {
                    BeanUtils.setProperty(arg, updater, username);
                    BeanUtils.setProperty(arg, updateTime, new Date());
                }
            }
        }
        Object object = pjp.proceed();
        return object;

    }

    @Around("daoCreate()")
    public Object doDaoCreate(ProceedingJoinPoint pjp) throws Throwable {
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        if (attributes == null) {
            return pjp.proceed();
        }
        Object[] objects = pjp.getArgs();
        if (objects != null && objects.length > 0) {
            for (Object arg : objects) {
                String username = getUserName();
                if (username != null) {
                    if (StringUtils.isBlank(BeanUtils.getProperty(arg, creater))) {
                        BeanUtils.setProperty(arg, creater, username);
                    }
                    if (StringUtils.isBlank(BeanUtils.getProperty(arg, createTime))) {
                        BeanUtils.setProperty(arg, createTime, new Date());
                    }
                }
            }
        }
        Object object = pjp.proceed();
        return object;
    }

    private String getUserName() {
        return UserUtils.getUsername();
    }
}

代碼介紹及註解說明

1.代碼介紹

核心代碼聲明瞭一個CommonDaoAspect切面類,實體類中聲明瞭4個核心方法和一個獲取用戶名信息的方法,UserUtils是項目中聲明的工具類,包含獲取用戶id、姓名等一些基礎信息,大家可以根據自己的實際情況去定義,不要照部就搬。4個核心方法中,daoUpdate和daoCreate上添加了@Pointcut註解,該註解通過聲明正則表達式來確定項目包中dao目錄下哪些方法執行該切面方法。doDaoUpdate和doDaoCreate方法上添加了@Around註解,註解中引入了上述兩個方法,表示環繞通知,在我們自己dao目錄下的對應文件目標方法完成前後做增強處理。

2.註解說明

  • @Aspect:聲明切面類,裏面可以定義切入點和通知
  • @Component:表明該類是spring管理的一個對象
  • @Pointcut:切入點,通過正則表達式聲明切入的時機,本文中是在目標方法(即項目中dao目錄下實體類中包含insert或update字符串的方法)執行時加入切入信息,即執行新增或更新時加入創建人和更新人等信息。
  • @Around:環繞通知,在目標方法完成前後做增強處理,本案例中表示在doCreate和doUpdate方法執行時添加參數信息
  • 如果您正在學習Spring Boot,推薦一個連載多年還在繼續更新的免費教程:http://blog.didispace.com/spring-boot-learning-2x/

注:execution(* com.xx.xxxx.*.dao.*.update*(..)) 表示在dao目錄下的任何文件中的以update開頭的方法

execution(* com.xx.xxxx.*.dao.*.insert*(..)) 表示在dao目錄下的任何文件中的以insert開頭的方法

由於才疏學淺,能力有限,對於文章中描述不準確或不恰當的地方,希望大家能夠及時指出和糾正,在編碼的路上一起努力前行。如果這篇文章對你有幫助的話,也希望能夠給點贊加關注哦,你的認可和喜歡是我持續輸出的最大動力。

- END -

往期推薦

架構師必備技能:Maven Archetype生成項目模板

如何更快地將string轉換成int/long

OAuth2 服務器Keycloak中的Realm

Java 17 將至,可能帶來哪些新特性呢?

機械妖姬上門要源碼後續結果來了!



喜歡本文歡迎轉發,關注我訂閱更多精彩

關注我回復「加羣」,加入Spring技術交流羣

本文分享自微信公衆號 - 程序猿DD(didispace)。
如有侵權,請聯繫 [email protected] 刪除。
本文參與“OSC源創計劃”,歡迎正在閱讀的你也加入,一起分享。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章