add set_short_cached_reference command

Author: mjarkk

README.md

@@ -84,7 +84,7 @@ Ok Response
 
 ### `send_cv`
 
-Send a scraped CV to RT-CV
+Send a scraped CV to RT-CV and triggers the `set_cached_reference` for the reference number of this cv if there where matches on this CV
 
 Example input
 
@@ -150,7 +150,7 @@ Ok Response
 
 ### `set_cached_reference`
 
-Save a reference number to the cache
+Save a reference number to the cache with a timeout of 3 days
 This cache is used to avoid sending the same CV twice or scraping data that has already been scraped
 
 *Note that "send_cv" also executes this function automatically*
@@ -167,6 +167,10 @@ Ok Response
 {"type":"ok"}
 ```
 
+### `set_short_cached_reference`
+
+Same as previouse one except this one has a time out of 12 hours
+
 ### `has_cached_reference`
 
 Is there a cache entry for a specific reference number?

api.go

@@ -228,21 +228,23 @@ func (a *API) GetUserSecret(key, encryptionKey string) (UserSecret, error) {
 	return result, err
 }
 
+// SetCacheEntry sets a cache entry for the reference number that expires after the duration
+func (a *API) SetCacheEntry(referenceNr string, duration time.Duration) {
+	a.Cache[referenceNr] = time.Now().Add(duration)
+}
+
 // CacheEntryExists returns true if the cache entry exists and is not expired
 func (a *API) CacheEntryExists(referenceNr string) bool {
 	cacheEntryInsertionTime, cacheEntryExists := a.Cache[referenceNr]
-	if cacheEntryExists {
-		expired := time.Now().After(
-			cacheEntryInsertionTime.Add(time.Hour * 72), // 3 days
-		)
-		if expired {
-			delete(a.Cache, referenceNr)
-		} else {
-			return true
-		}
+	if !cacheEntryExists {
+		return false
 	}
 
-	return false
+	expired := time.Now().After(cacheEntryInsertionTime)
+	if expired {
+		delete(a.Cache, referenceNr)
+	}
+	return !expired
 }
 
 // UserSecret represents the json layout of an user secret

main.go

@@ -224,9 +224,8 @@ func LoopAction(api *API, inputJSON string) (msgType MessageType, msgContent int
 
 		DebugToLogfile(referenceNrs)
 
-		now := time.Now()
 		for _, nr := range referenceNrs {
-			api.Cache[nr] = now
+			api.SetCacheEntry(nr, time.Hour*72) // 3 days
 		}
 
 		return MessageTypeOk, nil
@@ -255,7 +254,7 @@ func LoopAction(api *API, inputJSON string) (msgType MessageType, msgContent int
 
 		hasMatch := false
 		if api.MockMode {
-			api.Cache[referenceNr] = time.Now()
+			api.SetCacheEntry(referenceNr, time.Hour*72)
 			hasMatch = true
 		} else {
 			scanCVBody := json.RawMessage(`{"cv":` + string(input.Content) + `}`)
@@ -274,7 +273,7 @@ func LoopAction(api *API, inputJSON string) (msgType MessageType, msgContent int
 					hasMatch = response.HasMatches
 					if hasMatch {
 						// Only cache the CVs that where matched to something
-						api.Cache[referenceNr] = time.Now()
+						api.SetCacheEntry(referenceNr, time.Hour*72) // 3 days
 					}
 				}
 			}
@@ -317,7 +316,7 @@ func LoopAction(api *API, inputJSON string) (msgType MessageType, msgContent int
 		default:
 			return returnErr(errors.New("unknown secret"))
 		}
-	case "set_cached_reference", "has_cached_reference":
+	case "set_cached_reference", "set_short_cached_reference", "has_cached_reference":
 		referenceNr := ""
 		err = json.Unmarshal(input.Content, &referenceNr)
 		if err != nil {
@@ -328,11 +327,16 @@ func LoopAction(api *API, inputJSON string) (msgType MessageType, msgContent int
 			return returnErr(errors.New("reference number cannot be an empty string"))
 		}
 
-		if input.Type == "set_cached_reference" {
-			api.Cache[referenceNr] = time.Now()
+		switch input.Type {
+		case "set_cached_reference":
+			api.SetCacheEntry(referenceNr, time.Hour*72) // 3 days
+			return MessageTypeOk, nil
+		case "set_short_cached_reference":
+			api.SetCacheEntry(referenceNr, time.Hour*12) // 0.5 days
 			return MessageTypeOk, nil
 		}
 
+		// has_cached_reference
 		hasCachedReference := api.CacheEntryExists(referenceNr)
 		DebugToLogfile("has_cached_reference", referenceNr, ">", hasCachedReference)