package com.mengnankk.jmh.benchmark;

import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONB;
import tools.jackson.databind.ObjectMapper;
import tools.jackson.databind.json.JsonMapper;
import java.nio.charset.StandardCharsets;
import java.time.MonthDay;
import java.time.Year;
import java.time.YearMonth;
import org.openjdk.jmh.annotations.Benchmark;
import org.openjdk.jmh.annotations.Level;
import org.openjdk.jmh.annotations.Scope;
import org.openjdk.jmh.annotations.Setup;
import org.openjdk.jmh.annotations.State;
import org.openjdk.jmh.infra.Blackhole;

public class TimeBenchmark {

    @State(Scope.Benchmark)
    public static class BenchmarkState {
        ObjectMapper jacksonMapper;
        TimePojo pojo;
        byte[] jsonBytes;
        byte[] jsonbBytes;

        @Setup(Level.Trial)
        public void setup() {
            jacksonMapper = JsonMapper.builder().build();

            pojo = new TimePojo(
                    1,
                    "alpha",
                    Year.of(2026),
                    YearMonth.of(2026, 1),
                    MonthDay.of(1, 6)
            );

            String json = "{\"id\":1,\"name\":\"alpha\",\"year\":\"2026\",\"yearMonth\":\"2026-01\",\"monthDay\":\"--01-06\"}";
            jsonBytes = json.getBytes(StandardCharsets.UTF_8);
            jsonbBytes = JSONB.toBytes(pojo);
        }
    }

    @Benchmark
    public void fastjson2Serialize(BenchmarkState state, Blackhole bh) {
        bh.consume(JSON.toJSONBytes(state.pojo));
    }

    @Benchmark
    public void fastjson2Deserialize(BenchmarkState state, Blackhole bh) {
        bh.consume(JSON.parseObject(state.jsonBytes, TimePojo.class));
    }

    @Benchmark
    public void fastjson2SerializeJSONB(BenchmarkState state, Blackhole bh) {
        bh.consume(JSONB.toBytes(state.pojo));
    }

    @Benchmark
    public void fastjson2DeserializeJSONB(BenchmarkState state, Blackhole bh) {
        bh.consume(JSONB.parseObject(state.jsonbBytes, TimePojo.class));
    }

    @Benchmark
    public void jackson3Serialize(BenchmarkState state, Blackhole bh) throws Exception {
        bh.consume(state.jacksonMapper.writeValueAsBytes(state.pojo));
    }

    @Benchmark
    public void jackson3Deserialize(BenchmarkState state, Blackhole bh) throws Exception {
        bh.consume(state.jacksonMapper.readValue(state.jsonBytes, TimePojo.class));
    }

    public static class TimePojo {
        public int id;
        public String name;
        public Year year;
        public YearMonth yearMonth;
        public MonthDay monthDay;

        public TimePojo() {
        }

        public TimePojo(int id, String name, Year year, YearMonth yearMonth, MonthDay monthDay) {
            this.id = id;
            this.name = name;
            this.year = year;
            this.yearMonth = yearMonth;
            this.monthDay = monthDay;
        }
    }
}