Using JJWT with the Jakarta JSON Binding API

/*
 * Copyright (C) 2014 jsonwebtoken.io
 *
 * 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.jsonwebtoken.jsonb.io;

import io.jsonwebtoken.io.DeserializationException;
import io.jsonwebtoken.io.Deserializer;

import javax.json.bind.Jsonb;
import javax.json.bind.JsonbException;
import java.nio.charset.StandardCharsets;

import static java.util.Objects.requireNonNull;

/**
 * @since JJWT_RELEASE_VERSION
 */
public class JsonbDeserializer<T> implements Deserializer<T> {

    private final Class<T> returnType;
    private final Jsonb jsonb;

    @SuppressWarnings("unused") //used via reflection by RuntimeClasspathDeserializerLocator
    public JsonbDeserializer() {
        this(JsonbSerializer.DEFAULT_JSONB);
    }

    @SuppressWarnings({"unchecked", "WeakerAccess", "unused"}) // for end-users providing a custom ObjectMapper
    public JsonbDeserializer(Jsonb jsonb) {
        this(jsonb, (Class<T>) Object.class);
    }

    private JsonbDeserializer(Jsonb jsonb, Class<T> returnType) {
        requireNonNull(jsonb, "ObjectMapper cannot be null.");
        requireNonNull(returnType, "Return type cannot be null.");
        this.jsonb = jsonb;
        this.returnType = returnType;
    }

    @Override
    public T deserialize(byte[] bytes) throws DeserializationException {
        try {
            return readValue(bytes);
        } catch (JsonbException jsonbException) {
            String msg = "Unable to deserialize bytes into a " + returnType.getName() + " instance: " + jsonbException.getMessage();
            throw new DeserializationException(msg, jsonbException);
        }
    }

    protected T readValue(byte[] bytes) {
        return jsonb.fromJson(new String(bytes, StandardCharsets.UTF_8), returnType);
    }

}

/*
 * Copyright (C) 2014 jsonwebtoken.io
 *
 * 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.jsonwebtoken.jsonb.io;

import io.jsonwebtoken.io.Encoders;
import io.jsonwebtoken.io.SerializationException;
import io.jsonwebtoken.io.Serializer;
import io.jsonwebtoken.lang.Assert;

import javax.json.bind.Jsonb;
import javax.json.bind.JsonbBuilder;
import javax.json.bind.JsonbException;
import java.nio.charset.StandardCharsets;

import static java.util.Objects.requireNonNull;

/**
 * @since JJWT_RELEASE_VERSION
 */
public class JsonbSerializer<T> implements Serializer<T> {

    static final Jsonb DEFAULT_JSONB = JsonbBuilder.create();

    private final Jsonb jsonb;

    @SuppressWarnings("unused") //used via reflection by RuntimeClasspathDeserializerLocator
    public JsonbSerializer() {
        this(DEFAULT_JSONB);
    }

    @SuppressWarnings("WeakerAccess") //intended for end-users to use when providing a custom ObjectMapper
    public JsonbSerializer(Jsonb jsonb) {
        requireNonNull(jsonb, "Jsonb cannot be null.");
        this.jsonb = jsonb;
    }

    @Override
    public byte[] serialize(T t) throws SerializationException {
        requireNonNull(t, "Object to serialize cannot be null.");
        try {
            return writeValueAsBytes(t);
        } catch (JsonbException jsonbException) {
            String msg = "Unable to serialize object: " + jsonbException.getMessage();
            throw new SerializationException(msg, jsonbException);
        }
    }

    @SuppressWarnings("WeakerAccess") //for testing
    protected byte[] writeValueAsBytes(T t) {
        final Object obj;

        if (t instanceof byte[]) {
            obj = Encoders.BASE64.encode((byte[]) t);
        } else if (t instanceof char[]) {
            obj = new String((char[]) t);
        } else {
            obj = t;
        }

        return this.jsonb.toJson(obj).getBytes(StandardCharsets.UTF_8);
    }
}