PKGBUILD/pkg/logfile/csv.go

package logfile

import (
	"encoding/csv"
	"fmt"
	"os"
	"strconv"
	"time"

	"github.com/go-flucky/flucky/pkg/types"
)

type csvLogfile struct {
	logfile string
}

func (cl *csvLogfile) Append(compression bool, measuredValues []types.MeasuredValue) error {

	allMeasuredValues, err := cl.Read()
	if err != nil {
		return err
	}

	allMeasuredValues = append(allMeasuredValues, measuredValues...)

	if compression {
		allMeasuredValues = Compression(allMeasuredValues)
	}

	err = cl.Write(allMeasuredValues)
	if err != nil {
		return err

	}

	return nil
}

func (cl *csvLogfile) Read() ([]types.MeasuredValue, error) {
	if _, err := os.Stat(cl.logfile); os.IsNotExist(err) {
		return nil, fmt.Errorf("%v: %v", errorLogfileNotFound, cl.logfile)
	}

	f, err := os.Open(cl.logfile)
	if err != nil {
		return nil, fmt.Errorf("%v: %v", errorLogfileOpen, cl.logfile)
	}
	defer f.Close()

	r := csv.NewReader(f)
	records, err := r.ReadAll()
	if err != nil {
		return nil, fmt.Errorf("%v %v: %v", errorLogfileDecode, cl.logfile, err)
	}

	measuredValues := make([]types.MeasuredValue, 0)

	for _, record := range records {

		// Value
		value, err := strconv.ParseFloat(record[2], 64)
		if err != nil {
			return nil, fmt.Errorf("%v %v: %v", errorParseFloat, record[2], err)
		}

		// Times
		times := make([]time.Time, 0)
		for _, i := range []int{3, 4} {
			time, err := time.Parse(timeFormat, record[i])
			if err != nil {
				return nil, fmt.Errorf("%v %v: %v", errorParseTime, record[i], err)
			}
			times = append(times, time)
		}

		switch record[1] {
		case fmt.Sprint(types.MeasuredValueTypeHumidity):
			measuredValue := &types.Humidity{
				HumidityID: record[0], // 0
				// 1 - MeasuredValue
				HumidityValue:    value,     // 2
				HumidityFromDate: times[0],  // 3
				HumidityTillDate: times[1],  // 4
				SensorID:         record[5], // 5
			}

			// CreationDate
			creationDate, err := time.Parse(timeFormat, record[6])
			if err != nil {
				return nil, fmt.Errorf("%v %v: %v", errorParseTime, record[6], err)
			}
			measuredValue.CreationDate = &creationDate

			// UpdateDate
			if record[7] != "null" {
				updateDate, err := time.Parse(timeFormat, record[7])
				if err != nil {
					return nil, fmt.Errorf("%v %v: %v", errorParseTime, record[7], err)
				}
				measuredValue.UpdateDate = &updateDate
			}

			measuredValues = append(measuredValues, measuredValue)

		case fmt.Sprint(types.MeasuredValueTypeTemperature):
			measuredValue := &types.Temperature{
				TemperatureID: record[0], // 0
				// 1 - MeasuredValue
				TemperatureValue:    value,     // 2
				TemperatureFromDate: times[0],  // 3
				TemperatureTillDate: times[1],  // 4
				SensorID:            record[5], // 5
			}

			// CreationDate
			creationDate, err := time.Parse(timeFormat, record[6])
			if err != nil {
				return nil, fmt.Errorf("%v %v: %v", errorParseTime, record[6], err)
			}
			measuredValue.CreationDate = &creationDate

			// UpdateDate
			if record[7] != "null" {
				updateDate, err := time.Parse(timeFormat, record[7])
				if err != nil {
					return nil, fmt.Errorf("%v %v: %v", errorParseTime, record[7], err)
				}
				measuredValue.UpdateDate = &updateDate
			}

			measuredValues = append(measuredValues, measuredValue)
		default:
			continue
		}

	}

	return measuredValues, nil

}

func (cl *csvLogfile) Write(measuredValues []types.MeasuredValue) error {
	f, err := os.Create(cl.logfile)
	if err != nil {
		return fmt.Errorf("%v: %v", errorLogfileCreate, cl.logfile)
	}
	defer f.Close()

	writeCreationDate(measuredValues)

	w := csv.NewWriter(f)

	for _, measuredValue := range measuredValues {

		record := []string{
			measuredValue.GetID(),
			fmt.Sprintf("%v", measuredValue.GetMeasuredValueType()),
			fmt.Sprintf("%v", measuredValue.GetValue()),
			measuredValue.GetFromDate().Format(timeFormat),
			measuredValue.GetTillDate().Format(timeFormat),
			measuredValue.GetSensorID(),
		}

		if measuredValue.GetCreationDate() != nil {
			record = append(record, measuredValue.GetCreationDate().Format(timeFormat))
		} else {
			record = append(record, "null")
		}

		if measuredValue.GetUpdateDate() != nil {
			record = append(record, measuredValue.GetUpdateDate().Format(timeFormat))
		} else {
			record = append(record, "null")
		}

		w.Write(record)
	}

	w.Flush()

	return nil
}