p_s/setup_instruments/setup_instruments.go

   1 package setup_instruments
   2
   3 import (
   4         "database/sql"
   5         "log"
   6
   7         "github.com/sjmudd/pstop/lib"
   8 )
   9
  10 // Error 1142: UPDATE command denied to user
  11 const UPDATE_FAILED = "Error 1142"
  12
  13 type setup_instruments_row struct {
  14         NAME    string
  15         ENABLED string
  16         TIMED   string
  17 }
  18
  19 type SetupInstruments struct {
  20         update_succeeded bool
  21         rows []setup_instruments_row
  22 }
  23
  24 // Change settings to monitor wait/synch/mutex/%
  25 func (si *SetupInstruments) EnableMutexMonitoring(dbh *sql.DB) {
  26         si.rows = make([]setup_instruments_row, 0, 100)
  27
  28         // populate the rows which are not set
  29         sql := "SELECT NAME, ENABLED, TIMED FROM setup_instruments WHERE NAME LIKE 'wait/synch/mutex/%' AND ( enabled <> 'YES' OR timed <> 'YES' )"
  30
  31         lib.Logger.Println("Collecting p_s.setup_instruments wait/synch/mutex configuration settings")
  32
  33         rows, err := dbh.Query(sql)
  34         if err != nil {
  35                 log.Fatal(err)
  36
  37
  38         }
  39         defer rows.Close()
  40
  41         count := 0
  42         for rows.Next() {
  43                 var r setup_instruments_row
  44                 if err := rows.Scan(
  45                         &r.NAME,
  46                         &r.ENABLED,
  47                         &r.TIMED); err != nil {
  48                         log.Fatal(err)
  49                 }
  50                 // we collect all information even if it's mainly empty as we may reference it later
  51                 si.rows = append(si.rows, r)
  52                 count++
  53         }
  54         if err := rows.Err(); err != nil {
  55                 log.Fatal(err)
  56         }
  57         lib.Logger.Println("- found", count, "rows whose configuration need changing")
  58
  59         // update the rows which need to be set - do multiple updates but I don't care
  60         lib.Logger.Println("Updating p_s.setup_instruments to allow wait/synch/mutex configuration")
  61
  62         count = 0
  63         for i := range si.rows {
  64                 sql := "UPDATE setup_instruments SET enabled = 'YES', TIMED = 'YES' WHERE NAME = '" + si.rows[i].NAME + "'"
  65                 if _, err := dbh.Exec(sql); err == nil {
  66                         si.update_succeeded = true
  67                 } else {
  68                         if err.Error()[0:10] != UPDATE_FAILED {
  69                                 log.Fatal(err)
  70                         }
  71                         break
  72
  73                 }
  74                 count++
  75         }
  76         lib.Logger.Println(count, "rows changed in p_s.setup_instruments")
  77 }
  78
  79 // restore any changed rows back to their original state
  80 func (si *SetupInstruments) Restore(dbh *sql.DB) {
  81         // If the previous update didn't work then don't try to restore
  82         if ! si.update_succeeded {
  83                 lib.Logger.Println("Not restoring p_s.setup_instruments to its original settings as previous UPDATE had failed")
  84                 return
  85         } else {
  86                 lib.Logger.Println("Restoring p_s.setup_instruments to its original settings")
  87         }
  88
  89         // update the rows which need to be set - do multiple updates but I don't care
  90         count := 0
  91         for i := range si.rows {
  92                 sql := "UPDATE setup_instruments SET enabled = '" + si.rows[i].ENABLED + "', TIMED = '" + si.rows[i].TIMED + "' WHERE NAME = '" + si.rows[i].NAME + "'"
  93                 if _, err := dbh.Exec(sql); err != nil {
  94                         log.Fatal(err)
  95                 }
  96                 count++
  97         }
  98         lib.Logger.Println(count, "rows changed in p_s.setup_instruments")
  99 }