package com.amazonaws.xray.agent.runtime.models;

import com.blogspot.mydailyjava.weaklockfree.WeakConcurrentMap;
import org.checkerframework.checker.nullness.qual.Nullable;

import java.sql.PreparedStatement;

/**
 * Contains state information for each logical request/response transaction event.
 *
 * Events that are captured should capture the relevant information and store it here.
 */
public class XRayTransactionState {
    private String host;
    private String url;
    private String userAgent;
    private String clientIP;
    private String method;
    private String serviceType; // Type of AWS resource running application.
    private String traceHeader;
    private String origin;

    private static String serviceName;
    private static final WeakConcurrentMap<PreparedStatement, String> preparedStatementMap
            = new WeakConcurrentMap.WithInlinedExpunction<>();

    public XRayTransactionState withHost(String host) {
        this.host = host;
        return this;
    }

    public XRayTransactionState withUrl(String url) {
        this.url = url;
        return this;
    }

    public XRayTransactionState withUserAgent(String userAgent) {
        this.userAgent = userAgent;
        return this;
    }

    public XRayTransactionState withClientIP(String clientIP) {
        this.clientIP = clientIP;
        return this;
    }

    public XRayTransactionState withMethod(String method) {
        this.method = method;
        return this;
    }

    public XRayTransactionState withServiceType(String serviceType) {
        this.serviceType = serviceType;
        return this;
    }

    public XRayTransactionState withTraceheaderString(@Nullable String traceHeaderString) {
        this.traceHeader = traceHeaderString;
        return this;
    }

    public XRayTransactionState withOrigin(@Nullable String origin) {
        this.origin = origin;
        return this;
    }

    public String getHost() {
        return this.host;
    }

    public String getURL() {
        return this.url;
    }

    public String getUserAgent() {
        return this.userAgent;
    }

    public String getClientIP() {
        return this.clientIP;
    }

    public String getMethod() {
        return this.method;
    }

    public String getServiceType() {
        return this.serviceType;
    }

    @Nullable
    public String getTraceHeader() {
        return this.traceHeader;
    }

    @Nullable
    public String getOrigin() {
        return this.origin;
    }

    public static void setServiceName(String inServiceName) {
        serviceName = inServiceName;
    }

    public static String getServiceName() {
        return serviceName;
    }

    public static void putPreparedQuery(PreparedStatement ps, String query) {
        preparedStatementMap.put(ps, query);
    }

    public static String getPreparedQuery(PreparedStatement ps) {
        return preparedStatementMap.get(ps);
    }
}