mirror of https://github.com/dapr/cli.git
90 lines
3.2 KiB
Go
90 lines
3.2 KiB
Go
// ------------------------------------------------------------
|
|
// Copyright (c) Microsoft Corporation and Dapr Contributors.
|
|
// Licensed under the MIT License.
|
|
// ------------------------------------------------------------
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"runtime"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/dapr/cli/pkg/print"
|
|
"github.com/dapr/cli/pkg/standalone"
|
|
)
|
|
|
|
var (
|
|
publishAppID string
|
|
pubsubName string
|
|
publishTopic string
|
|
publishPayload string
|
|
publishPayloadFile string
|
|
publishSocket string
|
|
)
|
|
|
|
var PublishCmd = &cobra.Command{
|
|
Use: "publish",
|
|
Short: "Publish a pub-sub event. Supported platforms: Self-hosted",
|
|
Example: `
|
|
# Publish to sample topic in target pubsub via a publishing app
|
|
dapr publish --publish-app-id myapp --pubsub target --topic sample --data '{"key":"value"}'
|
|
|
|
# Publish to sample topic in target pubsub via a publishing app using Unix domain socket
|
|
dapr publish --enable-domain-socket --publish-app-id myapp --pubsub target --topic sample --data '{"key":"value"}'
|
|
`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
bytePayload := []byte{}
|
|
var err error
|
|
if publishPayloadFile != "" && publishPayload != "" {
|
|
print.FailureStatusEvent(os.Stderr, "Only one of --data and --data-file allowed in the same publish command")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if publishPayloadFile != "" {
|
|
bytePayload, err = ioutil.ReadFile(publishPayloadFile)
|
|
if err != nil {
|
|
print.FailureStatusEvent(os.Stderr, "Error reading payload from '%s'. Error: %s", publishPayloadFile, err)
|
|
os.Exit(1)
|
|
}
|
|
} else if publishPayload != "" {
|
|
bytePayload = []byte(publishPayload)
|
|
}
|
|
|
|
client := standalone.NewClient()
|
|
// TODO(@daixiang0): add Windows support
|
|
if publishSocket != "" {
|
|
if runtime.GOOS == "windows" {
|
|
print.FailureStatusEvent(os.Stderr, "The unix-domain-socket option is not supported on Windows")
|
|
os.Exit(1)
|
|
} else {
|
|
print.WarningStatusEvent(os.Stdout, "Unix domain sockets are currently a preview feature")
|
|
}
|
|
}
|
|
err = client.Publish(publishAppID, pubsubName, publishTopic, bytePayload, publishSocket)
|
|
if err != nil {
|
|
print.FailureStatusEvent(os.Stderr, fmt.Sprintf("Error publishing topic %s: %s", publishTopic, err))
|
|
os.Exit(1)
|
|
}
|
|
|
|
print.SuccessStatusEvent(os.Stdout, "Event published successfully")
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
PublishCmd.Flags().StringVarP(&publishAppID, "publish-app-id", "i", "", "The ID of the publishing app")
|
|
PublishCmd.Flags().StringVarP(&pubsubName, "pubsub", "p", "", "The name of the pub/sub component")
|
|
PublishCmd.Flags().StringVarP(&publishTopic, "topic", "t", "", "The topic to be published to")
|
|
PublishCmd.Flags().StringVarP(&publishPayload, "data", "d", "", "The JSON serialized data string (optional)")
|
|
PublishCmd.Flags().StringVarP(&publishPayloadFile, "data-file", "f", "", "A file containing the JSON serialized data (optional)")
|
|
PublishCmd.Flags().StringVarP(&publishSocket, "unix-domain-socket", "u", "", "Path to a unix domain socket dir. If specified, Dapr API servers will use Unix Domain Sockets")
|
|
PublishCmd.Flags().BoolP("help", "h", false, "Print this help message")
|
|
PublishCmd.MarkFlagRequired("publish-app-id")
|
|
PublishCmd.MarkFlagRequired("topic")
|
|
PublishCmd.MarkFlagRequired("pubsub")
|
|
RootCmd.AddCommand(PublishCmd)
|
|
}
|