Sources/XCMetricsBackendLib/Common/Models/Target.swift

// Copyright (c) 2020 Spotify AB.
//
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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.

import Fluent
import Vapor
import XCLogParser

public final class Target: Model, Content, PartitionedByDay {

    public typealias IDValue = String

    public static let schema = "build_targets"

    public init() {
        
    }

    @ID(custom: .id, generatedBy: IDProperty.Generator.user)
    public var id: String?

    @Field(key: "build_identifier")
    var buildIdentifier: String

    @Field(key: "name")
    var name: String

    @Field(key: "start_timestamp")
    var startTimestamp: Date

    @Field(key: "end_timestamp")
    var endTimestamp: Date

    @Field(key: "start_timestamp_microseconds")
    var startTimestampMicroseconds: Double

    @Field(key: "end_timestamp_microseconds")
    var endTimestampMicroseconds: Double

    @Field(key: "duration")
    var duration: Double

    @Field(key: "warning_count")
    var warningCount: Int32

    @Field(key: "error_count")
    var errorCount: Int32

    @Field(key: "fetched_from_cache")
    var fetchedFromCache: Bool

    @Field(key: "category")
    var category: String

    @Field(key: "compiled_count")
    var compiledCount: Int32

    @Field(key: "compilation_end_timestamp")
    var compilationEndTimestamp: Date

    @Field(key: "compilation_end_timestamp_microseconds")
    var compilationEndTimestampMicroseconds: Double

    @Field(key: "compilation_duration")
    var compilationDuration: Double

    @Field(key: "day")
    var day: Date?

}

extension Target {

    func withBuildStep(buildStep: BuildStep) -> Target {
        self.id = buildStep.identifier
        self.buildIdentifier = buildStep.parentIdentifier
        self.name = buildStep.title.replacingOccurrences(of: "Build target ", with: "")
        self.startTimestamp = Date(timeIntervalSince1970: buildStep.startTimestamp)
        self.endTimestamp = Date(timeIntervalSince1970: buildStep.endTimestamp)
        self.startTimestampMicroseconds = buildStep.startTimestamp
        self.endTimestampMicroseconds = buildStep.endTimestamp
        self.duration = buildStep.duration.xcm_roundToDecimal(9)
        self.warningCount = Int32(buildStep.warningCount)
        self.errorCount = Int32(buildStep.errorCount)
        self.fetchedFromCache = buildStep.fetchedFromCache
        self.compilationEndTimestamp = Date(timeIntervalSince1970: buildStep.compilationEndTimestamp)
        self.compilationEndTimestampMicroseconds = buildStep.compilationEndTimestamp
        self.compilationDuration = buildStep.compilationDuration.xcm_roundToDecimal(9)
        return self
    }

    func withCategory(_ category: String) -> Target {
        self.category = category
        return self
    }

    func withCompiledCount(_ compiledCount: Int32) -> Target {
        self.compiledCount = compiledCount
        return self
    }

}