From 1a8c0399f318bd9dc8c543223b197f6cc2272add Mon Sep 17 00:00:00 2001
From: Marvin Zhang <tikazyq@163.com>
Date: Sun, 10 Jul 2022 20:22:51 +0800
Subject: [PATCH] feat(mongo): added more methods of mongo.FindResult

---
 mongo/result.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/mongo/result.go b/mongo/result.go
index 2169950..c665c4e 100644
--- a/mongo/result.go
+++ b/mongo/result.go
@@ -9,6 +9,10 @@ import (
 type FindResultInterface interface {
 	One(val interface{}) (err error)
 	All(val interface{}) (err error)
+	GetCol() (col *Col)
+	GetSingleResult() (res *mongo.SingleResult)
+	GetCursor() (cur *mongo.Cursor)
+	GetError() (err error)
 }
 
 func NewFindResult() (fr *FindResult) {
@@ -28,6 +32,11 @@ type FindResult struct {
 	err error
 }
 
+func (fr *FindResult) GetError() (err error) {
+	//TODO implement me
+	panic("implement me")
+}
+
 func (fr *FindResult) One(val interface{}) (err error) {
 	if fr.err != nil {
 		return fr.err
@@ -59,3 +68,15 @@ func (fr *FindResult) All(val interface{}) (err error) {
 	}
 	return fr.cur.All(ctx, val)
 }
+
+func (fr *FindResult) GetCol() (col *Col) {
+	return fr.col
+}
+
+func (fr *FindResult) GetSingleResult() (res *mongo.SingleResult) {
+	return fr.res
+}
+
+func (fr *FindResult) GetCursor() (cur *mongo.Cursor) {
+	return fr.cur
+}