session: Remove incorrect UTC conversion, alter app_state/user_state to text (#623)

Author: hyprh

docs/mkdocs/en/session.md

@@ -595,7 +595,7 @@ CREATE TABLE app_states (
     id BIGSERIAL PRIMARY KEY,
     app_name VARCHAR(255) NOT NULL,
     key VARCHAR(255) NOT NULL,
-    value BYTEA NOT NULL,
+    value TEXT DEFAULT NULL,
     created_at TIMESTAMP NOT NULL,
     updated_at TIMESTAMP NOT NULL,
     expires_at TIMESTAMP,
@@ -609,7 +609,7 @@ CREATE TABLE user_states (
     app_name VARCHAR(255) NOT NULL,
     user_id VARCHAR(255) NOT NULL,
     key VARCHAR(255) NOT NULL,
-    value BYTEA NOT NULL,
+    value TEXT DEFAULT NULL,
     created_at TIMESTAMP NOT NULL,
     updated_at TIMESTAMP NOT NULL,
     expires_at TIMESTAMP,

docs/mkdocs/zh/session.md

@@ -601,7 +601,7 @@ CREATE TABLE app_states (
     id BIGSERIAL PRIMARY KEY,
     app_name VARCHAR(255) NOT NULL,
     key VARCHAR(255) NOT NULL,
-    value BYTEA NOT NULL,
+    value TEXT DEFAULT NULL,
     created_at TIMESTAMP NOT NULL,
     updated_at TIMESTAMP NOT NULL,
     expires_at TIMESTAMP,
@@ -615,7 +615,7 @@ CREATE TABLE user_states (
     app_name VARCHAR(255) NOT NULL,
     user_id VARCHAR(255) NOT NULL,
     key VARCHAR(255) NOT NULL,
-    value BYTEA NOT NULL,
+    value TEXT DEFAULT NULL,
     created_at TIMESTAMP NOT NULL,
     updated_at TIMESTAMP NOT NULL,
     expires_at TIMESTAMP,

examples/runner/main.go

@@ -145,7 +145,7 @@ func (c *multiTurnChat) setup(_ context.Context) error {
 			postgres.WithUser(pgUser),
 			postgres.WithPassword(pgPassword),
 			postgres.WithDatabase(pgDatabase),
-			postgres.WithTablePrefix("tRPC"),
+			postgres.WithTablePrefix("trpc_"),
 		)
 		if err != nil {
 			return fmt.Errorf("failed to create postgres session service: %w", err)

session/postgres/init.go

@@ -81,7 +81,7 @@ const (
 			id BIGSERIAL PRIMARY KEY,
 			app_name VARCHAR(255) NOT NULL,
 			key VARCHAR(255) NOT NULL,
-			value BYTEA DEFAULT NULL,
+			value TEXT DEFAULT NULL,
 			created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
 			updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
 			expires_at TIMESTAMP DEFAULT NULL,
@@ -94,7 +94,7 @@ const (
 			app_name VARCHAR(255) NOT NULL,
 			user_id VARCHAR(255) NOT NULL,
 			key VARCHAR(255) NOT NULL,
-			value BYTEA DEFAULT NULL,
+			value TEXT DEFAULT NULL,
 			created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
 			updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
 			expires_at TIMESTAMP DEFAULT NULL,
@@ -232,7 +232,7 @@ var expectedSchema = map[string]struct {
 			{"id", "bigint", false},
 			{"app_name", "character varying", false},
 			{"key", "character varying", false},
-			{"value", "bytea", true},
+			{"value", "text", true},
 			{"created_at", "timestamp without time zone", false},
 			{"updated_at", "timestamp without time zone", false},
 			{"expires_at", "timestamp without time zone", true},
@@ -249,7 +249,7 @@ var expectedSchema = map[string]struct {
 			{"app_name", "character varying", false},
 			{"user_id", "character varying", false},
 			{"key", "character varying", false},
-			{"value", "bytea", true},
+			{"value", "text", true},
 			{"created_at", "timestamp without time zone", false},
 			{"updated_at", "timestamp without time zone", false},
 			{"expires_at", "timestamp without time zone", true},

session/postgres/schema.sql

@@ -82,7 +82,7 @@ CREATE TABLE IF NOT EXISTS app_states (
   id BIGSERIAL PRIMARY KEY,
   app_name VARCHAR(255) NOT NULL,
   key VARCHAR(255) NOT NULL,
-  value BYTEA DEFAULT NULL,
+  value TEXT DEFAULT NULL,
   created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
   updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
   expires_at TIMESTAMP DEFAULT NULL,
@@ -106,7 +106,7 @@ CREATE TABLE IF NOT EXISTS user_states (
   app_name VARCHAR(255) NOT NULL,
   user_id VARCHAR(255) NOT NULL,
   key VARCHAR(255) NOT NULL,
-  value BYTEA DEFAULT NULL,
+  value TEXT DEFAULT NULL,
   created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
   updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
   expires_at TIMESTAMP DEFAULT NULL,

session/postgres/service.go

@@ -253,7 +253,7 @@ func (s *Service) CreateSession(
 		key.SessionID = uuid.New().String()
 	}
 
-	now := time.Now().UTC()
+	now := time.Now()
 	sessState := &SessionState{
 		ID:        key.SessionID,
 		State:     make(session.StateMap),
@@ -405,7 +405,7 @@ func (s *Service) UpdateAppState(ctx context.Context, appName string, state sess
 		return session.ErrAppNameRequired
 	}
 
-	now := time.Now().UTC()
+	now := time.Now()
 	var expiresAt *time.Time
 	if s.appStateTTL > 0 {
 		t := now.Add(s.appStateTTL)
@@ -453,7 +453,7 @@ func (s *Service) ListAppStates(ctx context.Context, appName string) (session.St
 		WHERE app_name = $1
 		AND (expires_at IS NULL OR expires_at > $2)
 		AND deleted_at IS NULL`, s.tableAppStates),
-		appName, time.Now().UTC())
+		appName, time.Now())
 
 	if err != nil {
 		return nil, fmt.Errorf("postgres session service list app states failed: %w", err)
@@ -476,7 +476,7 @@ func (s *Service) DeleteAppState(ctx context.Context, appName string, key string
 		_, err = s.pgClient.ExecContext(ctx,
 			fmt.Sprintf(`UPDATE %s SET deleted_at = $1
 			 WHERE app_name = $2 AND key = $3 AND deleted_at IS NULL`, s.tableAppStates),
-			time.Now().UTC(), appName, key)
+			time.Now(), appName, key)
 	} else {
 		// Hard delete: permanently remove record
 		_, err = s.pgClient.ExecContext(ctx,
@@ -497,7 +497,7 @@ func (s *Service) UpdateUserState(ctx context.Context, userKey session.UserKey,
 		return err
 	}
 
-	now := time.Now().UTC()
+	now := time.Now()
 	var expiresAt *time.Time
 	if s.userStateTTL > 0 {
 		t := now.Add(s.userStateTTL)
@@ -545,7 +545,7 @@ func (s *Service) ListUserStates(ctx context.Context, userKey session.UserKey) (
 		WHERE app_name = $1 AND user_id = $2
 		AND (expires_at IS NULL OR expires_at > $3)
 		AND deleted_at IS NULL`, s.tableUserStates),
-		userKey.AppName, userKey.UserID, time.Now().UTC())
+		userKey.AppName, userKey.UserID, time.Now())
 
 	if err != nil {
 		return nil, fmt.Errorf("postgres session service list user states failed: %w", err)
@@ -567,7 +567,7 @@ func (s *Service) DeleteUserState(ctx context.Context, userKey session.UserKey,
 		_, err = s.pgClient.ExecContext(ctx,
 			fmt.Sprintf(`UPDATE %s SET deleted_at = $1
 			 WHERE app_name = $2 AND user_id = $3 AND key = $4 AND deleted_at IS NULL`, s.tableUserStates),
-			time.Now().UTC(), userKey.AppName, userKey.UserID, key)
+			time.Now(), userKey.AppName, userKey.UserID, key)
 	} else {
 		_, err = s.pgClient.ExecContext(ctx,
 			fmt.Sprintf(`DELETE FROM %s
@@ -664,7 +664,7 @@ func (s *Service) getSession(
 		WHERE app_name = $1 AND user_id = $2 AND session_id = $3
 		AND (expires_at IS NULL OR expires_at > $4)
 		AND deleted_at IS NULL`, s.tableSessionStates)
-	stateArgs := []interface{}{key.AppName, key.UserID, key.SessionID, time.Now().UTC()}
+	stateArgs := []interface{}{key.AppName, key.UserID, key.SessionID, time.Now()}
 
 	err := s.pgClient.Query(ctx, func(rows *sql.Rows) error {
 		if rows.Next() {
@@ -708,7 +708,7 @@ func (s *Service) getSession(
 	// Query events (always filter deleted records)
 	// Note: limit here only controls how many events to return, not delete from database
 	events := []event.Event{}
-	now := time.Now().UTC()
+	now := time.Now()
 	var eventQuery string
 	var eventArgs []interface{}
 
@@ -761,7 +761,7 @@ func (s *Service) getSession(
 		WHERE app_name = $1 AND user_id = $2 AND session_id = $3
 		AND (expires_at IS NULL OR expires_at > $4)
 		AND deleted_at IS NULL`, s.tableSessionSummaries)
-	summaryArgs := []interface{}{key.AppName, key.UserID, key.SessionID, time.Now().UTC()}
+	summaryArgs := []interface{}{key.AppName, key.UserID, key.SessionID, time.Now()}
 
 	err = s.pgClient.Query(ctx, func(rows *sql.Rows) error {
 		for rows.Next() {
@@ -822,7 +822,7 @@ func (s *Service) listSessions(
 		AND (expires_at IS NULL OR expires_at > $3)
 		AND deleted_at IS NULL
 		ORDER BY updated_at DESC`, s.tableSessionStates)
-	listArgs := []interface{}{key.AppName, key.UserID, time.Now().UTC()}
+	listArgs := []interface{}{key.AppName, key.UserID, time.Now()}
 
 	err = s.pgClient.Query(ctx, func(rows *sql.Rows) error {
 		for rows.Next() {
@@ -890,7 +890,7 @@ func (s *Service) listSessions(
 }
 
 func (s *Service) addEvent(ctx context.Context, key session.Key, event *event.Event) error {
-	now := time.Now().UTC()
+	now := time.Now()
 
 	// Get current session state (always filter deleted records, but allow expired sessions)
 	var sessState *SessionState
@@ -1031,7 +1031,7 @@ func (s *Service) enforceEventLimit(ctx context.Context, tx *sql.Tx, key session
 // refreshSessionTTL updates the session's updated_at and expires_at timestamps.
 // This effectively "renews" the session, extending its lifetime by the configured TTL.
 func (s *Service) refreshSessionTTL(ctx context.Context, key session.Key) error {
-	now := time.Now().UTC()
+	now := time.Now()
 	expiresAt := now.Add(s.sessionTTL)
 
 	_, err := s.pgClient.ExecContext(ctx,
@@ -1051,7 +1051,7 @@ func (s *Service) deleteSessionState(ctx context.Context, key session.Key) error
 	err := s.pgClient.Transaction(ctx, func(tx *sql.Tx) error {
 		if s.opts.softDelete {
 			// Soft delete: set deleted_at timestamp
-			now := time.Now().UTC()
+			now := time.Now()
 
 			// Soft delete session state
 			_, err := tx.ExecContext(ctx,
@@ -1222,7 +1222,7 @@ func (s *Service) getEventsList(
 				LIMIT $6
 			) e ON true
 			ORDER BY s.session_id`, s.tableSessionEvents)
-		args = []interface{}{sessionIDs, sessionKeys[0].AppName, sessionKeys[0].UserID, time.Now().UTC(), afterTime, limit}
+		args = []interface{}{sessionIDs, sessionKeys[0].AppName, sessionKeys[0].UserID, time.Now(), afterTime, limit}
 	} else {
 		// Without limit: simple query with IN clause
 		query = fmt.Sprintf(`
@@ -1234,7 +1234,7 @@ func (s *Service) getEventsList(
 			AND created_at > $5
 			AND deleted_at IS NULL
 			ORDER BY session_id, created_at DESC`, s.tableSessionEvents)
-		args = []interface{}{sessionKeys[0].AppName, sessionKeys[0].UserID, sessionIDs, time.Now().UTC(), afterTime}
+		args = []interface{}{sessionKeys[0].AppName, sessionKeys[0].UserID, sessionIDs, time.Now(), afterTime}
 	}
 
 	// Execute query and group events by session
@@ -1324,7 +1324,7 @@ func (s *Service) getSummariesList(
 			summariesMap[sessionID][filterKey] = &sum
 		}
 		return nil
-	}, summaryQuery, sessionKeys[0].AppName, sessionKeys[0].UserID, sessionIDs, time.Now().UTC())
+	}, summaryQuery, sessionKeys[0].AppName, sessionKeys[0].UserID, sessionIDs, time.Now())
 
 	if err != nil {
 		return nil, fmt.Errorf("query summaries failed: %w", err)
@@ -1358,7 +1358,7 @@ func (s *Service) cleanupExpiredForUser(ctx context.Context, userKey session.Use
 // If userKey is nil, it cleans up all expired data globally.
 // If userKey is provided, it only cleans up expired data for that specific user.
 func (s *Service) cleanupExpiredData(ctx context.Context, userKey *session.UserKey) {
-	now := time.Now().UTC()
+	now := time.Now()
 
 	// Define cleanup tasks: table name, TTL, and whether it's session-scoped
 	type cleanupTask struct {

session/postgres/service_edge_test.go

@@ -60,7 +60,7 @@ func TestGetSession_RefreshTTL(t *testing.T) {
 	}
 	stateBytes, _ := json.Marshal(sessState)
 	stateRows := sqlmock.NewRows([]string{"state", "created_at", "updated_at"}).
-		AddRow(stateBytes, time.Now().UTC(), time.Now().UTC())
+		AddRow(stateBytes, time.Now(), time.Now())
 
 	mock.ExpectQuery("SELECT state, created_at, updated_at FROM session_states").
 		WithArgs("test-app", "test-user", "test-session", sqlmock.AnyArg()).
@@ -232,7 +232,7 @@ func TestAppendEvent_ToExpiredSession(t *testing.T) {
 		State: session.StateMap{},
 	}
 	stateBytes, _ := json.Marshal(sessState)
-	pastTime := time.Now().UTC().Add(-2 * time.Hour)
+	pastTime := time.Now().Add(-2 * time.Hour)
 	stateRows := sqlmock.NewRows([]string{"state", "expires_at"}).
 		AddRow(stateBytes, pastTime)