FilesExpand file tree

 master

/

Votes.php

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

127 lines (111 loc) · 3.32 KB

 master

/

Votes.php

Copy path

Top

File metadata and controls

• Code
• Blame

127 lines (111 loc) · 3.32 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

<?php

/*

* Copyright 2020 Google LLC.

*

* Licensed under the Apache License, Version 2.0 (the "License");

* you may not use this file except in compliance with the License.

* You may obtain a copy of the License at

*

* http://www.apache.org/licenses/LICENSE-2.0

*

* Unless required by applicable law or agreed to in writing, software

* distributed under the License is distributed on an "AS IS" BASIS,

* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

* See the License for the specific language governing permissions and

* limitations under the License.

*/

declare(strict_types=1);

namespace Google\Cloud\Samples\CloudSQL\Postgres;

use PDO;

use PDOException;

use RuntimeException;

/**

* Manage votes using the Cloud SQL database.

*/

class Votes

{

/**

* @var PDO

*/

private $connection;

/**

* @param PDO $connection A connection to the database.

*/

public function __construct(PDO $connection)

{

$this->connection = $connection;

}

/**

* Creates the table if it does not yet exist.

*

* @return void

*/

public function createTableIfNotExists()

{

try {

$stmt = $this->connection->prepare('SELECT 1 FROM votes');

$stmt->execute();

} catch (PDOException $e) {

$sql = "CREATE TABLE votes (

vote_id SERIAL NOT NULL,

time_cast TIMESTAMP NOT NULL,

vote_value VARCHAR(6) NOT NULL,

PRIMARY KEY (vote_id)

);";

$this->connection->exec($sql);

}

}

/**

* Returns a list of the last five votes

*

* @return array

*/

public function listVotes() : array

{

$sql = "SELECT vote_value, time_cast FROM votes ORDER BY time_cast DESC LIMIT 5";

$statement = $this->connection->prepare($sql);

$statement->execute();

return $statement->fetchAll(PDO::FETCH_ASSOC);

}

/**

* Get the number of votes cast for a given value.

*

* @param string $value

* @param int

*/

public function getCountByValue(string $value) : int

{

$sql = "SELECT COUNT(vote_id) as voteCount FROM votes WHERE vote_value = ?";

$statement = $this->connection->prepare($sql);

$statement->execute([$value]);

return (int) $statement->fetch(PDO::FETCH_COLUMN);

}

/**

* Insert a new vote into the database

*

* @param string $value The value to vote for.

* @return boolean

*/

public function insertVote(string $value) : bool

{

$conn = $this->connection;

$res = false;

# [START cloud_sql_postgres_pdo_connection]

// Use prepared statements to guard against SQL injection.

$sql = "INSERT INTO votes (time_cast, vote_value) VALUES (NOW(), :voteValue)";

try {

$statement = $conn->prepare($sql);

$statement->bindParam('voteValue', $value);

$res = $statement->execute();

} catch (PDOException $e) {

throw new RuntimeException(

"Could not insert vote into database. The PDO exception was " .

$e->getMessage(),

$e->getCode(),

$e

);

}

# [END cloud_sql_postgres_pdo_connection]

return $res;

}

}