1 files changed, 4 insertions, 31 deletions

diff --git a/listener/deal.go b/listener/deal.go
index 70c2827..af65ff8 100644
--- a/listener/deal.go
+++ b/listener/deal.go
@@ -10,17 +10,8 @@ import (
         "go.mongodb.org/mongo-driver/bson"
 )
 
-const (
-        dbName      string = "test"
-        pidColName  string = "pids"
-        fdColName   string = "fds"
-        fileColName string = "files"
-)
-
-var pidCol, fdCol, fileCol mongoClient
-
 func initPidCol() (err error) {
-        // TODO: 这里是否需要补全一下进程信息？
+        // 这里是否需要补全一下进程信息？
         dirs, err := os.ReadDir(fmt.Sprintf("/proc/%d/task", containerdPid))
         if err != nil {
                 return err
@@ -41,6 +32,9 @@ func initPidCol() (err error) {
                         process.Star = true
                 }
                 err = pidCol.InsertOne(process)
+                if err != nil {
+                        return err
+                }
         }
         return nil
 }
@@ -49,27 +43,6 @@ func deal() {
         defer wg.Done()
         var cooked Event
         var ok bool
-        var err error
-
-        if err = pidCol.init(dbName, pidColName); err != nil {
-                fmt.Fprintf(os.Stderr, "Error while initing the mongodb: %v\n", err)
-                return
-        }
-        if err = initPidCol(); err != nil {
-                fmt.Fprintf(os.Stderr, "Err while initing pidcol: %v\n", err)
-        }
-
-        if err = fdCol.init(dbName, fdColName); err != nil {
-                fmt.Fprintf(os.Stderr, "Error while initing the mongodb: %v\n", err)
-                return
-        }
-        if err = fileCol.init(dbName, fileColName); err != nil {
-                fmt.Fprintf(os.Stderr, "Error while initing the mongodb: %v\n", err)
-        }
-
-        defer pidCol.Disconnect()
-        defer fdCol.Disconnect()
-        defer fileCol.Disconnect()
 
         for {
                 cooked, ok = <-cookedChan