go_service/controllers/ImportV2.go

62 lines
1.4 KiB
Go

package controller
import (
gm "db_service/gorm_models"
"db_service/repositories"
"encoding/json"
"fmt"
"log"
"math/big"
"net/http"
"sync/atomic"
"time"
)
func StartProductImport(w http.ResponseWriter, _ *http.Request) {
if !atomic.CompareAndSwapUint32(&locker, stateUnlocked, stateLocked) {
w.WriteHeader(http.StatusTooManyRequests)
err := json.NewEncoder(w).Encode(map[string]string{
"msg": "Product import in progress!",
})
log.Println(err)
return
}
defer atomic.StoreUint32(&locker, stateUnlocked)
start := time.Now()
r := new(big.Int)
fmt.Println("start import", r.Binomial(1000, 10))
importRepo, err := repositories.ImporterInstance()
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
//wait until initialization data is loaded to memory
importRepo.ImportWGroup.Wait()
if err = importRepo.Start().Error; err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
elapsed := time.Since(start)
log.Printf("end import took %s", elapsed)
return
}
importRepo.ImportWGroup.Wait()
//scout index flush
if err = gm.Flush(); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
elapsed := time.Since(start)
log.Printf("end import took %s", elapsed)
return
}
elapsed := time.Since(start)
log.Printf("end import took %s", elapsed)
http.Error(w, fmt.Sprintf("end import took %s", elapsed), http.StatusOK)
}