【043期】面试官问：如何使用 Redis 实现电商系统的库存扣减？

private RedisTemplate<String, Object> redisTemplate;

/**

* 执行扣库存的脚本

*/

public static final String STOCK_LUA;

static {

/**

* @desc 扣减库存Lua脚本

* 库存（stock）-1：表示不限库存

* 库存（stock）0：表示没有库存

* 库存（stock）大于0：表示剩余库存

* @params 库存key

* @return

*   -3:库存未初始化

*   -2:库存不足

*   -1:不限库存

*   大于等于0:剩余库存（扣减之后剩余的库存）

*      redis缓存的库存(value)是-1表示不限库存，直接返回1

*/

StringBuilder sb = new StringBuilder();

sb.append(“if (redis.call(‘exists’, KEYS[1]) == 1) then”);

sb.append("    local stock = tonumber(redis.call(‘get’, KEYS[1]));");

sb.append("    local num = tonumber(ARGV[1]);");

sb.append("    if (stock == -1) then");

sb.append("        return -1;");

sb.append("    end;");

sb.append("    if (stock >= num) then");

sb.append("        return redis.call(‘incrby’, KEYS[1], 0 - num);");

sb.append("    end;");

sb.append("    return -2;");

sb.append(“end;”);

sb.append(“return -3;”);

STOCK_LUA = sb.toString();

}

/**

* @param key           库存key

* @param expire        库存有效时间,单位秒

* @param num           扣减数量

* @param stockCallback 初始化库存回调函数

* @return -2:库存不足; -1:不限库存; 大于等于0:扣减库存之后的剩余库存

*/

public long stock(String key, long expire, int num, IStockCallback stockCallback) {

long stock = stock(key, num);

// 初始化库存

if (stock == UNINITIALIZED_STOCK) {

RedisLock redisLock = new RedisLock(redisTemplate, key);

try {

// 获取锁

if (redisLock.tryLock()) {

// 双重验证，避免并发时重复回源到数据库

stock = stock(key, num);

if (stock == UNINITIALIZED_STOCK) {

// 获取初始化库存

final int initStock = stockCallback.getStock();

// 将库存设置到redis

redisTemplate.opsForValue().set(key, initStock, expire, TimeUnit.SECONDS);

// 调一次扣库存的操作

stock = stock(key, num);

}

}

} catch (Exception e) {

logger.error(e.getMessage(), e);

} finally {

redisLock.unlock();

}

}

return stock;

}

/**

* 加库存(还原库存)

* @param key    库存key

* @param num    库存数量

* @return

*/

public long addStock(String key, int num) {

return addStock(key, null, num);

}

/**

* 加库存

* @param key    库存key

* @param expire 过期时间（秒）

* @param num    库存数量

* @return

*/

public long addStock(String key, Long expire, int num) {

boolean hasKey = redisTemplate.hasKey(key);

// 判断key是否存在，存在就直接更新

if (hasKey) {

return redisTemplate.opsForValue().increment(key, num);

}

Assert.notNull(expire,“初始化库存失败，库存过期时间不能为null”);

RedisLock redisLock = new RedisLock(redisTemplate, key);

try {

if (redisLock.tryLock()) {

// 获取到锁后再次判断一下是否有key

hasKey = redisTemplate.hasKey(key);

if (!hasKey) {

// 初始化库存

redisTemplate.opsForValue().set(key, num, expire, TimeUnit.SECONDS);

}

}

} catch (Exception e) {

logger.error(e.getMessage(), e);

} finally {

redisLock.unlock();

}

return num;

}

/**

* 获取库存

* @param key 库存key

* @return -1:不限库存; 大于等于0:剩余库存

*/

public int getStock(String key) {

Integer stock = (Integer) redisTemplate.opsForValue().get(key);

return stock == null ? -1 : stock;

}

/**

* 扣库存

* @param key 库存key

* @param num 扣减库存数量

* @return 扣减之后剩余的库存【-3:库存未初始化; -2:库存不足; -1:不限库存; 大于等于0:扣减库存之后的剩余库存】

*/

private Long stock(String key, int num) {

// 脚本里的KEYS参数

List keys = new ArrayList<>();

keys.add(key);

// 脚本里的ARGV参数

List args = new ArrayList<>();

args.add(Integer.toString(num));

long result = redisTemplate.execute(new RedisCallback() {

@Override

public Long doInRedis(RedisConnection connection) throws DataAccessException {

Object nativeConnection = connection.getNativeConnection();

// 集群模式和单机模式虽然执行脚本的方法一样，但是没有共同的接口，所以只能分开执行

// 集群模式

if (nativeConnection instanceof JedisCluster) {

return (Long) ((JedisCluster) nativeConnection).eval(STOCK_LUA, keys, args);

}

// 单机模式

else if (nativeConnection instanceof Jedis) {

return (Long) ((Jedis) nativeConnection).eval(STOCK_LUA, keys, args);

}

return UNINITIALIZED_STOCK;

}

});

return result;

}

}

