欢乐颂,将夜猫腻小说,盗墓笔记txt全集下载

在上一篇文章中，我們使用內存做數據的存儲。在這一篇文章中，我們持久化數據到硬盤上，采用RocksDB作為存儲引擎。同時也增加了一個service層，將命令解析和存儲邏輯提取到這一層中。

RocksDB 先在Cargo.toml文件中加入rocksdb依賴：

然后，在src/storage目錄下創建rocksdb_storage.rs文件。代碼如下：

 1#[derive(Debug)]
 2pubstructRocksDbStorage(DB);
 3
 4implRocksDbStorage{
 5pubfnnew(path:implAsRef)->Self{
 6Self(DB::open_default(path).unwrap())
 7}
 8}
 9
10implStorageforRocksDbStorage{
11fnget(&self,key:&str)->Result,Box>{
12letv=self.0.get(key)?.unwrap();
13Ok(Some(v.into()))
14}
15
16fnset(&self,key:&str,value:bytes::Bytes)->Result,Box>{
17self.0.put(key,value.clone()).unwrap();
18Ok(Some(value))
19}
20}

Service層創建src/service目錄，然后創建mod.rs文件及cmd_service.rs文件。在mod.rs文件中加入如下代碼：

1pubmodcmd_service;
2
3pubtraitCmdService{
4//解析命令，返回Response
5fnexecute(self,store:&implStorage)->CmdResponse;
6}

在cmd_service.rs文件中為命令實現CmdService trait，代碼如下：

 1usecrate::{CmdResponse,CmdService,Get,Set};
 2
 3//為GET實現execute
 4implCmdServiceforGet{
 5fnexecute(self,store:&implcrate::Storage)->CmdResponse{
 6//從存儲中獲取數據，返回CmdResponse
 7matchstore.get(&self.key){
 8Ok(Some(value))=>value.into(),
 9Ok(None)=>"Notfound".into(),
10Err(e)=>e.into(),
11}
12}
13}
14
15//為SET實現execute
16implCmdServiceforSet{
17//存儲數據
18fnexecute(self,store:&implcrate::Storage)->CmdResponse{
19matchstore.set(&self.key,self.value){
20Ok(Some(value))=>value.into(),
21Ok(None)=>"Setfail".into(),
22Err(e)=>e.into(),
23}
24}
25}

在src/pb/mod.rs中實現從Bytes、&str、Box轉換為CmdResponse：

 1implFromforCmdResponse{
 2fnfrom(v:Bytes)->Self{
 3Self{
 4status:200u32,
 5message:"success".to_string(),
 6value:v,
 7}
 8}
 9}
10
11implFrom<&str>forCmdResponse{
12fnfrom(s:&str)->Self{
13Self{
14status:400u32,
15message:s.to_string(),
16..Default::default()
17}
18}
19}
20
21implFrom>forCmdResponse{
22fnfrom(e:Box)->Self{
23Self{
24status:500u32,
25message:e.to_string(),
26..Default::default()
27}
28}
29}

然后在src/service/mod.rs中加入service代碼：

 1//設置默認存儲為RocksDB
 2pubstructService{
 3store_svc:Arc>,
 4}
 5
 6//在多線程中進行clone
 7pubstructStoreService{
 8store:Store,
 9}
10
11implService{
12pubfnnew(store:Store)->Self{
13Self{
14store_svc:Arc::new(StoreService{store}),
15}
16}
17
18//執行命令
19pubasyncfnexecute(&self,cmd_req:CmdRequest)->CmdResponse{
20println!("===ExecuteCommandBefore===");
21letcmd_res=process_cmd(cmd_req,&self.store_svc.store).await;
22println!("===ExecuteCommandAfter===");
23cmd_res
24}
25}
26
27//實現Clonetrait
28implCloneforService{
29fnclone(&self)->Self{
30Self{
31store_svc:self.store_svc.clone(),
32}
33}
34}
35
36//處理請求命令，返回Response
37asyncfnprocess_cmd(cmd_req:CmdRequest,store:&implStorage)->CmdResponse{
38matchcmd_req.req_data{
39//處理GET命令
40Some(ReqData::Get(cmd_get))=>cmd_get.execute(store),
41//處理SET命令
42Some(ReqData::Set(cmd_set))=>cmd_set.execute(store),
43_=>"Invalidcommand".into(),
44}
45}

配置我們修改配置，在conf/server.conf中加入RocksDB路徑

......[rocksdb_path]path = '/tmp/kvserver'

在src/config.rs中加入如下代碼：

 1//Server端配置
 2#[derive(Debug,Serialize,Deserialize)]
 3pubstructServerConfig{
 4......
 5pubrocksdb_path:RocksdbPath,
 6}
 7
 8......
 9
10//RocksDB存儲目錄
11#[derive(Debug,Serialize,Deserialize)]
12pubstructRocksdbPath{
13pubpath:String,
14}

修改kv_server 在kv_server.rs中使用service執行命令，刪除process_cmd函數：

 1#[tokio::main]
 2asyncfnmain()->Result<(),?Box>{
 3......
 4
 5//初始化Service及存儲
 6letservice=Service::new(RocksDbStorage::new(rocksdb_path));
 7
 8loop{
 9......
10letsvc=service.clone();
11
12tokio::spawn(asyncmove{
13//使用Frame的LengthDelimitedCodec進行編解碼操作
14letmutstream=Framed::new(stream,LengthDelimitedCodec::new());
15whileletSome(Ok(mutbuf))=stream.next().await{
16......
17
18//執行請求命令
19letcmd_res=svc.execute(cmd_req).await;
20
21......
22}
23info!("Client{:?}disconnected",addr);
24});
25}
26}

審核編輯：劉清

聲明：本文內容及配圖由入駐作者撰寫或者入駐合作網站授權轉載。文章觀點僅代表作者本人，不代表電子發燒友網立場。文章及其配圖僅供工程師學習之用，如有內容侵權或者其他違規問題，請聯系本站處理。舉報投訴

數據存儲

數據存儲

+關注

關注
5

文章
993

瀏覽量
51569
編解碼

編解碼

+關注

關注
1

文章
144

瀏覽量
19945
Rust

Rust

+關注

關注
1

文章
233

瀏覽量
6909

原文標題：用Rust實現KV Server-4 持久化存儲及service層

文章出處：【微信號：Rust語言中文社區，微信公眾號：Rust語言中文社區】歡迎添加關注！文章轉載請注明出處。

在线观看www成人影院-在线观看www日本免费网站-在线观看www视频-在线观看操-欧美18在线-欧美1级

搜索歷史

持久化數據到硬盤上并采用RocksDB作為存儲引擎

評論