chore(rename): Renamed md5 to id for document field

breadrock1 · Jun 16, 2024 · 2d3040a · 2d3040a
1 parent 38da771
commit 2d3040a
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 9 deletions.
diff --git a/internal/ocr/raw/raw.go b/internal/ocr/raw/raw.go
@@ -25,7 +25,7 @@ func (re *Service) RecognizeFile(document *reader.Document) error {
 
 	stringData := string(bytesData)
 	if len(stringData) == 0 {
-		return fmt.Errorf("returned empty content data for: %s", document.DocumentMD5)
+		return fmt.Errorf("returned empty content data for: %s", document.DocumentID)
 	}
 
 	document.SetContentData(stringData)

diff --git a/internal/reader/document.go b/internal/reader/document.go
@@ -33,7 +33,7 @@ type Document struct {
 	FolderID            string        `json:"folder_id"`
 	FolderPath          string        `json:"folder_path"`
 	Content             string        `json:"content"`
-	DocumentMD5         string        `json:"document_md5"`
+	DocumentID          string        `json:"document_id"`
 	DocumentSSDEEP      string        `json:"document_ssdeep"`
 	DocumentName        string        `json:"document_name"`
 	DocumentPath        string        `json:"document_path"`
@@ -142,7 +142,7 @@ func (d *Document) ComputeMd5Hash() {
 }
 
 func (d *Document) ComputeMd5HashData(data []byte) {
-	d.DocumentMD5 = fmt.Sprintf("%x", md5.Sum(data))
+	d.DocumentID = fmt.Sprintf("%x", md5.Sum(data))
 }
 
 func (d *Document) ComputeSsdeepHash() {

diff --git a/internal/reader/reader.go b/internal/reader/reader.go
@@ -22,7 +22,7 @@ func New() *Service {
 
 func (s *Service) AddAwaitDocument(document *Document) {
 	s.mu.Lock()
-	s.AwaitingDocs[document.DocumentMD5] = document
+	s.AwaitingDocs[document.DocumentID] = document
 	s.mu.Unlock()
 }
 

diff --git a/internal/searcher/searcher.go b/internal/searcher/searcher.go
@@ -36,7 +36,7 @@ func (s *Service) StoreDocument(document *reader.Document) error {
 	}
 
 	reqBody := bytes.NewBuffer(jsonData)
-	targetURL := fmt.Sprintf("%s/storage/folders/%s/documents/%s", s.Address, document.FolderID, document.DocumentMD5)
+	targetURL := fmt.Sprintf("%s/storage/folders/%s/documents/%s", s.Address, document.FolderID, document.DocumentID)
 	log.Printf("Storing document %s to elastic", document.FolderID)
 
 	method := "PUT"
@@ -48,14 +48,14 @@ func (s *Service) StoreDocument(document *reader.Document) error {
 	}
 
 	reqBody = bytes.NewBuffer(jsonData)
-	targetURL = fmt.Sprintf("%s/storage/folders/%s/documents/%s", s.Address, "history", document.DocumentMD5)
+	targetURL = fmt.Sprintf("%s/storage/folders/%s/documents/%s", s.Address, "history", document.DocumentID)
 	_, err = sender.SendRequest(reqBody, &targetURL, &method, &mimeType, s.Timeout)
 	if err != nil {
 		log.Println("Failed while sending request: ", err)
 	}
 
 	reqBody = bytes.NewBuffer(jsonData)
-	targetURL = fmt.Sprintf("%s/storage/folders/%s/documents/%s?document_type=vectors", s.Address, document.FolderID, document.DocumentMD5)
+	targetURL = fmt.Sprintf("%s/storage/folders/%s/documents/%s?document_type=vectors", s.Address, document.FolderID, document.DocumentID)
 	_, err = sender.SendRequest(reqBody, &targetURL, &method, &mimeType, s.Timeout)
 	if err != nil {
 		log.Println("Failed while sending request: ", err)

diff --git a/internal/storage/storage.go b/internal/storage/storage.go
@@ -60,7 +60,7 @@ func (s *Service) Create(ctx context.Context, document *reader.Document) (int, e
 		document.FolderID,
 		document.FolderPath,
 		document.Content,
-		document.DocumentMD5,
+		document.DocumentID,
 		document.DocumentSSDEEP,
 		document.DocumentName,
 		document.DocumentPath,

diff --git a/internal/watcher/watcher.go b/internal/watcher/watcher.go
@@ -115,7 +115,7 @@ func (nw *NotifyWatcher) PopRecognizedDocument(documentID string) *reader.Docume
 
 func (nw *NotifyWatcher) AppendRecognizedDocument(document *reader.Document) {
 	nw.mu.Lock()
-	nw.RecognizedDocuments[document.DocumentMD5] = document
+	nw.RecognizedDocuments[document.DocumentID] = document
 	nw.mu.Unlock()
 }