Skip to content

Commit

Permalink
Add Background Sample
Browse files Browse the repository at this point in the history
  • Loading branch information
Bill Prin committed Apr 14, 2016
1 parent db0d35a commit 405c845
Show file tree
Hide file tree
Showing 4 changed files with 157 additions and 0 deletions.
9 changes: 9 additions & 0 deletions appengine/background/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
# Using Background Threads from Google App Engine

This example shows how to use manual or basic scaling to start App Engine background threads.

See the [documentation on modules](https://cloud.google.com/appengine/docs/python/modules/) for
more information.

Your app.yaml configuration must specify scaling as either manual or basic. The default
automatic scaling does not allow background threads.
10 changes: 10 additions & 0 deletions appengine/background/app.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
runtime: python27
api_version: 1
threadsafe: yes

manual_scaling:
instances: 1

handlers:
- url: .*
script: main.app
85 changes: 85 additions & 0 deletions appengine/background/main.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,85 @@
# Copyright 2016 Google Inc. All rights reserved.
#
# 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.

"""
Sample application that demonstrates how to use the App Engine background
threads.
app.yaml scaling must be set to manual or basic.
"""

# [START background-imp]
from google.appengine.api import background_thread
# [END background-imp]

import webapp2

val = 'Dog'


class MainHandler(webapp2.RequestHandler):
def get(self):
self.response.headers['Content-Type'] = 'text/plain'
self.response.write(str(val))


class SetDogHandler(webapp2.RequestHandler):
""" Resets the global val to 'Dog'"""

def get(self):
global val
val = 'Dog'
self.response.headers['Content-Type'] = 'text/plain'
self.response.write('Done')


class SetCatBackgroundHandler(webapp2.RequestHandler):
""" Demonstrates two ways to start new background threads
"""

def get(self):
"""
Demonstrates using a background thread to change the global
val from 'Dog' to 'Cat'
The auto GET parameter determines whether to start the thread
automatically or manually
"""
auto = self.request.get('auto')

# [START background-start]
# sample function to run in a background thread
def change_val(arg):
global val
val = arg

if auto:
# Start the new thread in one command
background_thread.start_new_background_thread(change_val, ['Cat'])
else:
# create a new thread and start it
t = background_thread.BackgroundThread(
target=change_val, args=['Cat'])
t.start()
# [END background-start]

self.response.headers['Content-Type'] = 'text/plain'
self.response.write('Done')

app = webapp2.WSGIApplication([
('/', MainHandler),
('/dog', SetDogHandler),
('/cat', SetCatBackgroundHandler),
], debug=True)
# [END all]
53 changes: 53 additions & 0 deletions appengine/background/main_test.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
# Copyright 2016 Google Inc. All rights reserved.
#
# 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.

import main

from mock import patch
import pytest
import webtest


@pytest.fixture
def app(cloud_config, testbed):
main.PROJECTID = cloud_config.project
return webtest.TestApp(main.app)


@patch("main.background_thread")
def test_background(thread, app):
app.get('/dog')
response = app.get('/')
assert response.status_int == 200
assert response.body == 'Dog'
app.get('/cat')
# no stub for system so manually set
main.val = 'Cat'
response = app.get('/')
assert response.status_int == 200
assert response.body == 'Cat'


@patch("main.background_thread")
def test_background_auto_start(thread, app):
app.get('/dog')
response = app.get('/')
assert response.status_int == 200
assert response.body == 'Dog'
app.get('/cat?auto=True')
# no stub for system so manually set
main.val = 'Cat'
response = app.get('/')
assert response.status_int == 200
assert response.body == 'Cat'

0 comments on commit 405c845

Please sign in to comment.