001/* 002 * (C) Copyright 2019 Nuxeo SA (http://nuxeo.com/) and others. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 * 016 * Contributors: 017 * bdelbosc 018 */ 019package org.nuxeo.ecm.core.work; 020 021import java.nio.charset.StandardCharsets; 022import java.util.Map; 023 024import org.nuxeo.runtime.api.Framework; 025import org.nuxeo.runtime.kv.KeyValueService; 026import org.nuxeo.runtime.kv.KeyValueStore; 027 028/** 029 * Filter that use a KeyValueStore to pass big record value 030 * 031 * @since 11.1 032 */ 033public class KeyValueStoreOverflowRecordFilter extends BaseOverflowRecordFilter { 034 035 private static final String INIT_KEY = "_init_"; 036 037 @Override 038 public void init(Map<String, String> options) { 039 super.init(options); 040 // Check at init time if the KV store works 041 getKeyValueStore().put(getPrefixedKey(INIT_KEY), INIT_KEY.getBytes(StandardCharsets.UTF_8), 60); 042 } 043 044 protected KeyValueStore getKeyValueStore() { 045 return Framework.getService(KeyValueService.class).getKeyValueStore(getStoreName()); 046 } 047 048 @Override 049 protected void storeValue(String recordKey, byte[] data) { 050 String key = getPrefixedKey(recordKey); 051 getKeyValueStore().put(key, data, getStoreTTL().toSeconds()); 052 } 053 054 @Override 055 protected byte[] fetchValue(String recordKey) { 056 String key = getPrefixedKey(recordKey); 057 return getKeyValueStore().get(key); 058 } 059 060}