/*
* Copyright 2016 the original author or authors.
*
* 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 org.gradle.caching.internal.tasks.origin;
import org.gradle.api.internal.TaskInternal;
import org.gradle.internal.UncheckedException;
import org.gradle.internal.id.UniqueId;
import org.gradle.internal.remote.internal.inet.InetAddressFactory;
import org.gradle.internal.scopeids.id.BuildScopeId;
import org.gradle.internal.time.TimeProvider;
import org.gradle.util.GradleVersion;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Arrays;
import java.util.List;
import java.util.Properties;
public class TaskOutputOriginFactory {
private static final Logger LOGGER = LoggerFactory.getLogger(TaskOutputOriginFactory.class);
private static final String BUILD_ID_KEY = "buildId";
private static final List<String> METADATA_KEYS = Arrays.asList(BUILD_ID_KEY, "type", "path", "gradleVersion", "creationTime", "executionTime", "rootPath", "operatingSystem", "hostName", "userName");
private final InetAddressFactory inetAddressFactory;
private final String userName;
private final String operatingSystem;
private final TimeProvider timeProvider;
private final GradleVersion gradleVersion;
private final BuildScopeId buildScopeId;
private final File rootDir;
public TaskOutputOriginFactory(TimeProvider timeProvider, InetAddressFactory inetAddressFactory, File rootDir, String userName, String operatingSystem, GradleVersion gradleVersion, BuildScopeId buildScopeId) {
this.inetAddressFactory = inetAddressFactory;
this.rootDir = rootDir;
this.userName = userName;
this.operatingSystem = operatingSystem;
this.timeProvider = timeProvider;
this.gradleVersion = gradleVersion;
this.buildScopeId = buildScopeId;
}
public TaskOutputOriginWriter createWriter(final TaskInternal task, final long elapsedTime) {
return new TaskOutputOriginWriter() {
@Override
public void execute(OutputStream outputStream) {
// TODO: Replace this with something better
Properties properties = new Properties();
properties.setProperty(BUILD_ID_KEY, buildScopeId.getId().asString());
properties.setProperty("type", task.getClass().getCanonicalName());
properties.setProperty("path", task.getPath());
properties.setProperty("gradleVersion", gradleVersion.getVersion());
properties.setProperty("creationTime", Long.toString(timeProvider.getCurrentTime()));
properties.setProperty("executionTime", Long.toString(elapsedTime));
properties.setProperty("rootPath", rootDir.getAbsolutePath());
properties.setProperty("operatingSystem", operatingSystem);
properties.setProperty("hostName", inetAddressFactory.getHostname());
properties.setProperty("userName", userName);
try {
properties.store(outputStream, "Generated origin information");
} catch (IOException e) {
UncheckedException.throwAsUncheckedException(e);
}
assert METADATA_KEYS.containsAll(properties.stringPropertyNames()) : "Update expected metadata property list";
}
};
}
public TaskOutputOriginReader createReader(final TaskInternal task) {
return new TaskOutputOriginReader() {
@Override
public TaskOutputOriginMetadata execute(InputStream inputStream) {
// TODO: Replace this with something better
Properties properties = new Properties();
try {
properties.load(inputStream);
} catch (IOException e) {
UncheckedException.throwAsUncheckedException(e);
}
if (!properties.stringPropertyNames().containsAll(METADATA_KEYS)) {
throw new IllegalStateException("Cached result format error, corrupted origin metadata.");
}
LOGGER.info("Origin for {}: {}", task, properties);
String originBuildIdString = properties.getProperty(BUILD_ID_KEY);
UniqueId originBuildId = UniqueId.from(originBuildIdString);
return new TaskOutputOriginMetadata(originBuildId);
}
};
}
}