fission-src/pkg/fission-cli/cmd/records/view.go

/*
Copyright 2019 The Fission 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 records

import (
	"fmt"
	"os"
	"text/tabwriter"

	"github.com/pkg/errors"

	"github.com/fission/fission/pkg/controller/client"
	"github.com/fission/fission/pkg/fission-cli/cliwrapper/cli"
	"github.com/fission/fission/pkg/fission-cli/util"
	redisCache "github.com/fission/fission/pkg/redis/build/gen"
)

type ViewSubCommand struct {
	client *client.Client
}

func View(flags cli.Input) error {
	c, err := util.GetServer(flags)
	if err != nil {
		return err
	}
	opts := ViewSubCommand{
		client: c,
	}
	return opts.do(flags)
}

func (opts *ViewSubCommand) do(flags cli.Input) error {
	return opts.run(flags)
}

func (opts *ViewSubCommand) run(flags cli.Input) error {
	var verbosity int
	if flags.Bool("v") && flags.Bool("vv") {
		return errors.New("conflicting verbosity levels, use either --v or --vv")
	}
	if flags.Bool("v") {
		verbosity = 1
	}
	if flags.Bool("vv") {
		verbosity = 2
	}

	function := flags.String("function")
	trigger := flags.String("trigger")
	from := flags.String("from")
	to := flags.String("to")

	//Refuse multiple filters for now
	if multipleFiltersSpecified(function, trigger, from+to) {
		return errors.New("maximum of one filter is currently supported, either --function, --trigger, or --from,--to")
	}

	if len(function) != 0 {
		return recordsByFunction(opts.client, function, verbosity)
	}
	if len(trigger) != 0 {
		return recordsByTrigger(opts.client, trigger, verbosity)
	}
	if len(from) != 0 && len(to) != 0 {
		return recordsByTime(opts.client, from, to, verbosity)
	}
	err := recordsAll(opts.client, verbosity)
	if err != nil {
		return errors.Wrap(err, "error viewing records")
	}
	return nil
}

func recordsAll(client *client.Client, verbosity int) error {
	records, err := client.RecordsAll()
	if err != nil {
		return errors.Wrap(err, "error viewing records")
	}
	showRecords(records, verbosity)
	return nil
}

func recordsByTrigger(client *client.Client, trigger string, verbosity int) error {
	records, err := client.RecordsByTrigger(trigger)
	if err != nil {
		return errors.Wrap(err, "error viewing records")
	}
	showRecords(records, verbosity)
	return nil
}

// TODO: More accurate function name (function filter)
func recordsByFunction(client *client.Client, function string, verbosity int) error {
	records, err := client.RecordsByFunction(function)
	if err != nil {
		return errors.Wrap(err, "error viewing records")
	}
	showRecords(records, verbosity)
	return nil
}

func recordsByTime(client *client.Client, from string, to string, verbosity int) error {
	records, err := client.RecordsByTime(from, to)
	if err != nil {
		return errors.Wrap(err, "error viewing records")
	}
	showRecords(records, verbosity)
	return nil
}

func showRecords(records []*redisCache.RecordedEntry, verbosity int) {
	w := tabwriter.NewWriter(os.Stdout, 0, 0, 1, ' ', 0)

	if verbosity == 1 {
		fmt.Fprintf(w, "%v\t%v\t%v\t%v\t%v\n",
			"REQUID", "REQUEST METHOD", "FUNCTION", "RESPONSE STATUS", "TRIGGER")
		for _, record := range records {
			fmt.Fprintf(w, "%v\t%v\t%v\t%v\t%v\n",
				record.ReqUID, record.Req.Method, record.Req.Header["X-Fission-Function-Name"], record.Resp.Status, record.Trigger)
		}
	} else if verbosity == 2 {
		for _, record := range records {
			fmt.Println(record)
		}
	} else {
		fmt.Fprintf(w, "%v\n",
			"REQUID")
		for _, record := range records {
			fmt.Fprintf(w, "%v\n",
				record.ReqUID)
		}
	}
	w.Flush()
}

func multipleFiltersSpecified(entries ...string) bool {
	var specified int
	for _, entry := range entries {
		if len(entry) > 0 {
			specified += 1
		}
	}
	return specified > 1
}