mirror of
https://github.com/kopia/kopia.git
synced 2026-01-21 04:47:57 -05:00
7ee30b76bb
* fix(repository): fixed handling of content.Info Previously content.Info was an interface which was implemented by: * index.InfoStruct * index.indexEntryInfoV1 * index.indexEntryInfoV2 The last 2 implementations were relying on memory-mapped files which in rare cases could be closed while Kopia was still processing them leading to #2599. This changes fixes the bug and strictly separates content.Info (which is now always a struct) from the other two (which were renamed as index.InfoReader and only used inside repo/content/...). In addition to being safer, this _should_ reduce memory allocations. * reduce the size of content.Info with proper alignment. * pr feedback * renamed index.InfoStruct to index.Info
127 lines
2.4 KiB
Go
127 lines
2.4 KiB
Go
package cli
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/alecthomas/kingpin/v2"
|
|
|
|
"github.com/kopia/kopia/snapshot"
|
|
)
|
|
|
|
type jsonOutput struct {
|
|
jsonOutput bool
|
|
jsonIndent bool
|
|
jsonVerbose bool // output non-essential stats as part of JSON
|
|
|
|
out io.Writer
|
|
}
|
|
|
|
func (c *jsonOutput) setup(svc appServices, cmd *kingpin.CmdClause) {
|
|
cmd.Flag("json", "Output result in JSON format to stdout").BoolVar(&c.jsonOutput)
|
|
cmd.Flag("json-indent", "Output result in indented JSON format to stdout").Hidden().BoolVar(&c.jsonIndent)
|
|
cmd.Flag("json-verbose", "Output non-essential data (e.g. statistics) in JSON format").Hidden().BoolVar(&c.jsonVerbose)
|
|
|
|
c.out = svc.stdout()
|
|
}
|
|
|
|
func (c *jsonOutput) cleanupSnapshotManifestForJSON(v *snapshot.Manifest) interface{} {
|
|
m := *v
|
|
|
|
if !c.jsonVerbose {
|
|
return struct {
|
|
*snapshot.Manifest
|
|
|
|
// trick to remove 'stats' completely.
|
|
Stats string `json:"stats,omitempty"`
|
|
}{Manifest: v}
|
|
}
|
|
|
|
return &m
|
|
}
|
|
|
|
func (c *jsonOutput) cleanupSnapshotManifestListForJSON(manifests []*snapshot.Manifest) interface{} {
|
|
var res []interface{}
|
|
|
|
for _, m := range manifests {
|
|
res = append(res, c.cleanupSnapshotManifestForJSON(m))
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
func (c *jsonOutput) cleanupForJSON(v interface{}) interface{} {
|
|
switch v := v.(type) {
|
|
case *snapshot.Manifest:
|
|
return c.cleanupSnapshotManifestForJSON(v)
|
|
case []*snapshot.Manifest:
|
|
return c.cleanupSnapshotManifestListForJSON(v)
|
|
default:
|
|
return v
|
|
}
|
|
}
|
|
|
|
func (c *jsonOutput) jsonBytes(v interface{}) []byte {
|
|
return c.jsonIndentedBytes(v, "")
|
|
}
|
|
|
|
func (c *jsonOutput) jsonIndentedBytes(v interface{}, indent string) []byte {
|
|
v = c.cleanupForJSON(v)
|
|
|
|
var (
|
|
b []byte
|
|
err error
|
|
)
|
|
|
|
if c.jsonIndent {
|
|
b, err = json.MarshalIndent(v, indent+"", indent+" ")
|
|
} else {
|
|
b, err = json.Marshal(v)
|
|
}
|
|
|
|
if err != nil {
|
|
panic("error serializing JSON, that should not happen: " + err.Error())
|
|
}
|
|
|
|
return b
|
|
}
|
|
|
|
type jsonList struct {
|
|
separator string
|
|
o *jsonOutput
|
|
}
|
|
|
|
func (l *jsonList) begin(o *jsonOutput) {
|
|
l.o = o
|
|
|
|
if o.jsonOutput {
|
|
fmt.Fprintf(l.o.out, "[")
|
|
|
|
if !o.jsonIndent {
|
|
l.separator = "\n "
|
|
}
|
|
}
|
|
}
|
|
|
|
func (l *jsonList) end() {
|
|
if l.o.jsonOutput {
|
|
if !l.o.jsonIndent {
|
|
fmt.Fprintf(l.o.out, "\n")
|
|
}
|
|
|
|
fmt.Fprintf(l.o.out, "]")
|
|
}
|
|
}
|
|
|
|
func (l *jsonList) emit(v interface{}) {
|
|
fmt.Fprintf(l.o.out, l.separator)
|
|
fmt.Fprintf(l.o.out, "%s", l.o.jsonBytes(v))
|
|
|
|
if l.o.jsonIndent {
|
|
l.separator = ","
|
|
} else {
|
|
l.separator = ",\n "
|
|
}
|
|
}
|