esgo2dump/internal/xfile/xfile.go

package xfile

import (
	"bufio"
	"encoding/json"
	"esgo2dump/internal/interfaces"
	"github.com/sirupsen/logrus"
	"os"
)

type client struct {
	f       *os.File
	scanner *bufio.Scanner
}

func (c *client) IsInput() bool {
	//TODO implement me
	panic("implement me")
}

func (c *client) IsFile() bool {
	return true
}

func (c *client) Write(docs []*interfaces.ESSource) (int, error) {
	var (
		err   error
		bs    []byte
		count = 0
	)

	for _, doc := range docs {
		if bs, err = json.Marshal(doc); err != nil {
			return count, err
		}

		bs = append(bs, '\n')

		if _, err = c.f.Write(bs); err != nil {
			return count, err
		}

		count++
	}

	return count, nil
}

func (c *client) Read(i int) ([]*interfaces.ESSource, error) {
	var (
		err   error
		count = 0
		list  = make([]*interfaces.ESSource, 0, i)
	)

	for c.scanner.Scan() {
		line := c.scanner.Text()

		logrus.Debugf("xfile.Read: line=%s", line)

		item := new(interfaces.ESSource)
		if err = json.Unmarshal([]byte(line), item); err != nil {
			return list, err
		}

		list = append(list, item)

		count++
		if count >= i {
			break
		}
	}

	if err = c.scanner.Err(); err != nil {
		return list, err
	}

	return list, nil
}

func (c *client) Close() error {
	return c.f.Close()
}

func NewClient(file *os.File, ioType interfaces.IO) (interfaces.DumpIO, error) {
	c := &client{f: file}

	if ioType == interfaces.IOInput {
		c.scanner = bufio.NewScanner(c.f)
	}

	return c, nil
}