aboutsummaryrefslogtreecommitdiffstats
path: root/container-core/src/main/java/com/yahoo/container/logging/VespaAccessLog.java
blob: 09d619e070b1136f442e227289cc936c93d3e71b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.container.logging;

import com.yahoo.component.AbstractComponent;
import com.yahoo.container.core.AccessLogConfig;

import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.TimeZone;

/**
 * @author Bjorn Borud
 * @author Oyvind Bakksjo
 */
public final class VespaAccessLog extends AbstractComponent implements RequestLogHandler, LogWriter<RequestLogEntry> {

    private static final ThreadLocal<SimpleDateFormat> dateFormat = ThreadLocal.withInitial(VespaAccessLog::createDateFormat);

    private final AccessLogHandler logHandler;

    public VespaAccessLog(AccessLogConfig config) {
        logHandler = new AccessLogHandler(config.fileHandler(), this);
    }

    private static SimpleDateFormat createDateFormat() {
        SimpleDateFormat format = new SimpleDateFormat("[dd/MMM/yyyy:HH:mm:ss Z]");
        format.setTimeZone(TimeZone.getTimeZone("UTC"));
        return format;
    }

    private static String getDate() {
        return dateFormat.get().format(new Date());
    }

    private String getRequest(final String httpMethod, final String rawPath, final String rawQuery, final String httpVersion) {
        return httpMethod + " " + (rawQuery != null ? rawPath + "?" + rawQuery : rawPath) + " " + httpVersion;
    }

    private String getUser(String user) {
        return (user == null) ? "-" : user;
    }

    private String toLogline(String ipAddr, String user, String request, String referer, String agent,
                          long durationMillis, long byteCount, HitCounts hitcounts, int returnCode)
    {
        long ms = Math.max(0L, durationMillis);
        StringBuilder sb = new StringBuilder()
                .append(ipAddr)
                .append(" - ")
                .append(getUser(user))
                .append(' ')
                .append(getDate())
                .append(" \"")
                .append(request)
                .append("\" ")
                .append(returnCode)
                .append(' ')
                .append(byteCount)
                .append(" \"")
                .append(referer)
                .append("\" \"")
                .append(agent)
                .append("\" ")
                .append(ms/1000)
                .append('.');
        decimalsOfSecondsFromMilliseconds(ms, sb);
        sb.append(' ')
                .append((hitcounts == null) ? 0 : hitcounts.getTotalHitCount())
                .append(" 0.0 ")
                .append((hitcounts == null) ? 0 : hitcounts.getSummaryCount());
        return sb.toString();
    }

    private void decimalsOfSecondsFromMilliseconds(long ms, StringBuilder sb) {
        long dec = ms % 1000;
        String numbers = String.valueOf(dec);
        if (dec <= 9) {
            sb.append("00");
        } else if (dec <= 99) {
            sb.append('0');
        }
        sb.append(numbers);
    }

    @Override public void deconstruct() { logHandler.shutdown(); }

    @Override
    public void log(RequestLogEntry entry) {
        logHandler.log(entry);
    }

    @Override
    public void write(RequestLogEntry entry, OutputStream outputStream) throws IOException {
        outputStream.write(
                toLogline(
                        entry.peerAddress().get(),
                        null,
                        getRequest(
                                entry.httpMethod().orElse(null),
                                entry.rawPath().orElse(null),
                                entry.rawQuery().orElse(null),
                                entry.httpVersion().orElse(null)),
                        entry.referer().orElse(null),
                        entry.userAgent().orElse(null),
                        entry.duration().get().toMillis(),
                        entry.responseSize().orElse(0L),
                        entry.hitCounts().orElse(null),
                        entry.statusCode().orElse(0)).getBytes(StandardCharsets.UTF_8));
    }
}