whisk/start_test.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.
 */
package whisk

import (
	"bytes"
	"encoding/json"
	"fmt"
	"os"
)

func hello(event json.RawMessage) (json.RawMessage, error) {
	var obj map[string]interface{}
	json.Unmarshal(event, &obj)
	name, ok := obj["name"].(string)
	if !ok {
		name = "Stranger"
	}
	fmt.Printf("name=%s\n", name)
	msg := map[string]string{"message": ("Hello, " + name + "!")}
	return json.Marshal(msg)
}

func Example_repl() {
	in := bytes.NewBufferString("{\"name\":\"Mike\"}\nerr\n")
	repl(hello, in, os.Stdout)
	// Output:
	// name=Mike
	// {"message":"Hello, Mike!"}
	// name=Stranger
	// {"message":"Hello, Stranger!"}
}

func ExampleStart() {
	StartWithArgs(hello, []string{"{\"name\":\"Mike\"}", "err"})
	// Output:
	// name=Mike
	// {"message":"Hello, Mike!"}
	// name=Stranger
	// {"message":"Hello, Stranger!"}
}