org.apache.flink.runtime.state.gemini.engine.page.compress.GCompressAlgorithm Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of flink-runtime_2.11 Show documentation
There is a newer version: 1.5.1
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.flink.runtime.state.gemini.engine.page.compress;

import org.apache.flink.configuration.Configuration;

/**
 * GCompressAlgorithm.
 */
public enum GCompressAlgorithm {
	None((byte) 1){
		@Override
		public CompressorCodec getCompressorCodec() {
			return null;
		}

		@Override
		public boolean enableCompress() {
			return false;
		}

		@Override
		public void setConfiguration(Configuration configuration) {
			//do nothing
		}
	},
	Lz4((byte) 2){
		private Configuration configuration;
		private ThreadLocal compressorCodecThreadLocal;
		@Override
		public CompressorCodec getCompressorCodec() {
			return compressorCodecThreadLocal.get();
		}

		@Override
		public boolean enableCompress() {
			return true;
		}

		@Override
		public void setConfiguration(Configuration configuration) {
			this.configuration = configuration;
			compressorCodecThreadLocal = ThreadLocal.withInitial(() -> new LZ4CompressorCodec(this.configuration));
		}
	},
	Snappy((byte) 3){
		private ThreadLocal compressorCodecThreadLocal = ThreadLocal.withInitial(() -> new SnappyCompressorCodec());
		@Override
		public CompressorCodec getCompressorCodec() {
			return compressorCodecThreadLocal.get();
		}

		@Override
		public boolean enableCompress() {
			return true;
		}

		@Override
		public void setConfiguration(Configuration configuration) {
			//do nothing
		}
	};

	private final byte code;

	GCompressAlgorithm(final byte c) {
		this.code = c;
	}

	public byte getCode() {
		return this.code;
	}

	public static GCompressAlgorithm valueOf(byte gvalueType) {
		switch (gvalueType) {
			case 1:
				return None;
			case 2:
				return Lz4;
			case 3:
				return Snappy;
			default:
				return None;
		}
	}

	public abstract CompressorCodec getCompressorCodec();

	public abstract boolean enableCompress();

	public abstract void setConfiguration(Configuration configuration);

	public static GCompressAlgorithm getCompressAlgorithm(String algorithm, Configuration configuration) {
		GCompressAlgorithm compressAlgorithm = GCompressAlgorithm.valueOf(algorithm);
		//TODO Snappy not support HeapByteBuffer now.
		if (compressAlgorithm == Snappy) {
			throw new UnsupportedOperationException("Snappy is not supported currently");
		}

		compressAlgorithm.setConfiguration(configuration);
		return compressAlgorithm;
	}
}