Procházet zdrojové kódy

Allow variable number of arguments for iter.Scan()

Staņislavs Koikovs před 12 roky
rodič
revize
effe8de7bf
1 změnil soubory, kde provedl 2 přidání a 2 odebrání
  1. 2 2
      session.go

+ 2 - 2
session.go

@@ -234,11 +234,11 @@ func (iter *Iter) Scan(dest ...interface{}) bool {
 	if iter.next != nil && iter.pos == iter.next.pos {
 		go iter.next.fetch()
 	}
-	if len(dest) != len(iter.columns) {
+	if len(dest) > len(iter.columns) {
 		iter.err = errors.New("count mismatch")
 		return false
 	}
-	for i := 0; i < len(iter.columns); i++ {
+	for i := 0; i < len(dest); i++ {
 		err := Unmarshal(iter.columns[i].TypeInfo, iter.rows[iter.pos][i], dest[i])
 		if err != nil {
 			iter.err = err