Add SS and MCP databases

main.go

@@ -0,0 +1,74 @@
+package main
+
+import (
+	"database/sql"
+	"encoding/csv"
+	"encoding/json"
+	"flag"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"slices"
+	"strings"
+
+	_ "github.com/go-sql-driver/mysql"
+)
+
+var (
+	csvPath                = flag.String("csv", "aips.csv", "Path to the CSV file containing AIP names")
+	mysqlConnString        = flag.String("mysql", "root:12345@tcp(localhost:62001)/MCP", "mysql connection string")
+	elasticSearch          = flag.String("elastic", "http://127.0.0.1:62002", "elastic search URL")
+	packageNamePlaceholder = "VEV2025.624" // Placeholder for AIP package name
+)
+
+func main() {
+	flag.Parse()
+	file, err := os.Open(*csvPath)
+	assertNoErr(err)
+	records, err := csv.NewReader(file).ReadAll()
+	assertNoErr(err)
+
+	r := []string{}
+	for _, record := range records {
+		r = append(r, strings.TrimSpace(record[0]))
+	}
+
+	// Elastic
+	elastic := *elasticSearch + "/aips/_search?"
+	res, err := http.Get(elastic)
+	assertNoErr(err)
+	data, err := io.ReadAll(res.Body)
+	assertNoErr(err)
+
+	var er ElasticAipResponse
+	err = json.Unmarshal(data, &er)
+	assertNoErr(err)
+
+	fmt.Printf("Total: %d\n", er.Hits.Total)
+
+	result, err := os.OpenFile("aip_names.txt", os.O_TRUNC|os.O_CREATE|os.O_WRONLY, 0644)
+	for _, index := range er.Hits.Hits {
+		if slices.ContainsFunc(r, func(record string) bool {
+			return strings.Contains(index.Source.Name, record)
+		}) {
+			fmt.Println("AIP Found: " + index.Source.Name)
+			_, err = result.WriteString(index.Source.Name + "\n")
+			assertNoErr(err)
+		}
+	}
+}
+
+func openDB(connStr string) *sql.DB {
+	db, err := sql.Open("mysql", connStr)
+	assertNoErr(err)
+	err = db.Ping()
+	assertNoErr(err)
+	return db
+}
+
+func assertNoErr(err error) {
+	if err != nil {
+		panic(err)
+	}
+}