Пример #1
0
func (app *App) handleModel(w http.ResponseWriter, r *http.Request) {
	switch r.Method {
	case "GET":
		vars := mux.Vars(r)
		name := vars["name"]
		user, ok := app.getUser(r)
		if !ok {
			http.Error(w, "Unauthorized", http.StatusUnauthorized)
			return
		}

		tx, err := app.db.Begin()
		if err != nil {
			app.dbError(w, r, err)
			return
		}
		defer tx.Rollback()
		model, err := db.GetModel(tx, user.Name, name)
		if err != nil {
			if db.IsNotFound(err) {
				http.Error(w, "Model not found", http.StatusNotFound)
			} else {
				app.dbError(w, r, err)
			}
			return
		}
		v := model.ActiveVersion
		activeVersion, err := db.GetModelVersion(tx, user.Name, name, v)
		if err != nil {
			if db.IsNotFound(err) {
				http.Error(w, "Model not found", http.StatusNotFound)
			} else {
				app.dbError(w, r, err)
			}
			return
		}
		app.writeJson(w, map[string]interface{}{
			"Model":   model,
			"Version": activeVersion,
		})
	default:
		http.Error(w, "I only respond to GETs", http.StatusNotImplemented)
	}
}
Пример #2
0
func (s *storage) Get(user, model string, version int) (*mps.DeployInfo, string, error) {
	tx, err := s.app.db.Begin()
	if err != nil {
		return nil, "", fmt.Errorf("database unavailabled: %v", err)
	}
	defer tx.Rollback()
	modelVersion, err := db.GetModelVersion(tx, user, model, version)
	if err != nil {
		if db.IsNotFound(err) {
			return nil, "", errors.New("model version not found")
		}
		return nil, "", fmt.Errorf("could not query model version: %v", err)
	}

	baseImage, err := db.GetBaseImage(tx, modelVersion.Lang)
	if err != nil && err != sql.ErrNoRows {
		return nil, "", fmt.Errorf("could not query baseImage: %v", err)
	}

	// TODO: Conda channels, CRAN mirror and apt-get sources
	info := &mps.DeployInfo{
		Username:         user,
		Modelname:        model,
		Version:          version,
		Lang:             modelVersion.Lang,
		LanguagePackages: modelVersion.LangPackages,
		UbuntuPackages:   modelVersion.UbuntuPackages,
		BaseImage:        baseImage,
	}
	if modelVersion.BundleFilename == "" {
		return nil, "", fmt.Errorf("no bundle file provided")
	}
	bundlePath := filepath.Join(s.app.bundleDir, modelVersion.BundleFilename)
	log.Println("bundle path is", bundlePath)
	return info, bundlePath, nil
}