Added basic command line argument for buildingId

main.go

@@ -21,6 +21,7 @@
 package main
 
 import (
+	"flag"
 	"log"
 )
 
@@ -29,4 +30,12 @@ import (
  */
 func main() {
 	log.Println("Hello world")
+
+	// Read command line argument for buildingId
+	buildingIdPointer := flag.Int64("building", 0, "set building ID")
+	flag.Parse()
+
+	// Initiate variables
+	buildingId := *buildingIdPointer
+	log.Println("buildingId: ", buildingId)
 }

main_test.go

@@ -21,12 +21,32 @@
 package main
 
 import (
+	"bytes"
+	"flag"
 	"testing"
 )
 
-func TestMain(t *testing.T) {
+type Config struct {
+  building    int64
+	args []string
+}
+
+func parseFlags(t *testing.T) {
 	err := error(nil)
 	if err != nil {
 		t.Fatal(err)
 	}
+
+	flags := flag.NewFlagSet("rcli", flag.ContinueOnError)
+  var buf bytes.Buffer
+  flags.SetOutput(&buf)
+
+	var stringArray = [2]string
+
+  var conf Config
+	flags.Int64Var(&conf.building, "building", 1, "set building ID")
+  err = flags.Parse()
+	if err != nil {
+    return nil, buf.String(), err
+  }
 }