Crie uma ferramenta de linha de comando com Swift e MongoDB

1
mongoimport --host localhost:27017 --type csv --db school --collection students --file example.csv  --headerline

1
swift package init

1
swift package init --type=executable

1
swift package generate-xcodeproj

1
open <your project name>.xcodeproj/ # change this to the name that was created by the previous command.

1
firstname,lastname,assigned
2
Michael,Basic,FALSE
3
Dan,Acquilone,FALSE
4
Eli,Zimmerman,FALSE
5
Liam,Tyler,FALSE
6
Jane,Alberts,FALSE
7
Ted,Williams,FALSE
8
Suzy,Langford,FALSE
9
Paulina,Stern,FALSE
10
Jared,Lentz,FALSE
11
June,Gifford,FALSE
12
Wilma,Atkinson,FALSE

1
{
2
"_id": {
3
   "$oid": "5f491a3bf983e96173253352" // this will come from our driver.
4
},
5
"firstname": "Michael",
6
"lastname": "Basic",
7
"date": {
8
   "$date": "2020-08-28T14:52:43.398Z" // this will be set by our Struct default value
9
},
10
"assigned": false
11
}

1
import Foundation
2
3
let path = "/Users/mlynn/Desktop/example.csv" // change this to the path of your csv file
4
do {
5
   let contents = try String(contentsOfFile: path, encoding: .utf8)
6
   let rows = contents.components(separatedBy: NSCharacterSet.newlines)
7
   for row in rows {
8
      if row != "" {
9
            print("Got Row: \(row)")
10
      }
11
   }
12
}

1
// swift-tools-version:5.2
2
// The swift-tools-version declares the minimum version of Swift required to build this package.
3
import PackageDescription
4
5
let package = Package(
6
   name: "csvimport-swift",
7
   platforms: [
8
      .macOS(.v10_15),
9
   ],
10
   dependencies: [
11
      .package(url: "https://github.com/mongodb/mongo-swift-driver.git", from: "1.0.1"),
12
   ],
13
   targets: [
14
      .target(
15
            name: "csvimport-swift",
16
            dependencies: [.product(name: "MongoSwiftSync", package: "mongo-swift-driver")]),
17
      .testTarget(
18
            name: "csvimport-swiftTests",
19
            dependencies: ["csvimport-swift"]),
20
   ]
21
)

1
platforms: [
2
      .macOS(.v10_15)
3
],

1
swift package build

1
[3/3] Linking cmd

1
import Foundation
2
import MongoSwiftSync
3
4
var murl: String = "mongodb+srv://<username>:\(ProcessInfo.processInfo.environment["PASS"]!)@myfirstcluster.zbcul.mongodb.net/<dbname>?retryWrites=true&w=majority"
5
let client = try MongoClient(murl)
6
7
let db = client.db("students")
8
let session = client.startSession(options: ClientSessionOptions(causalConsistency: true))
9
10
struct Person: Codable {
11
   let firstname: String
12
   let lastname: String
13
   let date: Date = Date()
14
   let assigned: Bool
15
   let _id: BSONObjectID
16
}
17
18
let path = "/Users/mlynn/Desktop/example.csv"
19
var tempAssigned: Bool
20
var count: Int = 0
21
var header: Bool = true
22
23
let personCollection = db.collection("people", withType: Person.self)
24
25
do {
26
   let contents = try String(contentsOfFile: path, encoding: .utf8)
27
   let rows = contents.components(separatedBy: NSCharacterSet.newlines)
28
   for row in rows {
29
      if row != "" {
30
            var values: [String] = []
31
            values = row.components(separatedBy: ",")
32
            if header == true {
33
               header = false
34
            } else {
35
               if String(values[2]).lowercased() == "false" || Bool(values[2]) == false {
36
                  tempAssigned = false
37
               } else {
38
                  tempAssigned = true
39
               }
40
               try personCollection.insertOne(Person(firstname: values[0], lastname: values[1], assigned: tempAssigned, _id: BSONObjectID()), session: session)
41
               count.self += 1
42
               print("Inserted: \(count) \(row)")
43
44
            }
45
      }
46
   }
47
}

1
var murl: String = "mongodb+srv://<username>:\(ProcessInfo.processInfo.environment["PASS"]!)@myfirstcluster.zbcul.mongodb.net/<dbname>?retryWrites=true&w=majority"
2
let client = try MongoClient(murl)
3
4
let db = client.db("students")
5
let session = client.startSession(options: ClientSessionOptions(causalConsistency: true))

1
struct Code: Codable {
2
   let code: String
3
   let assigned: Bool
4
   let date: Date = Date()
5
   let _id: BSONObjectID
6
}

1
let path = "/Users/mlynn/Desktop/example.csv"
2
var tempAssigned: Bool
3
var count: Int = 0
4
var header: Bool = true

1
let personCollection = db.collection("people", withType: Person.self)

1
let contents = try String(contentsOfFile: path, encoding: .utf8)        // get the contents of our csv file with the String built-in
2
let rows = contents.components(separatedBy: NSCharacterSet.newlines)    // get the individual rows separated by newline characters
3
for row in rows {                                                       // Loop through all rows in the file.
4
   if row != "" {                                                       // in case we have an empty row... skip it.
5
         var values: [String] = []                                      // create / reset the values array of type string - to null.
6
         values = row.components(separatedBy: ",")                      // assign the values array to the fields in the row of data
7
         if header == true {                                            // if it's the first row... skip it and.
8
            header = false                                              // Set the header to false so we do this only once.
9
         } else {
10
            if String(values[2]).lowercased() == "false" || Bool(values[2]) == false {
11
               tempAssigned = false                                     // Above: if its the string or boolean value false, so be it
12
            } else {
13
               tempAssigned = true                                      // otherwise, explicitly set it to true
14
            }
15
            try personCollection.insertOne(Person(firstname: values[0], lastname: values[1], assigned: tempAssigned, _id: BSONObjectID()), session: session)
16
            count.self += 1                                             // Above: use the insertOne method of the collection class form
17
            print("Inserted: \(count) \(row)")                          // the mongo-swift-driver and create a document with the Person ``Struct``.
18
         }
19
      }
20
   }