Added RocksDB package (#10084)

* Added rocksdb package

* Fixes based on @tristan0x and @pramodk comments

* Cleanup

* A couple minor fixes based on @adamjstewart 's advice

* Fixed flake8 errors

1 files changed, 61 insertions, 0 deletions

diff --git a/var/spack/repos/builtin/packages/rocksdb/package.py b/var/spack/repos/builtin/packages/rocksdb/package.py
new file mode 100644
index 0000000000..e076e6f3c3
--- /dev/null
+++ b/var/spack/repos/builtin/packages/rocksdb/package.py
@@ -0,0 +1,61 @@
+# Copyright 2013-2018 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+from spack import *
+
+
+class Rocksdb(MakefilePackage):
+    """RocksDB: A Persistent Key-Value Store for Flash and RAM Storage"""
+
+    homepage = "https://github.com/facebook/rocksdb"
+    url      = 'https://github.com/facebook/rocksdb/archive/v5.17.2.tar.gz'
+    git      = 'https://github.com/facebook/rocksdb.git'
+
+    version('develop', git=git, branch='master', submodules=True)
+    version('5.17.2',  sha256='101f05858650a810c90e4872338222a1a3bf3b24de7b7d74466814e6a95c2d28')
+    version('5.16.6',  sha256='f0739edce1707568bdfb36a77638fd5bae287ca21763ce3e56cf0bfae8fff033')
+    version('5.15.10', sha256='26d5d4259fa352ae1604b5b4d275f947cacc006f4f7d2ef0b815056601b807c0')
+
+    variant('bz2', default=False, description='Enable bz2 compression support')
+    variant('lz4', default=True, description='Enable lz4 compression support')
+    variant('snappy', default=False, description='Enable snappy compression support')
+    variant('static', default=True, description='Build static library')
+    variant('zlib', default=True, description='Enable zlib compression support')
+    variant('zstd', default=False, description='Enable zstandard compression support')
+
+    depends_on('bzip2', when='+bzip2')
+    depends_on('gflags')
+    depends_on('lz4', when='+lz4')
+    depends_on('snappy', when='+snappy')
+    depends_on('zlib', when='+zlib')
+    depends_on('zstd', when='+zstd')
+
+    phases = ['install']
+
+    def install(self, spec, prefix):
+        cflags = []
+        ldflags = []
+
+        if '+zlib' in self.spec:
+            cflags.append('-I' + self.spec['zlib'].prefix.include)
+            ldflags.append(self.spec['zlib'].libs.ld_flags)
+        if '+bz2' in self.spec:
+            cflags.append('-I' + self.spec['bz2'].prefix.include)
+            ldflags.append(self.spec['bz2'].libs.ld_flags)
+
+        for pkg in ['lz4', 'snappy', 'zstd']:
+            if '+' + pkg in self.spec:
+                cflags.append(self.spec[pkg].headers.cpp_flags)
+                ldflags.append(self.spec[pkg].libs.ld_flags)
+
+        cflags.append(self.spec['gflags'].headers.cpp_flags)
+        ldflags.append(self.spec['gflags'].libs.ld_flags)
+
+        env['CFLAGS'] = ' '.join(cflags)
+        env['PLATFORM_FLAGS'] = ' '.join(ldflags)
+        env['INSTALL_PATH'] = self.spec.prefix
+
+        buildtype = 'install-static' if '+static' in spec else 'install-shared'
+        make(buildtype)