sdk/go/arvados/collection.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: Apache-2.0
   4
   5 package arvados
   6
   7 import (
   8         "bufio"
   9         "fmt"
  10         "strings"
  11         "time"
  12
  13         "git.curoverse.com/arvados.git/sdk/go/blockdigest"
  14 )
  15
  16 // Collection is an arvados#collection resource.
  17 type Collection struct {
  18         UUID                      string     `json:"uuid,omitempty"`
  19         OwnerUUID                 string     `json:"owner_uuid,omitempty"`
  20         TrashAt                   *time.Time `json:"trash_at,omitempty"`
  21         ManifestText              string     `json:"manifest_text,omitempty"`
  22         UnsignedManifestText      string     `json:"unsigned_manifest_text,omitempty"`
  23         Name                      string     `json:"name,omitempty"`
  24         CreatedAt                 *time.Time `json:"created_at,omitempty"`
  25         ModifiedAt                *time.Time `json:"modified_at,omitempty"`
  26         PortableDataHash          string     `json:"portable_data_hash,omitempty"`
  27         ReplicationConfirmed      *int       `json:"replication_confirmed,omitempty"`
  28         ReplicationConfirmedAt    *time.Time `json:"replication_confirmed_at,omitempty"`
  29         ReplicationDesired        *int       `json:"replication_desired,omitempty"`
  30         StorageClassesDesired     []string   `json:"storage_classes_desired,omitempty"`
  31         StorageClassesConfirmed   []string   `json:"storage_classes_confirmed,omitempty"`
  32         StorageClassesConfirmedAt *time.Time `json:"storage_classes_confirmed_at,omitempty"`
  33         DeleteAt                  *time.Time `json:"delete_at,omitempty"`
  34         IsTrashed                 bool       `json:"is_trashed,omitempty"`
  35 }
  36
  37 func (c Collection) resourceName() string {
  38         return "collection"
  39 }
  40
  41 // SizedDigests returns the hash+size part of each data block
  42 // referenced by the collection.
  43 func (c *Collection) SizedDigests() ([]SizedDigest, error) {
  44         manifestText := c.ManifestText
  45         if manifestText == "" {
  46                 manifestText = c.UnsignedManifestText
  47         }
  48         if manifestText == "" && c.PortableDataHash != "d41d8cd98f00b204e9800998ecf8427e+0" {
  49                 // TODO: Check more subtle forms of corruption, too
  50                 return nil, fmt.Errorf("manifest is missing")
  51         }
  52         var sds []SizedDigest
  53         scanner := bufio.NewScanner(strings.NewReader(manifestText))
  54         scanner.Buffer(make([]byte, 1048576), len(manifestText))
  55         for scanner.Scan() {
  56                 line := scanner.Text()
  57                 tokens := strings.Split(line, " ")
  58                 if len(tokens) < 3 {
  59                         return nil, fmt.Errorf("Invalid stream (<3 tokens): %q", line)
  60                 }
  61                 for _, token := range tokens[1:] {
  62                         if !blockdigest.LocatorPattern.MatchString(token) {
  63                                 // FIXME: ensure it's a file token
  64                                 break
  65                         }
  66                         // FIXME: shouldn't assume 32 char hash
  67                         if i := strings.IndexRune(token[33:], '+'); i >= 0 {
  68                                 token = token[:33+i]
  69                         }
  70                         sds = append(sds, SizedDigest(token))
  71                 }
  72         }
  73         return sds, scanner.Err()
  74 }
  75
  76 // CollectionList is an arvados#collectionList resource.
  77 type CollectionList struct {
  78         Items          []Collection `json:"items"`
  79         ItemsAvailable int          `json:"items_available"`
  80         Offset         int          `json:"offset"`
  81         Limit          int          `json:"limit"`
  82 }