From da83bb9bfa6b8ea287e1cba4dd191e8ec73a9775 Mon Sep 17 00:00:00 2001
From: Tyler Antonio <contact@tanton.io>
Date: Wed, 8 Apr 2015 10:57:29 -0600
Subject: [PATCH] Updated addEntry() to overwrite existing metadata if they
already exist in the database
---
.../Metadata/MetaDataStorageHandlerPdo.php | 23 +++++++++++++------
1 file changed, 16 insertions(+), 7 deletions(-)
diff --git a/lib/SimpleSAML/Metadata/MetaDataStorageHandlerPdo.php b/lib/SimpleSAML/Metadata/MetaDataStorageHandlerPdo.php
index 894d66384..89c5def97 100644
--- a/lib/SimpleSAML/Metadata/MetaDataStorageHandlerPdo.php
+++ b/lib/SimpleSAML/Metadata/MetaDataStorageHandlerPdo.php
@@ -171,8 +171,7 @@ class SimpleSAML_Metadata_MetaDataStorageHandlerPdo extends SimpleSAML_Metadata_
}
/**
- * Replace the -'s to an _ in table names for Metadata sets
- * since SQL does not allow a - in a table name.
+ * Add metadata to the configured database
*
* @param string $index Entity ID
* @param string $set The set to add the metadata to
@@ -190,13 +189,23 @@ class SimpleSAML_Metadata_MetaDataStorageHandlerPdo extends SimpleSAML_Metadata_
$tableName = $this->getTableName($set);
- $stmt = $this->pdo->prepare("INSERT INTO $tableName (entity_id, entity_data) VALUES(:entity_id, :entity_data)");
+ $metadata = $this->pdo->prepare("SELECT entity_id, entity_data FROM $tableName WHERE entity_id = :entity_id");
+ $metadata->bindValue(":entity_id", $index, PDO::PARAM_STR);
+ $metadata->execute();
+ $retrivedEntityIDs = $metadata->fetch();
+
+ if(count($retrivedEntityIDs) > 0){
+ $stmt = $this->pdo->prepare("UPDATE $tableName SET entity_data = :entity_data WHERE entity_id = :entity_id");
+ }
+ else{
+ $stmt = $this->pdo->prepare("INSERT INTO $tableName (entity_id, entity_data) VALUES (:entity_id, :entity_data)");
+ }
+
$stmt->bindValue(":entity_id", $index, PDO::PARAM_STR);
$stmt->bindValue(":entity_data", json_encode($entityData), PDO::PARAM_STR);
- $stmt->execute();
-
- if ($result === FALSE) {
- throw new Exception("PDO metadata handler: Database error: " . var_export($this->pdo->errorInfo(), TRUE));
+
+ if ($stmt->execute() === FALSE) {
+ throw new Exception("PDO metadata handler: Database error: " . var_export($this->pdo->errorInfo(), TRUE));
}
return 1 === $stmt->rowCount();
}
--
GitLab