使用HashMap或者List模拟数据库插入和查询数据
理解代码必备
- 参数中使用
(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封装起来,然后插入和查询