Added search by layer hash

spacemeshos · Dec 28, 2020 · be63936 · be63936
1 parent d2fdc3e
commit be63936
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/api/httpserver/rest/search.go b/api/httpserver/rest/search.go
@@ -44,6 +44,13 @@ func (s *Service) SearchHandler(w http.ResponseWriter, r *http.Request) {
                 buf.WriteString(fmt.Sprintf("\"redirect\":\"/smeshers/%v\"", idStr))
                 break
             }
+            if s.storage.GetLayersCount(s.ctx, &bson.D{{"hash", idStr}}) > 0 {
+                layer, err := s.storage.GetLayer(s.ctx, &bson.D{{"hash", idStr}})
+                if (err == nil) {
+                    buf.WriteString(fmt.Sprintf("\"redirect\":\"/layers/%v\"", layer.Number))
+                    break
+                }
+            }
             return nil, http.StatusNotFound, errors.New("Not found")
         default:
             objId, err := primitive.ObjectIDFromHex(idStr);

diff --git a/storage/layer.go b/storage/layer.go
@@ -17,6 +17,7 @@ import (
 
 func (s *Storage) InitLayersStorage(ctx context.Context) error {
     _, err := s.db.Collection("layers").Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{{"number", 1}}, Options: options.Index().SetName("numberIndex").SetUnique(true)});
+    _, err = s.db.Collection("layers").Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{{"hash", 1}}, Options: options.Index().SetName("hashIndex").SetUnique(true)});
     return err
 }