/*
 * Copyright (c) 2016-2017 Flux Capacitor.
 *
 * 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.fluxcapacitor.common.api;

import lombok.AllArgsConstructor;
import lombok.NonNull;
import lombok.experimental.Wither;

@lombok.Data
@AllArgsConstructor
public class SerializedMessage implements SerializedObject<byte[], SerializedMessage> {

    @Wither
    @NonNull
    private Data<byte[]> data;
    private Metadata metadata;
    @Wither
    private Integer segment;
    private Long index;
    private String source;
    private String target;
    private Integer requestId;
    private Long timestamp;
    private String messageId;

    public SerializedMessage(Data<byte[]> data, Metadata metadata, String messageId) {
        this.data = data;
        this.metadata = metadata;
        this.timestamp = System.currentTimeMillis();
        this.messageId = messageId;
    }

    @Override
    public Data<byte[]> data() {
        return data;
    }

}

@@ -727,16 +727,41 @@
 ither;%0A%0A
+import java.time.Clock;%0A%0A
 @lombok.
@@ -874,16 +874,112 @@
 age%3E %7B%0A%0A
+    private static final ThreadLocal%3CClock%3E clock = ThreadLocal.withInitial(Clock::systemUTC);%0A%0A
     @Wit
@@ -1442,27 +1442,21 @@
 p = 
-System.currentTimeM
+clock.get().m
 illi
@@ -1546,24 +1546,24 @@
 %5D%3E data() %7B%0A
-
         retu
@@ -1578,10 +1578,196 @@
 %0A    %7D%0A%0A
+    public static void useCustomClock(Clock customClock) %7B%0A        clock.set(customClock);%0A    %7D%0A%0A    public static void useDefaultClock() %7B%0A        clock.set(Clock.systemUTC());%0A    %7D%0A%0A
 %7D%0A

Enable custom Clock to determine a timestamp for SerializedMessage