X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/2e03d03bc55b5a612c2bf04d878a72f2ee420d99..a9be3117466506dffc39617be1c58406c5914e4b:/lib/crunchrun/crunchrun.go

diff --git a/lib/crunchrun/crunchrun.go b/lib/crunchrun/crunchrun.go
index eadf22876f..ee9115d8d8 100644
--- a/lib/crunchrun/crunchrun.go
+++ b/lib/crunchrun/crunchrun.go
@@ -140,6 +140,7 @@ type ContainerRunner struct {
 	MkArvClient   func(token string) (IArvadosClient, IKeepClient, *arvados.Client, error)
 	finalState    string
 	parentTemp    string
+	costStartTime time.Time
 
 	keepstoreLogger  io.WriteCloser
 	keepstoreLogbuf  *bufThenWrite
@@ -1457,6 +1458,10 @@ func (runner *ContainerRunner) UpdateContainerFinal() error {
 	if runner.finalState == "Complete" && runner.OutputPDH != nil {
 		update["output"] = *runner.OutputPDH
 	}
+	var it arvados.InstanceType
+	if j := os.Getenv("InstanceType"); j != "" && json.Unmarshal([]byte(j), &it) == nil && it.Price > 0 {
+		update["cost"] = it.Price * time.Now().Sub(runner.costStartTime).Seconds() / time.Hour.Seconds()
+	}
 	return runner.DispatcherArvClient.Update("containers", runner.Container.UUID, arvadosclient.Dict{"container": update}, nil)
 }
 
@@ -1489,6 +1494,7 @@ func (runner *ContainerRunner) Run() (err error) {
 	runner.CrunchLog.Printf("Using FUSE mount: %s", v)
 	runner.CrunchLog.Printf("Using container runtime: %s", runner.executor.Runtime())
 	runner.CrunchLog.Printf("Executing container: %s", runner.Container.UUID)
+	runner.costStartTime = time.Now()
 
 	hostname, hosterr := os.Hostname()
 	if hosterr != nil {