This quickstart shows how you can use the FSSpec interface to connect to Alluxio. For more information on what to expect, please read the blog Accelerate data loading in large scale ML training with Ray and Alluxio.
Alluxio version >= 309
Launch Alluxio clusters with the example configuration
# only one master, one worker are running in this example
alluxio.master.hostname=localhost
alluxio.worker.hostname=localhost
# Critical properties for this example
# UFS address (e.g., the src of data to cache), change it to your bucket
alluxio.dora.client.ufs.root=s3://example_bucket/datasets/
# storage dir
alluxio.worker.page.store.dirs=/tmp/page_ufs
# size of storage dir
alluxio.worker.page.store.sizes=10GB
# use etcd to keep consistent hashing ring
alluxio.worker.membership.manager.type=ETCD
# default etcd endpoint
alluxio.etcd.endpoints=http://localhost:2379
# number of vnodes per worker on the ring
alluxio.user.consistent.hash.virtual.node.count.per.worker=5
# Other optional settings, good to have
alluxio.job.batch.size=200
alluxio.master.journal.type=NOOP
alluxio.master.scheduler.initial.wait.time=10s
alluxio.network.netty.heartbeat.timeout=5min
alluxio.underfs.io.threads=50
Python in range of [3.8, 3.9, 3.10] ray >= 2.8.2 fsspec released after 2023.6
Alluxio fsspec acts as a cache on top of an existing underlying data lake storage connection. The fsspec implementation corresponding to the underlying data lake storage needs to be installed. In the below Alluxio configuration example, Amazon S3 is the data lake storage where the dataset is read from.
To connect to an existing underlying storage, there are two requirements
- Install the underlying storage fsspec
- For all built-in storage fsspec, no extra python libraries are needed to be installed.
- For all third-party storage fsspec, the third-party fsspec python libraries are needed to be installed.
- Set credentials for the underlying data lake storage
Example: Deploy S3 as the underlying data lake storage Install third-party S3 fsspec
pip install s3fs
Directly install the latest published alluxiofs
pip install alluxiofs
[Optional] Install from the source code
git clone [email protected]:fsspec/alluxiofs.git
cd alluxiofs && python3 setup.py bdist_wheel && \
pip3 install dist/alluxiofs-<alluxiofs_version>-py3-none-any.whl
bin/alluxio job load --path s3://example_bucket/datasets/ --submit
This will trigger a load job asynchronously with a job ID specified. You can wait until the load finishes or check the progress of this loading process using the following command:
bin/alluxio job load --path s3://example_bucket/datasets/ --progress
Create the Alluxio Filesystem with data backed in S3
import fsspec
from alluxiofs import AlluxioFileSystem
# Register Alluxio to fsspec
fsspec.register_implementation("alluxiofs", AlluxioFileSystem, clobber=True)
# Create Alluxio filesystem
alluxio_fs = fsspec.filesystem("alluxiofs", etcd_hosts="localhost", etcd_port=2379, target_protocol="s3")
Similar to fsspec examples and alluxiofs examples. Note that all the read operations can only succeed if the parent folder has been loaded into Alluxio.
# list files
contents = alluxio_fs.ls("s3://apc999/datasets/nyc-taxi-csv/green-tripdata/", detail=True)
# Read files
with alluxio_fs.open("s3://apc999/datasets/nyc-taxi-csv/green-tripdata/green_tripdata_2021-01.csv", "rb") as f:
data = f.read()
import fsspec
import ray
from alluxiofs import AlluxioFileSystem
# Register the Alluxio fsspec implementation
fsspec.register_implementation("alluxiofs", AlluxioFileSystem, clobber=True)
alluxio_fs = fsspec.filesystem(
"alluxiofs", etcd_hosts="localhost", target_protocol="s3"
)
# Pass the initialized Alluxio filesystem to Ray and read the NYC taxi ride data set
ds = ray.data.read_csv("s3://example_bucket/datasets/example.csv", filesystem=alluxio_fs)
# Get a count of the number of records in the single CSV file
ds.count()
# Display the schema derived from the CSV file header record
ds.schema()
# Display the header record
ds.take(1)
# Display the first data record
ds.take(2)
# Read multiple CSV files:
ds2 = ray.data.read_csv("s3://apc999/datasets/csv_dir/", filesystem=alluxio_fs)
# Get a count of the number of records in the twelve CSV files
ds2.count()
# End of Python example
alluxiocommon package is a native enhancement module for alluxiofs based on PyO3 rust bindings. Currently it enhances big reads (multi-page reads from alluxio) by issuing multi-threaded requests to alluxio.
to enable it, first install alluxiocommon package:
pip install alluxiocommon
and when start the Alluxio fsspec instance, add an additional option flag:
alluxio_options = {"alluxio.common.extension.enable" : "True"}
alluxio_fs = fsspec.filesystem(
"alluxiofs", etcd_hosts="localhost", target_protocol="s3",
options=alluxio_options
)
import fsspec
from alluxiofs import AlluxioFileSystem
# Register the Alluxio fsspec implementation
fsspec.register_implementation("alluxiofs", AlluxioFileSystem, clobber=True)
alluxio_fs = fsspec.filesystem(
"alluxiofs", etcd_hosts="localhost", target_protocol="s3"
)
# Example 1
# Pass the initialized Alluxio filesystem to Pyarrow and read the data set from the example parquet file
import pyarrow.dataset as ds
dataset = ds.dataset("s3://example_bucket/datasets/example.parquet", filesystem=alluxio_fs)
# Get a count of the number of records in the parquet file
dataset.count_rows()
# Display the schema derived from the parquet file header record
dataset.schema
# Display the first record
dataset.take(0)
# Example 2
# Create a python-based PyArrow filesystem using FsspecHandler
py_fs = PyFileSystem(FSSpecHandler(alluxio_file_system))
# Read the data by using the Pyarrow filesystem interface
with py_fs.open_input_file("s3://example_bucket/datasets/example.parquet") as f:
alluxio_file_data = f.read()
# End of Python example