多线程与hibernate openSession绑定在当前线程

小鱼儿 2023-06-19 09:40 3阅读 0赞

先上代码：

package com.ifunpay.portal.task;
    
    import com.alibaba.fastjson.JSON;
    import com.ifunpay.portal.dao.order.OrderEntityDao;
    import com.ifunpay.portal.entity.order.OrderEntity;
    import com.ifunpay.portal.model.PaidInfo;
    import com.ifunpay.portal.service.PaymentManageService;
    import com.ifunpay.portal.service.payment.AbcAggPaymentManager;
    import com.ifunpay.util.enums.BankAppEnum;
    import com.ifunpay.util.enums.PaymentStatusEnum;
    import lombok.extern.log4j.Log4j2;
    import org.hibernate.FlushMode;
    import org.hibernate.Session;
    import org.hibernate.SessionFactory;
    import org.springframework.context.ApplicationContext;
    import org.springframework.orm.hibernate4.SessionFactoryUtils;
    import org.springframework.orm.hibernate4.SessionHolder;
    import org.springframework.stereotype.Component;
    import org.springframework.transaction.support.TransactionSynchronizationManager;
    import org.springframework.web.servlet.ModelAndView;
    
    import java.util.Map;
    
    
    /**
     * @Author: lilong
     * @Data: Created on 2019/12/3
     * @Desc: 线程池
     */
    @Log4j2
    @Component
    public class AbcAggTask implements Runnable {
        public static final String SUCCESS  = "SUCCESS";
    
        private  PaymentManageService paymentManageService;
    
        private OrderEntityDao orderEntityDao;
    
        private ApplicationContext applicationContext;
    
        protected String orderId;
        protected String commercialId;
        protected String privateKey;
    
        public AbcAggTask(String commercialId,String orderId,String privateKey,OrderEntityDao orderEntityDao,PaymentManageService paymentManageService,ApplicationContext applicationContext) {
            this.commercialId = commercialId;
            this.orderId = orderId;
            this.privateKey = privateKey;
            this.orderEntityDao = orderEntityDao;
            this.paymentManageService = paymentManageService;
            this.applicationContext = applicationContext;
        }
    
        @Override
        public void run() {
            SessionFactory sessionFactory = null;
            boolean participate = false ;
            try {
                //重点在这两行代码~绑定session到当前线程 
                sessionFactory = (SessionFactory)applicationContext.getBean("sessionFactory");
                participate = bindHibernateSessionToThread(sessionFactory);
                //业务逻辑代码
                OrderEntity order = orderEntityDao.findByOrderId(orderId);
                if (PaymentStatusEnum.NotPay.equals(order.getPaymentStatus())) {
                    log.info("abcAggCallBackIsNull...query....orderId:"+orderId);
                    for (int i = 1; i < 21; i++) {
                        if (i == 1) {
                            Thread.sleep(30000);
                        } else {
                            Thread.sleep(5000);
                        }
                        AbcAggPaymentManager abcAggPaymentManager = new AbcAggPaymentManager();
                        String result = abcAggPaymentManager.query(commercialId, orderId, privateKey);
                        Map<String,String> returnMap = JSON.parseObject(result,Map.class);
                        if (SUCCESS.equals(returnMap.get("code")) && SUCCESS.equals(returnMap.get("trade_state"))){ //订单支付成功
                            ModelAndView modelAndView = new ModelAndView();
                            modelAndView.setViewName("payment/callback");
                            PaidInfo paidInfo = new PaidInfo();
                            paidInfo.setModelAndView(modelAndView);
                            paidInfo.setBank(BankAppEnum.ABCAGG);
                            paidInfo.setAmount(order.getTotalAmount());
                            paidInfo.setStatus(PaymentStatusEnum.PaySucceeded);
                            paidInfo.setOrderId(order.getCode());
                            paidInfo.setPosId(null);
                            paidInfo.setSerialNumber(returnMap.get("transaction_id"));
                            paidInfo.getModelAndView().setViewName("error");
                            paymentManageService.notifyCallbackObservers(paidInfo);
                            break;
                        }else {
                            log.error("getAbcAggQrcodeFiled...:code:"+returnMap.get("code")+";message:"+returnMap.get("message"));
                        }
                    }
                }
            } catch (Exception e) {
                log.error("abcAggTaskFailed：", e);
            }finally {
                //关闭Session
                closeHibernateSessionFromThread(participate,sessionFactory);
            }
        }
    
        /**
     * @Author: lilong
     * @Data: Created on 2019/12/3
     * @Desc: open一个session然后将其绑定在当前线程
     */
        public static boolean bindHibernateSessionToThread(SessionFactory sessionFactory){
            if (TransactionSynchronizationManager.hasResource(sessionFactory)){
                return true;
            }else {
                Session session = sessionFactory.openSession();
                session.setFlushMode(FlushMode.MANUAL);
                SessionHolder sessionHolder = new SessionHolder(session);
                TransactionSynchronizationManager.bindResource(sessionFactory,sessionHolder);//将open的session 绑定带线程
            }
            return false;
        }
    
            /**
     * @Author: lilong
     * @Data: Created on 2019/12/3
     * @Desc: open的session都要手动关闭
     */
    
        public static void closeHibernateSessionFromThread(boolean participate,Object sessionFactory){
            if (!participate){
                SessionHolder sessionHolder = (SessionHolder) TransactionSynchronizationManager.unbindResource(sessionFactory);
                SessionFactoryUtils.closeSession(sessionHolder.getSession());
            }
        }
    }

先了解下这个~~~[openSession()和getCurrentSession的区别][openSession_getCurrentSession]，然后看代码!

[openSession_getCurrentSession]: https://blog.csdn.net/qq_19167629/article/details/90263663