parsetree_list.go

package pg_query

import (
	"encoding/hex"
	"encoding/json"
	"fmt"

	nodes "github.com/lfittl/pg_query_go/nodes"
)

type ParsetreeList struct {
	Statements []nodes.Node
}

func (input ParsetreeList) MarshalJSON() ([]byte, error) {
	type ParsetreeListAlias ParsetreeList
	return json.Marshal(input.Statements)
}

func (output *ParsetreeList) UnmarshalJSON(input []byte) (err error) {
	var list []json.RawMessage
	err = json.Unmarshal([]byte(input), &list)
	if err != nil {
		return
	}

	for _, nodeJson := range list {
		var node nodes.Node
		node, err = nodes.UnmarshalNodeJSON(nodeJson)
		if err != nil {
			return
		}
		output.Statements = append(output.Statements, node)
	}

	return
}

func (input ParsetreeList) Fingerprint() string {
	const fingerprintVersion uint = 1

	ctx := nodes.NewFingerprintHashContext()
	for _, node := range input.Statements {
		node.Fingerprint(ctx, nil, "")
	}

	return fmt.Sprintf("%02x%s", fingerprintVersion, hex.EncodeToString(ctx.Sum()))
}