RustでMomentoのコレクションデータ型を試す

#Rust

#momento

中村修太

2023.01.30

この記事は公開されてから1年以上経過しています。情報が古い可能性がありますので、ご注意ください。

Introduction

高速サーバレスキャッシュサービスMomentoの情報です。
最近のアップデートにより、コレクションデータ型がサポートされました。
今回はMomentoのRust用SDKを使って
コレクション型をつかったMomentoへのアクセスを試してみます。

[注意]
Momentoのコレクション機能は現在previewです。
2023年1月現在、この機能を使いたい場合は
support@momentohq.comに連絡する必要があるので
注意してください。

Environment

今回試した環境は以下のとおりです。

MacBook Pro (13-inch, M1, 2020)
OS : MacOS 12.4
Rust : 1.66.1
Momento SDK : v0.21.0

DevIOのMomento関連記事はここにあるので、
これらもご確認おねがいします。

Setup

まずはMomentoのセットアップです。
このあたりを参考に、
Momentoの認証トークンを取得しましょう。

トークンを取得したらコンソールで環境変数にセットしておきます。

% export MOMENTO_AUTH_TOKEN = <取得した認証トークン>

次に、CargoをつかってRustプロジェクトの作成と依存ライブラリの設定です。

% cargo new momento-rust && cd momento-rust

# Cargo.toml

[dependencies]
momento = "0.21.0"

[dependencies.uuid]
version = "1.2.2"
features = [
    "v4",                
    "macro-diagnostics", 
]

Try

ではコレクションをつかってみましょう。
基本的にはJavaScript用SDKと似ているのですが、
言語的な違いがあったりするので多少は違います。

とりあえず必要モジュールのimportとクライアントの作成をするコードです。

use std::collections::HashMap;
use std::env;
use std::time::Duration;
use uuid::Uuid;

use momento::{
    MomentoResult,
    response::MomentoDictionaryGetResponse,
    response::MomentoDictionaryGetStatus, 
    response::CollectionTtl, 
    response::SimpleCacheClient,
    response::MomentoError,
    SimpleCacheClientBuilder,
};

fn get_client() -> SimpleCacheClient {
    let auth_token = env::var("MOMENTO_AUTH_TOKEN")
        .expect("env var MOMENTO_AUTH_TOKEN must be set to your auth token");
    let item_default_ttl_seconds = 60;

    SimpleCacheClientBuilder::new(auth_token, Duration::from_secs(item_default_ttl_seconds))
        .unwrap()
        .build()
}

Dictionary

クライアントを取得したり、元になるDictionaryを作ります。

    //Get Momento Client
    let mut momento = get_client();

    //Create Cache
    let cache_name = Uuid::new_v4().to_string();
    match momento.create_cache(&cache_name).await {
        Ok(_) => println!("ok"),
        Err(e1) =>println!("error occurred"),
    }

    //Create Dictionary & Dictionary Name
    let mut dictionary = HashMap::new();
    dictionary.insert("key1".to_string(), "value1".to_string());
    dictionary.insert("key2".to_string(), "10".to_string());
    dictionary.insert("key3".to_string(), "value3".to_string());

    let dictionary_name:String = Uuid::new_v4().to_string();

キャッシュ名とDictionary名を指定して、
dictionary_set関数でHashMapをキャッシュに登録します。

    momento
        .dictionary_set(
            &cache_name,
            &*dictionary_name, //or dictionary_name.clone()
            dictionary,
            CollectionTtl::default(),
        )
        .await;

登録したDictionaryから値の取得です。
Vecで任意の数のキーを指定して取得することもできます。

    let resp = momento
        .dictionary_get(&cache_name, &*dictionary_name, vec!["key1","key3"])
        .await
        .unwrap();

レスポンスは下記のように個別の値を取得できます。

    ////MomentoDictionaryGetResponse
    match resp.result　 {
        MomentoDictionaryGetStatus::FOUND => {
            println!("dictionary found!");
            if let Some(dictionary) = &resp.dictionary {
                println!("dictionary entries:");
                for (k, v) in dictionary.iter() {
                    let k_str = String::from_utf8_lossy(k);
                    let v_str = String::from_utf8_lossy(v);
                    println!("0: {:?} => {:?}", k_str, v_str);
                }
            }        
        }
        MomentoDictionaryGetStatus::MISSING => println!("dictionary missing!"),
        _ => println!("error occurred"),
    };

dictionary_incrementを使えば引数に指定したキーの値を増やします。

 //key2の値を50増やす
 let value = momento
         .dictionary_increment(&cache_name, &*dictionary_name, "key2", 50, CollectionTtl::default())
         .await
         .expect("Failed to increment dictionary key")
         .value;
    println!("{}",value); //60

Dictionaryから値を削除するにはdictionary_deleteを使います。

    //remove some fields
    momento.dictionary_delete(
             &cache_name,
             &*dictionary_name,
             Fields::Some(vec!["key1"]),
         ).await.unwrap();


    // remove entire dictionary
    momento.dictionary_delete(
             &cache_name,
             &*dictionary_name,
             Fields::<Vec<u8>>::All,
         ).await.unwrap();

Set

次はSetを使ってみます。
set_union関数でキャッシュ名、セット名とコレクションを渡します。

    let elements = vec!["a", "b", "c", "a", "d"];

    momento
         .set_union(&cache_name, set_name.clone(), elements, CollectionTtl::default())
         .await
         .expect("error");

set_fetch関数を使えばSet全体を取得します。

    let response = momento
         .set_fetch(&cache_name, set_name.clone())
         .await
         .expect("Failed to fetch");
     if let Some(set) = response.value {
         println!("set entries:");
         for entry in &set {
             println!("{:?}", String::from_utf8_lossy(entry));
         }
     } else {
         println!("Collection not found");
     }