array.go

package types

import (
	"fmt"
	"reflect"
)

type Array struct {
	v reflect.Value

	append AppenderFunc
	scan   ScannerFunc
}

var _ ValueAppender = (*Array)(nil)
var _ ValueScanner = (*Array)(nil)

func NewArray(vi interface{}) *Array {
	v := reflect.ValueOf(vi)
	if !v.IsValid() {
		panic(fmt.Errorf("pg: Array(nil)"))
	}
	return &Array{
		v: v,

		append: ArrayAppender(v.Type()),
		scan:   ArrayScanner(v.Type()),
	}
}

func (a *Array) AppendValue(b []byte, quote int) []byte {
	if a.append == nil {
		panic(fmt.Errorf("pg: Array(unsupported %s)", a.v.Type()))
	}
	return a.append(b, a.v, quote)
}

func (a *Array) ScanValue(rd Reader, n int) error {
	if a.scan == nil {
		return fmt.Errorf("pg: Array(unsupported %s)", a.v.Type())
	}
	return a.scan(a.v, rd, n)
}

func (a *Array) Value() interface{} {
	if a.v.IsValid() {
		return a.v.Interface()
	}
	return nil
}