瀏覽代碼

Allow variable number of arguments for iter.Scan()

Staņislavs Koikovs 12 年之前
父節點
當前提交
effe8de7bf
共有 1 個文件被更改,包括 2 次插入2 次删除
  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