コード例 #1
0
ファイル: tee.go プロジェクト: ando-masaki/someutils
// Exec actually performs the tee
func (tee *SomeTee) Invoke(invocation *someutils.Invocation) (error, int) {
	invocation.ErrPipe.Drain()
	invocation.AutoHandleSignals()
	flag := os.O_CREATE
	if tee.isAppend {
		flag = flag | os.O_APPEND
	}
	writeables := uggo.ToWriteableOpeners(tee.args, flag, 0666)
	files, err := uggo.OpenAll(writeables)
	if err != nil {
		return err, 1
	}
	writers := []io.Writer{invocation.MainPipe.Out}
	for _, file := range files {
		writers = append(writers, file)
	}
	multiwriter := io.MultiWriter(writers...)
	_, err = io.Copy(multiwriter, invocation.MainPipe.In)
	if err != nil {
		return err, 1
	}
	for _, file := range files {
		err = file.Close()
		if err != nil {
			return err, 1
		}
	}
	return nil, 0

}
コード例 #2
0
ファイル: tee.go プロジェクト: jayvansantos/someutils
func Tee(call []string) error {
	options := TeeOptions{}
	flagSet := uggo.NewFlagSetDefault("tee", "", VERSION)
	flagSet.AliasedBoolVar(&options.isAppend, []string{"a", "append"}, false, "Append instead of overwrite")

	err := flagSet.Parse(call[1:])
	if err != nil {
		return err
	}
	if flagSet.ProcessHelpOrVersion() {
		return nil
	}
	flag := os.O_CREATE
	if options.isAppend {
		flag = flag | os.O_APPEND
	}
	writeables := flagSet.ArgsAsWriteables(flag, 0666)
	files, err := uggo.OpenAll(writeables)
	if err != nil {
		return err
	}
	writers := []io.Writer{os.Stdout}
	for _, file := range files {
		writers = append(writers, file)
	}
	multiwriter := io.MultiWriter(writers...)
	_, err = io.Copy(multiwriter, os.Stdin)
	if err != nil {
		return err
	}
	for _, file := range files {
		err = file.Close()
		if err != nil {
			return err
		}
	}
	return nil
}