最近在实现一个微信小程序新用户注册领取红包的功能,业务逻辑是:
新用户领取红包,首先通过用户id红包oid两个字段查询资金流水表,如果有记录就说明领取过了,不允许领取,否则往红包记录表中insert一条记录。以此来保证数据的唯一性。
压力测试时,遇到问题:当多个线程携带同个token访问该url时,就会出现插入多条记录的情况,测试代码如下:
public class ThreadRunner implements Runnable {
@Override
public void run() {
String url = _redpack_api_url;//领取红包的api
System.out.println(HttpClientUtil.sendGet(url));
}
public static void main(String[] args) {
ExecutorService pool = Executors.newFixedThreadPool(10);
for(int i = 0 ; i < 10 ; i++){
pool.submit(new ThreadRunner());
}
pool.shutdown();
}
}
问题原因:自己流水表中包含其他业务类型的资金流水,所以没有对用户id和红包活动id做唯一键限制。在多线程频繁访问时,会出现,第一次领取红包的记录还没有插入数据库,其他线程开始检查资金流水,这是检查结果时没有领取过红包,所以两个线程都可以成功领取红包。
根本原因就是资金流水表没有设置唯一键。
解决方法:
由于业务限制,不可以在资金流水表中对用户oid和红包oid设置唯一键。
新建一张红包领取记录表,设置用户oid和红包oid为唯一键。每次领取红包会首先往红包领取记录表里面insert一条,如果insert成功再去资金流水表里面insert记录。
这样可以保证,无论前面的业务逻辑校验是否允许添加领取记录,都能够在数据库层级保证,领取记录是一条,否则所有操作回滚。