玩具kv数据库

谁借莪１个温暖的怀抱￠ 2022-10-01 14:59 159阅读 0赞

# 介绍 #

用java写一个简陋的kv数据库（俩小时的货），用来复习一下java流知识、线程、socket等知识。

客户端：
    	很简单的写了一下
    	功能：就是发送用户的命令，还有接收数据显示出来
    服务端：
    	redis类：读写和操作数据库（就是封装了一堆map）
    	socket类：接收命令，发送数据
    
    打开这两个，然后在客户端打命令即可。
    
    使用规范：
    ***************************************
    create tableName//创建表（最多一百个，可以改）
    例：create stu//创建名为stu的表
    ***************************************
    use tableName//操作表
    例：use stu//操作名字为stu的表
    ***************************************
    showall//打印所有表名
    ***************************************
    增：add key value
    删：delete key
    改：change key
    查：get key
    是否存在：contains key
    ***************************************
    end//结束

# redis.java: #

大概思路：

属性：

private Map<String, String>\[\] r // 存储所有表  
private String\[\] tableName//存储所有表名  
private int num = 0;// 表数量  
private int target;// 当前操作的表

方法：

增删改查

更换操作的表

打印

写入硬盘

从硬盘读出数据

接受命令并执行，给反馈

package redisDemo01;
    
    import java.io.*;
    import java.util.*;
    
    public class redis {
    	Scanner in = new Scanner(System.in);
    	private Map<String, String>[] r = new Map[100];// 最多创建100个表
    	private String[] tableName = new String[100];
    	private int num = 0;// 表数量
    	private int target;// 当前操作的表
    	// 创建表
    	public void create(String name) {
    		r[num] = new HashMap<String, String>();
    		tableName[num] = name;
    		target = num;
    		num++;
    	}
    	
    	// 操作表
    	public void use(int target) {
    		this.target = target;
    	}
    
    	public void use(String name) {
    		for (int i = 0; i < num; i++) {
    			if (tableName[i].equals(name)) {
    				target = i;
    				break;
    			}
    		}
    	}
    
    	// 判断是否存在
    	public boolean contains(String key) {
    		return r[target].containsKey(key);
    	}
    
    	// 增
    	public void add(String key, String value) {
    		r[target].put(key, value);
    	}
    
    	// 删
    	public void delete(String key) {
    		r[target].remove(key);
    	}
    
    	// 改
    	public void change(String key, String value) {
    		r[target].put(key, value);
    	}
    
    	// 查
    	public String get(String key) {
    		return r[target].get(key);
    	}
    
    	// 打印表
    	public String show() {
    		String ans = "";
    		for (String key : r[target].keySet()) {
    			ans = ans + key + " " + r[target].get(key) + " ";
    		}
    		return ans;
    	}
    	public String showAllTableName() {
    		String anString="";
    		for(int i=0;i<num;i++)
    			anString=anString+i+":"+tableName[i];
    		return anString;
    	}
    	//写入硬盘
    	public void write() {
    		try {
    			Writer writer = new FileWriter(new File("member.txt"));
    			int i=0;//表的下标
    			while (i<num) {
    				writer.write("table " + tableName[i] + "\r\n");//表名
    				for (String key : r[i].keySet()) {
    					writer.write(key + " " + r[i].get(key) + "\r\n");//数据
    				}
    				i++;
    			}
    			writer.close();
    		} catch (IOException e) {
    			e.printStackTrace();
    		}
    	}
    	//读数据
    	public void read() {
    		try {
    
    			FileReader fileReader = new FileReader("member.txt");
    			BufferedReader bufferedReader = new BufferedReader(fileReader);
    			String str;
    
    			while ((str = bufferedReader.readLine()) != null) {
    				String[] a = str.split(" ");
    				if (a[0].equals("table")) {
    					create(a[1]);//新表
    				} else {
    					add(a[0], a[1]);//表里的数据
    				}
    			}
    			bufferedReader.close();
    			fileReader.close();
    		} catch (IOException e) {
    			e.printStackTrace();
    		}
    	}
    
    	// 接收命令
    	public String command(String aString) {
    		String ans = "success";
    		String[] cmd = aString.split(" ");
    
    		for (int i = 0; i < cmd.length; i++) {
    			cmd[i] = cmd[i].replaceAll(" ", "");
    		}
    
    		String command = cmd[0];
    		if (command.equals("create")) {
    			create(cmd[1]);
    		} else if (command.equals("use")) {
    			use(cmd[1]);
    		} else if (command.equals("add")) {
    			add(cmd[1], cmd[2]);
    		} else if (command.equals("contains")) {
    			if (contains(cmd[1]))
    				ans = "This key exists";
    			else
    				ans = "This key does not exists";
    		} else if (command.equals("get")) {
    			ans = "key:" + cmd[1] + "  value:" + get(cmd[1]);
    		} else if (command.equals("delete")) {
    			delete(cmd[1]);
    		} else if (command.equals("show")) {
    			ans = show();
    		} else if (command.equals("showall")) {
    			ans = showAllTableName();
    		} else if (command.equals("end")) {
    			write();
    			return "end";
    		}else {
    			return "false,tryAgain";
    		}
    		return ans;
    	}
    }

