keyを指定できるように変更

Author: cateiru

example/main.go

@@ -13,12 +13,8 @@ func rootHandler(w http.ResponseWriter, r *http.Request) {
 
 func createHandler(w http.ResponseWriter, r *http.Request) {
 	ip := r.RemoteAddr
-	result, err := memory_retention.CreateKey(ip)
-	if err != nil {
-		fmt.Fprintln(w, err)
-		return
-	}
-	fmt.Fprintln(w, result)
+	memory_retention.CreateKey(ip)
+	fmt.Fprintln(w, ip)
 }
 
 func addHandler(w http.ResponseWriter, r *http.Request) {

memory_retention.go

@@ -1,12 +1,8 @@
 package memory_retention
 
 import (
-	"crypto/sha256"
-	"encoding/hex"
 	"fmt"
-	"strings"
 	"sync"
-	"time"
 )
 
 var mutex = &sync.Mutex{}
@@ -22,24 +18,11 @@ var keyMap = make(map[string]bool)
 // Create a key.
 //
 // Arguments:
-// seed {string} - Seed value for generating the key.
-//
-// Returns:
-// {string} - The generated hash value. Use this value to add/remove values.
-func CreateKey(seed string) (string, error) {
-	var strBuild strings.Builder
-
-	strBuild.WriteString(seed)
-	strBuild.WriteString(time.Now().String())
-
-	result := sha256.Sum256([]byte(strBuild.String()))
-	hash := hex.EncodeToString(result[:])
-
+// hash {string} - hash value.
+func CreateKey(hash string) {
 	mutex.Lock()
 	keyMap[hash] = true
 	mutex.Unlock()
-
-	return hash, nil
 }
 
 // Deletes the specified key.