fetch.go

package clickhouse

import (
	"fmt"
	"strconv"
	"time"
)

//TODO maybe need to add GetArray<Type> func(column string) (Type, error) where Type is all listen types above

// Columns returns columns list
func (result Result) Columns() (columns []string) {
	for column := range result.data {
		columns = append(columns, column)
	}

	return columns
}

// Exist returns true if field is exist or false
func (result Result) Exist(column string) bool {
	cfg.logger.debug(fmt.Sprintf("Try to check if exist by `%s`", column))

	_, ok := result.data[column]

	return ok
}

// String returns value of string
func (result Result) String(column string) (value string, err error) {
	cfg.logger.debug(fmt.Sprintf("Try to get value by `%s`", column))

	value, ok := result.data[column]
	if !ok {
		err = fmt.Errorf("can't get value by `%s`", column)

		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return
	}

	cfg.logger.debug(fmt.Sprintf("Success get `%s` = %s", column, value))

	return
}

// Bytes returns value of bytes
func (result Result) Bytes(column string) (bytes []byte, err error) {
	value, err := result.String(column)
	if err != nil {
		return
	}

	bytes = []byte(value)

	return
}

func (result Result) getUInt(column string, bitSize int) (ui64 uint64, err error) {
	value, err := result.String(column)
	if err != nil {
		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return
	}

	ui64, err = strconv.ParseUint(value, 10, bitSize)
	if err != nil {
		err = fmt.Errorf("can't convert value %s to uint%d: %s", value, bitSize, err.Error())

		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return
	}

	return
}

// Bool returns value as bool
func (result Result) Bool(column string) (f bool, err error) {
	i, err := result.getUInt(column, 8)
	if err != nil {
		return
	}

	return i == 1, nil
}

// UInt8 returns value as uint8
func (result Result) UInt8(column string) (ui8 uint8, err error) {
	i, err := result.getUInt(column, 8)

	return uint8(i), err
}

// UInt16 returns value as uint16
func (result Result) UInt16(column string) (ui16 uint16, err error) {
	i, err := result.getUInt(column, 16)

	return uint16(i), err
}

// UInt32 returns value as uint32
func (result Result) UInt32(column string) (ui32 uint32, err error) {
	i, err := result.getUInt(column, 32)

	return uint32(i), err
}

// UInt64 returns value as uint64
func (result Result) UInt64(column string) (ui64 uint64, err error) {
	i, err := result.getUInt(column, 64)

	return uint64(i), err
}

func (result Result) getInt(column string, bitSize int) (i64 int64, err error) {
	value, err := result.String(column)
	if err != nil {
		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return 0, err
	}

	i64, err = strconv.ParseInt(value, 10, bitSize)
	if err != nil {
		err := fmt.Errorf("can't convert value %s to int%d: %s", value, bitSize, err.Error())

		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return 0, err
	}

	return i64, nil
}

// Int8 returns value as int8
func (result Result) Int8(column string) (i8 int8, err error) {
	i, err := result.getInt(column, 8)

	return int8(i), err
}

// Int16 returns value as int16
func (result Result) Int16(column string) (i16 int16, err error) {
	i, err := result.getInt(column, 16)

	return int16(i), err
}

// Int32 returns value as int32
func (result Result) Int32(column string) (i32 int32, err error) {
	i, err := result.getInt(column, 32)

	return int32(i), err
}

// Int64 returns value as int64
func (result Result) Int64(column string) (i64 int64, err error) {
	i, err := result.getInt(column, 64)

	return int64(i), err
}

func (result Result) getFloat(column string, bitSize int) (f64 float64, err error) {
	value, err := result.String(column)
	if err != nil {
		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return 0, err
	}

	f64, err = strconv.ParseFloat(value, bitSize)
	if err != nil {
		err := fmt.Errorf("can't convert value %s to float%d: %s", value, bitSize, err.Error())

		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return 0, err
	}

	return f64, nil
}

// Float32 returns value as float32
func (result Result) Float32(column string) (f32 float32, err error) {
	f, err := result.getFloat(column, 32)

	return float32(f), err
}

// Float64 returns value as float64
func (result Result) Float64(column string) (f64 float64, err error) {
	f, err := result.getFloat(column, 64)

	return float64(f), err
}

// Date returns value as date
func (result Result) Date(column string) (t time.Time, err error) {
	value, err := result.String(column)
	if err != nil {
		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return time.Time{}, err
	}

	t, err = time.Parse("2006-01-02", value)
	if err != nil {
		err := fmt.Errorf("can't convert value %s to date: %s", value, err.Error())

		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return time.Time{}, err
	}

	return t, nil
}

// DateTime returns value as datetime
func (result Result) DateTime(column string) (t time.Time, err error) {
	value, err := result.String(column)
	if err != nil {
		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return time.Time{}, err
	}

	t, err = time.Parse("2006-01-02 15:04:05", value)
	if err != nil {
		err := fmt.Errorf("can't convert value %s to datetime: %s", value, err.Error())

		cfg.logger.error(fmt.Sprintf("Catch error %s", err.Error()))

		return time.Time{}, err
	}

	return t, nil
}