fixes

Signed-off-by: Grant Linville <grant@acorn.io>

Author: g-linville

pkg/apis/options/options.go

@@ -160,6 +160,7 @@ func NewFlagSet() *pflag.FlagSet {
 	flagSet.Int("postgres-max-idle-conns", 10, "Maximum number of idle connections to postgres")
 	flagSet.Int("postgres-max-open-conns", 100, "Maximum number of open connections to postgres")
 	flagSet.Int("postgres-conn-max-lifetime", 3600, "Maximum lifetime of a connection to postgres in seconds")
+	flagSet.String("postgres-table-name-prefix", "", "Prefix for postgres table names")
 	flagSet.String("signature-key", "", "GAP-Signature request signature key (algorithm:secretkey)")
 	flagSet.Bool("gcp-healthchecks", false, "Enable GCP/GKE healthcheck endpoints")
 

pkg/sessions/postgres/lock.go

@@ -18,14 +18,6 @@ type SessionLock struct {
 	UpdatedAt time.Time
 }
 
-// Lock implements the sessions.Lock interface using a custom table.
-// This implementation uses a dedicated table to track locks with expiration times.
-//
-// Important notes about this implementation:
-// 1. Locks have built-in expiration functionality
-// 2. Locks are automatically cleaned up when they expire
-// 3. The lock key is stored directly in the table
-// 4. If the database connection is lost, the lock will be automatically released
 type Lock struct {
 	db              *gorm.DB
 	key             string

pkg/sessions/postgres/postgres_store_test.go

@@ -92,6 +92,8 @@ var _ = Describe("PostgreSQL SessionStore Tests", func() {
 
 		BeforeEach(func() {
 			store = &SessionStore{db: db, tableNamePrefix: "oauth2_proxy_test"}
+			db.Table("oauth2_proxy_test_sessions").AutoMigrate(&Session{})
+			db.Table("oauth2_proxy_test_session_locks").AutoMigrate(&SessionLock{})
 		})
 
 		It("obtains and releases locks", func() {