# socket.java #

属性：

String cmd;// 命令  
redis r0 // 数据库  
String returnText // 给客户看的信息

方法：

主要任务是和客户端通信，接受命令，发出客户看到的界面数据。

package redisDemo01;
    
    import java.io.*;
    import java.net.*;
    
    public class socket implements Runnable {
    	String cmd;// 命令
    	redis r0 = new redis();// 数据库
    	String returnText = "success";// 给客户看的信息
    
    	public void run() {
    		try {
    			r0.read();
    			ServerSocket serverSocket = new ServerSocket(12016);
    			do {
    				Socket socket = serverSocket.accept();
    				InputStream is = socket.getInputStream();
    				OutputStream os = socket.getOutputStream();
    				byte[] cache = new byte[1024];
    
    				is.read(cache);
    				cmd = new String(cache);
    				cmd = cmd.trim();// 去掉多余空格，这个bug调了半天
    				System.out.println("redis  >  " + cmd);
    
    				returnText = r0.command(cmd);//操作数据库
    				os.write(returnText.getBytes());// 给客户打印的数据
    
    				os.flush();
    				is.close();
    				os.close();
    			} while (!returnText.equals("end"));
    			// serverSocket.close();
    		} catch (Exception e) {
    			e.printStackTrace();
    		}
    	}
    }

然后你run一下socket，服务端这边就完事了。

# ClientThread.java #

接受客户命令然后发给服务端

接受服务端处理完的数据

用自己的方式呈现给客户（惭愧惭愧）

package Demo;
    
    import java.io.*;
    import java.net.*;
    import java.util.*;
    
    public class ClientThread implements Runnable {
    	@Override
    	public void run() {
    		try {
    			while (true) {
    				Scanner s = new Scanner(System.in);
    				String temp = s.nextLine();
    				Socket socket = new Socket("127.0.0.1", 12016);
    				// 获取输入输出流
    				InputStream is = socket.getInputStream();
    				OutputStream os = socket.getOutputStream();
    				// 发送
    				os.write(temp.getBytes());
    				os.flush();
    				// 接收
    				byte[] cache = new byte[1024];
    				is.read(cache);
    				String a = new String(cache);
    				if (temp.equals("show")) {//打印所有k——v
    					a = a.trim();//去空格
    					String[] c = a.split(" ");
    					for (int i = 0; i < c.length; i++) {
    						c[i] = c[i].replaceAll(" ", "");
    					}
    					for (int i = 0; i < c.length - 1; i += 2)//依次打印k——v
    						System.out.printf("key:%10s  value:%10s\n", c[i], c[i + 1]);
    				} else
    					System.out.println(a);
    				os.close();
    				socket.close();
    				if (temp.equals("end"))
    					break;
    			}
    		} catch (Exception e) {
    			e.printStackTrace();
    		}
    	}
    }

界面贼丑但是就是随便练个手啦。。。