Skip to content

Commit

Permalink
Merge pull request #3402 from acharyasreej/issue-3400
Browse files Browse the repository at this point in the history
api: Expose vSphere API to evict subscribed content library
  • Loading branch information
dougm committed Apr 12, 2024
2 parents d528fec + 70a7617 commit 0b6df0d
Show file tree
Hide file tree
Showing 7 changed files with 181 additions and 2 deletions.
15 changes: 15 additions & 0 deletions govc/USAGE.md
Original file line number Diff line number Diff line change
Expand Up @@ -217,6 +217,7 @@ but appear via `govc $cmd -h`:
- [library.cp](#librarycp)
- [library.create](#librarycreate)
- [library.deploy](#librarydeploy)
- [library.evict](#libraryevict)
- [library.export](#libraryexport)
- [library.import](#libraryimport)
- [library.info](#libraryinfo)
Expand Down Expand Up @@ -3559,6 +3560,20 @@ Options:
-profile= Storage profile
```

## library.evict

```
Usage: govc library.evict [OPTIONS] LIBRARY NAME | ITEM NAME
Evict library NAME or item NAME.
Examples:
govc library.evict subscribed-library
govc library.evict subscribed-library/item
Options:
```

## library.export

```
Expand Down
79 changes: 79 additions & 0 deletions govc/library/evict.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,79 @@
/*
Copyright (c) 2024-2024 VMware, Inc. All Rights Reserved.
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 library

import (
"context"
"flag"
"fmt"

"github.com/vmware/govmomi/govc/cli"
"github.com/vmware/govmomi/govc/flags"
"github.com/vmware/govmomi/vapi/library"
)

type evict struct {
*flags.ClientFlag
}

func init() {
cli.Register("library.evict", &evict{})
}

func (cmd *evict) Register(ctx context.Context, f *flag.FlagSet) {
cmd.ClientFlag, ctx = flags.NewClientFlag(ctx)
cmd.ClientFlag.Register(ctx, f)
}

func (cmd *evict) Usage() string {
return "LIBRARY NAME | ITEM NAME"
}

func (cmd *evict) Description() string {
return `Evict library NAME or item NAME.
Examples:
govc library.evict subscribed-library
govc library.evict subscribed-library/item`
}

func (cmd *evict) Run(ctx context.Context, f *flag.FlagSet) error {
if f.NArg() != 1 {
return flag.ErrHelp
}

c, err := cmd.RestClient()
if err != nil {
return err
}

m := library.NewManager(c)

res, err := flags.ContentLibraryResult(ctx, c, "", f.Arg(0))
if err != nil {
return err
}

switch t := res.GetResult().(type) {
case library.Library:
return m.EvictSubscribedLibrary(ctx, &t)
case library.Item:
return m.EvictSubscribedLibraryItem(ctx, &t)
default:
return fmt.Errorf("%q is a %T", f.Arg(0), t)
}
}
1 change: 1 addition & 0 deletions govc/library/info.go
Original file line number Diff line number Diff line change
Expand Up @@ -223,6 +223,7 @@ func (r infoResultsWriter) writeItem(
}
fmt.Fprintf(w, " Type:\t%s\n", v.Type)
fmt.Fprintf(w, " Size:\t%s\n", units.ByteSize(v.Size))
fmt.Fprintf(w, " Cached:\t%t\n", v.Cached)
fmt.Fprintf(w, " Created:\t%s\n", v.CreationTime.Format(time.ANSIC))
fmt.Fprintf(w, " Modified:\t%s\n", v.LastModifiedTime.Format(time.ANSIC))
fmt.Fprintf(w, " Version:\t%s\n", v.Version)
Expand Down
46 changes: 46 additions & 0 deletions govc/test/library.bats
Original file line number Diff line number Diff line change
Expand Up @@ -607,3 +607,49 @@ EOF
assert_success
assert_matches INVALID_URL
}

@test "library.evict" {
vcsim_env

run govc library.create -pub published-content
assert_success
id="$output"

url="https://$(govc env GOVC_URL)/cls/vcsp/lib/$id"

run govc library.info published-content
assert_success
assert_matches "Publication:"
assert_matches "$url"

run govc library.import published-content "$GOVC_IMAGES/ttylinux-latest.ova"
assert_success

run govc library.create -sub "$url" -sub-ondemand=true subscribed-content
assert_success

run govc library.info subscribed-content
assert_success
assert_matches "Subscription:"
assert_matches "$url"

run govc library.ls subscribed-content/ttylinux-latest/
assert_success
assert_matches "/subscribed-content/ttylinux-latest/ttylinux-pc_i486-16.1.ovf"

run govc library.sync subscribed-content/ttylinux-latest
assert_success

# assert cached is false after item sync
cached=$(govc library.info subscribed-content/ttylinux-latest | grep Cached: | awk '{print $2}')
assert_equal "true" "$cached"

run govc library.evict subscribed-content/ttylinux-latest
assert_success

# assert cached is false after library item evict
cached=$(govc library.info subscribed-content/ttylinux-latest | grep Cached: | awk '{print $2}')
assert_equal "false" "$cached"
}


10 changes: 9 additions & 1 deletion vapi/library/library.go
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
Copyright (c) 2018-2023 VMware, Inc. All Rights Reserved.
Copyright (c) 2018-2024 VMware, Inc. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -322,3 +322,11 @@ func (c *Manager) DeleteSubscriber(ctx context.Context, library *Library, subscr
url := c.Resource(internal.Subscriptions).WithID(library.ID).WithAction("delete")
return c.Do(ctx, url.Request(http.MethodPost, &id), nil)
}

// EvictSubscribedLibrary evicts the cached content of an on-demand subscribed library.
// This operation allows the cached content of a subscribed library to be removed to free up storage capacity.
func (c *Manager) EvictSubscribedLibrary(ctx context.Context, library *Library) error {
path := internal.SubscribedLibraryPath
url := c.Resource(path).WithID(library.ID).WithAction("evict")
return c.Do(ctx, url.Request(http.MethodPost), nil)
}
8 changes: 8 additions & 0 deletions vapi/library/library_item.go
Original file line number Diff line number Diff line change
Expand Up @@ -206,3 +206,11 @@ func (c *Manager) FindLibraryItems(
var res []string
return res, c.Do(ctx, url.Request(http.MethodPost, spec), &res)
}

// EvictSubscribedLibraryItem evicts the cached content of a library item in an on-demand subscribed library.
// This operation allows the cached content of a subscribed library item to be removed to free up storage capacity.
func (c *Manager) EvictSubscribedLibraryItem(ctx context.Context, item *Item) error {
path := internal.SubscribedLibraryItem
url := c.Resource(path).WithID(item.ID).WithAction("evict")
return c.Do(ctx, url.Request(http.MethodPost), nil)
}
24 changes: 23 additions & 1 deletion vapi/simulator/simulator.go
Original file line number Diff line number Diff line change
Expand Up @@ -905,6 +905,19 @@ func (s *handler) library(w http.ResponseWriter, r *http.Request) {
}
}

func (content *content) cached(val bool) {
for _, item := range content.Item {
item.cached(val)
}
}

func (item *item) cached(val bool) {
item.Cached = val
for _, file := range item.File {
file.Cached = types.NewBool(val)
}
}

func (s *handler) publish(w http.ResponseWriter, r *http.Request, sids []internal.SubscriptionDestination, l *content, vmtx *item) bool {
var ids []string
if len(sids) == 0 {
Expand Down Expand Up @@ -991,10 +1004,14 @@ func (s *handler) libraryID(w http.ResponseWriter, r *http.Request) {
case "sync":
if l.Type == "SUBSCRIBED" {
l.LastSyncTime = types.NewTime(time.Now())
l.cached(true)
OK(w)
} else {
http.NotFound(w, r)
}
case "evict":
l.cached(false)
OK(w)
}
case http.MethodGet:
OK(w, l)
Expand Down Expand Up @@ -1242,8 +1259,9 @@ func (s *handler) libraryItemID(w http.ResponseWriter, r *http.Request) {

OK(w, id)
case "sync":
if l.Type == "SUBSCRIBED" {
if l.Type == "SUBSCRIBED" || l.Publication != nil {
item.LastSyncTime = types.NewTime(time.Now())
item.cached(true)
OK(w)
} else {
http.NotFound(w, r)
Expand All @@ -1255,6 +1273,9 @@ func (s *handler) libraryItemID(w http.ResponseWriter, r *http.Request) {
OK(w)
}
}
case "evict":
item.cached(false)
OK(w, id)
}
case http.MethodGet:
OK(w, item)
Expand Down Expand Up @@ -2309,6 +2330,7 @@ func (s *handler) libraryItemTemplateID(w http.ResponseWriter, r *http.Request)
return
}

item.cached(true)
ref, err := s.cloneVM(item.Template.Value, spec.Name, p, spec.DiskStorage)
if err != nil {
BadRequest(w, err.Error())
Expand Down

0 comments on commit 0b6df0d

Please sign in to comment.