调用

/**

* @author yuhao.wang

*/

@RestController

public class StockController {

@Autowired

private StockService stockService;

@RequestMapping(value = “stock”, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)

public Object stock() {

// 商品ID

long commodityId = 1;

// 库存ID

String redisKey = “redis_key:stock:” + commodityId;

long stock = stockService.stock(redisKey, 60 * 60, 2, () -> initStock(commodityId));

return stock >= 0;

}

/**

* 获取初始的库存

* @return

*/

private int initStock(long commodityId) {

// TODO 这里做一些初始化库存的操作

return 1000;

}

@RequestMapping(value = “getStock”, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)

public Object getStock() {

// 商品ID

long commodityId = 1;

// 库存ID

String redisKey = “redis_key:stock:” + commodityId;

return stockService.getStock(redisKey);

}

@RequestMapping(value = “addStock”, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)

public Object addStock() {

// 商品ID

long commodityId = 2;

// 库存ID

String redisKey = “redis_key:stock:” + commodityId;

return stockService.addStock(redisKey, 2);

}

}

参考

–

• http://www.cnblogs.com/billyxp/p/3701124.html

• http://blog.youkuaiyun.com/jiao_fuyou/article/details/15504777

• https://www.jianshu.com/p/48c1a92fbf3a

作者： 拥抱心中的梦想

juejin.cn/post/6844903581678600206

往期精选  点击标题可

【033期】面试官问：说一说 Spring 中接口 bean 是如何注入的吗？

【034期】美团面试题：JVM 堆内存溢出后，其他线程是否可继续工作？

【035期】面试官问：什么是耦合？解耦合的方法有哪几种？

Spring全套教学资料

Spring是Java程序员的《葵花宝典》，其中提供的各种大招，能简化我们的开发，大大提升开发效率！目前99%的公司使用了Spring，大家可以去各大招聘网站看一下，Spring算是必备技能，所以一定要掌握。

目录：

部分内容：

Spring源码

• 第一部分 Spring 概述
• 第二部分 核心思想
• 第三部分 手写实现 IoC 和 AOP（自定义Spring框架）
• 第四部分 Spring IOC 高级应用
  基础特性
  高级特性
• 第五部分 Spring IOC源码深度剖析
  设计优雅
  设计模式
  注意：原则、方法和技巧
• 第六部分 Spring AOP 应用
  声明事务控制
• 第七部分 Spring AOP源码深度剖析
  必要的笔记、必要的图、通俗易懂的语言化解知识难点

脚手框架：SpringBoot技术

它的目标是简化Spring应用和服务的创建、开发与部署，简化了配置文件，使用嵌入式web服务器，含有诸多开箱即用的微服务功能，可以和spring cloud联合部署。

Spring Boot的核心思想是约定大于配置，应用只需要很少的配置即可，简化了应用开发模式。

• SpringBoot入门
• 配置文件
• 日志
• Web开发
• Docker
• SpringBoot与数据访问
• 启动配置原理
• 自定义starter

微服务架构：Spring Cloud Alibaba

同 Spring Cloud 一样，Spring Cloud Alibaba 也是一套微服务解决方案，包含开发分布式应用微服务的必需组件，方便开发者通过 Spring Cloud 编程模型轻松使用这些组件来开发分布式应用服务。

• 微服务架构介绍
• Spring Cloud Alibaba介绍
• 微服务环境搭建
• 服务治理
• 服务容错
• 服务网关
• 链路追踪
• ZipKin集成及数据持久化
• 消息驱动
• 短信服务
• Nacos Confifig—服务配置
• Seata—分布式事务
• Dubbo—rpc通信

Spring MVC

目录：

部分内容：

件，方便开发者通过 Spring Cloud 编程模型轻松使用这些组件来开发分布式应用服务。

• 微服务架构介绍
• Spring Cloud Alibaba介绍
• 微服务环境搭建
• 服务治理
• 服务容错
• 服务网关
• 链路追踪
• ZipKin集成及数据持久化
• 消息驱动
• 短信服务
• Nacos Confifig—服务配置
• Seata—分布式事务
• Dubbo—rpc通信

[外链图片转存中…(img-gu9imbDA-1718906242889)]

[外链图片转存中…(img-FX4tFUrf-1718906242889)]

Spring MVC

目录：

[外链图片转存中…(img-9pMJv6Tf-1718906242890)]

[外链图片转存中…(img-E2w1Vson-1718906242890)]

[外链图片转存中…(img-1LJ8gDac-1718906242891)]

部分内容：

[外链图片转存中…(img-CmCTHeQL-1718906242891)]

[外链图片转存中…(img-OZ0IPMLb-1718906242891)]