segui久久综合精品,国产国语**毛片高清视频

在上一篇文章中，我們使用內(nèi)存做數(shù)據(jù)的存儲(chǔ)。在這一篇文章中，我們持久化數(shù)據(jù)到硬盤(pán)上，采用RocksDB作為存儲(chǔ)引擎。同時(shí)也增加了一個(gè)service層，將命令解析和存儲(chǔ)邏輯提取到這一層中。

RocksDB 先在Cargo.toml文件中加入rocksdb依賴(lài)：

然后，在src/storage目錄下創(chuàng)建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層創(chuàng)建src/service目錄，然后創(chuàng)建mod.rs文件及cmd_service.rs文件。在mod.rs文件中加入如下代碼：

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

在cmd_service.rs文件中為命令實(shí)現(xiàn)CmdService trait，代碼如下：

 1usecrate::{CmdResponse,CmdService,Get,Set};
 2
 3//為GET實(shí)現(xiàn)execute
 4implCmdServiceforGet{
 5fnexecute(self,store:&implcrate::Storage)->CmdResponse{
 6//從存儲(chǔ)中獲取數(shù)據(jù)，返回CmdResponse
 7matchstore.get(&self.key){
 8Ok(Some(value))=>value.into(),
 9Ok(None)=>"Notfound".into(),
10Err(e)=>e.into(),
11}
12}
13}
14
15//為SET實(shí)現(xiàn)execute
16implCmdServiceforSet{
17//存儲(chǔ)數(shù)據(jù)
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中實(shí)現(xiàn)從Bytes、&str、Box轉(zhuǎn)換為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//設(shè)置默認(rèn)存儲(chǔ)為RocksDB
 2pubstructService{
 3store_svc:Arc>,
 4}
 5
 6//在多線(xiàn)程中進(jìn)行clone
 7pubstructStoreService{
 8store:Store,
 9}
10
11implService{
12pubfnnew(store:Store)->Self{
13Self{
14store_svc:Arc::new(StoreService{store}),
15}
16}
17
18//執(zhí)行命令
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//實(shí)現(xiàn)Clonetrait
28implCloneforService{
29fnclone(&self)->Self{
30Self{
31store_svc:self.store_svc.clone(),
32}
33}
34}
35
36//處理請(qǐng)求命令，返回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存儲(chǔ)目錄
11#[derive(Debug,Serialize,Deserialize)]
12pubstructRocksdbPath{
13pubpath:String,
14}

修改kv_server 在kv_server.rs中使用service執(zhí)行命令，刪除process_cmd函數(shù)：

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

審核編輯：劉清

聲明：本文內(nèi)容及配圖由入駐作者撰寫(xiě)或者入駐合作網(wǎng)站授權(quán)轉(zhuǎn)載。文章觀(guān)點(diǎn)僅代表作者本人，不代表電子發(fā)燒友網(wǎng)立場(chǎng)。文章及其配圖僅供工程師學(xué)習(xí)之用，如有內(nèi)容侵權(quán)或者其他違規(guī)問(wèn)題，請(qǐng)聯(lián)系本站處理。舉報(bào)投訴

數(shù)據(jù)存儲(chǔ)

數(shù)據(jù)存儲(chǔ)

+關(guān)注

關(guān)注
5

文章
977

瀏覽量
50976
編解碼

編解碼

+關(guān)注

關(guān)注
1

文章
140

瀏覽量
19635
Rust

Rust

+關(guān)注

關(guān)注
1

文章
229

瀏覽量
6626