FilesExpand file tree

 php72-scripts

/

QueryCommand.php

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

116 lines (108 loc) · 3.81 KB

 php72-scripts

/

QueryCommand.php

Copy path

Top

File metadata and controls

• Code
• Blame

116 lines (108 loc) · 3.81 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

<?php

/**

* Copyright 2016 Google Inc.

*

* 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.

*/

namespace Google\Cloud\Samples\BigQuery;

use Symfony\Component\Console\Command\Command;

use Symfony\Component\Console\Input\InputInterface;

use Symfony\Component\Console\Input\InputArgument;

use Symfony\Component\Console\Input\InputOption;

use Symfony\Component\Console\Output\OutputInterface;

use Symfony\Component\Console\Question\Question;

use Google\Cloud\Core\Exception\BadRequestException;

use Exception;

/**

* Command line utility to run a BigQuery query.

*

* Usage: php bigquery.php query

*/

class QueryCommand extends Command

{

use ProjectIdTrait;

protected function configure()

{

$this

->setName('query')

->setDescription('Run a BigQuery query')

->setHelp(<<<EOF

The <info>%command.name%</info> command queries your dataset.

<info>%command.full_name% "SELECT TOP(corpus, 3) as title, COUNT(*) as unique_words FROM [publicdata:samples.shakespeare]"</info>

EOF

)

->addArgument(

'query',

InputArgument::OPTIONAL,

'The query to run'

)

->addOption(

'project',

null,

InputOption::VALUE_REQUIRED,

'The Google Cloud Platform project name to use for this invocation. ' .

'If omitted then the current gcloud project is assumed. '

)

->addOption(

'as-job',

null,

InputOption::VALUE_NONE,

'run the query by creating a query job'

)

->addOption(

'legacy-sql',

null,

InputOption::VALUE_NONE,

'run the query using legacy SQL instead of standard SQL syntax'

)

;

}

protected function execute(InputInterface $input, OutputInterface $output)

{

$question = $this->getHelper('question');

if (!$projectId = $input->getOption('project')) {

$projectId = $this->getProjectIdFromGcloud();

}

$message = sprintf('<info>Running query for project %s</info>', $projectId);

$output->writeln($message);

if (!$query = $input->getArgument('query')) {

if ($input->isInteractive()) {

$q = new Question('Enter your query: ');

$query = $question->ask($input, $output, $q);

} else {

throw new Exception('You must supply a query argument');

}

}

try {

if (!$input->getOption('as-job')) {

run_query(

$projectId,

$query,

$input->getOption('legacy-sql')

);

} else {

run_query_as_job(

$projectId,

$query,

$input->getOption('legacy-sql')

);

}

} catch (BadRequestException $e) {

$response = $e->getServiceException()->getResponse();

$errorJson = json_decode((string) $response->getBody(), true);

$error = $errorJson['error']['errors'][0]['message'];

$output->writeln(sprintf('<error>%s</error>', $error));

throw $e;

}

}

}