orus-io / yago Goto Github PK

// Model of the database
type Model struct {
	Meta *yago.Metadata

	RoutingEntry RoutingEntryModel
}

// RoutingEntry is an entry of the routing table
//yago:routing_table
type RoutingEntry struct {
	Key   string `yago:"primary_key"`
	Value string `yago:"."`
}

// NewModel initialize a Model
func NewModel() *Model {
	meta := yago.NewMetadata()
	return &Model{
		Meta:         meta,
		RoutingEntry: NewRoutingEntryModel(meta),
	}
}

// Scan a struct
func (mapper RoutingEntryMapper) Scan(rows *sql.Rows, instance yago.MappedStruct) error {
	s, ok := instance.(*RoutingEntry)
	if !ok {
		panic(fmt.Sprintf(
			"Wrong struct type passed to the mapper. Expected &RoutingEntry{}, got %s",
			reflect.TypeOf(instance).Name(),
		))
	}
	if err := rows.Scan(
		&s.Key,
		&s.Value,
	); err != nil {
		return err
	}
	return nil
}

// Scan a struct
func (mapper RoutingEntryMapper) Scan(rows *sql.Rows, instance yago.MappedStruct) error {
	s, ok := instance.(*RoutingEntry)
	if !ok {
		panic(fmt.Sprintf(
			"Wrong struct type passed to the mapper. Expected &RoutingEntry{}, got %s",
			reflect.TypeOf(instance).Name(),
		))
	}
	err := rows.Scan(
		&s.Key,
		&s.Value,
	)
	return err
}