[gorocks] add function call for creating column family with TTL

db.go

@@ -153,6 +153,10 @@ func OpenDbColumnFamiliesWithTTL(
 		return nil, nil, errors.New("must provide the same number of column family names and options")
 	}
 
+	if numColumnFamilies != len(cfTtls) {
+		return nil, nil, errors.New("must provide the same number of column family names and ttls")
+	}
+
 	cName := C.CString(name)
 	defer C.free(unsafe.Pointer(cName))
 
@@ -725,6 +729,22 @@ func (db *DB) CreateColumnFamily(opts *Options, name string) (*ColumnFamilyHandl
 	return NewNativeColumnFamilyHandle(cHandle), nil
 }
 
+// CreateColumnFamilyWithTTL creates a new column family with a TTL.
+func (db *DB) CreateColumnFamilyWithTTL(opts *Options, name string, ttl int) (*ColumnFamilyHandle, error) {
+	var (
+		cErr  *C.char
+		cName = C.CString(name)
+		cTtl = C.int(ttl)
+	)
+	defer C.free(unsafe.Pointer(cName))
+	cHandle := C.rocksdb_create_column_family_with_ttl(db.c, opts.c, cName, cTtl, &cErr)
+	if cErr != nil {
+		defer C.rocksdb_free(unsafe.Pointer(cErr))
+		return nil, errors.New(C.GoString(cErr))
+	}
+	return NewNativeColumnFamilyHandle(cHandle), nil
+}
+
 // DropColumnFamily drops a column family.
 func (db *DB) DropColumnFamily(c *ColumnFamilyHandle) error {
 	var cErr *C.char

db_test.go

@@ -24,12 +24,35 @@ func TestOpenDbColumnFamiliesWithTTL(t *testing.T) {
 	opts.SetCreateIfMissing(true)
 	opts.SetCreateIfMissingColumnFamilies(true)
 
-	db, _, err := OpenDbColumnFamiliesWithTTL(opts, dir, []string{"default", "mycf"}, []*Options{opts, opts}, []int{3600})
+	db, _, err := OpenDbColumnFamiliesWithTTL(opts, dir, []string{"default", "mycf"}, []*Options{opts, opts}, []int{3600, 3600})
 	defer db.Close()
 
 	ensure.Nil(t, err)
 }
 
+func TestCreateColumnFamilyWithTTL(t *testing.T) {
+	db := newTestDBWithTTL(t, "TestCreateColumnFamilyWithTTL", nil)
+	defer db.Close()
+
+	var (
+		givenKey = []byte("hello")
+		givenVal = []byte("world")
+		o        = NewDefaultOptions()
+		wo       = NewDefaultWriteOptions()
+		ro       = NewDefaultReadOptions()
+	)
+
+	cf, err := db.CreateColumnFamilyWithTTL(o, "cf", 3600)
+	ensure.Nil(t, err)
+
+	ensure.Nil(t, db.PutCF(wo, cf, givenKey, givenVal))
+
+	v, err := db.GetCF(ro, cf, givenKey)
+	defer v.Free()
+	ensure.Nil(t, err)
+	ensure.DeepEqual(t, v.Data(), givenVal)
+}
+
 func TestDBCRUD(t *testing.T) {
 	db := newTestDB(t, "TestDBGet", nil)
 	defer db.Close()
@@ -158,6 +181,24 @@ func newTestDB(t *testing.T, name string, applyOpts func(opts *Options)) *DB {
 	return db
 }
 
+func newTestDBWithTTL(t *testing.T, name string, applyOpts func(opts *Options)) *DB {
+	dir, err := ioutil.TempDir("", "gorocksdb-"+name)
+	ensure.Nil(t, err)
+
+	opts := NewDefaultOptions()
+	// test the ratelimiter
+	rateLimiter := NewRateLimiter(1024, 100*1000, 10)
+	opts.SetRateLimiter(rateLimiter)
+	opts.SetCreateIfMissing(true)
+	if applyOpts != nil {
+		applyOpts(opts)
+	}
+	db, err := OpenDbWithTTL(opts, dir, 3600)
+	ensure.Nil(t, err)
+
+	return db
+}
+
 func newSecondaryTestDB(t *testing.T, name string) *DB {
 	secondaryDir := name + "-secondary"