io.github.panxiaochao.sensitive.serializer.jackson.SensitiveJackJsonSerializer Maven / Gradle / Ivy

Go to download
/*
 * Copyright © 2022-2024 Lypxc ([email protected])
 *
 * Licensed 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 io.github.panxiaochao.sensitive.serializer.jackson;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.BeanProperty;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.ser.ContextualSerializer;
import io.github.panxiaochao.core.enums.CommonResponseEnum;
import io.github.panxiaochao.core.exception.ServerRuntimeException;
import io.github.panxiaochao.core.utils.ObjectUtil;
import io.github.panxiaochao.sensitive.annotation.Sensitive;
import io.github.panxiaochao.sensitive.strategy.IHandler;
import io.github.panxiaochao.sensitive.strategy.IStrategy;
import io.github.panxiaochao.sensitive.utils.InvokeMethodUtil;
import org.springframework.util.Assert;

import java.util.Objects;

/**
 * 
 * Jackson 脱敏序列化
 * 
 *
 * @author Lypxc
 * @since 2024-06-11
 * @version 1.0
 */
public class SensitiveJackJsonSerializer extends JsonSerializer implements ContextualSerializer {

	/**
	 * 策略
	 */
	private final IStrategy strategy;

	/**
	 * 自定义策略 className
	 */
	private final String strategyClassName;

	public SensitiveJackJsonSerializer() {
		this(null, null);
	}

	public SensitiveJackJsonSerializer(IStrategy strategy, String strategyClassName) {
		this.strategy = strategy;
		this.strategyClassName = strategyClassName;
	}

	@Override
	public void serialize(String value, JsonGenerator gen, SerializerProvider serializers) {
		try {
			// 默认方法
			if (strategyClassName.equals(IHandler.class.getName())) {
				gen.writeString(strategy.use().apply(value));
			}
			else {
				Object invokeValue = InvokeMethodUtil.invoke(strategyClassName, value);
				gen.writeString(invokeValue.toString());
			}
		}
		catch (Exception e) {
			throw new ServerRuntimeException(CommonResponseEnum.INTERNAL_SERVER_ERROR,
					"The field [" + gen.getOutputContext().getCurrentName() + "] serialize is error! ");
		}
	}

	@Override
	public JsonSerializer createContextual(SerializerProvider prov, BeanProperty property) {
		if (ObjectUtil.isNotEmpty(property) && Objects.equals(String.class, property.getType().getRawClass())) {
			Sensitive sensitive = property.getAnnotation(Sensitive.class);
			if (null == sensitive) {
				sensitive = property.getContextAnnotation(Sensitive.class);
			}
			if (null != sensitive) {
				Assert.notNull(sensitive.strategy(), "The strategy must not be null");
				Assert.notNull(sensitive.handler(), "The handler must not be null");
				return new SensitiveJackJsonSerializer(sensitive.strategy(), sensitive.handler().getName());
			}
		}
		return new SensitiveJackJsonSerializer();
	}

}