feat(report):已完成用户抄表记录的上传处理。

service/end_user.go

@@ -2,10 +2,17 @@ package service
 
 import (
 	"electricity_bill_calc/config"
+	"electricity_bill_calc/excel"
+	"electricity_bill_calc/exceptions"
 	"electricity_bill_calc/global"
 	"electricity_bill_calc/model"
+	"fmt"
+	"io"
 
+	"github.com/samber/lo"
+	"github.com/shopspring/decimal"
 	"xorm.io/builder"
+	"xorm.io/xorm/schemas"
 )
 
 type _EndUserService struct{}
@@ -47,3 +54,165 @@ func (_EndUserService) AllEndUserRecord(reportId string) ([]model.EndUserDetail,
 		Find(&users)
 	return users, err
 }
+
+func (es _EndUserService) BatchImportNonPVRegister(reportId string, file io.Reader) *exceptions.BatchError {
+	errs := exceptions.NewBatchError()
+	users, err := es.AllEndUserRecord(reportId)
+	if err != nil {
+		errs.AddError(es.newVirtualExcelAnalysisError(err))
+		return errs
+	}
+	indexedUsers := lo.Reduce(
+		users,
+		func(acc map[string]model.EndUserDetail, elem model.EndUserDetail, index int) map[string]model.EndUserDetail {
+			acc[elem.MeterId] = elem
+			return acc
+		},
+		make(map[string]model.EndUserDetail, 0),
+	)
+	analyzer, err := excel.NewEndUserNonPVExcelAnalyzer(file)
+	if err != nil {
+		errs.AddError(es.newVirtualExcelAnalysisError(err))
+		return errs
+	}
+	imports, excelErrs := analyzer.Analysis(*new(model.EndUserImport))
+	if len(excelErrs) > 0 {
+		for _, e := range excelErrs {
+			errs.AddError(e)
+		}
+		return errs
+	}
+	tx := global.DBConn.NewSession()
+	if err = tx.Begin(); err != nil {
+		errs.AddError(es.newVirtualExcelAnalysisError(err))
+		return errs
+	}
+	defer tx.Close()
+
+	for _, im := range imports {
+		if elem, ok := indexedUsers[im.MeterId]; ok {
+			elem.CurrentPeriodOverall = im.CurrentPeriodOverall
+			elem.AdjustOverall = im.AdjustOverall
+			elem.CurrentPeriodCritical = decimal.Zero
+			elem.CurrentPeriodPeak = decimal.Zero
+			elem.CurrentPeriodValley = decimal.Zero
+			elem.AdjustCritical = decimal.Zero
+			elem.AdjustPeak = decimal.Zero
+			elem.AdjustValley = decimal.Zero
+			elem.CalculatePeriod()
+			_, err = tx.ID(schemas.NewPK(elem.ReportId, elem.ParkId, elem.MeterId)).
+				Cols(
+					"current_period_overall",
+					"adjust_overall",
+					"current_period_critical",
+					"current_period_peak",
+					"current_period_flat",
+					"current_perios_valley",
+					"adjust_critical",
+					"adjust_peak",
+					"adjust_flat",
+					"adjust_valley",
+				).
+				Update(elem)
+			if err != nil {
+				errs.AddError(es.newVirtualExcelAnalysisError(err))
+			}
+		} else {
+			errs.AddError(exceptions.NewNotFoundError(fmt.Sprintf("表计 %s 未找到", im.MeterId)))
+		}
+	}
+	if errs.Len() > 0 {
+		tx.Rollback()
+		return errs
+	}
+
+	err = tx.Commit()
+	if err != nil {
+		tx.Rollback()
+		errs.AddError(es.newVirtualExcelAnalysisError(err))
+	}
+	return errs
+}
+
+func (_EndUserService) newVirtualExcelAnalysisError(err error) *excel.ExcelAnalysisError {
+	return &excel.ExcelAnalysisError{Col: -1, Row: -1, Err: excel.AnalysisError{Err: err}}
+}
+
+func (es _EndUserService) BatchImportPVRegister(reportId string, file io.Reader) *exceptions.BatchError {
+	errs := exceptions.NewBatchError()
+	users, err := es.AllEndUserRecord(reportId)
+	if err != nil {
+		errs.AddError(es.newVirtualExcelAnalysisError(err))
+		return errs
+	}
+	indexedUsers := lo.Reduce(
+		users,
+		func(acc map[string]model.EndUserDetail, elem model.EndUserDetail, index int) map[string]model.EndUserDetail {
+			acc[elem.MeterId] = elem
+			return acc
+		},
+		make(map[string]model.EndUserDetail, 0),
+	)
+	analyzer, err := excel.NewEndUserNonPVExcelAnalyzer(file)
+	if err != nil {
+		errs.AddError(es.newVirtualExcelAnalysisError(err))
+		return errs
+	}
+	imports, excelErrs := analyzer.Analysis(*new(model.EndUserImport))
+	if len(excelErrs) > 0 {
+		for _, e := range excelErrs {
+			errs.AddError(e)
+		}
+		return errs
+	}
+	tx := global.DBConn.NewSession()
+	if err = tx.Begin(); err != nil {
+		errs.AddError(es.newVirtualExcelAnalysisError(err))
+		return errs
+	}
+	defer tx.Close()
+
+	for _, im := range imports {
+		if elem, ok := indexedUsers[im.MeterId]; ok {
+			elem.CurrentPeriodOverall = im.CurrentPeriodOverall
+			elem.AdjustOverall = im.AdjustOverall
+			elem.CurrentPeriodCritical = im.CurrentPeriodCritical.Decimal
+			elem.CurrentPeriodPeak = im.CurrentPeriodPeak.Decimal
+			elem.CurrentPeriodValley = im.CurrentPeriodValley.Decimal
+			elem.AdjustCritical = im.AdjustCritical.Decimal
+			elem.AdjustPeak = im.AdjustPeak.Decimal
+			elem.AdjustValley = im.AdjustValley.Decimal
+			elem.CalculatePeriod()
+			_, err = tx.ID(schemas.NewPK(elem.ReportId, elem.ParkId, elem.MeterId)).
+				Cols(
+					"current_period_overall",
+					"adjust_overall",
+					"current_period_critical",
+					"current_period_peak",
+					"current_period_flat",
+					"current_perios_valley",
+					"adjust_critical",
+					"adjust_peak",
+					"adjust_flat",
+					"adjust_valley",
+				).
+				Update(elem)
+			if err != nil {
+				errs.AddError(es.newVirtualExcelAnalysisError(err))
+			}
+		} else {
+			errs.AddError(es.newVirtualExcelAnalysisError(exceptions.NewNotFoundError(fmt.Sprintf("表计 %s 未找到", im.MeterId))))
+		}
+	}
+	if errs.Len() > 0 {
+		tx.Rollback()
+		return errs
+	}
+
+	err = tx.Commit()
+	if err != nil {
+		tx.Rollback()
+		errs.AddError(es.newVirtualExcelAnalysisError(err))
+	}
+	return errs
+}