Skip to main content

A JIT implementation for Marshmallow to speed up dumping and loading objects.

Project description

*************************************************************
:fire:toastedmarshmallow:fire:: Makes Marshmallow Toasty Fast
*************************************************************

Toastedmarshmallow implements a JIT for marshmallow that speeds up dumping
objects 10-25X (depending on your schema). Toastedmarshmallow allows you to
have the great API that
`Marshmallow <https://github.com/marshmallow-code/marshmallow>`_ provides
without having to sacrifice performance!

::

Benchmark Result:
Original Time: 2682.61 usec/dump
Optimized Time: 176.38 usec/dump
Optimized (Cython) Time: 125.77 usec/dump
Speed up: 15.21x
Cython Speed up: 21.33x

Even ``PyPy`` benefits from ``toastedmarshmallow``!

::

Benchmark Result:
Original Time: 189.78 usec/dump
Optimized Time: 20.03 usec/dump
Speed up: 9.48x

Enabling Toastedmarshmallow on an existing Schema is just one line of code:

.. code-block:: python

from datetime import date
import toastedmarshmallow
from marshmallow import Schema, fields, pprint

class ArtistSchema(Schema):
name = fields.Str()

class AlbumSchema(Schema):
title = fields.Str()
release_date = fields.Date()
artist = fields.Nested(ArtistSchema())

schema = AlbumSchema()
# Specify the jit method as toastedmarshmallow's jit
schema.jit = toastedmarshmallow.Jit
# And that's it! Your dump methods are 15x faster!

You can also enable Toastedmarshmallow globally by setting the environment
variable ``MARSHMALLOW_SCHEMA_DEFAULT_JIT`` to ``toastedmarshmallow.Jit`` .

Installing toastedmarshmallow
-----------------------------

.. code-block:: bash

pip install toastedmarshmallow

This will *also* install a slightly-forked ``marshmallow`` that includes some
hooks Toastedmarshmallow needs enable the JIT to run before falling back
to the original marshmallow code. These changes are minimal making it easier
to track upstream. You can find the changes
`Here <https://github.com/marshmallow-code/marshmallow/pull/629>`_.

This means you should **remove** marshmallow from your requirements and replace
it with Toastedmarshmallow.

How it works
------------

Toastedmarshmallow works by generating code at runtime to optimize dumping
objects without going through layers and layers of reflection. The generated
code optimistically assumes the objects being passed in are schematically valid,
falling back to the original marshmallow code on failure.

For example, taking ``AlbumSchema`` from above, Toastedmarshmallow will
generate the following 3 methods:

.. code-block:: python

def InstanceSerializer(obj):
res = {}
value = obj.release_date; value = value() if callable(value) else value; res["release_date"] = _field_release_date__serialize(value, "release_date", obj)
value = obj.artist; value = value() if callable(value) else value; res["artist"] = _field_artist__serialize(value, "artist", obj)
value = obj.title; value = value() if callable(value) else value; value = str(value) if value is not None else None; res["title"] = value
return res

def DictSerializer(obj):
res = {}
if "release_date" in obj:
value = obj["release_date"]; value = value() if callable(value) else value; res["release_date"] = _field_release_date__serialize(value, "release_date", obj)
if "artist" in obj:
value = obj["artist"]; value = value() if callable(value) else value; res["artist"] = _field_artist__serialize(value, "artist", obj)
if "title" in obj:
value = obj["title"]; value = value() if callable(value) else value; value = str(value) if value is not None else None; res["title"] = value
return res

def HybridSerializer(obj):
res = {}
try:
value = obj["release_date"]
except (KeyError, AttributeError, IndexError, TypeError):
value = obj.release_date
value = value; value = value() if callable(value) else value; res["release_date"] = _field_release_date__serialize(value, "release_date", obj)
try:
value = obj["artist"]
except (KeyError, AttributeError, IndexError, TypeError):
value = obj.artist
value = value; value = value() if callable(value) else value; res["artist"] = _field_artist__serialize(value, "artist", obj)
try:
value = obj["title"]
except (KeyError, AttributeError, IndexError, TypeError):
value = obj.title
value = value; value = value() if callable(value) else value; value = str(value) if value is not None else None; res["title"] = value
return res

Toastedmarshmallow will invoke the proper serializer based upon the input.

Since Toastedmarshmallow is generating code at runtime, it's critical you
re-use Schema objects. If you're creating a new Schema object every time you
serialize/deserialize an object you'll likely have much worse performance.

:zap::microscope: Experimental :microscope::zap:
--------------------------------------------------

Toastedmarshmallow also has an experimental Cython based jit. It takes the
generated code above and runs it through Cython first, getting another 1.5x
win. Generally the generated Python code is fast enough, but this is a useful
option when you've got to squeeze out every last bit of performance.

To use the Cython jit, replace `Jit` with `CythonJit`:

.. code-block:: python

schema.jit = toastedmarshmallow.CythonJit


Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

toastedmarshmallow-0.1.0-py2.py3-none-any.whl (52.4 kB view details)

Uploaded Python 2 Python 3

File details

Details for the file toastedmarshmallow-0.1.0-py2.py3-none-any.whl.

File metadata

File hashes

Hashes for toastedmarshmallow-0.1.0-py2.py3-none-any.whl
Algorithm Hash digest
SHA256 18ef2d75b7bb13f1cec7d3777ee780b7f5242fcba539f6ce30582ab234b17da1
MD5 5e4b6e1424d7190e7a3f0dd892efc21b
BLAKE2b-256 ba717867f7ba48a7271ae31c1cf4a3405ce9f5211048c0d506e35a4fc79d2a91

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page