75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
|
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)
|
||
|
|
}
|
||
|
|
}
|