sql_row_scanner.go

/*
 * Copyright (c) 2018 Miguel Ángel Ortuño.
 * See the LICENSE file for more information.
 */

package storage

import (
	"strings"

	"github.com/ortuman/jackal/storage/model"
)

type rowScanner interface {
	Scan(...interface{}) error
}

type rowsScanner interface {
	rowScanner
	Next() bool
}

func scanRosterItemEntity(ri *model.RosterItem, scanner rowScanner) error {
	var groups string
	if err := scanner.Scan(&ri.Username, &ri.JID, &ri.Name, &ri.Subscription, &groups, &ri.Ask, &ri.Ver); err != nil {
		return err
	}
	ri.Groups = strings.Split(groups, ";")
	return nil
}

func scanRosterItemEntities(scanner rowsScanner) ([]model.RosterItem, error) {
	var ret []model.RosterItem
	for scanner.Next() {
		var ri model.RosterItem
		if err := scanRosterItemEntity(&ri, scanner); err != nil {
			return nil, err
		}
		ret = append(ret, ri)
	}
	return ret, nil
}

func scanBlockListItemEntities(scanner rowsScanner) ([]model.BlockListItem, error) {
	var ret []model.BlockListItem
	for scanner.Next() {
		var it model.BlockListItem
		scanner.Scan(&it.Username, &it.JID)
		ret = append(ret, it)
	}
	return ret, nil
}