feat(api, core): add metrics table

Author: gusye1234

src/server/api/go/internal/bootstrap/container.go

@@ -59,6 +59,7 @@ func BuildContainer() *do.Injector {
 				&model.ToolReference{},
 				&model.ToolSOP{},
 				&model.ExperienceConfirmation{},
+				&model.Metric{},
 			)
 		}
 

src/server/api/go/internal/modules/model/metric.go

@@ -0,0 +1,23 @@
+package model
+
+import (
+	"time"
+
+	"github.com/google/uuid"
+)
+
+type Metric struct {
+	ID        uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
+	ProjectID uuid.UUID `gorm:"type:uuid;not null;index:idx_metric_project_id_tag_created_at,priority:1" json:"project_id"`
+
+	Tag       string `gorm:"type:text;not null;index:idx_metric_project_id_tag_created_at,priority:2" json:"tag"`
+	Increment int    `gorm:"not null;default:0" json:"increment"`
+
+	CreatedAt time.Time `gorm:"autoCreateTime;not null;default:CURRENT_TIMESTAMP;index:idx_metric_project_id_tag_created_at,priority:3" json:"created_at"`
+	UpdatedAt time.Time `gorm:"autoUpdateTime;not null;default:CURRENT_TIMESTAMP" json:"updated_at"`
+
+	// Metric <-> Project
+	Project *Project `gorm:"foreignKey:ProjectID;references:ID;constraint:OnDelete:CASCADE,OnUpdate:CASCADE;" json:"-"`
+}
+
+func (Metric) TableName() string { return "metrics" }

src/server/api/go/internal/modules/model/project.go

@@ -27,6 +27,9 @@ type Project struct {
 
 	// Project <-> ToolReference
 	ToolReferences []ToolReference `gorm:"constraint:OnDelete:CASCADE,OnUpdate:CASCADE;" json:"-"`
+
+	// Project <-> Metric
+	Metrics []Metric `gorm:"constraint:OnDelete:CASCADE,OnUpdate:CASCADE;" json:"-"`
 }
 
 func (Project) TableName() string { return "projects" }

src/server/core/acontext_core/schema/orm/__init__.py

@@ -10,6 +10,7 @@
 from .tool_reference import ToolReference
 from .tool_sop import ToolSOP
 from .experience_confirmation import ExperienceConfirmation
+from .metric import Metric
 
 __all__ = [
     "ORM_BASE",
@@ -28,4 +29,5 @@
     "ToolReference",
     "ToolSOP",
     "ExperienceConfirmation",
+    "Metric",
 ]

src/server/core/acontext_core/schema/orm/base.py

@@ -2,7 +2,6 @@
 from sqlalchemy.dialects.postgresql import UUID
 from datetime import datetime
 from dataclasses import dataclass, field, fields
-from pydantic import ValidationError
 from sqlalchemy.orm import registry, RelationshipProperty
 from sqlalchemy import Column
 from sqlalchemy.sql import func

src/server/core/acontext_core/schema/orm/message.py

@@ -1,4 +1,3 @@
-import json
 from dataclasses import dataclass, field
 from sqlalchemy import String, ForeignKey, Index, CheckConstraint, Column
 from sqlalchemy.orm import relationship

src/server/core/acontext_core/schema/orm/metric.py

@@ -0,0 +1,44 @@
+from dataclasses import dataclass, field
+from sqlalchemy import Column, ForeignKey, Index, Integer, String
+from sqlalchemy.orm import relationship
+from sqlalchemy.dialects.postgresql import UUID
+from typing import TYPE_CHECKING
+from .base import ORM_BASE, CommonMixin
+from ..utils import asUUID
+
+if TYPE_CHECKING:
+    from .project import Project
+
+
+@ORM_BASE.mapped
+@dataclass
+class Metric(CommonMixin):
+    __tablename__ = "metrics"
+
+    __table_args__ = (
+        Index(
+            "idx_metric_project_id_tag_created_at", "project_id", "tag", "created_at"
+        ),
+    )
+
+    project_id: asUUID = field(
+        metadata={
+            "db": Column(
+                UUID(as_uuid=True),
+                ForeignKey("projects.id", ondelete="CASCADE"),
+                nullable=False,
+            )
+        }
+    )
+
+    tag: str = field(metadata={"db": Column(String, nullable=False)})
+
+    increment: int = field(
+        default=0,
+        metadata={"db": Column(Integer, nullable=False, default=0)},
+    )
+
+    # Relationships
+    project: "Project" = field(
+        init=False, metadata={"db": relationship("Project", back_populates="metrics")}
+    )

src/server/core/acontext_core/schema/orm/project.py

@@ -1,8 +1,7 @@
-import uuid
 from dataclasses import dataclass, field
-from sqlalchemy import String, Index, Column, ForeignKey
+from sqlalchemy import String, Index, Column
 from sqlalchemy.orm import relationship
-from sqlalchemy.dialects.postgresql import JSONB, UUID
+from sqlalchemy.dialects.postgresql import JSONB
 from typing import TYPE_CHECKING, List, Optional
 from .base import ORM_BASE, CommonMixin
 
@@ -11,6 +10,7 @@
     from .session import Session
     from .task import Task
     from .tool_reference import ToolReference
+    from .metric import Metric
 
 
 @ORM_BASE.mapped
@@ -67,3 +67,12 @@ class Project(CommonMixin):
             )
         },
     )
+
+    metrics: List["Metric"] = field(
+        default_factory=list,
+        metadata={
+            "db": relationship(
+                "Metric", back_populates="project", cascade="all, delete-orphan"
+            )
+        },
+    )