jh_peng/test_data
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

V0

Browse Source
This commit is contained in:
jh_peng 2024-11-19 09:59:40 +08:00
parent d95a730b7b
commit 4839476d26
1 changed files with 70 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

126
repository/test.data/report.go
View File

@ -612,43 +612,46 @@ func SaveCP(report *model.Report) { //pbi, product, lot, waferID string,
}
func SaveFT(report *model.Report) { //pbi, product, lot, subBatch string,
var ftFile *model.FileHandled
var ftFiles []*model.FileHandled
global.PostGreSQL.Where("pbi = ? AND product = ? AND lot = ? AND sub_batch = ? AND step = ?",
report.PBI, report.Product, report.Lot, report.SubBatch, "FT").Find(&ftFile)
if _, err := os.Stat(ftFile.Path); err != nil {
log.Println("该文件不存在")
}
ft, err := os.Open(ftFile.Path)
if err != nil {
log.Println(ftFile.Path, ":", err)
return
}
defer ft.Close()
scanner := bufio.NewScanner(ft)
report.PBI, report.Product, report.Lot, report.SubBatch, "FT").Find(&ftFiles)
datas := make([][]string, 0)
fieldMap := make(map[string]int)
for scanner.Scan() {
line := scanner.Text()
s := strings.Split(line, ",")
datas = append(datas, s[:len(s)-1])
if len(datas) == 1 {
for index, cell := range s {
if index == len(s)-1 {
continue
sbinHbinMap := make(map[string]string)
for _, ftFile := range ftFiles {
fileDatas := make([][]string, 0)
if _, err := os.Stat(ftFile.Path); err != nil {
log.Println("该文件不存在")
}
ft, err := os.Open(ftFile.Path)
if err != nil {
log.Println(ftFile.Path, ":", err)
return
}
scanner := bufio.NewScanner(ft)
for scanner.Scan() {
line := scanner.Text()
s := strings.Split(line, ",")
fileDatas = append(fileDatas, s[:len(s)-1])
if len(fileDatas) == 1 {
for index, cell := range s {
if index == len(s)-1 {
continue
}
fieldMap[cell] = index
}
fieldMap[cell] = index
}
}
}
if len(datas) >= 1 {
datas = datas[1:]
if len(datas) >= 1 {
datas = append(datas, fileDatas[1:]...)
}
ft.Close()
//sbinHbinMap := make(map[string]map[string]string)
sbinHbinByte := []byte(ftFile.SbinHbin)
_ = json.Unmarshal(sbinHbinByte, &sbinHbinMap)
}
//sbinHbinMap := make(map[string]map[string]string)
sbinHbinMap := make(map[string]string)
sbinHbinByte := []byte(ftFile.SbinHbin)
_ = json.Unmarshal(sbinHbinByte, &sbinHbinMap)
binCounter := make(map[string]int)
siteCounters := make(map[string]map[string]int)
for _, data := range datas {
@ -659,35 +662,41 @@ func SaveFT(report *model.Report) { //pbi, product, lot, subBatch string,
siteCounters[data[fieldMap["SITE_NUM"]]][data[fieldMap["SOFT_BIN"]]]++
}
var rtFile *model.FileHandled
var rtFiles []*model.FileHandled
var passCounter int
if !errors.Is(global.PostGreSQL.Where("pbi = ? AND product = ? AND lot = ? AND sub_batch = ? AND step = ?",
report.PBI, report.Product, report.Lot, report.SubBatch, "RT").
Find(&rtFile).Error, gorm.ErrRecordNotFound) {
rt, err := os.Open(rtFile.Path)
if err != nil {
log.Println(err)
}
defer rt.Close()
rtScanner := bufio.NewScanner(rt)
Find(&rtFiles).Error, gorm.ErrRecordNotFound) {
rtDatas := make([][]string, 0)
rtFieldMap := make(map[string]int)
for rtScanner.Scan() {
line := rtScanner.Text()
s := strings.Split(line, ",")
rtDatas = append(rtDatas, s[:len(s)-1])
if len(rtDatas) == 1 {
for index, cell := range s {
if index == len(s)-1 {
continue
for _, rtFile := range rtFiles {
fileDatas := make([][]string, 0)
rt, err := os.Open(rtFile.Path)
if err != nil {
log.Println(err)
}
rtScanner := bufio.NewScanner(rt)
for rtScanner.Scan() {
line := rtScanner.Text()
s := strings.Split(line, ",")
fileDatas = append(fileDatas, s[:len(s)-1])
if len(fileDatas) == 1 {
for index, cell := range s {
if index == len(s)-1 {
continue
}
rtFieldMap[cell] = index
}
rtFieldMap[cell] = index
}
}
rt.Close()
if len(rtDatas) >= 1 {
rtDatas = append(rtDatas, fileDatas[1:]...)
}
}
if len(rtDatas) >= 1 {
rtDatas = rtDatas[1:]
}
for _, rtData := range rtDatas {
if rtData[rtFieldMap["SOFT_BIN"]] == "1" {
passCounter++
@ -715,13 +724,18 @@ func SaveFT(report *model.Report) { //pbi, product, lot, subBatch string,
})
}
var siteReports []*model.ReportSites
for site := range siteCounters {
siteReports = append(siteReports, &model.ReportSites{
ReportID: report.ID,
Site: site,
})
}
global.PostGreSQL.Create(&siteReports)
global.PostGreSQL.Transaction(func(tx *gorm.DB) error {
for site := range siteCounters {
siteReports = append(siteReports, &model.ReportSites{
ReportID: report.ID,
Site: site,
})
tx.Where("report_id = ?", report.ID).Delete(&model.ReportSites{})
tx.Create(&siteReports)
}
return nil
})
siteReports = []*model.ReportSites{}
global.PostGreSQL.Where("report_id = ?", report.ID).Find(&siteReports)
for _, siteReport := range siteReports {