forked from mooncake-track/Mooncake
30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
import os
|
|
import time
|
|
from distributed_object_store import distributed_object_store
|
|
|
|
def startProvider():
|
|
# Initialize the store
|
|
store = distributed_object_store()
|
|
# Use TCP protocol by default for testing
|
|
protocol = os.getenv("PROTOCOL", "tcp")
|
|
device_name = os.getenv("DEVICE_NAME", "ibp6s0")
|
|
local_hostname = os.getenv("LOCAL_HOSTNAME", "localhost:12355")
|
|
metadata_server = os.getenv("METADATA_ADDR", "127.0.0.1:2379")
|
|
global_segment_size = 3200 * 1024 * 1024
|
|
local_buffer_size = 512 * 1024 * 1024
|
|
master_server_address = os.getenv("MASTER_SERVER", "127.0.0.1:50051")
|
|
retcode = store.setup(local_hostname,
|
|
metadata_server,
|
|
global_segment_size,
|
|
local_buffer_size,
|
|
protocol,
|
|
device_name,
|
|
master_server_address)
|
|
if retcode:
|
|
exit(1)
|
|
time.sleep(100) # Give some time for initialization
|
|
|
|
|
|
if __name__ == '__main__':
|
|
startProvider()
|