Advertisement
Guest User

Untitled

a guest
Nov 8th, 2018
134
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Go 2.66 KB | None | 0 0
  1. func createFactor(tabWeight weights, listFields initer.PostForm, risk string) factors {
  2.         var factor factors
  3.         testTab := make([]weights, 0)
  4.         testTab = append(testTab, tabWeight)
  5.         if !tabWeight.IsEmpty() {
  6.                 factor.Weights = testTab
  7.         }
  8.         factor.Type = listFields.Type
  9.         factor.Value = risk
  10.         factor.Label = listFields.Label
  11.         factor.InputType = listFields.InputType
  12.  
  13.         return factor
  14. }
  15.  
  16. func createFx(risk Reader.RiskExcel) []byte {
  17.         dbPort := viper.GetString("DB_PORT")
  18.         dbUser := viper.GetString("DB_USER")
  19.         dbName := viper.GetString("DB_NAME")
  20.         dbPwd := viper.GetString("DB_PWD")
  21.         dbType := viper.GetString("DB_TYPE")
  22.         dbAdr := viper.GetString("DB_ADR")
  23.  
  24.         connectStr := "host=" + dbAdr + " port=" + dbPort + " user=" + dbUser + " dbname=" + dbName + " password=" + dbPwd + " sslmode=disable"
  25.         db, err := gorm.Open(dbType, connectStr)
  26.         if err != nil {
  27.                 fmt.Println(err)
  28.         }
  29.         defer db.Close()
  30.  
  31.         var listFrequency []initer.Values
  32.         var listSeverity []initer.Values
  33.  
  34.         db.Where("type = ?", "frequency").Find(&listFrequency)
  35.         db.Where("type = ?", "severity").Find(&listSeverity)
  36.  
  37.         tabWeight := createSliceWeight(listFrequency, listSeverity, risk.Occurrence)
  38.  
  39.         var listFields []initer.PostForm
  40.         db.Find(&listFields)
  41.  
  42.         factorsPost := createSliceFactors(listFields, risk, tabWeight)
  43.  
  44.         dbAdr := viper.GetString("DB_ADR")
  45.  
  46.         connectStr := "host=" + dbAdr + " port=" + dbPort + " user=" + dbUser + " dbname=" + dbName + " password=" + dbPwd + " sslmode=disable"
  47.         db, err := gorm.Open(dbType, connectStr)
  48.         if err != nil {
  49.                 fmt.Println(err)
  50.         }
  51.         defer db.Close()
  52.  
  53.         var listFrequency []initer.Values
  54.         var listSeverity []initer.Values
  55.  
  56.         db.Where("type = ?", "frequency").Find(&listFrequency)
  57.         db.Where("type = ?", "severity").Find(&listSeverity)
  58.  
  59.         tabWeight := createSliceWeight(listFrequency, listSeverity, risk.Occurrence)
  60.  
  61.         var listFields []initer.PostForm
  62.         db.Find(&listFields)
  63.  
  64.         factorsPost := createSliceFactors(listFields, risk, tabWeight)
  65.  
  66.         var fxField riskToPost
  67.         fxField.Factors = factorsPost
  68.         fxField.Mitigations = make([]string, 0)
  69.         jsonFxField, err := json.Marshal(fxField)
  70. //      jsonFxField = strings.Replace(string(jsonFxField), "null", "[]", -1)
  71.         if err != nil {
  72.                 fmt.Println(err)
  73.         }
  74.         fmt.Println(string(jsonFxField))
  75.         return jsonFxField
  76. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement