理解代码必备
- 参数中使用
(String... keys)
在 Java 中,String… keys 这种语法被称为可变参数(Varargs),是 Java 5 引入的特性,用于简化方法参数的定义。它允许方法接收任意数量(包括零个)的同一类型参数
用此可变参数可以省去创建String[]数组的功夫,而且不写和写几个都行,更加灵活
可变参数使用限制:
一个方法只能有一个可变参数
可变参数必须是方法的最后一个参数 - 对于
CountDownLatch
的使用
CountDownLatch 是 Java 并发包(java.util.concurrent)中的一个同步工具类,用于协调多个线程的执行顺序。它允许一个或多个线程等待其他线程完成操作后再继续执行,类似于一个 “倒计时门闩”
会对其进行计数,用一次减一进行计数,这就是为什么要用try,finally threadPool.execute(Runnable task)
threadPool.execute(Runnable task) 是 Java 线程池(ExecutorService)的核心方法之一,用于提交一个任务(Runnable 对象)到线程池执行。它是线程池管理多线程的基础操作,下面详细解释其功能和用法。
HashMap模拟数据库
package KaKe;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
public class SimpleDatabase {
private Map<String,String> dataStore;//HashMap模拟数据库存储数据
private ExecutorService threadPool;//线程池管理查询线程
//无参构造器
public SimpleDatabase(){
//给HashMap存储数据库创建对象
dataStore = new HashMap<>();
//线程池中创建三个线程
threadPool = Executors.newFixedThreadPool(3);
}
//模拟数据库插入
public void insert(String key,String value){
dataStore.put(key,value);
System.out.println("key:"+key+",value:"+value);
}
//多线程查询
public void queryWithThreads(String... keys){
if(keys==null || keys.length==0){
System.out.println("查询条件为空");
return;
}
CountDownLatch latch = new CountDownLatch(keys.length);//给keys数组中的每个key计数
System.out.println("启动多线程查询");
for (String key : keys) {
//提交任务到线程池中执行
threadPool.execute(()->{
try{
//模拟查询操作
String value = dataStore.get(key);
String result = value != null ?
"线程["+Thread.currentThread().getName()+"]查询到:key="+key+",value="+value:
"线程["+Thread.currentThread().getName()+"]查询不到:key="+key;
System.out.println(result);
}finally {
latch.countDown();//执行完毕,计数减一
}
});
}
try{
latch.await();//等待所有线程执行完毕
System.out.println("所有查询线程执行完成");
}catch (InterruptedException e) {
e.printStackTrace();
}
}
public void shutdown(){
if (threadPool!=null && !threadPool.isShutdown()){
threadPool.shutdown();
System.out.println("数据库连接关闭");
}
}
public static void main(String[] args) {
SimpleDatabase db = new SimpleDatabase();
//插入数据
db.insert("user1","张三");
db.insert("user2","李四");
db.insert("user3","王五");
db.queryWithThreads("user1","user2","user3");
db.shutdown();
}
}
key:user1,value:张三
key:user2,value:李四
key:user3,value:王五
启动多线程查询
线程[pool-1-thread-3]查询到:key=user3,value=王五
线程[pool-1-thread-2]查询到:key=user2,value=李四
线程[pool-1-thread-1]查询到:key=user1,value=张三
所有查询线程执行完成
数据库连接关闭
List模拟数据库
package KaKe;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
public class ListBasedDatabase {
private List<Map.Entry<String,String>> dataStore;
private ExecutorService threadPool;
public ListBasedDatabase(){
//使用线程安全的ArrayList保证线程安全
dataStore = new CopyOnWriteArrayList<>();
threadPool = Executors.newFixedThreadPool(3);
}
public void insert(String key,String value){
for (Map.Entry<String, String> entry : dataStore) {
if (entry.getKey().equals(key)){
System.out.println("插入失败:key="+key+"已存在");
return;
}
}
dataStore.add(Map.entry(key, value));
System.out.println("插入成功:key="+key+"value="+value);
}
public void queryWithThreads(String... keys){
if (keys==null || keys.length==0){
System.out.println("查询条件为空");
return;
}
CountDownLatch latch = new CountDownLatch(keys.length);
System.out.println("启动线程查询:");
for (String key : keys) {
threadPool.execute(()->{
try{
String result = searchInList(key);
System.out.println("线程["+Thread.currentThread().getName()+"]:"+result);
}finally {
latch.countDown();
}
});
}
try{
latch.await();
System.out.println("所有线程查询执行完成");
}catch(InterruptedException e){
e.printStackTrace();
}
}
//在List中查找数据
private String searchInList(String key){
for (Map.Entry<String,String> entry: dataStore){
if (entry.getKey().equals(key)){
return "查询到数据:key="+key+",value="+entry.getValue();
}
}
return "未找到key="+key+"的数据";
}
public void shutdown(){
if (threadPool!=null && !threadPool.isShutdown() ){
threadPool.shutdown();
System.out.println("数据库关闭连接");
}
}
public static void main(String[] args) {
ListBasedDatabase db = new ListBasedDatabase();
// 插入数据
db.insert("user1", "张三");
db.insert("user2", "李四");
db.insert("user3", "王五");
db.insert("user4", "赵六");
db.insert("user5", "董七");
db.queryWithThreads("user1","user2","user3","user4","user5");
db.shutdown();
}
}
插入成功:key=user1value=张三
插入成功:key=user2value=李四
插入成功:key=user3value=王五
插入成功:key=user4value=赵六
插入成功:key=user5value=董七
启动线程查询:
线程[pool-1-thread-1]:查询到数据:key=user1,value=张三
线程[pool-1-thread-3]:查询到数据:key=user3,value=王五
线程[pool-1-thread-1]:查询到数据:key=user4,value=赵六
线程[pool-1-thread-2]:查询到数据:key=user2,value=李四
线程[pool-1-thread-3]:查询到数据:key=user5,value=董七
所有线程查询执行完成
数据库关闭连接
总结:
其实本质上还是使用HashMap ,只不过后者是用List将其HashMap封装起来,然后插入和查询