Moge-Row commited on
Commit
0b8b186
·
verified ·
1 Parent(s): a6bdc93

Update internal/handler/keys.go

Browse files
Files changed (1) hide show
  1. internal/handler/keys.go +9 -6
internal/handler/keys.go CHANGED
@@ -61,23 +61,26 @@ func GenerateKey(n string) string {
61
  b := make([]byte, 8)
62
  for i := range b { b[i] = chars[rand.Intn(len(chars))] }
63
  k := "RWPX-" + string(b)
 
64
  s.mu.Lock(); defer s.mu.Unlock()
65
- s.Keys[k] = &KeyData{Name: n, Key: k}
 
 
66
  go BackupKeyToSheet(n, k)
67
  save()
68
  return k
69
  }
70
-
71
  func HandleGenKey(w http.ResponseWriter, r *http.Request) {
72
  n := r.FormValue("name")
73
  if n == "" { n = "User" }
74
  existingKey := r.FormValue("key")
75
  var k string
76
  if existingKey != "" {
77
- s.mu.Lock()
78
- s.Keys[existingKey] = &KeyData{Name: n, Key: existingKey}
79
- save()
80
- s.mu.Unlock()
 
81
  go BackupKeyToSheet(n, existingKey)
82
  k = existingKey
83
  } else {
 
61
  b := make([]byte, 8)
62
  for i := range b { b[i] = chars[rand.Intn(len(chars))] }
63
  k := "RWPX-" + string(b)
64
+
65
  s.mu.Lock(); defer s.mu.Unlock()
66
+
67
+ s.Keys[n+":"+k] = &KeyData{Name: n, Key: k}
68
+
69
  go BackupKeyToSheet(n, k)
70
  save()
71
  return k
72
  }
 
73
  func HandleGenKey(w http.ResponseWriter, r *http.Request) {
74
  n := r.FormValue("name")
75
  if n == "" { n = "User" }
76
  existingKey := r.FormValue("key")
77
  var k string
78
  if existingKey != "" {
79
+ s.mu.Lock()
80
+ s.Keys[n+":"+existingKey] = &KeyData{Name: n, Key: existingKey}
81
+ save()
82
+ s.mu.Unlock()
83
+ }
84
  go BackupKeyToSheet(n, existingKey)
85
  k = existingKey
86
  } else {