python/func/python/dict_state.py at main · faasm/python · GitHub

History

52 lines (38 loc) · 1.14 KB

Raw

import pickle

from pyfaasm.core import (

read_state,

read_state_size,

write_state,

chain,

await_call,

)

KEY_A = "dict_a"

KEY_B = "dict_b"

def get_dict_from_state(key):

dict_size = read_state_size(key)

pickled_dict = read_state(key, dict_size)

return pickle.loads(pickled_dict)

def write_dict_to_state(key, dict_in):

pickled_dict = pickle.dumps(dict_in)

write_state(key, pickled_dict)

# This function is chained

def check_pickle(input_bytes):

d = get_dict_from_state(KEY_A)

d["epsilon"] = "zeta"

write_dict_to_state(KEY_B, d)

def faasm_main():

# Write initial dictionary to state

dict_a = {"alpha": "beta", "gamma": "delta"}

write_dict_to_state(KEY_A, dict_a)

# Make the chained call

call_id = chain(check_pickle, b"")

await_call(call_id)

# Load from state again

dict_b = get_dict_from_state(KEY_B)

# Check expectation

expected = {"alpha": "beta", "gamma": "delta", "epsilon": "zeta"}

if dict_b != expected:

print("Expected {} but got {}".format(expected, dict_b))

exit(1)

print("Dictionary as expected: {}".format(dict_b))

return 0

python/func/python/dict_state.py at main · faasm/python · GitHub

Uh oh!

Expand file tree

Breadcrumbs

dict_state.py

Latest commit

History

Breadcrumbs

dict_state.py

File metadata and controls